Merge pull request #56 from zhujian0805/main

Normalize Droid model defaults

Author: zhujian0805

code_assistant_manager/tools/droid.py

@@ -211,19 +211,19 @@ def _build_models_json(self, entries: List[str]) -> List[dict]:
             parts = entry.split("|")
             if len(parts) < 5:
                 continue
-            display, base_url, api_key, provider, max_tokens = parts[:5]
+            display, base_url, api_key, provider, *_ = parts
             model_id = display.split("[")[0].strip()
-            try:
-                max_tokens_val = int(max_tokens)
-            except ValueError:
-                max_tokens_val = 0
+            provider_value = (provider or "").strip() or "generic-chat-completion-api"
+            if "claude" in model_id.lower():
+                provider_value = "anthropic"
+            max_tokens_val = 64000
             models.append(
                 {
                     "displayName": display,
                     "model": model_id,
                     "baseUrl": base_url,
                     "apiKey": api_key,
-                    "provider": provider or "generic-chat-completion-api",
+                    "provider": provider_value,
                     "maxOutputTokens": max_tokens_val,
                 }
             )

tests/test_tools.py

@@ -186,14 +186,17 @@ def test_droid_build_models_json(self, config_manager):
         """Test building models JSON for Droid."""
         tool = DroidTool(config_manager)
         entries = [
-            "model1 [ep1]|https://api.com|key|provider|16384",
-            "model2 [ep2]|https://api2.com|key2|provider|65536",
+            "model1 [ep1]|https://api.com|key||16384",
+            "claude-3 [ep2]|https://api2.com|key2|provider|65536",
         ]
         models = tool._build_models_json(entries)
         assert len(models) == 2
         assert models[0]["model"] == "model1"
         assert models[0]["baseUrl"] == "https://api.com"
-        assert models[1]["maxOutputTokens"] == 65536
+        assert models[0]["maxOutputTokens"] == 64000
+        assert models[1]["maxOutputTokens"] == 64000
+        assert models[0]["provider"] == "generic-chat-completion-api"
+        assert models[1]["provider"] == "anthropic"
 
     @patch.dict(os.environ, {"CODE_ASSISTANT_MANAGER_NONINTERACTIVE": "1"})
     def test_droid_process_endpoint_prefers_api_key_env_reference(self, config_manager):