up

2026-06-17 23:10:20 +08:00
parent fd0a3bc1fe
commit 9f088c265d
7 changed files with 1408 additions and 2 deletions
@@ -1,6 +1,7 @@
 package main

 import (
+	"errors"
 	"net/http"
 	"strconv"
 	"time"
@@ -12,12 +13,24 @@ import (
 func registerAdminLLMRoutes(r *gin.RouterGroup, store *store) {
 	group := r.Group("/llm")
 	{
+		// LLM Message Queue
 		group.GET("/messages", handleListLLMMessages(store))
 		group.GET("/messages/:id", handleGetLLMMessage(store))
 		group.PUT("/messages/:id/status", handleUpdateLLMMessageStatus(store))
 		group.DELETE("/messages/:id", handleDeleteLLMMessage(store))
 		group.DELETE("/bots/:bot_id/messages", handleDeleteLLMMessagesByBot(store))
 		group.POST("/messages/cleanup", handleCleanupDeletedLLMMessages(store))
+
+		// LLM Providers
+		group.GET("/providers", handleListLLMProviders(store))
+		group.GET("/providers/:name", handleGetLLMProvider(store))
+		group.POST("/providers", handleCreateLLMProvider(store))
+		group.PUT("/providers/:name", handleUpdateLLMProvider(store))
+		group.DELETE("/providers/:name", handleDeleteLLMProvider(store))
+
+		// LLM Tool Router
+		group.GET("/tool-router", handleGetLLMToolRouter(store))
+		group.PUT("/tool-router", handleUpdateLLMToolRouter(store))
 	}
 }

@@ -184,3 +197,301 @@ func handleCleanupDeletedLLMMessages(store *store) gin.HandlerFunc {
 		c.JSON(http.StatusOK, gin.H{"status": "ok", "deleted_count": count})
 	}
 }
+
+// ============================================
+// LLM Provider Handlers
+// ============================================
+
+func handleListLLMProviders(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		includeInactive := c.Query("include_inactive") == "true"
+
+		rows, err := store.ListLLMProviders(includeInactive)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		items := make([]map[string]any, 0, len(rows))
+		for _, row := range rows {
+			items = append(items, llmProviderDTO(row))
+		}
+
+		c.JSON(http.StatusOK, gin.H{"items": items})
+	}
+}
+
+func handleGetLLMProvider(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		name := c.Param("name")
+		if name == "" {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "provider name is required"})
+			return
+		}
+
+		record, err := store.GetLLMProvider(name)
+		if err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				c.JSON(http.StatusNotFound, gin.H{"error": "provider not found"})
+				return
+			}
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		c.JSON(http.StatusOK, gin.H{"item": llmProviderDTO(*record)})
+	}
+}
+
+func handleCreateLLMProvider(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		var req struct {
+			Name               string `json:"name"`
+			Active             bool   `json:"active"`
+			APIKey             string `json:"api_key"`
+			BaseURL            string `json:"base_url"`
+			Model              string `json:"model"`
+			Timeout            int    `json:"timeout"`
+			ContextWindowTokens int   `json:"context_window_tokens"`
+		}
+		if err := c.ShouldBindJSON(&req); err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request body"})
+			return
+		}
+
+		if req.Name == "" {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "name is required"})
+			return
+		}
+
+		record := &llmProviderRecord{
+			Name:               req.Name,
+			Active:             req.Active,
+			APIKey:             req.APIKey,
+			BaseURL:            req.BaseURL,
+			Model:              req.Model,
+			Timeout:            req.Timeout,
+			ContextWindowTokens: req.ContextWindowTokens,
+		}
+
+		if err := store.CreateLLMProvider(record); err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		c.JSON(http.StatusOK, gin.H{"status": "ok", "item": llmProviderDTO(*record)})
+	}
+}
+
+func handleUpdateLLMProvider(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		name := c.Param("name")
+		if name == "" {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "provider name is required"})
+			return
+		}
+
+		var req struct {
+			Active             *bool   `json:"active"`
+			APIKey             *string `json:"api_key"`
+			BaseURL            *string `json:"base_url"`
+			Model              *string `json:"model"`
+			Timeout            *int    `json:"timeout"`
+			ContextWindowTokens *int   `json:"context_window_tokens"`
+		}
+		if err := c.ShouldBindJSON(&req); err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request body"})
+			return
+		}
+
+		updates := make(map[string]any)
+		if req.Active != nil {
+			updates["active"] = *req.Active
+		}
+		if req.APIKey != nil {
+			updates["api_key"] = *req.APIKey
+		}
+		if req.BaseURL != nil {
+			updates["base_url"] = *req.BaseURL
+		}
+		if req.Model != nil {
+			updates["model"] = *req.Model
+		}
+		if req.Timeout != nil {
+			updates["timeout"] = *req.Timeout
+		}
+		if req.ContextWindowTokens != nil {
+			updates["context_window_tokens"] = *req.ContextWindowTokens
+		}
+
+		if len(updates) == 0 {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "no fields to update"})
+			return
+		}
+
+		if err := store.UpdateLLMProvider(name, updates); err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				c.JSON(http.StatusNotFound, gin.H{"error": "provider not found"})
+				return
+			}
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		record, err := store.GetLLMProvider(name)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		c.JSON(http.StatusOK, gin.H{"status": "ok", "item": llmProviderDTO(*record)})
+	}
+}
+
+func handleDeleteLLMProvider(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		name := c.Param("name")
+		if name == "" {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "provider name is required"})
+			return
+		}
+
+		if err := store.DeleteLLMProvider(name); err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		c.JSON(http.StatusOK, gin.H{"status": "ok"})
+	}
+}
+
+func llmProviderDTO(row llmProviderRecord) map[string]any {
+	return map[string]any{
+		"name":                 row.Name,
+		"active":               row.Active,
+		"base_url":             row.BaseURL,
+		"model":                row.Model,
+		"timeout":              row.Timeout,
+		"context_window_tokens": row.ContextWindowTokens,
+		"created_at":           row.CreatedAt,
+		"updated_at":           row.UpdatedAt,
+	}
+}
+
+// ============================================
+// LLM Tool Router Handlers
+// ============================================
+
+func handleGetLLMToolRouter(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		record, err := store.GetLLMToolRouter()
+		if err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				c.JSON(http.StatusNotFound, gin.H{"error": "tool router config not found"})
+				return
+			}
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		c.JSON(http.StatusOK, gin.H{"item": llmToolRouterDTO(*record)})
+	}
+}
+
+func handleUpdateLLMToolRouter(store *store) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		record, err := store.GetLLMToolRouter()
+		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+
+		var req struct {
+			Enabled      *bool   `json:"enabled"`
+			OpenAIName   *string `json:"openai_name"`
+			Timeout      *int    `json:"timeout"`
+			MaxTokens    *int    `json:"max_tokens"`
+			SystemPrompt *string `json:"system_prompt"`
+		}
+		if err := c.ShouldBindJSON(&req); err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request body"})
+			return
+		}
+
+		updates := make(map[string]any)
+		if req.Enabled != nil {
+			updates["enabled"] = *req.Enabled
+		}
+		if req.OpenAIName != nil {
+			updates["openai_name"] = *req.OpenAIName
+		}
+		if req.Timeout != nil {
+			updates["timeout"] = *req.Timeout
+		}
+		if req.MaxTokens != nil {
+			updates["max_tokens"] = *req.MaxTokens
+		}
+		if req.SystemPrompt != nil {
+			updates["system_prompt"] = *req.SystemPrompt
+		}
+
+		if len(updates) == 0 {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "no fields to update"})
+			return
+		}
+
+		if record == nil {
+			// 创建新配置
+			newRecord := &llmToolRouterRecord{
+				Enabled:      req.Enabled != nil && *req.Enabled,
+				OpenAIName:   "",
+				Timeout:      30,
+				MaxTokens:    512,
+				SystemPrompt: "",
+			}
+			if req.OpenAIName != nil {
+				newRecord.OpenAIName = *req.OpenAIName
+			}
+			if req.Timeout != nil {
+				newRecord.Timeout = *req.Timeout
+			}
+			if req.MaxTokens != nil {
+				newRecord.MaxTokens = *req.MaxTokens
+			}
+			if req.SystemPrompt != nil {
+				newRecord.SystemPrompt = *req.SystemPrompt
+			}
+			if err := store.CreateLLMToolRouter(newRecord); err != nil {
+				c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+				return
+			}
+			record = newRecord
+		} else {
+			// 更新现有配置
+			if err := store.UpdateLLMToolRouter(record.ID, updates); err != nil {
+				c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+				return
+			}
+			record, err = store.GetLLMToolRouter()
+			if err != nil {
+				c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+				return
+			}
+		}
+
+		c.JSON(http.StatusOK, gin.H{"status": "ok", "item": llmToolRouterDTO(*record)})
+	}
+}
+
+func llmToolRouterDTO(row llmToolRouterRecord) map[string]any {
+	return map[string]any{
+		"id":            row.ID,
+		"enabled":       row.Enabled,
+		"openai_name":   row.OpenAIName,
+		"timeout":       row.Timeout,
+		"max_tokens":    row.MaxTokens,
+		"system_prompt": row.SystemPrompt,
+		"created_at":    row.CreatedAt,
+		"updated_at":    row.UpdatedAt,
+	}
+}