config

type AuthConfig struct {
	Secret   string `mapstructure:"secret"`
	Required bool   `mapstructure:"required"`
}

type AvailableIndexes struct {
	IVFFLAT bool `mapstructure:"ivfflat"`
	HSNW    bool `mapstructure:"hsnw"`
}

type AzureOpenAIConfig struct {
	LLMDeployment       string `mapstructure:"llm_deployment"`
	EmbeddingDeployment string `mapstructure:"embedding_deployment"`
}

type Config struct {
	LLM           LLM                 `mapstructure:"llm"`
	NLP           NLP                 `mapstructure:"nlp"`
	Memory        MemoryConfig        `mapstructure:"memory"`
	Extractors    ExtractorsConfig    `mapstructure:"extractors"`
	Store         StoreConfig         `mapstructure:"store"`
	Server        ServerConfig        `mapstructure:"server"`
	Log           LogConfig           `mapstructure:"log"`
	Auth          AuthConfig          `mapstructure:"auth"`
	OpenTelemetry OpenTelemetryConfig `mapstructure:"opentelemetry"`
	DataConfig    DataConfig          `mapstructure:"data"`
	Development   bool                `mapstructure:"development"`
	CustomPrompts CustomPromptsConfig `mapstructure:"custom_prompts"`
}

type CustomPromptsConfig struct {
	SummarizerPrompts ExtractorPromptsConfig `mapstructure:"summarizer_prompts"`
}

type DataConfig struct {
	// PurgeEvery is the period between hard deletes, in minutes.
	// If set to 0, hard deletes will not be performed.
	PurgeEvery int `mapstructure:"purge_every"`
}

type DocumentExtractorsConfig struct {
	Embeddings EmbeddingsConfig `mapstructure:"embeddings"`
}

type EmbeddingsConfig struct {
	Enabled    bool   `mapstructure:"enabled"`
	Dimensions int    `mapstructure:"dimensions"`
	Service    string `mapstructure:"service"`
	// ChunkSize is the number of documents to embed in a single task.
	ChunkSize int `mapstructure:"chunk_size"`
}

type EntityExtractorConfig struct {
	Enabled bool `mapstructure:"enabled"`
}

type ExtractorPromptsConfig struct {
	OpenAI    string `mapstructure:"openai"`
	Anthropic string `mapstructure:"anthropic"`
}

type ExtractorsConfig struct {
	Messages  MessageExtractorsConfig  `mapstructure:"messages"`
	Documents DocumentExtractorsConfig `mapstructure:"documents"`
}

type IntentExtractorConfig struct {
	Enabled bool `mapstructure:"enabled"`
}

type LLM struct {
	Service             string            `mapstructure:"service"`
	Model               string            `mapstructure:"model"`
	AnthropicAPIKey     string            `mapstructure:"anthropic_api_key"`
	OpenAIAPIKey        string            `mapstructure:"openai_api_key"`
	AzureOpenAIEndpoint string            `mapstructure:"azure_openai_endpoint"`
	AzureOpenAIModel    AzureOpenAIConfig `mapstructure:"azure_openai"`
	OpenAIEndpoint      string            `mapstructure:"openai_endpoint"`
	OpenAIOrgID         string            `mapstructure:"openai_org_id"`
}

type LogConfig struct {
	Level string `mapstructure:"level"`
}

type MemoryConfig struct {
	MessageWindow int `mapstructure:"message_window"`
}

type MessageExtractorsConfig struct {
	Summarizer SummarizerConfig      `mapstructure:"summarizer"`
	Embeddings EmbeddingsConfig      `mapstructure:"embeddings"`
	Entities   EntityExtractorConfig `mapstructure:"entities"`
	Intent     IntentExtractorConfig `mapstructure:"intent"`
}

type NLP struct {
	ServerURL string `mapstructure:"server_url"`
}

type OpenTelemetryConfig struct {
	Enabled bool `mapstructure:"enabled"`
}

type PostgresConfig struct {
	DSN              string           `mapstructure:"dsn"`
	AvailableIndexes AvailableIndexes `mapstructure:"available_indexes"`
}

type ServerConfig struct {
	Host           string `mapstructure:"host"`
	Port           int    `mapstructure:"port"`
	WebEnabled     bool   `mapstructure:"web_enabled"`
	MaxRequestSize int64  `mapstructure:"max_request_size"`
}

type StoreConfig struct {
	Type     string         `mapstructure:"type"`
	Postgres PostgresConfig `mapstructure:"postgres"`
}

type SummarizerConfig struct {
	Enabled    bool                  `mapstructure:"enabled"`
	Embeddings EmbeddingsConfig      `mapstructure:"embeddings"`
	Entities   EntityExtractorConfig `mapstructure:"entities"`
}