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

refactor: simplify api_key mapping and consolidate tests

Browse Source 
Apply review suggestions: use concise `or` pattern for API key
resolution, consolidate tests into parameterized subTest, move
import to module level per PEP 8.
This commit is contained in:
javierdejesusda
2026-03-24 14:52:51 +01:00
parent f5026009f9
commit 047b38971c
2 changed files with 18 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tradingagents/llm_clients/google_client.py
+3 -4
View File
@@ -32,10 +32,9 @@ class GoogleClient(BaseLLMClient):
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"]
google_api_key = self.kwargs.get("api_key") or self.kwargs.get("google_api_key")
if google_api_key:
llm_kwargs["google_api_key"] = google_api_key
# Map thinking_level to appropriate API param based on model
# Gemini 3 Pro: low, high