config

type AnthropicConfig struct {
	APIKey  string `yaml:"apiKey"`
	Model   string `yaml:"model" default:"claude-3-opus-20240229"`
	BaseURL string `yaml:"baseURL" default:"https://api.anthropic.com/v1"`
}

type AzureOpenAIConfig struct {
	APIKey     string `yaml:"apiKey"`
	Model      string `yaml:"model"`
	Endpoint   string `yaml:"endpoint"`
	APIVersion string `yaml:"apiVersion" default:"2024-03-01-preview"`
}

type CohereConfig struct {
	APIKey  string `yaml:"apiKey"`
	Model   string `yaml:"model" default:"embed-english-v3.0"`
	BaseURL string `yaml:"baseURL" default:"https://api.cohere.ai/v1"`
}

type Config struct {
	RAG         RAGConfig         `yaml:"rag"`
	Embedding   EmbeddingConfig   `yaml:"embedding"`
	LLM         LLMConfig         `yaml:"llm"`
	VectorStore VectorStoreConfig `yaml:"vectorstore"`
	Logging     LoggingConfig     `yaml:"logging"`
}

type EmbeddingConfig struct {
	Provider string       `yaml:"provider" default:"openai"`
	OpenAI   OpenAIConfig `yaml:"openai"`
	Ollama   OllamaConfig `yaml:"ollama"`
	Cohere   CohereConfig `yaml:"cohere"`
	Voyage   VoyageConfig `yaml:"voyage"`
}

type LLMConfig struct {
	Provider    string            `yaml:"provider" default:"openai"`
	OpenAI      OpenAIConfig      `yaml:"openai"`
	Anthropic   AnthropicConfig   `yaml:"anthropic"`
	Ollama      OllamaConfig      `yaml:"ollama"`
	AzureOpenAI AzureOpenAIConfig `yaml:"azure_openai"`
}

type Loader struct {
	// contains filtered or unexported fields
}

type LoggingConfig struct {
	Level  string `yaml:"level" default:"info"`
	Format string `yaml:"format" default:"json"`
}

type MemoryConfig struct {
	MaxSize int `yaml:"maxSize" default:"10000"`
}

type MilvusConfig struct {
	Host     string `yaml:"host" default:"localhost"`
	Port     string `yaml:"port" default:"19530"`
	Username string `yaml:"username"`
	Password string `yaml:"password"`
	Database string `yaml:"database" default:"default"`
}

type OllamaConfig struct {
	Model   string `yaml:"model" default:"qwen3:7b"`
	BaseURL string `yaml:"baseURL" default:"http://localhost:11434"`
}

type OpenAIConfig struct {
	APIKey  string `yaml:"apiKey"`
	Model   string `yaml:"model" default:"text-embedding-ada-002"`
	BaseURL string `yaml:"baseURL" default:"https://api.openai.com/v1"`
}

type PineconeConfig struct {
	APIKey      string `yaml:"apiKey"`
	Environment string `yaml:"environment"`
}

type QdrantConfig struct {
	URL    string `yaml:"url" default:"http://localhost:6333"`
	APIKey string `yaml:"apiKey"`
}

type RAGConfig struct {
	TopK                  int     `yaml:"topK" default:"5"`
	ChunkSize             int     `yaml:"chunkSize" default:"1000"`
	ChunkOverlap          int     `yaml:"chunkOverlap" default:"100"`
	UseSemanticChunking   bool    `yaml:"useSemanticChunking" default:"false"`
	UseHyDE               bool    `yaml:"useHyDE" default:"false"`
	UseRAGFusion          bool    `yaml:"useRAGFusion" default:"false"`
	UseContextCompression bool    `yaml:"useContextCompression" default:"false"`
	RAGFusionQueries      int     `yaml:"ragFusionQueries" default:"4"`
	RAGFusionWeight       float64 `yaml:"ragFusionWeight" default:"0.5"`
}

type ValidationError struct {
	Field   string
	Message string
}

type Validator struct {
	// contains filtered or unexported fields
}

type VectorStoreConfig struct {
	Type     string         `yaml:"type" default:"memory"`
	Memory   MemoryConfig   `yaml:"memory"`
	Milvus   MilvusConfig   `yaml:"milvus"`
	Qdrant   QdrantConfig   `yaml:"qdrant"`
	Weaviate WeaviateConfig `yaml:"weaviate"`
	Pinecone PineconeConfig `yaml:"pinecone"`
}

type VoyageConfig struct {
	APIKey  string `yaml:"apiKey"`
	Model   string `yaml:"model" default:"voyage-2"`
	BaseURL string `yaml:"baseURL" default:"https://api.voyageai.com/v1"`
}

type WeaviateConfig struct {
	URL    string `yaml:"url" default:"http://localhost:8080"`
	APIKey string `yaml:"apiKey"`
}