types

type Credentials struct {
	Type   string            `json:"type"`   // apikey, token, oauth, etc.
	Values map[string]string `json:"values"` // key-value pairs
}

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

type DeleteRequest = DeleteModelRequest

type EngineConfig struct {
	Host         string `json:"host"`
	Scheme       string `json:"scheme"` // http/https
	EnginePath   string `json:"engine_path"`
	ExecFile     string `json:"exec_file"`
	ExecPath     string `json:"exec_path"`
	DownloadURL  string `json:"download_url"`
	DownloadPath string `json:"download_path"`
	DeviceType   string `json:"device_type,omitempty"` // GPU type
}

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 EngineStatus struct {
	Name    string `json:"name"`
	Version string `json:"version"`
	Status  string `json:"status"` // running, stopped, installing, error
	Message string `json:"message,omitempty"`
}

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

type HealthStatus struct {
	Status  string            `json:"status"` // UP, DOWN, UNKNOWN
	Message string            `json:"message,omitempty"`
	Details map[string]string `json:"details,omitempty"`
}

type InvokeRequest struct {
	ServiceName string            `json:"service_name"`
	RequestBody []byte            `json:"request_body"`
	Headers     map[string]string `json:"headers,omitempty"`
}

type InvokeResponse struct {
	ResponseBody []byte            `json:"response_body"`
	Headers      map[string]string `json:"headers,omitempty"`
	StatusCode   int               `json:"status_code,omitempty"`
}

type ListModelsResponse struct {
	Models []ModelInfo `json:"models"`
}

type ListResponse = ListModelsResponse

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

type LoadRequest = LoadModelRequest

type LocalTime time.Time

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

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

type OllamaResources struct {
	Executable  string `json:"executable,omitempty" yaml:"executable,omitempty"`
	ModelsDir   string `json:"models_dir,omitempty" yaml:"models_dir,omitempty"`
	DownloadDir string `json:"download_dir,omitempty" yaml:"download_dir,omitempty"`
}

type PlatformConfig struct {
	Executable   string   `json:"executable" yaml:"executable"`
	Dependencies []string `json:"dependencies,omitempty" yaml:"dependencies,omitempty"`
}

type PluginError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Details string `json:"details,omitempty"`
}

type PluginManifest struct {
	Version   string                    `json:"version" yaml:"version"`
	Provider  ProviderInfo              `json:"provider" yaml:"provider"`
	Services  []ServiceDef              `json:"services" yaml:"services"`
	Platforms map[string]PlatformConfig `json:"platforms,omitempty" yaml:"platforms,omitempty"`
	Resources *ResourcesConfig          `json:"resources,omitempty" yaml:"resources,omitempty"`

	// PluginDir is the plugin directory path (injected at runtime, not serialized).
	PluginDir string `json:"-" yaml:"-"`
}

type PluginStatus struct {
	Name    string `json:"name"`
	Version string `json:"version"`
	Status  string `json:"status"` // running, stopped, error
	Message string `json:"message,omitempty"`
}

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

type ProviderInfo struct {
	Name        string `json:"name" yaml:"name"`
	DisplayName string `json:"display_name" yaml:"display_name"`
	Version     string `json:"version" yaml:"version"`
	Type        string `json:"type" yaml:"type"` // local/remote
	Author      string `json:"author,omitempty" yaml:"author,omitempty"`
	Description string `json:"description,omitempty" yaml:"description,omitempty"`
	Homepage    string `json:"homepage,omitempty" yaml:"homepage,omitempty"`
	EngineHost  string `json:"engine_host,omitempty" yaml:"engine_host,omitempty"` // Engine base URL (required for local type)
}

type PullModelRequest struct {
	Model     string `json:"model"`
	Insecure  bool   `json:"insecure,omitempty"`
	Username  string `json:"username,omitempty"`
	Password  string `json:"password,omitempty"`
	Stream    *bool  `json:"stream,omitempty"`
	Name      string `json:"name,omitempty"`       // alias (for backward compatibility)
	ModelType string `json:"model_type,omitempty"` // model type (OpenVINO-specific)
}

