client

type BaseProvider struct {
	log.Logger

	Client *http.Client
}

type Chat struct {
	ID         string    `json:"id"`
	CreateTime time.Time `json:"create_time"`
	UpdateTime time.Time `json:"update_time"`
	Messages   []*Message
}

type ChatRepo interface {
	ListChatsAsync(
		ctx context.Context,
		keyword, model, provider *string,
		limit int,
	) <-chan OpResp
	GetChatAsync(ctx context.Context, chatID string) <-chan OpResp
	AddChatAsync(ctx context.Context, chat *Chat) <-chan OpResp
	UpdateChatAsync(ctx context.Context, chat *Chat) <-chan OpResp
	DeleteChatAsync(ctx context.Context, chatID string) <-chan OpResp

	// Sync versions for convenience
	ListChats(ctx context.Context, keyword, model, provider *string, limit int) ([]*Chat, error)
	Chat(ctx context.Context, chatID string) (*Chat, error)
	AddChat(ctx context.Context, chat *Chat) (*Chat, error)
	UpdateChat(ctx context.Context, chat *Chat) (*Chat, error)
	DeleteChat(ctx context.Context, chatID string) (bool, error)

	Close() error
}

type ChatSvr struct {
	log.Logger
	// contains filtered or unexported fields
}

type Config struct {

	// Model Provider
	Provider      string `mapstructure:"provider"`
	BaseURL       string `mapstructure:"base_url"`
	CustomAPIPath string `mapstructure:"custom_api_path"`
	Model         string `mapstructure:"model"`
	APIKey        string `mapstructure:"api_key"`

	StorageType string `mapstructure:"storage_type,omitempty"`
	// MCP
	MCPSvrPath string `mapstructure:"mcp_server_path"` // MCP Server 配置文件路径
	// Prompt
	PromptPath string `mapstructure:"prompt_path"` // Prompt 配置文件路径

	MaxTurns        uint   `mapstructure:"max_turns"`        // 最多调用 MCP Server 的次数
	MaxTokens       uint64 `mapstructure:"max_tokens"`       // 最大 token 数
	ReasoningEffort string `mapstructure:"reasoning_effort"` // 推理努力度 => high | medium | low | minimal
	Stream          bool   `mapstructure:"stream"`           // 是否使用流式输出
	// contains filtered or unexported fields
}

type ContentPart struct {
	Text string `json:"text"`
	Type string `json:"type"`

	CacheControl map[string]any `json:"cache_control,omitempty"` // Claude-3
}

type FileRepo struct {
	// contains filtered or unexported fields
}

type LLMStreamRet struct {
	Err  error // First
	Done bool  // Second

	ID    string
	Model string

	Content  string
	StreamCh <-chan StreamChunk
}

type ListChatsOption struct {
	// contains filtered or unexported fields
}

type MCPConfigSvr struct {
	log.Logger
	// contains filtered or unexported fields
}

type MCPSvrConfigFileRepo struct {
	log.Logger
	// contains filtered or unexported fields
}

type MCPSvrConfigRepo interface {
	MCPServerConfigByName(name string) (*MCPSvrItem, error)
	AllMCPServerConfigs() []*MCPSvrItem
	UpdateMCPServerConfigByName(item *MCPSvrItem) error
	DeleteMCPServerConfigByName(name string) error
}

type MCPSvrItem struct {
	Name     string `json:"name"`
	Type     string `json:"type"` // "stdio", "sse", "streamableHttp"
	IsActive bool   `json:"isActive"`

	Description string `json:"description,omitempty"` // Description of the MCP Server

	BaseURL string `json:"baseUrl,omitempty"` // The URL endpoint for SSE / StreamableHttp server connection

	//nolint:lll
	Command string   `json:"command,omitempty"` // The command to execute the server (e.g., 'node', 'python') - used for stdio
	Args    []string `json:"args,omitempty"`    // Command line arguments for the server - used for stdio

	//nolint:lll
	AutoConfirm []string `json:"autoConfirm,omitempty"` // List of tool names that should be auto-confirmed without user prompt
}

type MCPSvrManager struct {
	log.Logger
	// contains filtered or unexported fields
}

type MCPToolUse struct {
	ServerName string
	ToolsName  string
	Arguments  map[string]any
}

type Manager struct {
	log.Logger

	MCPMgr *MCPSvrManager
	// contains filtered or unexported fields
}

type Message struct {
	Role    string `json:"role"`
	Content any    `json:"content"` // string or []ContentPart

	Timestamp     *time.Time `json:"timestamp,omitempty"`
	UnixTimestamp int64      `json:"unix_timestamp,omitempty"`

	ReasoningContent string         `json:"reasoning_content,omitempty"`
	ReasoningEffort  string         `json:"reasoning_effort,omitempty"`
	Links            []string       `json:"links,omitempty"`
	Images           []string       `json:"images,omitempty"`
	Model            string         `json:"model,omitempty"`
	Provider         string         `json:"provider,omitempty"`
	ID               string         `json:"id,omitempty"`
	ParentID         string         `json:"parent_id,omitempty"`
	Server           string         `json:"server,omitempty"`
	Tool             string         `json:"tool,omitempty"`
	Arguments        map[string]any `json:"arguments,omitempty"`
}

