types

type ConversionStepDef struct {
	Converter string `yaml:"converter" json:"converter"`
	Config    any    `yaml:"config" json:"config"`
}

type DataMigrateVersion struct {
	ID        int       `gorm:"primaryKey;column:id;autoIncrement" json:"id"`
	Version   string    `gorm:"column:version;not null" json:"version"`
	CreatedAt LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
}

type DeleteRequest struct {
	Model string `json:"model"`
}

type DropAction struct{}

type EngineRecommendConfig struct {
	Host           string `json:"host"`
	Origin         string `json:"origin"`
	Scheme         string `json:"scheme"`
	RecommendModel string `json:"recommend_model"`
	DownloadUrl    string `json:"download_url"`
	DownloadPath   string `json:"download_path"`
	EnginePath     string `json:"engine_path"`
	ExecPath       string `json:"exec_path"`
	ExecFile       string `json:"exec_file"`
	DeviceType     string `json:"device_type"`
}

type EngineVersionResponse struct {
	Version string `json:"version"`
}

type FlavorConversionDef struct {
	Prologue   []string            `yaml:"prologue,omitempty" json:"prologue,omitempty"`
	Epilogue   []string            `yaml:"epilogue,omitempty" json:"epilogue,omitempty"`
	Conversion []ConversionStepDef `yaml:"conversion,omitempty" json:"conversion,omitempty"`
}

type HTTPContent struct {
	Body   []byte
	Header http.Header
}

type HTTPErrorResponse struct {
	StatusCode int
	Header     http.Header
	Body       []byte
}

type ListModelResponse struct {
	Name       string       `json:"name"`
	Model      string       `json:"model"`
	ModifiedAt time.Time    `json:"modified_at"`
	Size       int64        `json:"size"`
	Digest     string       `json:"digest"`
	Details    ModelDetails `json:"details,omitempty"`
}

type ListResponse struct {
	Models []ListModelResponse `json:"models"`
}

type LoadRequest struct {
	Model string `json:"model"`
}

type LocalTime time.Time

type Model struct {
	ID            int       `gorm:"primaryKey;column:id;autoIncrement" json:"id"`
	ModelName     string    `gorm:"column:model_name;not null" json:"model_name"`
	ProviderName  string    `gorm:"column:provider_name" json:"provider_name"`
	Status        string    `gorm:"column:status;not null" json:"status"`
	ServiceName   string    `gorm:"column:service_name" json:"service_name"`
	ServiceSource string    `gorm:"column:service_source" json:"service_source"`
	IsDefault     bool      `gorm:"column:is_default" json:"is_default"`
	CreatedAt     LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt     LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
}

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

type OllamaLoadModelRequest struct {
	Model string `json:"model"`
}

type OllamaUnloadModelRequest struct {
	Model     string `json:"model"`
	KeepAlive int64  `json:"keep_alive"`
}

type PathDiskSizeInfo struct {
	FreeSize  int `json:"free_size"`
	TotalSize int `json:"total_size"`
	UsageSize int `json:"usage_size"`
}

type ProgressResponse struct {
	Status    string `json:"status"`
	Digest    string `json:"digest,omitempty"`
	Total     int64  `json:"total,omitempty"`
	Completed int64  `json:"completed,omitempty"`
}

type PullModelRequest struct {
	Model     string `json:"model"`
	Insecure  bool   `json:"insecure,omitempty"`
	Username  string `json:"username"`
	Password  string `json:"password"`
	Stream    *bool  `json:"stream,omitempty"`
	ModelType string `json:"model_type,omitempty"`

	// Deprecated: set the model name with Model instead
	Name string `json:"name"`
}

type PullProgressFunc func(ProgressResponse) error

type RagChunk struct {
	ID         string                `json:"id"`
	FileID     string                `json:"file_id"`
	Content    string                `json:"content"`
	ChunkIndex int                   `json:"index"`
	Embedding  datastore.Float32List `json:"embedding"`
	CreatedAt  LocalTime             `json:"created_at"`
	UpdatedAt  LocalTime             `json:"updated_at"`
}

type RagFile struct {
	ID         int       `gorm:"primaryKey;column:id;autoIncrement" json:"id"`
	FileName   string    `gorm:"column:file_name" json:"file_name"`
	FileID     string    `gorm:"column:file_id" json:"file_id"`
	FileType   string    `gorm:"column:file_type" json:"file_type"`
	FilePath   string    `gorm:"column:file_path" json:"file_path"`
	Status     int       `gorm:"column:status" json:"status"` // 1-processing | 2-done | 3-failed
	EmbedModel string    `gorm:"column:embed_model" json:"embed_model"`
	CreatedAt  LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt  LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
}

