farcasclaudiu/TradingAgents
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/TradingAgents.git synced 2026-06-29 03:01:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix(llm_clients): standardize Google API key to unified api_key param

Browse Source 
GoogleClient now accepts the unified `api_key` parameter used by
OpenAI and Anthropic clients, mapping it to the provider-specific
`google_api_key` that ChatGoogleGenerativeAI expects. Legacy
`google_api_key` still works for backward compatibility.

Resolves TODO.md item #2 (inconsistent parameter handling).
This commit is contained in:
javierdejesusda
2026-03-24 14:35:02 +01:00
parent 589b351f2a
commit f5026009f9
3 changed files with 49 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tradingagents/llm_clients/google_client.py
+7 -1
View File
@@ -27,10 +27,16 @@ class GoogleClient(BaseLLMClient):
"""Return configured ChatGoogleGenerativeAI instance."""
llm_kwargs = {"model": self.model}
for key in ("timeout", "max_retries", "google_api_key", "callbacks", "http_client", "http_async_client"):
for key in ("timeout", "max_retries", "callbacks", "http_client", "http_async_client"):
if key in self.kwargs:
llm_kwargs[key] = self.kwargs[key]
# Unified api_key maps to provider-specific google_api_key
if "api_key" in self.kwargs:
llm_kwargs["google_api_key"] = self.kwargs["api_key"]
elif "google_api_key" in self.kwargs:
llm_kwargs["google_api_key"] = self.kwargs["google_api_key"]
# Map thinking_level to appropriate API param based on model
# Gemini 3 Pro: low, high
# Gemini 3 Flash: minimal, low, medium, high