feat(frontend):添加AI智能体聊天功能

- 在前端 App.vue 中添加 AI智能体聊天入口
- 在后端 App.d.ts 和 App.js 中添加 ChatWithAgent 函数- 在 app_common.go 中实现 ChatWithAgent 方法，使用 agent.NewStockAiAgentApi().Chat 进行聊天
- 更新 go.mod，添加与 AI 聊天相关的依赖

Author: ArvinLovegood

app_common.go

@@ -1,6 +1,8 @@
 package main
 
 import (
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+	"go-stock/backend/agent"
 	"go-stock/backend/data"
 	"go-stock/backend/models"
 )
@@ -62,3 +64,10 @@ func (a App) SearchStock(words string) map[string]any {
 func (a App) GetHotStrategy() map[string]any {
 	return data.NewSearchStockApi("").HotStrategy()
 }
+
+func (a App) ChatWithAgent(question string, aiConfigId int, sysPromptId *int) {
+	ch := agent.NewStockAiAgentApi().Chat(question, aiConfigId, sysPromptId)
+	for msg := range ch {
+		runtime.EventsEmit(a.ctx, "agent-message", msg)
+	}
+}

backend/agent/agent.go

@@ -0,0 +1,92 @@
+package agent
+
+import (
+	"context"
+	"go-stock/backend/agent/tools"
+	"go-stock/backend/data"
+	"go-stock/backend/logger"
+	"time"
+
+	"github.com/cloudwego/eino-ext/components/model/ark"
+	"github.com/cloudwego/eino-ext/components/model/deepseek"
+	"github.com/cloudwego/eino-ext/components/model/openai"
+	"github.com/cloudwego/eino/components/model"
+	"github.com/cloudwego/eino/components/tool"
+	"github.com/cloudwego/eino/compose"
+	"github.com/cloudwego/eino/flow/agent/react"
+	"github.com/cloudwego/eino/schema"
+)
+
+// GetStockAiAgent @Author spark
+// @Date 2025/8/4 16:17
+// @Desc
+// -----------------------------------------------------------------------------------
+func GetStockAiAgent(ctx *context.Context, aiConfig data.AIConfig) *react.Agent {
+	logger.SugaredLogger.Infof("GetStockAiAgent aiConfig: %v", aiConfig)
+	temperature := float32(aiConfig.Temperature)
+	var toolableChatModel model.ToolCallingChatModel
+	var err error
+	if aiConfig.BaseUrl == "https://ark.cn-beijing.volces.com/api/v3" {
+		toolableChatModel, err = ark.NewChatModel(context.Background(), &ark.ChatModelConfig{
+			BaseURL:     aiConfig.BaseUrl,
+			Model:       aiConfig.ModelName,
+			APIKey:      aiConfig.ApiKey,
+			MaxTokens:   &aiConfig.MaxTokens,
+			Temperature: &temperature,
+		})
+
+	} else if aiConfig.BaseUrl == "https://api.deepseek.com" {
+		toolableChatModel, err = deepseek.NewChatModel(*ctx, &deepseek.ChatModelConfig{
+			BaseURL:     aiConfig.BaseUrl,
+			Model:       aiConfig.ModelName,
+			APIKey:      aiConfig.ApiKey,
+			Timeout:     time.Duration(aiConfig.TimeOut) * time.Second,
+			MaxTokens:   aiConfig.MaxTokens,
+			Temperature: temperature,
+		})
+
+	} else {
+		toolableChatModel, err = openai.NewChatModel(*ctx, &openai.ChatModelConfig{
+			BaseURL:     aiConfig.BaseUrl,
+			Model:       aiConfig.ModelName,
+			APIKey:      aiConfig.ApiKey,
+			Timeout:     time.Duration(aiConfig.TimeOut) * time.Second,
+			MaxTokens:   &aiConfig.MaxTokens,
+			Temperature: &temperature,
+		})
+	}
+
+	if err != nil {
+		logger.SugaredLogger.Error(err.Error())
+		return nil
+	}
+	// 初始化所需的 tools
+	aiTools := compose.ToolsNodeConfig{
+		Tools: []tool.BaseTool{
+			tools.GetQueryEconomicDataTool(),
+			tools.GetQueryStockPriceInfoTool(),
+			tools.GetQueryStockCodeInfoTool(),
+			tools.GetQueryMarketNewsTool(),
+			tools.GetChoiceStockByIndicatorsTool(),
+			tools.GetStockKLineTool(),
+			tools.GetInteractiveAnswerDataTool(),
+			tools.GetFinancialReportTool(),
+			tools.GetQueryStockNewsTool(),
+			tools.GetIndustryResearchReportTool(),
+		},
+	}
+	// 创建 agent
+	agent, err := react.NewAgent(*ctx, &react.AgentConfig{
+		ToolCallingModel: toolableChatModel,
+		ToolsConfig:      aiTools,
+		MaxStep:          len(aiTools.Tools)*3 + 2,
+		MessageModifier: func(ctx context.Context, input []*schema.Message) []*schema.Message {
+			return input
+		},
+	})
+	if err != nil {
+		logger.SugaredLogger.Error(err.Error())
+		return nil
+	}
+	return agent
+}

backend/agent/agent_api.go

@@ -0,0 +1,91 @@
+package agent
+
+import (
+	"context"
+	"errors"
+	"github.com/cloudwego/eino/compose"
+	"github.com/cloudwego/eino/flow/agent"
+	"github.com/cloudwego/eino/flow/agent/react"
+	"github.com/cloudwego/eino/schema"
+	"github.com/samber/lo"
+	"go-stock/backend/agent/tool_logger"
+	"go-stock/backend/data"
+	"go-stock/backend/logger"
+	"io"
+)
+
+// @Author spark
+// @Date 2025/8/7 9:07
+// @Desc
+// -----------------------------------------------------------------------------------
+type StockAiAgent struct {
+	*react.Agent
+}
+
+func NewStockAiAgentApi() *StockAiAgent {
+	return &StockAiAgent{}
+}
+
+func (receiver StockAiAgent) newStockAiAgent(ctx *context.Context, aiConfigId int) *StockAiAgent {
+	settingConfig := data.GetSettingConfig()
+	aiConfig, ok := lo.Find(settingConfig.AiConfigs, func(item *data.AIConfig) bool {
+		return uint(aiConfigId) == item.ID
+	})
+	if !ok {
+		return nil
+	}
+	return &StockAiAgent{
+		Agent: GetStockAiAgent(ctx, *aiConfig),
+	}
+}
+
+func (receiver StockAiAgent) Chat(question string, aiConfigId int, sysPromptId *int) chan *schema.Message {
+	ch := make(chan *schema.Message, 512)
+	ctx := context.Background()
+	stockAiAgent := receiver.newStockAiAgent(&ctx, aiConfigId)
+
+	sysPrompt := ""
+	if sysPromptId == nil || *sysPromptId == 0 {
+		sysPrompt = "你现在扮演一位拥有20年实战经验的顶级股票投资大师，精通价值投资、趋势交易、量化分析等多种策略。你擅长结合宏观经济、行业周期和企业基本面进行全方位、精准的多维分析，尤其对A股、港股、美股市场有深刻理解，始终秉持“风险控制第一”的原则，善于用通俗易懂的方式传授投资智慧。"
+	} else {
+		sysPrompt = data.NewPromptTemplateApi().GetPromptTemplateByID(*sysPromptId)
+	}
+	agentOption := []agent.AgentOption{
+		agent.WithComposeOptions(compose.WithCallbacks(&tool_logger.LoggerCallback{MessageChanel: ch})),
+		//react.WithChatModelOptions(ark.WithCache(cacheOption)),
+	}
+
+	go func() {
+		defer close(ch)
+		sr, err := stockAiAgent.Stream(ctx, []*schema.Message{
+			{
+				Role:    schema.System,
+				Content: sysPrompt,
+			},
+			{
+				Role:    schema.User,
+				Content: question,
+			},
+		}, agentOption...)
+		if err != nil {
+			logger.SugaredLogger.Errorf("stream error: %v", err)
+			return
+		}
+		defer sr.Close()
+		for {
+			msg, err := sr.Recv()
+			if err != nil {
+				if errors.Is(err, io.EOF) {
+					// finish
+					break
+				}
+				// error
+				logger.SugaredLogger.Errorf("failed to recv: %v", err)
+				break
+			}
+			logger.SugaredLogger.Infof("stream: %s", msg.String())
+			ch <- msg
+		}
+	}()
+	return ch
+}

backend/agent/agent_test.go

@@ -0,0 +1,84 @@
+package agent
+
+import (
+	"context"
+	"errors"
+	"go-stock/backend/agent/tool_logger"
+	"go-stock/backend/data"
+	"go-stock/backend/db"
+	"go-stock/backend/logger"
+	"io"
+	"strings"
+	"testing"
+
+	"github.com/cloudwego/eino/compose"
+	"github.com/cloudwego/eino/flow/agent"
+	"github.com/cloudwego/eino/schema"
+)
+
+// @Author spark
+// @Date 2025/8/4 17:32
+// @Desc
+//-----------------------------------------------------------------------------------
+
+func TestGetStockAiAgent(t *testing.T) {
+	ctx := context.Background()
+	db.Init("../../data/stock.db")
+	config := data.GetSettingConfig()
+	aiAgent := GetStockAiAgent(&ctx, *config.AiConfigs[0])
+
+	opt := []agent.AgentOption{
+		agent.WithComposeOptions(compose.WithCallbacks(&tool_logger.LoggerCallback{})),
+		//react.WithChatModelOptions(ark.WithCache(cacheOption)),
+	}
+
+	sr, err := aiAgent.Stream(ctx, []*schema.Message{
+		{
+			Role:    schema.System,
+			Content: config.Settings.Prompt + "",
+		},
+		{
+			Role:    schema.User,
+			Content: "结合以上提供的宏观经济数据/市场指数行情/国内外市场资讯/电报/会议/事件/投资者关注的问题，\n结合宏观经济，事件驱动，政策支持，投资者关注的问题，分析当前市场情绪和热点 找出有潜力/优质的板块/行业/概念/标的/主题，\n多因子深度分析计算上涨或下跌的逻辑和概率，\n最后按风险和投资周期给出具体推荐标的操作建议",
+		},
+	}, opt...)
+	if err != nil {
+		logger.SugaredLogger.Errorf("stream error: %v", err)
+		return
+	}
+
+	defer sr.Close() // remember to close the stream
+
+	md := strings.Builder{}
+	for {
+		msg, err := sr.Recv()
+		if err != nil {
+			if errors.Is(err, io.EOF) {
+				// finish
+				break
+			}
+			// error
+			logger.SugaredLogger.Errorf("failed to recv: %v", err)
+			return
+		}
+		//logger.SugaredLogger.Infof("stream recv: %v", msg)
+		if msg.ReasoningContent != "" {
+			md.WriteString(msg.ReasoningContent)
+		}
+		if msg.Content != "" {
+			md.WriteString(msg.Content)
+		}
+	}
+	logger.SugaredLogger.Info(md.String())
+	//logger.SugaredLogger.Infof("stream done:\n%s", md.String())
+}
+
+func TestAgent(t *testing.T) {
+	db.Init("../../data/stock.db")
+
+	ch := NewStockAiAgentApi().Chat("分析一下海立股份，使用工具", 1, nil)
+	for message := range ch {
+		logger.SugaredLogger.Infof("res:%s", message.String())
+	}
+
+}

backend/agent/tool_logger/tool_logger.go

@@ -0,0 +1,98 @@
+package tool_logger
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"go-stock/backend/logger"
+	"io"
+
+	"github.com/cloudwego/eino/callbacks"
+	"github.com/cloudwego/eino/components/model"
+	"github.com/cloudwego/eino/flow/agent/react"
+	"github.com/cloudwego/eino/schema"
+)
+
+// @Author spark
+// @Date 2025/8/5 10:21
+// @Desc
+//-----------------------------------------------------------------------------------
+
+type LoggerCallback struct {
+	MessageChanel            chan *schema.Message
+	callbacks.HandlerBuilder // 可以用 callbacks.HandlerBuilder 来辅助实现 callback
+}
+
+func (cb *LoggerCallback) OnStart(ctx context.Context, info *callbacks.RunInfo, input callbacks.CallbackInput) context.Context {
+	logger.SugaredLogger.Infof("==================")
+	inputStr, _ := json.MarshalIndent(input, "", "  ") // nolint: byted_s_returned_err_check
+	logger.SugaredLogger.Infof("[OnStart] %s\n", string(inputStr))
+
+	modelCallbackInput := model.ConvCallbackInput(input)
+	if modelCallbackInput != nil {
+		for _, message := range modelCallbackInput.Messages {
+			cb.MessageChanel <- message
+		}
+	}
+	return ctx
+}
+
+func (cb *LoggerCallback) OnEnd(ctx context.Context, info *callbacks.RunInfo, output callbacks.CallbackOutput) context.Context {
+	logger.SugaredLogger.Infof("=========[OnEnd]=========")
+	outputStr, _ := json.MarshalIndent(output, "", "  ") // nolint: byted_s_returned_err_check
+	logger.SugaredLogger.Infof(string(outputStr))
+	return ctx
+}
+
+func (cb *LoggerCallback) OnError(ctx context.Context, info *callbacks.RunInfo, err error) context.Context {
+	logger.SugaredLogger.Infof("=========[OnError]=========")
+	logger.SugaredLogger.Infof("%s", err.Error())
+	return ctx
+}
+
+func (cb *LoggerCallback) OnEndWithStreamOutput(ctx context.Context, info *callbacks.RunInfo,
+	output *schema.StreamReader[callbacks.CallbackOutput]) context.Context {
+
+	var graphInfoName = react.GraphName
+
+	go func() {
+		defer func() {
+			if err := recover(); err != nil {
+				logger.SugaredLogger.Infof("[OnEndStream] panic err:", err)
+			}
+		}()
+
+		defer output.Close() // remember to close the stream in defer
+
+		logger.SugaredLogger.Infof("=========[OnEndStream]=========")
+		for {
+			frame, err := output.Recv()
+			if errors.Is(err, io.EOF) {
+				// finish
+				break
+			}
+			if err != nil {
+				logger.SugaredLogger.Infof("internal error: %s\n", err)
+				return
+			}
+
+			s, err := json.Marshal(frame)
+			if err != nil {
+				logger.SugaredLogger.Infof("internal error: %s\n", err)
+				return
+			}
+
+			if info.Name == graphInfoName { // 仅打印 graph 的输出, 否则每个 stream 节点的输出都会打印一遍
+				logger.SugaredLogger.Infof("%s: %s\n", info.Name, string(s))
+			}
+		}
+
+	}()
+	return ctx
+}
+
+func (cb *LoggerCallback) OnStartWithStreamInput(ctx context.Context, info *callbacks.RunInfo,
+	input *schema.StreamReader[callbacks.CallbackInput]) context.Context {
+	defer input.Close()
+	return ctx
+}