Update cli/utils.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

cli/utils.py

@@ -269,7 +269,7 @@ def select_llm_provider() -> tuple[str, str]:
         ("Google", "https://generativelanguage.googleapis.com/v1"),
         ("Anthropic", "https://api.anthropic.com/"),
         ("xAI", "https://api.x.ai/v1"),
-        ("Z.AI", "zai", "https://api.z.ai/api/coding/paas/v4"),
+        ("Zai", "https://api.z.ai/api/coding/paas/v4"),
         ("Openrouter", "https://openrouter.ai/api/v1"),
         ("Ollama", "http://localhost:11434/v1"),
     ]

tradingagents/llm_clients/zai_client.py

@@ -22,14 +22,16 @@ class ZAIClient(BaseLLMClient):
             "base_url": self.base_url or self.DEFAULT_BASE_URL,
         }
 
-        api_key = os.environ.get("ZAI_API_KEY")
+        api_key = self.kwargs.get("api_key")
+        if api_key is None:
+            api_key = os.environ.get("ZAI_API_KEY")
+
         if api_key:
             llm_kwargs["api_key"] = api_key
 
         for key in (
             "timeout",
             "max_retries",
-            "api_key",
             "callbacks",
             "http_client",
             "http_async_client",