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

feat: update Ollama and OpenRouter model options

Browse Source 
- Ollama: Add Qwen3 (8B), GPT-OSS (20B), GLM-4.7-Flash (30B)
- OpenRouter: Add NVIDIA Nemotron 3 Nano, Z.AI GLM 4.5 Air
- Add explicit Ollama provider handling in OpenAI client for consistency
This commit is contained in:
Yijia Xiao
2026-01-26 22:24:59 +00:00
parent d4dadb82fc
commit a3761bdd66
2 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tradingagents/llm_clients/openai_client.py
+7
View File
@@ -50,6 +50,13 @@ class OpenAIClient(BaseLLMClient):
api_key = os.environ.get("XAI_API_KEY")
if api_key:
llm_kwargs["api_key"] = api_key
elif self.provider == "openrouter":
llm_kwargs["base_url"] = "https://openrouter.ai/api/v1"
api_key = os.environ.get("OPENROUTER_API_KEY")
if api_key:
llm_kwargs["api_key"] = api_key
elif self.provider == "ollama":
llm_kwargs["base_url"] = "http://localhost:11434/v1"
elif self.base_url:
llm_kwargs["base_url"] = self.base_url