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

}