Merge branch 'dev' of https://github.com/tinkle-community/nofx

api/server.go

@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"nofx/auth"
 	"nofx/config"
+	"nofx/decision"
 	"nofx/manager"
 	"strconv"
 	"strings"
@@ -109,6 +110,10 @@ func (s *Server) setupRoutes() {
 			protected.GET("/user/signal-sources", s.handleGetUserSignalSource)
 			protected.POST("/user/signal-sources", s.handleSaveUserSignalSource)
 
+			// 系统提示词模板管理
+			protected.GET("/prompt-templates", s.handleGetPromptTemplates)
+			protected.GET("/prompt-templates/:name", s.handleGetPromptTemplate)
+
 			// 竞赛总览
 			protected.GET("/competition", s.handleCompetition)
 			
@@ -200,18 +205,19 @@ func (s *Server) getTraderFromQuery(c *gin.Context) (*manager.TraderManager, str
 
 // AI交易员管理相关结构体
 type CreateTraderRequest struct {
-	Name            string  `json:"name" binding:"required"`
-	AIModelID       string  `json:"ai_model_id" binding:"required"`
-	ExchangeID      string  `json:"exchange_id" binding:"required"`
-	InitialBalance  float64 `json:"initial_balance"`
-	BTCETHLeverage  int     `json:"btc_eth_leverage"`
-	AltcoinLeverage int     `json:"altcoin_leverage"`
-	TradingSymbols  string  `json:"trading_symbols"`
-	CustomPrompt    string  `json:"custom_prompt"`
-	OverrideBasePrompt bool `json:"override_base_prompt"`
-	IsCrossMargin   *bool   `json:"is_cross_margin"` // 指针类型，nil表示使用默认值true
-	UseCoinPool     bool    `json:"use_coin_pool"`
-	UseOITop        bool    `json:"use_oi_top"`
+	Name                 string  `json:"name" binding:"required"`
+	AIModelID            string  `json:"ai_model_id" binding:"required"`
+	ExchangeID           string  `json:"exchange_id" binding:"required"`
+	InitialBalance       float64 `json:"initial_balance"`
+	BTCETHLeverage       int     `json:"btc_eth_leverage"`
+	AltcoinLeverage      int     `json:"altcoin_leverage"`
+	TradingSymbols       string  `json:"trading_symbols"`
+	CustomPrompt         string  `json:"custom_prompt"`
+	OverrideBasePrompt   bool    `json:"override_base_prompt"`
+	SystemPromptTemplate string  `json:"system_prompt_template"` // 系统提示词模板名称
+	IsCrossMargin        *bool   `json:"is_cross_margin"`        // 指针类型，nil表示使用默认值true
+	UseCoinPool          bool    `json:"use_coin_pool"`
+	UseOITop             bool    `json:"use_oi_top"`
 }
 
 type ModelConfig struct {
@@ -319,23 +325,30 @@ func (s *Server) handleCreateTrader(c *gin.Context) {
 		}
 	}
 	
+	// 设置系统提示词模板默认值
+	systemPromptTemplate := "default"
+	if req.SystemPromptTemplate != "" {
+		systemPromptTemplate = req.SystemPromptTemplate
+	}
+
     // 创建交易员配置（数据库实体）
     trader := &config.TraderRecord{
-		ID:                  traderID,
-		UserID:              userID,
-		Name:                req.Name,
-		AIModelID:           req.AIModelID,
-		ExchangeID:          req.ExchangeID,
-		InitialBalance:      req.InitialBalance,
-		BTCETHLeverage:      btcEthLeverage,
-		AltcoinLeverage:     altcoinLeverage,
-		TradingSymbols:      req.TradingSymbols,
-		UseCoinPool:         req.UseCoinPool,
-		UseOITop:            req.UseOITop,
-		CustomPrompt:        req.CustomPrompt,
-		OverrideBasePrompt:  req.OverrideBasePrompt,
-		IsCrossMargin:       isCrossMargin,
-		ScanIntervalMinutes: 3, // 默认3分钟
+		ID:                   traderID,
+		UserID:               userID,
+		Name:                 req.Name,
+		AIModelID:            req.AIModelID,
+		ExchangeID:           req.ExchangeID,
+		InitialBalance:       req.InitialBalance,
+		BTCETHLeverage:       btcEthLeverage,
+		AltcoinLeverage:      altcoinLeverage,
+		TradingSymbols:       req.TradingSymbols,
+		UseCoinPool:          req.UseCoinPool,
+		UseOITop:             req.UseOITop,
+		CustomPrompt:         req.CustomPrompt,
+		OverrideBasePrompt:   req.OverrideBasePrompt,
+		SystemPromptTemplate: systemPromptTemplate,
+		IsCrossMargin:        isCrossMargin,
+		ScanIntervalMinutes:  3, // 默认3分钟
 		IsRunning:           false,
 	}
 
@@ -1407,3 +1420,37 @@ func (s *Server) Start() error {
 
 	return s.router.Run(addr)
 }
+
+// handleGetPromptTemplates 获取所有系统提示词模板列表
+func (s *Server) handleGetPromptTemplates(c *gin.Context) {
+	// 导入 decision 包
+	templates := decision.GetAllPromptTemplates()
+	
+	// 转换为响应格式
+	response := make([]map[string]interface{}, 0, len(templates))
+	for _, tmpl := range templates {
+		response = append(response, map[string]interface{}{
+			"name": tmpl.Name,
+		})
+	}
+	
+	c.JSON(http.StatusOK, gin.H{
+		"templates": response,
+	})
+}
+
+// handleGetPromptTemplate 获取指定名称的提示词模板内容
+func (s *Server) handleGetPromptTemplate(c *gin.Context) {
+	templateName := c.Param("name")
+	
+	template, err := decision.GetPromptTemplate(templateName)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("模板不存在: %s", templateName)})
+		return
+	}
+	
+	c.JSON(http.StatusOK, gin.H{
+		"name":    template.Name,
+		"content": template.Content,
+	})
+}