type PullProgressFunc func(ProgressResponse) error

type RecommendModelData struct {
	Id              string    `json:"id"`
	Service         string    `json:"service_name"`
	ApiFlavor       string    `json:"api_flavor"`
	Flavor          string    `json:"flavor"`
	Method          string    `json:"method" default:"POST"`
	Desc            string    `json:"desc"`
	Url             string    `json:"url"`
	AuthType        string    `json:"auth_type"`
	AuthApplyUrl    string    `json:"auth_apply_url"`
	AuthFields      []string  `json:"auth_fields"`
	Name            string    `json:"name"`
	ServiceProvider string    `json:"service_provider_name"`
	Size            string    `json:"size"`
	IsRecommended   bool      `json:"is_recommended" default:"false"`
	Status          string    `json:"status"`
	Avatar          string    `json:"avatar"`
	CanSelect       bool      `json:"can_select" default:"false"`
	Class           []string  `json:"class"`
	OllamaId        string    `json:"ollama_id"`
	ParamsSize      float32   `json:"params_size"`
	InputLength     int       `json:"input_length"`
	OutputLength    int       `json:"output_length"`
	Source          string    `json:"source"`
	IsDefault       string    `json:"is_default" default:"false"`
	Think           bool      `json:"think"`
	ThinkSwitch     bool      `json:"think_switch"`
	Tools           bool      `json:"tools"` // 是否支持工具调用
	Context         float32   `json:"context"`
	CreatedAt       LocalTime `json:"created_at"`
}

type ResourcesConfig struct {
	// DataDir is the data root directory, supports environment variable expansion.
	DataDir string `json:"data_dir,omitempty" yaml:"data_dir,omitempty"`

	// Ollama contains Ollama-specific resource configuration (optional, depends on plugin type).
	Ollama *OllamaResources `json:"ollama,omitempty" yaml:"ollama,omitempty"`
}

type ServiceCapabilities struct {
	// SupportStreaming indicates if HTTP protocol supports streaming response.
	SupportStreaming bool `json:"support_streaming,omitempty" yaml:"support_streaming,omitempty"`

	// SupportBidirectional indicates if bidirectional streaming is supported.
	SupportBidirectional bool `json:"support_bidirectional,omitempty" yaml:"support_bidirectional,omitempty"`
}

type ServiceDef struct {
	ServiceName    string   `json:"service_name" yaml:"service_name"`
	TaskType       string   `json:"task_type" yaml:"task_type"`
	Protocol       string   `json:"protocol" yaml:"protocol"`
	ExposeProtocol string   `json:"expose_protocol" yaml:"expose_protocol"`
	Endpoint       string   `json:"endpoint,omitempty" yaml:"endpoint,omitempty"`
	AuthType       string   `json:"auth_type,omitempty" yaml:"auth_type,omitempty"`
	DefaultModel   string   `json:"default_model,omitempty" yaml:"default_model,omitempty"`
	SupportModels  []string `json:"support_models,omitempty" yaml:"support_models,omitempty"`

	// ConfigRef references an existing template configuration, format: "flavor:service"
	ConfigRef string `json:"config_ref,omitempty" yaml:"config_ref,omitempty"`

	// Timeout is the service invocation timeout in seconds.
	// - Not set (0): Use default timeout (60s for unary, 300s for streaming)
	// - Positive value (N > 0): Custom timeout of N seconds
	// - Negative value (N < 0): No timeout limit (recommended for time-consuming services like text-to-image, model downloads)
	// AOG uses context.WithTimeout for positive values and context.WithCancel for negative values.
	Timeout int `json:"timeout,omitempty" yaml:"timeout,omitempty"`

	Middlewares  []string            `json:"middlewares,omitempty" yaml:"middlewares,omitempty"`
	Capabilities ServiceCapabilities `json:"capabilities,omitempty" yaml:"capabilities,omitempty"`
}

type StreamChunk struct {
	Data     []byte            // chunk data
	Error    error             // error information
	IsFinal  bool              // whether this is the final chunk
	Metadata map[string]string // metadata (e.g., content-type)
}

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