config

type APIKey struct {
	Key   string `yaml:"key"`
	Scope string `yaml:"scope"`
}

type Backend struct {
	Name    string `yaml:"name"`
	BaseURL string `yaml:"base_url"`
	APIKey  string `yaml:"api_key"`
}

type ChatModel struct {
	Name      string `yaml:"name"`
	Backend   string `yaml:"backend"`
	Model     string `yaml:"model"`
	MaxTokens int    `yaml:"max_tokens"`
	// ExtraBody is merged into the /v1/chat/completions request body.
	// Needed e.g. to disable reasoning on thinking-first models
	// (llama.cpp/vLLM: chat_template_kwargs: {enable_thinking: false}),
	// whose reasoning otherwise consumes max_tokens and returns empty
	// content.
	ExtraBody map[string]any `yaml:"extra_body"`
}

type Chunking struct {
	Strategy      string `yaml:"strategy"`
	MaxTokens     int    `yaml:"max_tokens"`
	OverlapTokens int    `yaml:"overlap_tokens"`
}

type Config struct {
	Server         Server          `yaml:"server"`
	Storage        Storage         `yaml:"storage"`
	Inference      Inference       `yaml:"inference"`
	Defaults       Defaults        `yaml:"defaults"`
	KnowledgeBases []KnowledgeBase `yaml:"knowledge_bases"`
}

type Contextual struct {
	Enabled bool   `yaml:"enabled"`
	Model   string `yaml:"model"` // ref into inference.chat_models
}

type Defaults struct {
	Chunking       Chunking `yaml:"chunking"`
	EmbeddingModel string   `yaml:"embedding_model"`
	RerankModel    string   `yaml:"rerank_model"`
	OCR            *OCR     `yaml:"ocr"`
}

type EmbeddingModel struct {
	Name       string `yaml:"name"`
	Backend    string `yaml:"backend"`
	Model      string `yaml:"model"`
	Dimensions int    `yaml:"dimensions"`
	BatchSize  int    `yaml:"batch_size"`
	MaxRetries int    `yaml:"max_retries"`
}

type Inference struct {
	Backends        []Backend        `yaml:"backends"`
	EmbeddingModels []EmbeddingModel `yaml:"embedding_models"`
	RerankModels    []RerankModel    `yaml:"rerank_models"`
	ChatModels      []ChatModel      `yaml:"chat_models"`
}

type KnowledgeBase struct {
	Name           string    `yaml:"name"`
	Description    string    `yaml:"description"`
	Writable       bool      `yaml:"writable"`
	Sources        []Source  `yaml:"sources"`
	Chunking       *Chunking `yaml:"chunking"`
	EmbeddingModel string    `yaml:"embedding_model"`
	RerankModel    string    `yaml:"rerank_model"`
	// RerankDefault decides whether searches rerank when no per-request
	// override is given: "on" (default) or "off". With "off" the reranker
	// stays available to requests that ask for rerank: true.
	RerankDefault string      `yaml:"rerank_default"`
	Contextual    *Contextual `yaml:"contextual"`
	OCR           *OCR        `yaml:"ocr"`
}

type OCR struct {
	Enabled bool   `yaml:"enabled"`
	Model   string `yaml:"model"` // ref into inference.chat_models (must be vision-capable)
}

type RerankModel struct {
	Name    string `yaml:"name"`
	Backend string `yaml:"backend"`
	Model   string `yaml:"model"`
}

type Server struct {
	Bind    string   `yaml:"bind"`
	DataDir string   `yaml:"data_dir"`
	APIKeys []APIKey `yaml:"api_keys"`
	// OpsRequireAuth gates /metrics and the per-KB detail of /readyz behind a
	// key. Nil means "follow api_keys": on when keys are configured, off when
	// they are not. /healthz is always open.
	OpsRequireAuth *bool  `yaml:"ops_require_auth"`
	LogLevel       string `yaml:"log_level"`
	LogFormat      string `yaml:"log_format"`
}

type Source struct {
	Name string `yaml:"name"`
	Type string `yaml:"type"` // localdir | git | s3

	// localdir
	Path       string   `yaml:"path"`
	Include    []string `yaml:"include"`
	Exclude    []string `yaml:"exclude"`
	Watch      bool     `yaml:"watch"`
	DebounceMS int      `yaml:"debounce_ms"`

	// git
	URL          string        `yaml:"url"`
	Branch       string        `yaml:"branch"`
	Paths        []string      `yaml:"paths"` // also used by s3
	PollInterval time.Duration `yaml:"poll_interval"`
	TokenEnv     string        `yaml:"token_env"`

	// s3
	Bucket    string `yaml:"bucket"`
	Prefix    string `yaml:"prefix"`
	Region    string `yaml:"region"`
	Endpoint  string `yaml:"endpoint"`   // custom endpoint for MinIO/compatible
	PathStyle bool   `yaml:"path_style"` // path-style addressing (MinIO)
}

type Storage struct {
	Provider string         `yaml:"provider"`
	Options  map[string]any `yaml:"options"`
}