models

type DeepseekOptions struct {
	// Deepseek API 密钥
	APIKey string `json:"apiKey"`
}

type ListOllamaTagsResponse struct {
	Models []OllamaModel `json:"models,omitempty"`
}

type ListOpenAICompatibleModelsResponse struct {
	Data []OpenAICompatibleModel `json:"data,omitempty"`
}

type ModelProvider struct {
	Ollama           *OllamaOptions           `json:"ollama,omitempty"`
	Deepseek         *DeepseekOptions         `json:"deepseek,omitempty"`
	OpenAICompatible *OpenAICompatibleOptions `json:"openaiCompatible,omitempty"`
}

type OllamaModel struct {
	Name  string `json:"name"`
	Model string `json:"model"`

	ModifiedAt time.Time          `json:"modified_at,omitempty"`
	Size       int64              `json:"size,omitempty"`
	Digest     string             `json:"digest,omitempty"`
	Details    OllamaModelDetails `json:"details,omitempty"`
}

type OllamaModelDetails struct {
	ParentModel       string   `json:"parent_model,omitempty"`
	Format            string   `json:"format,omitempty"`
	Family            string   `json:"family,omitempty"`
	Families          []string `json:"families,omitempty"`
	ParameterSize     string   `json:"parameter_size,omitempty"`
	QuantizationLevel string   `json:"quantization_level,omitempty"`
}

type OllamaOptions struct {
	// Ollama 服务端地址
	//
	// 默认 http://localhost:11434
	ServerAddress string `json:"serverAddress,omitempty"`
	// 模型响应超时时间，秒
	//
	// 默认 300
	Timeout int `json:"timeout,omitempty"`
	// 模型名列表
	//
	// 空表示使用 Ollama 已下载的所有模型
	Models []string `json:"models,omitempty"`
}

type OpenAICompatibleModel struct {
	ID string `json:"id"`
}

type OpenAICompatibleOptions struct {
	Name    string `json:"name"`
	BaseURL string `json:"baseURL"`
	APIKey  string `json:"apiKey"`
}