diff options
Diffstat (limited to 'internal/config')
| -rw-r--r-- | internal/config/config.go | 17 | ||||
| -rw-r--r-- | internal/config/config_test.go | 43 |
2 files changed, 60 insertions, 0 deletions
diff --git a/internal/config/config.go b/internal/config/config.go index 80777ff..640e933 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -63,6 +63,20 @@ func (r RunnersConfig) GeminiEnabled() bool { return r.Gemini == nil || *r.Ge func (r RunnersConfig) LocalEnabled() bool { return r.Local == nil || *r.Local } func (r RunnersConfig) ContainerEnabled() bool { return r.Container == nil || *r.Container } +// ProviderConfig configures a native (or OpenAI-compatible) LLM provider +// backend for the multi-provider harness — see docs/api-keys-setup.md. Phase +// 1 only adds this type so config has somewhere for it to land; it is not yet +// read by runner construction (no native provider adapters exist yet besides +// the openaicompat one LocalModel already configures). Keyed by provider name +// in the [providers.<name>] TOML table, e.g. [providers.groq], +// [providers.anthropic]. +type ProviderConfig struct { + Endpoint string `toml:"endpoint"` + APIKey string `toml:"api_key"` + DefaultModel string `toml:"default_model"` + TimeoutSeconds int `toml:"timeout_seconds"` +} + type Config struct { DataDir string `toml:"data_dir"` DBPath string `toml:"-"` @@ -89,6 +103,9 @@ type Config struct { LocalModel LocalModel `toml:"local_model"` Runners RunnersConfig `toml:"runners"` Budget BudgetConfig `toml:"budget"` + // Providers configures native/OpenAI-compatible LLM backends by name (see + // ProviderConfig). Unused by runner construction in Phase 1. + Providers map[string]ProviderConfig `toml:"providers"` // ExternalBindAllowed must be explicitly true to bind a non-loopback address. ExternalBindAllowed bool `toml:"external_bind_allowed"` } diff --git a/internal/config/config_test.go b/internal/config/config_test.go index 0d47ca0..6991354 100644 --- a/internal/config/config_test.go +++ b/internal/config/config_test.go @@ -165,3 +165,46 @@ func TestLocalModel_UseForElaborate_ExplicitTrue(t *testing.T) { t.Error("explicit true should opt in") } } + +func TestProvidersConfig_TOMLRoundTrip(t *testing.T) { + dir := t.TempDir() + path := filepath.Join(dir, "config.toml") + toml := ` +[providers.groq] +endpoint = "https://api.groq.com/openai/v1" +api_key = "gsk_abc" +default_model = "llama-3.3-70b-versatile" +timeout_seconds = 30 + +[providers.anthropic] +api_key = "sk-ant-xyz" +default_model = "claude-sonnet-4-5" +` + if err := os.WriteFile(path, []byte(toml), 0600); err != nil { + t.Fatal(err) + } + t.Setenv("HOME", dir) + + cfg, err := LoadFile(path) + if err != nil { + t.Fatalf("unexpected error: %v", err) + } + if len(cfg.Providers) != 2 { + t.Fatalf("expected 2 providers, got %d: %+v", len(cfg.Providers), cfg.Providers) + } + groq, ok := cfg.Providers["groq"] + if !ok { + t.Fatal("expected [providers.groq] table") + } + if groq.Endpoint != "https://api.groq.com/openai/v1" || groq.APIKey != "gsk_abc" || + groq.DefaultModel != "llama-3.3-70b-versatile" || groq.TimeoutSeconds != 30 { + t.Errorf("groq provider config: %+v", groq) + } + anthropic, ok := cfg.Providers["anthropic"] + if !ok { + t.Fatal("expected [providers.anthropic] table") + } + if anthropic.Endpoint != "" || anthropic.APIKey != "sk-ant-xyz" || anthropic.DefaultModel != "claude-sonnet-4-5" { + t.Errorf("anthropic provider config: %+v", anthropic) + } +} |
