fix bug
This commit is contained in:
42
internal/logic/strategy/rule/ai.go
Normal file
42
internal/logic/strategy/rule/ai.go
Normal file
@@ -0,0 +1,42 @@
|
||||
package rule
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"git.apinb.com/bsm-sdk/core/utils"
|
||||
"git.apinb.com/quant/gostock/internal/config"
|
||||
"git.apinb.com/quant/gostock/internal/logic/deepseek"
|
||||
)
|
||||
|
||||
var (
|
||||
MarkdataPath = "./markdata/"
|
||||
)
|
||||
|
||||
func (r *RuleFactory) RunAi(code string) {
|
||||
mdPath := MarkdataPath + code + ".md"
|
||||
if !utils.PathExists(mdPath) {
|
||||
r.Model.AiScrore = -1
|
||||
r.Model.AddDesc(fmt.Sprintf("%s markdown 文件未找友", mdPath))
|
||||
return
|
||||
}
|
||||
|
||||
c := deepseek.Config{
|
||||
APIKey: config.Spec.DeepSeekApiKey, // 替换为你的API Key
|
||||
BaseURL: "https://api.deepseek.com",
|
||||
Model: "deepseek-chat", // 或 "deepseek-coder"
|
||||
MaxTokens: 2000,
|
||||
Temperature: 0.7,
|
||||
}
|
||||
|
||||
prompt := "你是一名资深量化投研分析师。请根据markdown格式的附件内容,给出详细总结与买入评分(0-10),输出JSON,字段:summary(500字内中文总结)、score(0-10数字)、action(买入/谨慎/观望)、risk(一句风险提示)。"
|
||||
result, err := deepseek.ProcessMarkdownQA(c, mdPath, prompt)
|
||||
if err != nil {
|
||||
r.Model.AiScrore = -1
|
||||
r.Model.AddDesc(fmt.Sprintf("处理失败: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 输出JSON格式的结果
|
||||
fmt.Println(string(result))
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user