main.py

import os
import time
import datetime
from tradingagents.graph.trading_graph import TradingAgentsGraph
from tradingagents.default_config import DEFAULT_CONFIG

from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Create a custom config
config = DEFAULT_CONFIG.copy()
config["deep_think_llm"] = "gpt-5-mini"  # Use a different model
config["quick_think_llm"] = "gpt-5-mini"  # Use a different model
config["max_debate_rounds"] = 1  # Increase debate rounds

config["llm_provider"] = "deepseek"  # Switch to DeepSeek provider
config["backend_url"] = "https://api.deepseek.com/v1" 
config["deep_think_llm"] = "deepseek-reasoner"  
config["quick_think_llm"] = "deepseek-chat"  
config["enable_on_chain_submission"] = True
config["on_chain_submit_hold_decisions"] = True  # True: HOLD 也提交到 RiskRouter
config["use_trader_v2"] = True  # True: 仅 BUY/SELL; False: BUY/SELL/HOLD

start_time = time.time()
# Initialize with custom config
ta = TradingAgentsGraph(debug=True, selected_analysts=['market', 'news', 'quant'], config=config, parallel_mode=True)

# XBTUSD ETHUSD SOLUSD BNBUSD
final_state, decision = ta.propagate("ETHUSD", datetime.datetime.now(datetime.timezone.utc))
end_time = time.time()

print(decision)
print('\n',"=" * 50,'\n')
print(f"Execution time: {end_time - start_time:.2f} seconds")
# Memorize mistakes and reflect
# ta.reflect_and_remember(1000) # parameter is the position returns