feat: 增加管理员 AI 模型配置 (#200)

2026-08-14 09:49:11 +08:00
parent ad44f83ea6
commit 018da3f732
24 changed files with 1546 additions and 8 deletions
@@ -0,0 +1,104 @@
+package web
+
+import (
+	"net/http"
+	"net/url"
+	"strconv"
+	"time"
+
+	"github.com/gin-gonic/gin"
+
+	"cmautobuy/admin/service"
+)
+
+func (h *Handler) AIConfigList(c *gin.Context) {
+	result, err := service.ListAIProviderConfigs(h.db, currentUser(c), h.aiSecrets)
+	if err != nil {
+		fail(c, http.StatusInternalServerError, "读取 AI 模型配置失败，配置没有被改动。")
+		return
+	}
+	c.HTML(http.StatusOK, "ai/list", page(c, "ai-settings", "AI 模型配置", gin.H{
+		"Rows": result.Items, "SecretStoreError": result.SecretStoreError,
+		"Message": c.Query("msg"), "Error": c.Query("error"),
+	}))
+}
+
+func (h *Handler) AIConfigSave(c *gin.Context) {
+	timeoutSeconds, _ := strconv.Atoi(c.PostForm("timeout_seconds"))
+	maxConcurrency, _ := strconv.Atoi(c.PostForm("max_concurrency"))
+	threshold, err := service.ParseConfidenceThresholdBPS(c.PostForm("confidence_threshold"))
+	if err != nil {
+		redirectAIConfig(c, "", err.Error())
+		return
+	}
+	_, err = service.SaveAIProviderConfig(h.db, currentUser(c), service.AIProviderInput{
+		ProviderID: c.PostForm("provider_id"), Name: c.PostForm("name"), BaseURL: c.PostForm("base_url"),
+		Model: c.PostForm("model"), TimeoutSeconds: timeoutSeconds, MaxConcurrency: maxConcurrency,
+		ConfidenceThresholdBPS: threshold,
+	}, h.aiPolicy, time.Now())
+	if err != nil {
+		if service.IsValidationError(err) {
+			redirectAIConfig(c, "", err.Error())
+			return
+		}
+		fail(c, http.StatusInternalServerError, "保存 AI 服务商失败，原有效配置保持不变。")
+		return
+	}
+	redirectAIConfig(c, "AI 服务商配置已保存；请保存密钥并测试连接后再启用", "")
+}
+
+func (h *Handler) AIConfigSecretSave(c *gin.Context) {
+	if err := service.SetAIProviderSecret(h.db, currentUser(c), h.aiSecrets, c.PostForm("provider_id"), c.PostForm("api_key"), time.Now()); err != nil {
+		redirectAIConfig(c, "", err.Error())
+		return
+	}
+	redirectAIConfig(c, "API Key 已安全替换；当前服务商已停用，请重新测试后启用", "")
+}
+
+func (h *Handler) AIConfigSecretClear(c *gin.Context) {
+	if err := service.ClearAIProviderSecret(h.db, currentUser(c), h.aiSecrets, c.PostForm("provider_id"), time.Now()); err != nil {
+		redirectAIConfig(c, "", err.Error())
+		return
+	}
+	redirectAIConfig(c, "API Key 已清除，服务商已停用", "")
+}
+
+func (h *Handler) AIConfigTest(c *gin.Context) {
+	if err := service.TestAIProviderConnection(c.Request.Context(), h.db, currentUser(c), h.aiSecrets,
+		c.PostForm("provider_id"), h.aiPolicy, nil, time.Now()); err != nil {
+		redirectAIConfig(c, "", err.Error())
+		return
+	}
+	redirectAIConfig(c, "连接测试成功；现在可以启用这个服务商", "")
+}
+
+func (h *Handler) AIConfigEnable(c *gin.Context) {
+	if err := service.EnableAIProviderConfig(h.db, currentUser(c), h.aiSecrets, c.PostForm("provider_id"), time.Now()); err != nil {
+		redirectAIConfig(c, "", err.Error())
+		return
+	}
+	redirectAIConfig(c, "AI 服务商已启用，新批次将使用这份配置", "")
+}
+
+func (h *Handler) AIConfigDisable(c *gin.Context) {
+	if err := service.DisableAIProviderConfig(h.db, currentUser(c), c.PostForm("provider_id"), time.Now()); err != nil {
+		redirectAIConfig(c, "", err.Error())
+		return
+	}
+	redirectAIConfig(c, "AI 服务商已停用，不会影响已经启动的批次", "")
+}
+
+func redirectAIConfig(c *gin.Context, message, errorMessage string) {
+	values := url.Values{}
+	if message != "" {
+		values.Set("msg", message)
+	}
+	if errorMessage != "" {
+		values.Set("error", errorMessage)
+	}
+	target := "/settings/ai"
+	if encoded := values.Encode(); encoded != "" {
+		target += "?" + encoded
+	}
+	c.Redirect(http.StatusSeeOther, target)
+}