charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: clean up code as per PEP 8 and review suggestions

This commit is contained in:
mogita 2025-08-16 16:21:03 +08:00
parent 570d91f515
commit 811d57fb00
No known key found for this signature in database
GPG Key ID: A0AA1B9C57A48ECF
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tradingagents/graph/trading_graph.py
View File

@ -13,6 +13,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
from langgraph.prebuilt import ToolNode
from tradingagents.agents import *
from tradingagents.utils.provider_utils import get_api_key_for_provider
from tradingagents.default_config import DEFAULT_CONFIG
from tradingagents.agents.utils.memory import FinancialSituationMemory
from tradingagents.agents.utils.agent_states import (
@ -60,16 +61,10 @@ class TradingAgentsGraph:
# Initialize LLMs
provider = self.config["llm_provider"].lower()
if provider == "openai" or provider == "ollama" or provider == "openrouter":
from tradingagents.utils.provider_utils import get_api_key_for_provider
if provider in ("openai", "ollama", "openrouter") or provider.startswith("custom"):
api_key = get_api_key_for_provider(self.config)
self.deep_thinking_llm = ChatOpenAI(model=self.config["deep_think_llm"], openai_api_base=self.config["backend_url"], openai_api_key=api_key)
self.quick_thinking_llm = ChatOpenAI(model=self.config["quick_think_llm"], openai_api_base=self.config["backend_url"], openai_api_key=api_key)
elif provider.startswith("custom"):
# Custom provider uses OpenAI-compatible interface
custom_api_key = os.getenv("CUSTOM_API_KEY")
self.deep_thinking_llm = ChatOpenAI(model=self.config["deep_think_llm"], openai_api_base=self.config["backend_url"], openai_api_key=custom_api_key)
self.quick_thinking_llm = ChatOpenAI(model=self.config["quick_think_llm"], openai_api_base=self.config["backend_url"], openai_api_key=custom_api_key)
self.deep_thinking_llm = ChatOpenAI(model=self.config["deep_think_llm"], base_url=self.config["backend_url"], api_key=api_key)
self.quick_thinking_llm = ChatOpenAI(model=self.config["quick_think_llm"], base_url=self.config["backend_url"], api_key=api_key)
elif provider == "anthropic":
self.deep_thinking_llm = ChatAnthropic(model=self.config["deep_think_llm"], base_url=self.config["backend_url"])
self.quick_thinking_llm = ChatAnthropic(model=self.config["quick_think_llm"], base_url=self.config["backend_url"])