Config Package (pkg/config)

主要功能

基礎使用

package main
import (
	"log"
	"github.com/maoxiaoyue/hypgo/pkg/config"
)
func main() {
	// 假設有 config.yaml
	cfg, err := config.LoadConfig("config.yaml")
	if err != nil {
		log.Fatalf("無法讀取配置: %v", err)
	}
	// 確保設定的數值都有預設值墊底
	cfg.ApplyDefaults()
	log.Printf("Server 將啟動於 %s", cfg.GetServerConfig().GetAddr())
}

配置結構

server:
 addr: ":8080"
 protocol: "HTTP/1.1"
database:
 driver: "mysql"
 dsn: "user:pass@tcp(127.0.0.1:3306)/dbname"
logger:
 level: "info"
 format: "json"

預設值參考

LLM 配置 (llm.yaml)

配置結構

type LLMConfig struct {
 Mode string `yaml:"mode"` // "none", "rag", "ollama", "api"
 RAG RAGConfig `yaml:"rag"` // mode=rag 時使用
 Ollama OllamaConfig `yaml:"ollama"` // mode=ollama 時使用
 API APIConfig `yaml:"api"` // mode=api 時使用
}

四種模式

Ollama 配置

mode: ollama
ollama:
 url: "http://localhost:11434" # 預設值
 model: "llama3" # 必填
 timeout: 30 # 秒(預設 30)
 max_tokens: 256 # 預設 256

API 配置

mode: api
api:
 provider: "openai" # 必填
 model: "gpt-4o-mini" # 必填
 api_key: "${OPENAI_API_KEY}" # 必填(支援 ${ENV_VAR})
 base_url: "" # 各 provider 有預設值
 timeout: 30 # 秒(預設 30)
 max_tokens: 256 # 預設 256

RAG 配置

mode: rag
rag:
 embedding_model: "nomic-embed-text" # 必填
 embedding_url: "http://localhost:11434" # 預設 Ollama
 vector_store: "chroma" # 必填:chroma/qdrant/milvus/faiss
 vector_store_url: "http://localhost:8000" # 必填
 collection: "my_codebase" # 集合名稱
 top_k: 5 # 檢索數量(預設 5)
 generator_model: "llama3" # 必填
 generator_url: "http://localhost:11434" # 預設 Ollama

載入方式

// 從指定路徑載入(檔案不存在時回傳 mode=none 預設值)
cfg, err := config.LoadLLMConfig("config/llm.yaml")
// 空路徑 → mode=none
cfg, err := config.LoadLLMConfig("")
// 檢查是否啟用
if cfg.IsEnabled() {
 // mode != "none"
}

安全設計

範例配置檔