type MessageOption struct {
	ReasoningContent string
	ReasoningEffort  string

	Links  []string
	Images []string

	Provider  string
	Model     string
	ID        string
	ParentID  string
	Server    string
	Tool      string
	Arguments map[string]any
}

type OllamaChatRequest struct {
	Model    string           `json:"model"`
	Messages []map[string]any `json:"messages"`
	Stream   bool             `json:"stream"`
}

type OllamaFormatProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type OllamaStreamResponse struct {
	Model     string    `json:"model"`      // 本次请求所使用的模型
	CreatedAt time.Time `json:"created_at"` // 响应创建的 UTC 时间戳 2025-08-28T03:42:30.559748Z
	Message   Message   `json:"message"`    // 包含模型生成内容的对象
	Done      bool      `json:"done"`       // 用于指示生成过程是否已完成

	DoneReason         string `json:"done_reason"`          // 诊断字段描述了 Ollama 模型生成任务的最终状态: stop(完整回答) / length(回答被截断)
	TotalDuration      int64  `json:"total_duration"`       // 整个请求所花费的总时间(ns)，从收到请求到生成完最后一个 token
	LoadDuration       int64  `json:"load_duration"`        // 加载模型到内存所花费的时间(ns)
	PromptEvalCount    int    `json:"prompt_eval_count"`    // 在处理用户输入（Prompt）时，模型评估（处理）的 token 数量
	PromptEvalDuration int64  `json:"prompt_eval_duration"` // 处理用户输入（Prompt）所花费的时间(ns), 可理解为 `模型理解问题` 花费的时间
	EvalCount          int    `json:"eval_count"`           // 模型生成回答时，总共评估（生成）的 token 数量 => 代表了模型输出的长度
	EvalDuration       int64  `json:"eval_duration"`        // 生成所有回答 token 所花费的总时间(ns), 模型“思考并写出答案”所用的时间
}

type OpResp struct {
	Data  any
	Error error
}

type OpenAIChatRequest struct {
	Model    string           `json:"model"`
	Messages []map[string]any `json:"messages"`
	Stream   bool             `json:"stream"`

	IncludeReasoning bool `json:"include_reasoning,omitempty"` // deepseek-r1
	Thinking         bool `json:"thinking,omitempty"`          // deepseekv3.1

	ReasoningEffort string `json:"reasoning_effort,omitempty"`
	MaxTokens       uint64 `json:"max_tokens,omitempty"`
}

type OpenAIFormatProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type OpenAIStreamChoice struct {
	Index        int                      `json:"index"`
	Delta        *OpenAIStreamChoiceDelta `json:"delta"`
	FinishReason string                   `json:"finish_reason"` // 在最后一个数据块出现
}

type OpenAIStreamChoiceDelta struct {
	Content          string `json:"content"`
	ReasoningContent string `json:"reasoning_content"` // DeepSeek-R1 模型的推理内容
	Role             string `json:"role"`              // 通常只在第一个数据块出现
}

type OpenAIStreamResponse struct {
	ID                string                `json:"id"`
	Object            string                `json:"object"`
	Created           int64                 `json:"created"`
	Model             string                `json:"model"`
	SystemFingerprint string                `json:"system_fingerprint"`
	Choices           []*OpenAIStreamChoice `json:"choices"`
}

type PromptFileRepo struct {
	log.Logger
	// contains filtered or unexported fields
}

type PromptItem struct {
	Name        string `mapstructure:"name"`                  // Unique identifier for the prompt
	Content     string `mapstructure:"content"`               // The content of the prompt
	Description string `mapstructure:"description,omitempty"` // Optional description of the prompt's purpose
}

type PromptRepo interface {
	PromptByName(name string) (*PromptItem, error)
	AllPrompts() []*PromptItem
	UpdatePromptByName(item *PromptItem) error
	DeletePromptByName(name string) error
}

type PromptSvr struct {
	log.Logger
	// contains filtered or unexported fields
}

type Provider interface {
	CallStreamableChatCompletions(
		messages []*Message,
		prompt *string,
	) *Message

	BuildRequest(
		_ context.Context,
		_ chan StreamChunk,
		_ []*Message,
		_ *string,
	) (*http.Request, error)
}

type StreamChunk struct {
	ID string // 每一条工具调用请求都有一个唯一的 ID => 在返回结果时，必须将这个 ID 附上，以便模型能够准确地将返回的结果与它当初的请求对应起来
	// Provider string
	Model string

	Content string // The content of the chunk
	Done    bool   // Whether the stream is done
	Error   error  // Any error that occurred
}

type TaijiChatRequest struct {
	QueryID     string     `json:"query_id"`
	Model       string     `json:"model"`
	Messages    []*Message `json:"messages"`
	Temperature string     `json:"temperature"` // 调节概率值，取值区间为 (0.0, 2.0]，默认为 1.0
	TopP        float64    `json:"top_p"`       // 采样累积概率的阈值, 取值区间为 [0.0, 1.0]，默认值 1.0
	MaxTokens   uint64     `json:"max_tokens"`
	Stream      bool       `json:"stream"`

	Thinking bool `json:"thinking,omitempty"` // DeepSeek-V3_1
}

type TaijiProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}