package config

import cfg "github.com/eschao/config"

type Config struct {
	// LLM
	OpenAIAPIKey      string `env:"OPENAI_API_KEY"`
	OpenAIEndpoint    string `env:"OPENAI_ENDPOINT"`
	OpenAIModel       string `env:"OPENAI_MODEL"`
	OpenRouteAPIKey   string `env:"OPENROUTE_API_KEY"`
	OpenRouteEndpoint string `env:"OPENROUTE_ENDPOINT"`
	OpenRouteModel    string `env:"OPENROUTE_MODEL"`

	OllamaEndpoint string `env:"OLLAMA_ENDPOINT"`
	OllamaModel    string `env:"OLLAMA_MODEL"`

	// Embeddings
	OpenAIEmbeddingAPIKey   string `env:"OPENAI_EMBEDDING_API_KEY"`
	OpenAIEmbeddingEndpoint string `env:"OPENAI_EMBEDDING_ENDPOINT"`
	OpenAIEmbeddingModel    string `env:"OPENAI_EMBEDDING_MODEL"`
	OllamaEmbeddingEndpoint string `env:"OLLAMA_EMBEDDING_ENDPOINT"`
	OllamaEmbeddingModel    string `env:"OLLAMA_EMBEDDING_MODEL"`

	// Database
	MilvusHost string `env:"MILVUS_HOST"`
}

func NewConfig() Config {
	config := Config{
		MilvusHost:              "192.168.10.56:19530",
		OllamaEmbeddingEndpoint: "http://192.168.10.56:11434",
		OllamaEmbeddingModel:    "bge-m3",
		OllamaEndpoint:          "http://192.168.10.56:11434/api/chat",
		OllamaModel:             "qwen3:latest",
	}
	cfg.ParseEnv(&config)
	return config
}