v0.0.5: context compaction, minimal config, retry/search/code symbols/diff/status/test commands

internal/config/config_test.go

@@ -16,8 +16,9 @@ func TestLoadExpandsEnvAndDefaults(t *testing.T) {
 providers:
   test:
     base_url: https://example.com
-    model: test-model
     api_key: ${AGENTU_TEST_KEY}
+    models:
+      - id: test-model
 agent:
   working_dir: .
 `), 0o644); err != nil {
@@ -43,6 +44,115 @@ agent:
 	}
 }
 
+func TestLoadMinimalModelObjectConfig(t *testing.T) {
+	dir := t.TempDir()
+	t.Setenv("AGENTU_TEST_KEY", "sk-test")
+	path := filepath.Join(dir, "agentu.yaml")
+	if err := os.WriteFile(path, []byte(`
+providers:
+  test:
+    base_url: https://example.com
+    api_key: ${AGENTU_TEST_KEY}
+    models:
+      - id: model-a
+        thinking: none
+        context: 256000
+`), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	cfg, err := Load(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	provider := cfg.Providers["test"]
+	if provider.Model != "model-a" {
+		t.Fatalf("model = %q", provider.Model)
+	}
+	if len(provider.Models) != 1 || provider.Models[0] != "model-a" {
+		t.Fatalf("models = %#v", provider.Models)
+	}
+	if provider.ContextTokens != 256000 {
+		t.Fatalf("context tokens = %d", provider.ContextTokens)
+	}
+	if !provider.ThinkingConfigured || provider.Thinking != "none" || provider.ThinkingParam != "thinking" {
+		t.Fatalf("thinking config = %#v", provider)
+	}
+	if cfg.Agent.ToolTimeout != 30*time.Second {
+		t.Fatalf("tool timeout = %s", cfg.Agent.ToolTimeout)
+	}
+	if cfg.Agent.SystemPrompt == "" {
+		t.Fatal("system prompt default was not applied")
+	}
+}
+
+func TestLoadRejectsStringModelEntries(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "agentu.yaml")
+	if err := os.WriteFile(path, []byte(`
+providers:
+  test:
+    base_url: https://example.com
+    api_key: sk-test
+    models:
+      - test-model
+`), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	_, err := Load(path)
+	if err == nil || !strings.Contains(err.Error(), "cannot unmarshal") {
+		t.Fatalf("err = %v", err)
+	}
+}
+
+func TestLoadMaxContextTokens(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "agentu.yaml")
+	if err := os.WriteFile(path, []byte(`
+providers:
+  test:
+    base_url: https://example.com
+    api_key: sk-test
+    models:
+      - id: test-model
+agent:
+  max_context_tokens: 120000
+`), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	cfg, err := Load(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if cfg.Agent.MaxContextTokens != 120000 {
+		t.Fatalf("max context tokens = %d", cfg.Agent.MaxContextTokens)
+	}
+}
+
+func TestLoadRejectsNegativeMaxContextTokens(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "agentu.yaml")
+	if err := os.WriteFile(path, []byte(`
+providers:
+  test:
+    base_url: https://example.com
+    api_key: sk-test
+    models:
+      - id: test-model
+agent:
+  max_context_tokens: -1
+`), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	_, err := Load(path)
+	if err == nil || !strings.Contains(err.Error(), "agent.max_context_tokens") {
+		t.Fatalf("err = %v", err)
+	}
+}
+
 func TestLoadUsesDefaultConfigPath(t *testing.T) {
 	dir := t.TempDir()
 	t.Setenv("HOME", dir)
@@ -56,8 +166,9 @@ func TestLoadUsesDefaultConfigPath(t *testing.T) {
 providers:
   test:
     base_url: https://example.com
-    model: test-model
     api_key: ${AGENTU_TEST_KEY}
+    models:
+      - id: test-model
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}
@@ -93,8 +204,9 @@ func TestLoadDefaultSystemPromptDiscouragesEagerTools(t *testing.T) {
 providers:
   test:
     base_url: https://example.com
-    model: test-model
     api_key: ${AGENTU_TEST_KEY}
+    models:
+      - id: test-model
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}
@@ -119,17 +231,15 @@ providers:
   loveuer:
     base_url: https://example.com
     api_key: ${AGENTU_TEST_KEY}
-    model: model-a
     models:
-      - model-a
-      - model-b
-    thinking: xhigh
-    thinking_param: thinking
+      - id: model-a
+        thinking: xhigh
+      - id: model-b
   backup:
     base_url: https://backup.example.com
     api_key: ${AGENTU_TEST_KEY}
     models:
-      - model-c
+      - id: model-c
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}
@@ -162,11 +272,13 @@ providers:
   two:
     base_url: https://two.example.com
     api_key: ${AGENTU_TEST_KEY}
-    model: model-b
+    models:
+      - id: model-b
   one:
     base_url: https://one.example.com
     api_key: ${AGENTU_TEST_KEY}
-    model: model-a
+    models:
+      - id: model-a
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}
@@ -188,9 +300,10 @@ func TestLoadRejectsInvalidThinking(t *testing.T) {
 providers:
   test:
     base_url: https://example.com
-    model: test-model
     api_key: ${AGENTU_TEST_KEY}
-    thinking: huge
+    models:
+      - id: test-model
+        thinking: huge
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}
@@ -224,8 +337,9 @@ func TestLoadRejectsLegacyProviderConfig(t *testing.T) {
 	if err := os.WriteFile(path, []byte(`
 provider:
   base_url: https://example.com
-  model: test-model
   api_key: sk-test
+  models:
+    - id: test-model
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}
@@ -244,8 +358,9 @@ active_provider: loveuer
 providers:
   loveuer:
     base_url: https://example.com
-    model: test-model
     api_key: sk-test
+    models:
+      - id: test-model
 `), 0o644); err != nil {
 		t.Fatal(err)
 	}