123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
package controllers
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"lokistack/models"
|
|
"lokistack/services"
|
|
"lokistack/utils"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// AIController AIOps：模型配置 + 日志分析 + 审计历史
|
|
type AIController struct{}
|
|
|
|
func NewAIController() *AIController { return &AIController{} }
|
|
|
|
// GetConfig 返回模型配置（APIKey 掩码）+ 今日已用 token。
|
|
func (ctrl *AIController) GetConfig(c *gin.Context) {
|
|
userID := c.GetUint("userId")
|
|
cfg := services.MaskedAIConfig()
|
|
utils.Success(c, gin.H{
|
|
"config": cfg,
|
|
"tokenUsedToday": services.AITokenUsedToday(userID),
|
|
})
|
|
}
|
|
|
|
// SaveConfig 保存模型配置。apiKey 传空或掩码 = 沿用旧值。
|
|
func (ctrl *AIController) SaveConfig(c *gin.Context) {
|
|
var req struct {
|
|
models.AIConfig
|
|
APIKey string `json:"apiKey"`
|
|
}
|
|
if !utils.BindAndValidate(c, &req) {
|
|
return
|
|
}
|
|
if err := services.SaveAIConfig(req.AIConfig, req.APIKey); err != nil {
|
|
utils.Error(c, 400, err.Error())
|
|
return
|
|
}
|
|
utils.Success(c, gin.H{"config": services.MaskedAIConfig()})
|
|
}
|
|
|
|
// TestConfig 连通性测试：用已保存（或本次提交）的配置发一句最小请求。
|
|
func (ctrl *AIController) TestConfig(c *gin.Context) {
|
|
var req struct {
|
|
APIKey string `json:"apiKey"`
|
|
BaseURL string `json:"baseUrl"`
|
|
Model string `json:"model"`
|
|
}
|
|
_ = c.ShouldBindJSON(&req)
|
|
|
|
cfg := services.GetAIConfig()
|
|
if req.BaseURL != "" {
|
|
cfg.BaseURL = req.BaseURL
|
|
}
|
|
if req.Model != "" {
|
|
cfg.Model = req.Model
|
|
}
|
|
// key 解析:传新值才覆盖;掩码/空一律沿用已保存的真实 key(GetAIConfig 返回的是解密后的真值)
|
|
if req.APIKey != "" && !containsMask(req.APIKey) {
|
|
cfg.APIKey = req.APIKey
|
|
}
|
|
|
|
result, err := services.TestAIConnection(cfg)
|
|
if err != nil {
|
|
utils.Error(c, 400, err.Error())
|
|
return
|
|
}
|
|
utils.Success(c, result)
|
|
}
|
|
|
|
func containsMask(s string) bool {
|
|
for i := 0; i+2 < len(s); i++ {
|
|
if s[i] == '*' && s[i+1] == '*' && s[i+2] == '*' {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// ListRuns 审计历史（分页）。
|
|
func (ctrl *AIController) ListRuns(c *gin.Context) {
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "20"))
|
|
runs, total := services.ListAIRuns(page, pageSize)
|
|
utils.Success(c, gin.H{"list": runs, "total": total, "page": page, "pageSize": pageSize})
|
|
}
|
|
|
|
// GetRun 单条审计详情（含脱敏后的完整 payload）。
|
|
func (ctrl *AIController) GetRun(c *gin.Context) {
|
|
id, err := utils.IntParam(c, "id")
|
|
if err != nil {
|
|
utils.BadRequest(c, "ID 无效")
|
|
return
|
|
}
|
|
run := services.GetAIRun(uint(id))
|
|
if run == nil {
|
|
utils.Error(c, 404, "记录不存在")
|
|
return
|
|
}
|
|
utils.Success(c, run)
|
|
}
|
|
|
|
// Chat 全站 AI 助手浮窗(多轮对话 + 页面上下文注入;Pro 闸门)
|
|
func (ctrl *AIController) Chat(c *gin.Context) {
|
|
var req struct {
|
|
History []services.ChatMessage `json:"history" binding:"required,min=1,max=40"`
|
|
Page string `json:"page"`
|
|
Context string `json:"context"`
|
|
}
|
|
if !utils.BindAndValidate(c, &req) {
|
|
return
|
|
}
|
|
cfg := services.GetAIConfig()
|
|
reply, err := services.RunAIChat(c.Request.Context(), cfg, c.GetUint("userId"), req.History, req.Page, req.Context)
|
|
if err != nil {
|
|
utils.Error(c, 400, err.Error())
|
|
return
|
|
}
|
|
utils.Success(c, gin.H{"reply": reply})
|
|
}
|