summaryrefslogtreecommitdiff
path: root/internal/appconfig/config_env_model_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/appconfig/config_env_model_test.go')
-rw-r--r--internal/appconfig/config_env_model_test.go32
1 files changed, 20 insertions, 12 deletions
diff --git a/internal/appconfig/config_env_model_test.go b/internal/appconfig/config_env_model_test.go
index f34416d..7038819 100644
--- a/internal/appconfig/config_env_model_test.go
+++ b/internal/appconfig/config_env_model_test.go
@@ -44,22 +44,30 @@ func TestEnv_SurfaceModelOverrides(t *testing.T) {
t.Setenv("HEXAI_TEMPERATURE_CLI", "0.22")
t.Setenv("HEXAI_PROVIDER_CLI", "ollama")
cfg := Load(log.New(os.Stderr, "test ", 0))
- if cfg.CompletionModel != "gpt-c" {
- t.Fatalf("expected completion model override, got %q", cfg.CompletionModel)
+ if len(cfg.CompletionConfigs) != 1 {
+ t.Fatalf("expected single completion entry, got %+v", cfg.CompletionConfigs)
}
- if cfg.CompletionTemperature == nil || *cfg.CompletionTemperature != 0.44 {
- t.Fatalf("expected completion temperature override, got %v", cfg.CompletionTemperature)
+ comp := cfg.CompletionConfigs[0]
+ if comp.Model != "gpt-c" {
+ t.Fatalf("expected completion model override, got %+v", comp)
}
- if cfg.CompletionProvider != "copilot" {
- t.Fatalf("expected completion provider override, got %q", cfg.CompletionProvider)
+ if comp.Temperature == nil || *comp.Temperature != 0.44 {
+ t.Fatalf("expected completion temperature override, got %+v", comp)
}
- if cfg.CLIModel != "gpt-cli" {
- t.Fatalf("expected cli model override, got %q", cfg.CLIModel)
+ if comp.Provider != "copilot" {
+ t.Fatalf("expected completion provider override, got %+v", comp)
}
- if cfg.CLITemperature == nil || *cfg.CLITemperature != 0.22 {
- t.Fatalf("expected cli temperature override, got %v", cfg.CLITemperature)
+ if len(cfg.CLIConfigs) != 1 {
+ t.Fatalf("expected single CLI entry, got %+v", cfg.CLIConfigs)
}
- if cfg.CLIProvider != "ollama" {
- t.Fatalf("expected cli provider override, got %q", cfg.CLIProvider)
+ cli := cfg.CLIConfigs[0]
+ if cli.Model != "gpt-cli" {
+ t.Fatalf("expected cli model override, got %+v", cli)
+ }
+ if cli.Temperature == nil || *cli.Temperature != 0.22 {
+ t.Fatalf("expected cli temperature override, got %+v", cli)
+ }
+ if cli.Provider != "ollama" {
+ t.Fatalf("expected cli provider override, got %+v", cli)
}
}