43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
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))
|
||
|
||
}
|