feat(backend):添加持仓成本价(costPrice)变量到用户提问问题模板

- 在 openai_api.go 文件中,增加了对持仓成本价的处理
- 通过查询数据库获取股票的持仓成本价,并加入到替换模板中
- 更新了问题模板的替换逻辑,支持新的成本价变量
This commit is contained in:
spark 2025-02-15 15:30:46 +08:00
parent a20d4e721d
commit dab51f7a70

View File

@ -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)