39 lines
1.2 KiB
Go
39 lines
1.2 KiB
Go
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
|
|
}
|