diff --git a/cmd/supervisor/main.go b/cmd/supervisor/main.go index c32d500..17e75e5 100644 --- a/cmd/supervisor/main.go +++ b/cmd/supervisor/main.go @@ -104,7 +104,9 @@ func main() { IngestBaseURL: cfg.IngestBaseURL, })) reg.Register(brain.New(brain.Config{ - IngestBaseURL: cfg.IngestBaseURL, + IngestBaseURL: cfg.IngestBaseURL, + IngestSvcURL: cfg.IngestSvcURL, + KBRetrievalURL: cfg.KBRetrievalURL, })) reg.Register(org.New(org.Config{ TierFn: tierFn, diff --git a/internal/config/config.go b/internal/config/config.go index 2a7ff3f..0865160 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -9,6 +9,8 @@ type Config struct { ConfigDir string // SUPERVISOR_CONFIG_DIR, default ./config/supervisor ModelsFile string // SUPERVISOR_MODELS_FILE, default /../models.yaml IngestBaseURL string // INGEST_BASE_URL, default http://localhost:3300 + IngestSvcURL string // INGEST_SVC_URL — base URL for brain_ingest (/ingest, /ingest-path) + KBRetrievalURL string // KB_RETRIEVAL_URL — base URL for brain_search SessionsDir string // SUPERVISOR_SESSIONS_DIR, default ./brain/sessions BrainDir string // SUPERVISOR_BRAIN_DIR, default ./brain } @@ -22,6 +24,8 @@ func Load() (Config, error) { } cfg.ModelsFile = envOr("SUPERVISOR_MODELS_FILE", cfg.ConfigDir+"/../models.yaml") cfg.IngestBaseURL = envOr("INGEST_BASE_URL", "http://localhost:3300") + cfg.IngestSvcURL = envOr("INGEST_SVC_URL", "") + cfg.KBRetrievalURL = envOr("KB_RETRIEVAL_URL", "") cfg.SessionsDir = envOr("SUPERVISOR_SESSIONS_DIR", "./brain/sessions") cfg.BrainDir = envOr("SUPERVISOR_BRAIN_DIR", "./brain") return cfg, nil