type RagServiceConfig struct {
	ChunkSize            int     `json:"chunk_size"`
	ChunkOverlap         int     `json:"chunk_overlap"`
	EmbeddingDim         int     `json:"embedding_dim"`
	TopK                 int     `json:"top_k"`
	ScoreThreshold       float64 `json:"score_threshold"`
	EmbedModel           string  `json:"embed_model"`
	DuplicationThreshold float64 `json:"duplication_threshold"`
}

type RecommendConfig struct {
	ModelEngine string `json:"model_engine"`
	ModelName   string `json:"model_name"`
}

type ScheduleDetails struct {
	Id           uint64
	IsRunning    bool
	ListMark     *list.Element
	TimeEnqueue  time.Time
	TimeRun      time.Time
	TimeComplete time.Time
}

type Service struct {
	Name           string    `gorm:"primaryKey;column:name" json:"name"`
	HybridPolicy   string    `gorm:"column:hybrid_policy;not null;default:default" json:"hybrid_policy"`
	RemoteProvider string    `gorm:"column:remote_provider;not null;default:''" json:"remote_provider"` // v0.6 deprecated
	LocalProvider  string    `gorm:"column:local_provider;not null;default:''" json:"local_provider"`   // v0.6 deprecated
	Status         int       `gorm:"column:status;not null;default:1" json:"status"`
	CanInstall     int       `gorm:"column:can_install;not null;default:0" json:"can_install"`
	Avatar         string    `gorm:"column:avatar;not null;default:''" json:"avatar"`
	CreatedAt      LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt      LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
}

type ServiceProvider struct {
	ID            int       `gorm:"primaryKey;autoIncrement" json:"id"`
	ProviderName  string    `gorm:"column:provider_name" json:"provider_name"`
	ServiceName   string    `gorm:"column:service_name" json:"service_name"`
	ServiceSource string    `gorm:"column:service_source;default:local" json:"service_source"`
	Desc          string    `gorm:"column:desc" json:"desc"`
	Method        string    `gorm:"column:method" json:"method"`
	URL           string    `gorm:"column:url" json:"url"`
	AuthType      string    `gorm:"column:auth_type" json:"auth_type"`
	AuthKey       string    `gorm:"column:auth_key" json:"auth_key"`
	Flavor        string    `gorm:"column:flavor" json:"flavor"`
	Protocol      string    `gorm:"column:protocol;default:HTTP" json:"protocol"`
	ExtraHeaders  string    `gorm:"column:extra_headers;default:'{}'" json:"extra_headers"`
	ExtraJSONBody string    `gorm:"column:extra_json_body;default:'{}'" json:"extra_json_body"`
	Properties    string    `gorm:"column:properties;default:'{}'" json:"properties"`
	Status        int       `gorm:"column:status;not null;default:0" json:"status"`
	Scope         string    `gorm:"column:scope;default:system" json:"scope"` // 'system' or 'custom'
	CreatedAt     LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt     LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
}

type ServiceProviderProperties struct {
	MaxInputTokens        int      `json:"max_input_tokens"`
	SupportedResponseMode []string `json:"supported_response_mode"`
	ModeIsChangeable      bool     `json:"mode_is_changeable"`
	Models                []string `json:"models"`
	XPU                   []string `json:"xpu"`
}

type ServiceRequest struct {
	AskStreamMode         bool          `json:"stream"`
	Model                 string        `json:"model"`
	HybridPolicy          string        `json:"hybrid_policy"`
	RemoteServiceProvider string        `json:"remote_service_provider"`
	FromFlavor            string        `json:"-"`
	Service               string        `json:"-"`
	Priority              int           `json:"-"`
	RequestSegments       int           `json:"request_segments"`
	RequestExtraUrl       string        `json:"extra_url"`
	HTTP                  HTTPContent   `json:"-"`
	OriginalRequest       *http.Request `json:"-"`
	WebSocketConnID       string        `json:"-"` // WebSocket connection ID, used to associate multiple messages on the same connection
	Think                 bool          `json:"think"`
}

type ServiceResult struct {
	Type       ServiceResultType
	TaskId     uint64
	Error      error
	StatusCode int
	HTTP       HTTPContent
}

type ServiceResultType int

type ServiceTarget struct {
	Location        string
	Stream          bool
	Model           string
	ToFavor         string
	XPU             string
	Protocol        string // Service protocol type, such as GRPC_STREAM
	ExposeProtocol  string
	ServiceProvider *ServiceProvider
}

type SpeechToTextParams struct {
	// 基本参数
	TaskID      string `json:"task_id,omitempty"` // 任务ID
	Service     string `json:"service,omitempty"`
	Model       string `json:"model,omitempty"`        // 模型名称
	Language    string `json:"language,omitempty"`     // 语言，如"zh"、"en"
	AudioFormat string `json:"audio_format,omitempty"` // 音频格式
	SampleRate  int    `json:"sample_rate,omitempty"`  // 采样率

	// 高级参数
	UseVAD       bool   `json:"use_vad,omitempty"`       // 是否使用VAD
	ReturnFormat string `json:"return_format,omitempty"` // 返回格式
	MaxDuration  int    `json:"max_duration,omitempty"`  // 最大处理时长(秒)

	// 状态参数
	Action          string `json:"action,omitempty"`            // 当前任务状态 run-task  finish-task
	TotalAudioBytes int    `json:"total_audio_bytes,omitempty"` // 总音频字节数
	LastAudioTime   int64  `json:"last_audio_time,omitempty"`   // 最后一次接收音频的时间
}

