This commit is contained in:
MarkLo 2025-12-11 01:31:44 +08:00
parent 71fb9844ee
commit 9ef0f2265d
1 changed files with 8 additions and 20 deletions

View File

@ -95,26 +95,14 @@ class TradingAgentsXGraph:
# Helper to initialize LLM based on URL/Provider # Helper to initialize LLM based on URL/Provider
def _create_llm(model: str, base_url: str, api_key: str): def _create_llm(model: str, base_url: str, api_key: str):
# Determine provider based on Base URL # Use ChatOpenAI for all providers including Anthropic's OpenAI-compatible endpoint
if "anthropic.com" in base_url: # Anthropic's /v1 endpoint is OpenAI-compatible, so we use ChatOpenAI
# Claude 4.5 API restriction: cannot use both temperature and top_p return ChatOpenAI(
# Only use temperature, set top_p to None explicitly model=model,
return ChatAnthropic( base_url=base_url,
model=model, openai_api_key=api_key,
base_url=base_url, max_tokens=16000 # Prevent report truncation
api_key=api_key, )
max_tokens=16000, # Prevent report truncation
temperature=0.7, # Use temperature for randomness control
top_p=None # Explicitly set to None to avoid conflict
)
else:
# Default to ChatOpenAI for OpenAI, Grok, DeepSeek, Qwen, and other OpenAI-compatible APIs
return ChatOpenAI(
model=model,
base_url=base_url,
openai_api_key=api_key,
max_tokens=16000 # Prevent report truncation
)
# Initialize LLMs independently # Initialize LLMs independently
print(f"DEBUG: Initializing Deep Thinking LLM: Model={self.config['deep_think_llm']}, BaseURL={deep_base_url}, Key={deep_api_key[:10]}...") print(f"DEBUG: Initializing Deep Thinking LLM: Model={self.config['deep_think_llm']}, BaseURL={deep_base_url}, Key={deep_api_key[:10]}...")