diff --git a/docker-compose.yml b/docker-compose.yml index 3a5d4e29..d28135b3 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -4,7 +4,7 @@ services: env_file: - .env volumes: - - ./results:/home/appuser/app/results + - tradingagents_data:/home/appuser/.tradingagents tty: true stdin_open: true @@ -22,7 +22,7 @@ services: environment: - LLM_PROVIDER=ollama volumes: - - ./results:/home/appuser/app/results + - tradingagents_data:/home/appuser/.tradingagents depends_on: - ollama tty: true @@ -31,4 +31,5 @@ services: - ollama volumes: + tradingagents_data: ollama_data: diff --git a/tradingagents/default_config.py b/tradingagents/default_config.py index 26a4e4d2..a9b75e4b 100644 --- a/tradingagents/default_config.py +++ b/tradingagents/default_config.py @@ -1,12 +1,11 @@ import os +_TRADINGAGENTS_HOME = os.path.join(os.path.expanduser("~"), ".tradingagents") + DEFAULT_CONFIG = { "project_dir": os.path.abspath(os.path.join(os.path.dirname(__file__), ".")), - "results_dir": os.getenv("TRADINGAGENTS_RESULTS_DIR", "./results"), - "data_cache_dir": os.path.join( - os.path.abspath(os.path.join(os.path.dirname(__file__), ".")), - "dataflows/data_cache", - ), + "results_dir": os.getenv("TRADINGAGENTS_RESULTS_DIR", os.path.join(_TRADINGAGENTS_HOME, "logs")), + "data_cache_dir": os.getenv("TRADINGAGENTS_CACHE_DIR", os.path.join(_TRADINGAGENTS_HOME, "cache")), # LLM settings "llm_provider": "openai", "deep_think_llm": "gpt-5.4", diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index 8e18f9c4..78bc13e5 100644 --- a/tradingagents/graph/trading_graph.py +++ b/tradingagents/graph/trading_graph.py @@ -66,10 +66,8 @@ class TradingAgentsGraph: set_config(self.config) # Create necessary directories - os.makedirs( - os.path.join(self.config["project_dir"], "dataflows/data_cache"), - exist_ok=True, - ) + os.makedirs(self.config["data_cache_dir"], exist_ok=True) + os.makedirs(self.config["results_dir"], exist_ok=True) # Initialize LLMs with provider-specific thinking configuration llm_kwargs = self._get_provider_kwargs()