mirror of
https://github.com/ArvinLovegood/go-stock.git
synced 2025-07-19 00:00:09 +08:00
feat(backend):添加持仓成本价(costPrice)变量到用户提问问题模板
- 在 openai_api.go 文件中,增加了对持仓成本价的处理 - 通过查询数据库获取股票的持仓成本价,并加入到替换模板中 - 更新了问题模板的替换逻辑,支持新的成本价变量
This commit is contained in:
parent
a20d4e721d
commit
dab51f7a70
@ -114,10 +114,21 @@ func (o OpenAi) NewChatStream(stock, stockCode string) <-chan string {
|
||||
"content": o.Prompt,
|
||||
},
|
||||
}
|
||||
question := strutil.ReplaceWithMap(o.QuestionTemplate, map[string]string{
|
||||
|
||||
replaceTemplates := map[string]string{
|
||||
"{{stockName}}": RemoveAllBlankChar(stock),
|
||||
"{{stockCode}}": RemoveAllBlankChar(stockCode),
|
||||
})
|
||||
}
|
||||
|
||||
followedStock := &FollowedStock{
|
||||
StockCode: stockCode,
|
||||
}
|
||||
db.Dao.Model(&followedStock).Where("stock_code = ?", stockCode).First(followedStock)
|
||||
if followedStock.CostPrice > 0 {
|
||||
replaceTemplates["{{costPrice}}"] = fmt.Sprintf("%.2f", followedStock.CostPrice)
|
||||
}
|
||||
|
||||
question := strutil.ReplaceWithMap(o.QuestionTemplate, replaceTemplates)
|
||||
logger.SugaredLogger.Infof("NewChatStream stock:%s stockCode:%s", stock, stockCode)
|
||||
logger.SugaredLogger.Infof("Prompt:%s", o.Prompt)
|
||||
logger.SugaredLogger.Infof("User Prompt config:%v", o.QuestionTemplate)
|
||||
|
Loading…
x
Reference in New Issue
Block a user