package models

import "time"

// LLMConfig 大语言模型配置（单行配置，类似 LDAPConfig） // Logan.Li
// 支持 OpenAI 兼容接口（OpenAI / DeepSeek / 通义千问 / Moonshot / Ollama / 自定义）
type LLMConfig struct {
	ID           uint      `gorm:"primarykey" json:"id"`
	Provider     string    `gorm:"size:50;not null;default:openai" json:"provider"`      // openai/deepseek/qwen/moonshot/ollama/custom
	Model        string    `gorm:"size:100;not null;default:gpt-4o-mini" json:"model"`    // 模型名，如 gpt-4o-mini / deepseek-chat
	APIKey       string    `gorm:"size:500" json:"api_key"`                               // API Key（GET 时会被脱敏）
	Endpoint     string    `gorm:"size:500" json:"endpoint"`                              // 基础地址，如 https://api.openai.com/v1
	Temperature  float64   `gorm:"default:0.7" json:"temperature"`                        // 采样温度 0-2
	MaxTokens    int       `gorm:"default:2000" json:"max_tokens"`                        // 最大输出 token
	SystemPrompt string    `gorm:"type:text" json:"system_prompt"`                        // 系统提示词（AIOps 运维助手角色）
	Enabled      bool      `gorm:"default:false" json:"enabled"`                          // 是否启用
	CreatedAt    time.Time `json:"created_at"`
	UpdatedAt    time.Time `json:"updated_at"`
}

// TableName 表名 // Logan.Li
func (LLMConfig) TableName() string { return "llm_configs" }

// SaveLLMConfigRequest 保存 LLM 配置请求 // Logan.Li
type SaveLLMConfigRequest struct {
	Provider     string  `json:"provider"`
	Model        string  `json:"model"`
	APIKey       string  `json:"api_key"`        // 若为脱敏值（含 ****）则后端保留原值
	Endpoint     string  `json:"endpoint"`
	Temperature  float64 `json:"temperature"`
	MaxTokens    int     `json:"max_tokens"`
	SystemPrompt string  `json:"system_prompt"`
	Enabled      bool    `json:"enabled"`
}

// ChatMessage 对话消息 // Logan.Li
type ChatMessage struct {
	Role    string `json:"role"`    // system / user / assistant
	Content string `json:"content"`
}

// ChatRequest AI 对话请求 // Logan.Li
type ChatRequest struct {
	Messages []ChatMessage `json:"messages" binding:"required,min=1,dive"`
}

// ChatResponse AI 对话响应 // Logan.Li
type ChatResponse struct {
	Content    string `json:"content"`
	Model      string `json:"model"`
	TokensUsed int    `json:"tokens_used"`
}