fix(providers): apply Kimi Coding default headers

2026-06-24 10:43:16 +08:00
parent 9d6c606cc9
commit f9afc9389b
5 changed files with 77 additions and 16 deletions
@@ -0,0 +1,50 @@
+from nanobot.config.schema import Config, ProviderConfig
+from nanobot.providers.factory import _provider_extra_headers, provider_signature
+from nanobot.providers.registry import find_by_name
+
+
+def test_kimi_coding_uses_default_user_agent_header() -> None:
+    spec = find_by_name("kimi_coding")
+
+    assert spec is not None
+    assert _provider_extra_headers(spec, ProviderConfig()) == {
+        "User-Agent": "claude-code/0.1.0",
+    }
+
+
+def test_provider_config_extra_headers_override_defaults() -> None:
+    spec = find_by_name("kimi_coding")
+    provider = ProviderConfig.model_validate({
+        "extraHeaders": {
+            "User-Agent": "custom-client/1.0",
+            "X-Test": "1",
+        },
+    })
+
+    assert _provider_extra_headers(spec, provider) == {
+        "User-Agent": "custom-client/1.0",
+        "X-Test": "1",
+    }
+
+
+def test_provider_signature_tracks_default_extra_headers() -> None:
+    config = Config.model_validate({
+        "providers": {
+            "kimiCoding": {
+                "apiKey": "sk-kimi-test",
+            },
+        },
+        "modelPresets": {
+            "primary": {
+                "provider": "kimi_coding",
+                "model": "kimi-for-coding",
+            },
+        },
+        "agents": {
+            "defaults": {
+                "modelPreset": "primary",
+            },
+        },
+    })
+
+    assert {"User-Agent": "claude-code/0.1.0"} in provider_signature(config)