From bd07db13aeb9f441aa0efa023caaf67a5fa55049 Mon Sep 17 00:00:00 2001 From: kimheesu Date: Tue, 1 Jul 2025 11:11:09 +0900 Subject: [PATCH] update README.md --- .gitignore | 3 ++- README.md | 3 ++- main.py | 15 ++++++--------- tradingagents/default_config.py | 14 ++++++-------- 4 files changed, 16 insertions(+), 19 deletions(-) diff --git a/.gitignore b/.gitignore index a73fa468..8495ef63 100644 --- a/.gitignore +++ b/.gitignore @@ -6,4 +6,5 @@ src/ eval_results/ eval_data/ *.egg-info/ -results/ \ No newline at end of file +results/ +.env \ No newline at end of file diff --git a/README.md b/README.md index cac18691..d8fd2209 100644 --- a/README.md +++ b/README.md @@ -119,9 +119,10 @@ You will also need the FinnHub API for financial data. All of our code is implem export FINNHUB_API_KEY=$YOUR_FINNHUB_API_KEY ``` -You will need the OpenAI API for all the agents. +You will need the OpenAI API or GEMINI API for all the agents. ```bash export OPENAI_API_KEY=$YOUR_OPENAI_API_KEY +export GEMINI_API_KEY=$YOUR_GEMINI_API_KEY ``` ### CLI Usage diff --git a/main.py b/main.py index 2d9d4cbc..2754c640 100644 --- a/main.py +++ b/main.py @@ -1,17 +1,14 @@ from tradingagents.graph.trading_graph import TradingAgentsGraph from tradingagents.default_config import DEFAULT_CONFIG -from dotenv import load_dotenv -import os -load_dotenv() # Create a custom config config = DEFAULT_CONFIG.copy() -config["llm_provider"] = os.getenv("LLM_PROVIDER", "openai") # Use a different model -config["backend_url"] = os.getenv("BACKEND_URL", "https://api.openai.com/v1") # Use a different backend -config["deep_think_llm"] = os.getenv("DEEP_THINK_LLM", "o4-mini") # Use a different model -config["quick_think_llm"] = os.getenv("QUICK_THINK_LLM", "gpt-4o-mini") # Use a different model -config["max_debate_rounds"] = int(os.getenv("MAX_DEBATE_ROUNDS", 1)) # Increase debate rounds -config["online_tools"] = bool(os.getenv("ONLINE_TOOLS", "True")) # Increase debate rounds +config["llm_provider"] = "google" # Use a different model +config["backend_url"] = "https://generativelanguage.googleapis.com/v1" # Use a different backend +config["deep_think_llm"] = "gemini-2.5-pro" # Use a different model +config["quick_think_llm"] = "gemini-2.5-flash-lite-preview-06-17" # Use a different model +config["max_debate_rounds"] = 1 # Increase debate rounds +config["online_tools"] = True # Increase debate rounds # Initialize with custom config ta = TradingAgentsGraph(debug=True, config=config) diff --git a/tradingagents/default_config.py b/tradingagents/default_config.py index 094ac9db..089e9c24 100644 --- a/tradingagents/default_config.py +++ b/tradingagents/default_config.py @@ -1,6 +1,4 @@ import os -from dotenv import load_dotenv -load_dotenv() DEFAULT_CONFIG = { "project_dir": os.path.abspath(os.path.join(os.path.dirname(__file__), ".")), @@ -11,14 +9,14 @@ DEFAULT_CONFIG = { "dataflows/data_cache", ), # LLM settings - "llm_provider": os.getenv("LLM_PROVIDER", "openai"), - "deep_think_llm": os.getenv("DEEP_THINK_LLM", "o4-mini"), - "quick_think_llm": os.getenv("QUICK_THINK_LLM", "gpt-4o-mini"), - "backend_url": os.getenv("BACKEND_URL", "https://api.openai.com/v1"), + "llm_provider": "openai", + "deep_think_llm": "o4-mini", + "quick_think_llm": "gpt-4o-mini", + "backend_url": "https://api.openai.com/v1", # Debate and discussion settings - "max_debate_rounds": int(os.getenv("MAX_DEBATE_ROUNDS", 1)), + "max_debate_rounds": 1, "max_risk_discuss_rounds": 1, "max_recur_limit": 100, # Tool settings - "online_tools": bool(os.getenv("ONLINE_TOOLS", "True")), + "online_tools": True, }