type StreamMode struct {
	Mode   StreamModeType
	Header http.Header
}

type StreamModeType int

type SupportModel struct {
	Id            string    `json:"id"`
	OllamaId      string    `json:"Ollama_id"`
	Name          string    `json:"name"`
	Avatar        string    `json:"avatar"`
	Description   string    `json:"description"`
	Class         []string  `json:"class"`
	Flavor        string    `json:"flavor"`
	ApiFlavor     string    `json:"api_flavor"`
	Size          string    `json:"size"`
	ParamSize     float32   `json:"params_size"`
	MaxInput      int       `json:"max_input"`
	MaxOutput     int       `json:"max_output"`
	InputLength   int       `json:"input_length"`
	OutputLength  int       `json:"output_length"`
	ServiceSource string    `json:"service_source"`
	ServiceName   string    `json:"service_name"`
	CreatedAt     LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt     LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
	Think         bool      `json:"think"`
	ThinkSwitch   bool      `json:"think_switch"`
	Tools         bool      `json:"tools"` // 是否支持工具调用
	Context       float32   `json:"context"`
}

type TextToSpeechRequest struct {
	Text   string `json:"text"`
	Voice  string `json:"voice"`
	Params string `json:"params"`
}

type UnloadModelRequest struct {
	Models []string `json:"model"`
}

type VersionUpdateRecord struct {
	ID           int       `gorm:"primaryKey;column:id;autoIncrement" json:"id"`
	Version      string    `gorm:"column:version;not null" json:"version"`
	ReleaseNotes string    `gorm:"column:release_notes;not null" json:"release_notes"`
	Status       int       `gorm:"column:status;not null" json:"status"`
	CreatedAt    LocalTime `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
	UpdatedAt    LocalTime `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
}

type WebSocketActionMessage struct {
	Task       string               `json:"task"`                 // 任务类型，如"speech-to-text-ws"
	Action     string               `json:"action"`               // 动作类型，如"run-task"或"finish-task"
	TaskID     string               `json:"task_id,omitempty"`    // 任务ID，由服务端生成并返回，用于后续消息关联
	Model      string               `json:"model,omitempty"`      // 模型名称
	Parameters *WebSocketParameters `json:"parameters,omitempty"` // 可选参数
}

type WebSocketEventHeader struct {
	Action       string `json:"action"`
	Streaming    string `json:"streaming"`
	TaskID       string `json:"task_id"`                 // 任务ID
	Event        string `json:"event"`                   // 事件类型
	ErrorCode    string `json:"error_code,omitempty"`    // 错误码，仅在task-failed事件中出现
	ErrorMessage string `json:"error_message,omitempty"` // 错误信息，仅在task-failed事件中出现
}

type WebSocketEventMessage struct {
	Header  WebSocketEventHeader `json:"header"`            // 事件头部
	Payload interface{}          `json:"payload,omitempty"` // 根据事件类型不同，包含不同的数据
}

type WebSocketFinishTaskAction struct {
	WebSocketActionMessage
}

type WebSocketParameters struct {
	Service      string `json:"service,omitempty"`       // 服务名称，如: "speech-to-text" "speech-to-text-ws"
	Format       string `json:"format,omitempty"`        // 音频格式: pcm/wav/mp3
	SampleRate   int    `json:"sample_rate,omitempty"`   // 采样率，通常为16000
	Language     string `json:"language,omitempty"`      // 语言，如"zh"、"en"
	UseVAD       bool   `json:"use_vad,omitempty"`       // 是否使用VAD
	ReturnFormat string `json:"return_format,omitempty"` // 返回格式，如"text"、"json"、"srt"
}

type WebSocketRecognitionOutput struct {
	Sentence WebSocketSentence `json:"sentence"` // 识别的句子
}

type WebSocketResultEvent struct {
	Header  WebSocketEventHeader   `json:"header"`
	Payload WebSocketResultPayload `json:"payload"`
}

type WebSocketResultPayload struct {
	Output WebSocketRecognitionOutput `json:"output"` // 识别输出
}

type WebSocketRunTaskAction struct {
	WebSocketActionMessage
}

type WebSocketSentence struct {
	BeginTime *int   `json:"begin_time"` // 开始时间，毫秒
	EndTime   *int   `json:"end_time"`   // 结束时间，毫秒，可能为null
	Text      string `json:"text"`       // 识别的文本
}