Revert "Docker support and Ollama support (#47)" (#57)

This reverts commit 78ea029a0b.
This commit is contained in:
Yijia Xiao
2025-06-26 00:07:58 -04:00
committed by GitHub
parent 78ea029a0b
commit 26c5ba5a78
23 changed files with 19 additions and 2141 deletions

55
main.py
View File

@@ -1,46 +1,21 @@
import os
from tradingagents.graph.trading_graph import TradingAgentsGraph
from tradingagents.default_config import DEFAULT_CONFIG
from dotenv import load_dotenv
def run_analysis(config_overrides=None):
"""
Initializes and runs a trading cycle with configurable overrides.
"""
load_dotenv() # Load .env file variables
# Create a custom config
config = DEFAULT_CONFIG.copy()
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.0-flash" # Use a different model
config["quick_think_llm"] = "gemini-2.0-flash" # Use a different model
config["max_debate_rounds"] = 1 # Increase debate rounds
config["online_tools"] = True # Increase debate rounds
config = DEFAULT_CONFIG.copy()
# Initialize with custom config
ta = TradingAgentsGraph(debug=True, config=config)
# Override with environment variables if set
config["llm_provider"] = os.environ.get("LLM_PROVIDER", config.get("llm_provider", "google"))
config["backend_url"] = os.environ.get("LLM_BACKEND_URL", config.get("backend_url", "https://generativelanguage.googleapis.com/v1"))
config["deep_think_llm"] = os.environ.get("LLM_DEEP_THINK_MODEL", config.get("deep_think_llm", "gemini-2.0-flash"))
config["quick_think_llm"] = os.environ.get("LLM_QUICK_THINK_MODEL", config.get("quick_think_llm", "gemini-2.0-flash"))
config["max_debate_rounds"] = int(os.environ.get("MAX_DEBATE_ROUNDS", config.get("max_debate_rounds", 1)))
config["online_tools"] = os.environ.get("ONLINE_TOOLS", str(config.get("online_tools", True))).lower() == 'true'
# forward propagate
_, decision = ta.propagate("NVDA", "2024-05-10")
print(decision)
# Apply overrides from function argument
if config_overrides:
config.update(config_overrides)
print("Using configuration:")
for key, value in config.items():
print(f"{key}: {value}")
# Initialize with the final config
ta = TradingAgentsGraph(debug=True, config=config)
# Forward propagate
_, decision = ta.propagate("NVDA", "2024-05-10")
return decision
if __name__ == "__main__":
# Example of running the trading analysis
# You can override specific configurations here if needed, e.g.:
# decision = run_trading_cyrun_analysiscle(config_overrides={"max_debate_rounds": 2})
decision = run_analysis()
print(decision)
# Memorize mistakes and reflect
# ta.reflect_and_remember(1000) # parameter is the position returns
# Memorize mistakes and reflect
# ta.reflect_and_remember(1000) # parameter is the position returns