Реализована операции Milvus для управления документами и встраиванием, включая функции вставки, запроса и удаления. Внедрите архитектуру RAG с LLM и сервисами встраивания. Добавьте обработку текста для фрагментации и конкатенации. Создайте автономный скрипт для настройки и управления Milvus. Разработайте комплексные тесты API для обработки документов и взаимодействия с LLM, включая имитации для сервисов. Расширьте возможности конфигурации пользователя с помощью дополнительных настроек YAML.
This commit is contained in:
38
config/config.go
Normal file
38
config/config.go
Normal file
@@ -0,0 +1,38 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user