fix: allow .env variables to override DEFAULT_CONFIG values
Merged origin/main and resolved all 8 conflicting files: - CLAUDE.md: merged MISTAKES.md ref + Project Tracking section + env override docs - cli/main.py: kept top-level json import, kept try/except in run_pipeline - tool_runner.py: kept descriptive comments for MAX_TOOL_ROUNDS - alpha_vantage_common.py: kept thread-safe rate limiter, robust error handling - interface.py: kept broader exception catch (AlphaVantageError + ConnectionError + TimeoutError) - default_config.py: kept _env()/_env_int() env var overrides with load_dotenv() at module level - scanner_graph.py: kept debug mode fix (stream for debug, invoke for result) - macro_bridge.py: kept get_running_loop() over deprecated get_event_loop() Co-authored-by: aguzererler <6199053+aguzererler@users.noreply.github.com>
This commit is contained in:
parent
9ac773a69d
commit
2193ff3fa1
19
.env.example
19
.env.example
|
|
@ -1,6 +1,23 @@
|
||||||
# LLM Providers (set the one you use)
|
# LLM Provider API Keys (set the ones you use)
|
||||||
OPENAI_API_KEY=
|
OPENAI_API_KEY=
|
||||||
GOOGLE_API_KEY=
|
GOOGLE_API_KEY=
|
||||||
ANTHROPIC_API_KEY=
|
ANTHROPIC_API_KEY=
|
||||||
XAI_API_KEY=
|
XAI_API_KEY=
|
||||||
OPENROUTER_API_KEY=
|
OPENROUTER_API_KEY=
|
||||||
|
|
||||||
|
# Data Provider API Keys
|
||||||
|
ALPHA_VANTAGE_API_KEY=
|
||||||
|
|
||||||
|
# ── Configuration overrides ──────────────────────────────────────────
|
||||||
|
# Any setting in DEFAULT_CONFIG can be overridden with a
|
||||||
|
# TRADINGAGENTS_<KEY> environment variable. Unset or empty values
|
||||||
|
# are ignored (the hardcoded default is kept).
|
||||||
|
#
|
||||||
|
# Examples:
|
||||||
|
# TRADINGAGENTS_LLM_PROVIDER=openrouter
|
||||||
|
# TRADINGAGENTS_QUICK_THINK_LLM=deepseek/deepseek-chat-v3-0324
|
||||||
|
# TRADINGAGENTS_DEEP_THINK_LLM=deepseek/deepseek-r1-0528
|
||||||
|
# TRADINGAGENTS_BACKEND_URL=https://openrouter.ai/api/v1
|
||||||
|
# TRADINGAGENTS_RESULTS_DIR=./my_results
|
||||||
|
# TRADINGAGENTS_MAX_DEBATE_ROUNDS=2
|
||||||
|
# TRADINGAGENTS_VENDOR_SCANNER_DATA=alpha_vantage
|
||||||
|
|
|
||||||
15
CLAUDE.md
15
CLAUDE.md
|
|
@ -102,16 +102,13 @@ OpenAI, Anthropic, Google, xAI, OpenRouter, Ollama
|
||||||
- `PROGRESS.md` — Feature progress, what works, TODOs
|
- `PROGRESS.md` — Feature progress, what works, TODOs
|
||||||
- `MISTAKES.md` — Past bugs and lessons learned (9 documented mistakes)
|
- `MISTAKES.md` — Past bugs and lessons learned (9 documented mistakes)
|
||||||
|
|
||||||
## Current LLM Configuration (Hybrid)
|
## LLM Configuration
|
||||||
|
|
||||||
```
|
Per-tier provider overrides in `tradingagents/default_config.py`:
|
||||||
quick_think: qwen3.5:27b via Ollama (http://192.168.50.76:11434)
|
- Each tier (`quick_think`, `mid_think`, `deep_think`) can have its own `_llm_provider` and `_backend_url`
|
||||||
mid_think: qwen3.5:27b via Ollama (http://192.168.50.76:11434)
|
- Falls back to top-level `llm_provider` and `backend_url` when per-tier values are None
|
||||||
deep_think: deepseek/deepseek-r1-0528 via OpenRouter
|
- All config values overridable via `TRADINGAGENTS_<KEY>` env vars
|
||||||
```
|
- Keys for LLM providers: `.env` file (e.g., `OPENROUTER_API_KEY`, `ALPHA_VANTAGE_API_KEY`)
|
||||||
|
|
||||||
Config: `tradingagents/default_config.py` (per-tier `_llm_provider` keys)
|
|
||||||
Keys: `.env` file (`OPENROUTER_API_KEY`, `ALPHA_VANTAGE_API_KEY`)
|
|
||||||
|
|
||||||
## Running the Scanner
|
## Running the Scanner
|
||||||
|
|
||||||
|
|
|
||||||
21
cli/main.py
21
cli/main.py
|
|
@ -1,5 +1,6 @@
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
import datetime
|
import datetime
|
||||||
|
import json
|
||||||
import typer
|
import typer
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
|
|
@ -1201,8 +1202,6 @@ def run_scan(date: Optional[str] = None):
|
||||||
raise typer.Exit(1)
|
raise typer.Exit(1)
|
||||||
|
|
||||||
# Save reports
|
# Save reports
|
||||||
import json as _json
|
|
||||||
|
|
||||||
for key in ["geopolitical_report", "market_movers_report", "sector_performance_report",
|
for key in ["geopolitical_report", "market_movers_report", "sector_performance_report",
|
||||||
"industry_deep_dive_report", "macro_scan_summary"]:
|
"industry_deep_dive_report", "macro_scan_summary"]:
|
||||||
content = result.get(key, "")
|
content = result.get(key, "")
|
||||||
|
|
@ -1217,7 +1216,7 @@ def run_scan(date: Optional[str] = None):
|
||||||
|
|
||||||
# Try to parse and show watchlist table
|
# Try to parse and show watchlist table
|
||||||
try:
|
try:
|
||||||
summary_data = _json.loads(summary)
|
summary_data = json.loads(summary)
|
||||||
stocks = summary_data.get("stocks_to_investigate", [])
|
stocks = summary_data.get("stocks_to_investigate", [])
|
||||||
if stocks:
|
if stocks:
|
||||||
table = Table(title="Stocks to Investigate", box=box.ROUNDED)
|
table = Table(title="Stocks to Investigate", box=box.ROUNDED)
|
||||||
|
|
@ -1235,16 +1234,16 @@ def run_scan(date: Optional[str] = None):
|
||||||
s.get("thesis_angle", ""),
|
s.get("thesis_angle", ""),
|
||||||
)
|
)
|
||||||
console.print(table)
|
console.print(table)
|
||||||
except (_json.JSONDecodeError, KeyError):
|
except (json.JSONDecodeError, KeyError):
|
||||||
pass # Summary wasn't valid JSON — already printed as markdown
|
pass # Summary wasn't valid JSON — already printed as markdown
|
||||||
|
|
||||||
|
|
||||||
console.print(f"\n[green]Results saved to {save_dir}[/green]")
|
console.print(f"\n[green]Results saved to {save_dir}[/green]")
|
||||||
|
|
||||||
|
|
||||||
def run_pipeline():
|
def run_pipeline():
|
||||||
"""Full pipeline: scan -> filter -> per-ticker deep dive."""
|
"""Full pipeline: scan -> filter -> per-ticker deep dive."""
|
||||||
import asyncio
|
import asyncio
|
||||||
import json as _json
|
|
||||||
from tradingagents.pipeline.macro_bridge import (
|
from tradingagents.pipeline.macro_bridge import (
|
||||||
parse_macro_output,
|
parse_macro_output,
|
||||||
filter_candidates,
|
filter_candidates,
|
||||||
|
|
@ -1293,10 +1292,14 @@ def run_pipeline():
|
||||||
output_dir = Path("results/macro_pipeline")
|
output_dir = Path("results/macro_pipeline")
|
||||||
|
|
||||||
console.print(f"\n[cyan]Running TradingAgents for {len(candidates)} tickers...[/cyan]")
|
console.print(f"\n[cyan]Running TradingAgents for {len(candidates)} tickers...[/cyan]")
|
||||||
with Live(Spinner("dots", text="Analyzing..."), console=console, transient=True):
|
try:
|
||||||
results = asyncio.run(
|
with Live(Spinner("dots", text="Analyzing..."), console=console, transient=True):
|
||||||
run_all_tickers(candidates, macro_context, config, analysis_date)
|
results = asyncio.run(
|
||||||
)
|
run_all_tickers(candidates, macro_context, config, analysis_date)
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
console.print(f"[red]Pipeline failed: {e}[/red]")
|
||||||
|
raise typer.Exit(1)
|
||||||
|
|
||||||
save_results(results, macro_context, output_dir)
|
save_results(results, macro_context, output_dir)
|
||||||
|
|
||||||
|
|
|
||||||
10
main.py
10
main.py
|
|
@ -1,11 +1,13 @@
|
||||||
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
||||||
from tradingagents.default_config import DEFAULT_CONFIG
|
|
||||||
|
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
# Load environment variables from .env file
|
# Load environment variables from .env file BEFORE importing any
|
||||||
|
# tradingagents modules so TRADINGAGENTS_* vars are visible to
|
||||||
|
# DEFAULT_CONFIG at import time.
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
|
|
||||||
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
||||||
|
from tradingagents.default_config import DEFAULT_CONFIG
|
||||||
|
|
||||||
# Create a custom config
|
# Create a custom config
|
||||||
config = DEFAULT_CONFIG.copy()
|
config = DEFAULT_CONFIG.copy()
|
||||||
config["deep_think_llm"] = "gpt-5-mini" # Use a different model
|
config["deep_think_llm"] = "gpt-5-mini" # Use a different model
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ dependencies = [
|
||||||
"langgraph>=0.4.8",
|
"langgraph>=0.4.8",
|
||||||
"pandas>=2.3.0",
|
"pandas>=2.3.0",
|
||||||
"parsel>=1.10.0",
|
"parsel>=1.10.0",
|
||||||
|
"python-dotenv>=1.0.0",
|
||||||
"pytz>=2025.2",
|
"pytz>=2025.2",
|
||||||
"questionary>=2.1.0",
|
"questionary>=2.1.0",
|
||||||
"rank-bm25>=0.2.2",
|
"rank-bm25>=0.2.2",
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,108 @@
|
||||||
|
"""Tests that TRADINGAGENTS_* environment variables override DEFAULT_CONFIG."""
|
||||||
|
|
||||||
|
import importlib
|
||||||
|
import os
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
class TestEnvOverridesDefaults:
|
||||||
|
"""Verify that setting TRADINGAGENTS_<KEY> env vars changes DEFAULT_CONFIG."""
|
||||||
|
|
||||||
|
def _reload_config(self):
|
||||||
|
"""Force-reimport default_config so the module-level dict is rebuilt."""
|
||||||
|
import tradingagents.default_config as mod
|
||||||
|
|
||||||
|
importlib.reload(mod)
|
||||||
|
return mod.DEFAULT_CONFIG
|
||||||
|
|
||||||
|
def test_llm_provider_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_LLM_PROVIDER": "openrouter"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["llm_provider"] == "openrouter"
|
||||||
|
|
||||||
|
def test_backend_url_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_BACKEND_URL": "http://localhost:1234"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["backend_url"] == "http://localhost:1234"
|
||||||
|
|
||||||
|
def test_deep_think_llm_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_DEEP_THINK_LLM": "deepseek/deepseek-r1"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["deep_think_llm"] == "deepseek/deepseek-r1"
|
||||||
|
|
||||||
|
def test_quick_think_llm_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_QUICK_THINK_LLM": "gpt-4o-mini"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["quick_think_llm"] == "gpt-4o-mini"
|
||||||
|
|
||||||
|
def test_mid_think_llm_none_by_default(self):
|
||||||
|
"""mid_think_llm defaults to None (falls back to quick_think_llm)."""
|
||||||
|
with patch.dict(os.environ, {}, clear=False):
|
||||||
|
# Remove the env var if it happens to be set
|
||||||
|
os.environ.pop("TRADINGAGENTS_MID_THINK_LLM", None)
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["mid_think_llm"] is None
|
||||||
|
|
||||||
|
def test_mid_think_llm_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_MID_THINK_LLM": "gpt-4o"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["mid_think_llm"] == "gpt-4o"
|
||||||
|
|
||||||
|
def test_empty_env_var_keeps_default(self):
|
||||||
|
"""An empty string is treated the same as unset (keeps the default)."""
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_LLM_PROVIDER": ""}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["llm_provider"] == "openai"
|
||||||
|
|
||||||
|
def test_empty_env_var_keeps_none_default(self):
|
||||||
|
"""An empty string for a None-default field stays None."""
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_DEEP_THINK_LLM_PROVIDER": ""}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["deep_think_llm_provider"] is None
|
||||||
|
|
||||||
|
def test_per_tier_provider_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_DEEP_THINK_LLM_PROVIDER": "anthropic"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["deep_think_llm_provider"] == "anthropic"
|
||||||
|
|
||||||
|
def test_per_tier_backend_url_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_MID_THINK_BACKEND_URL": "http://my-ollama:11434"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["mid_think_backend_url"] == "http://my-ollama:11434"
|
||||||
|
|
||||||
|
def test_max_debate_rounds_int(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_MAX_DEBATE_ROUNDS": "3"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["max_debate_rounds"] == 3
|
||||||
|
|
||||||
|
def test_max_debate_rounds_bad_value(self):
|
||||||
|
"""Non-numeric string falls back to hardcoded default."""
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_MAX_DEBATE_ROUNDS": "abc"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["max_debate_rounds"] == 1
|
||||||
|
|
||||||
|
def test_results_dir_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_RESULTS_DIR": "/tmp/my_results"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["results_dir"] == "/tmp/my_results"
|
||||||
|
|
||||||
|
def test_vendor_scanner_data_override(self):
|
||||||
|
with patch.dict(os.environ, {"TRADINGAGENTS_VENDOR_SCANNER_DATA": "alpha_vantage"}):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["data_vendors"]["scanner_data"] == "alpha_vantage"
|
||||||
|
|
||||||
|
def test_defaults_unchanged_when_no_env_set(self):
|
||||||
|
"""Without any TRADINGAGENTS_* vars, defaults are the original hardcoded values."""
|
||||||
|
# Clear all TRADINGAGENTS_ vars
|
||||||
|
env_clean = {k: v for k, v in os.environ.items() if not k.startswith("TRADINGAGENTS_")}
|
||||||
|
with patch.dict(os.environ, env_clean, clear=True):
|
||||||
|
cfg = self._reload_config()
|
||||||
|
assert cfg["llm_provider"] == "openai"
|
||||||
|
assert cfg["deep_think_llm"] == "gpt-5.2"
|
||||||
|
assert cfg["mid_think_llm"] is None
|
||||||
|
assert cfg["quick_think_llm"] == "gpt-5-mini"
|
||||||
|
assert cfg["backend_url"] == "https://api.openai.com/v1"
|
||||||
|
assert cfg["max_debate_rounds"] == 1
|
||||||
|
assert cfg["data_vendors"]["scanner_data"] == "yfinance"
|
||||||
|
|
@ -12,7 +12,9 @@ from typing import Any, List
|
||||||
from langchain_core.messages import AIMessage, ToolMessage
|
from langchain_core.messages import AIMessage, ToolMessage
|
||||||
|
|
||||||
|
|
||||||
MAX_TOOL_ROUNDS = 5 # safety limit to avoid infinite loops
|
# Most LLM tool-calling patterns resolve within 2-3 rounds;
|
||||||
|
# 5 provides headroom for complex scenarios while preventing runaway loops.
|
||||||
|
MAX_TOOL_ROUNDS = 5
|
||||||
|
|
||||||
|
|
||||||
def run_tool_loop(
|
def run_tool_loop(
|
||||||
|
|
|
||||||
|
|
@ -2,6 +2,8 @@ import os
|
||||||
import requests
|
import requests
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import json
|
import json
|
||||||
|
import threading
|
||||||
|
import time as _time
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from io import StringIO
|
from io import StringIO
|
||||||
|
|
||||||
|
|
@ -73,8 +75,6 @@ class ThirdPartyParseError(AlphaVantageError):
|
||||||
|
|
||||||
# ─── Rate-limited request helper ─────────────────────────────────────────────
|
# ─── Rate-limited request helper ─────────────────────────────────────────────
|
||||||
|
|
||||||
import threading
|
|
||||||
import time as _time
|
|
||||||
|
|
||||||
_rate_lock = threading.Lock()
|
_rate_lock = threading.Lock()
|
||||||
_call_timestamps: list[float] = []
|
_call_timestamps: list[float] = []
|
||||||
|
|
@ -83,14 +83,30 @@ _RATE_LIMIT = 75 # calls per minute (Alpha Vantage premium)
|
||||||
|
|
||||||
def _rate_limited_request(function_name: str, params: dict, timeout: int = 30) -> dict | str:
|
def _rate_limited_request(function_name: str, params: dict, timeout: int = 30) -> dict | str:
|
||||||
"""Make an API request with rate limiting (75 calls/min for premium key)."""
|
"""Make an API request with rate limiting (75 calls/min for premium key)."""
|
||||||
|
sleep_time = 0.0
|
||||||
with _rate_lock:
|
with _rate_lock:
|
||||||
now = _time.time()
|
now = _time.time()
|
||||||
# Remove timestamps older than 60 seconds
|
# Remove timestamps older than 60 seconds
|
||||||
_call_timestamps[:] = [t for t in _call_timestamps if now - t < 60]
|
_call_timestamps[:] = [t for t in _call_timestamps if now - t < 60]
|
||||||
if len(_call_timestamps) >= _RATE_LIMIT:
|
if len(_call_timestamps) >= _RATE_LIMIT:
|
||||||
sleep_time = 60 - (now - _call_timestamps[0]) + 0.1
|
sleep_time = 60 - (now - _call_timestamps[0]) + 0.1
|
||||||
_time.sleep(sleep_time)
|
|
||||||
|
# Sleep outside the lock to avoid blocking other threads
|
||||||
|
if sleep_time > 0:
|
||||||
|
_time.sleep(sleep_time)
|
||||||
|
|
||||||
|
# Re-check and register under lock to avoid races where multiple
|
||||||
|
# threads calculate similar sleep times and then all fire at once.
|
||||||
|
with _rate_lock:
|
||||||
|
now = _time.time()
|
||||||
|
_call_timestamps[:] = [t for t in _call_timestamps if now - t < 60]
|
||||||
|
if len(_call_timestamps) >= _RATE_LIMIT:
|
||||||
|
# Another thread filled the window while we slept — wait again
|
||||||
|
extra_sleep = 60 - (now - _call_timestamps[0]) + 0.1
|
||||||
|
_time.sleep(extra_sleep)
|
||||||
_call_timestamps.append(_time.time())
|
_call_timestamps.append(_time.time())
|
||||||
|
|
||||||
|
|
||||||
return _make_api_request(function_name, params, timeout=timeout)
|
return _make_api_request(function_name, params, timeout=timeout)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -131,6 +147,8 @@ def _make_api_request(function_name: str, params: dict, timeout: int = 30) -> di
|
||||||
)
|
)
|
||||||
except requests.exceptions.ConnectionError as exc:
|
except requests.exceptions.ConnectionError as exc:
|
||||||
raise ThirdPartyError(f"Connection error: function={function_name}, error={exc}")
|
raise ThirdPartyError(f"Connection error: function={function_name}, error={exc}")
|
||||||
|
except requests.exceptions.RequestException as exc:
|
||||||
|
raise ThirdPartyError(f"Request failed: function={function_name}, error={exc}")
|
||||||
|
|
||||||
# HTTP-level errors
|
# HTTP-level errors
|
||||||
if response.status_code == 401:
|
if response.status_code == 401:
|
||||||
|
|
@ -146,7 +164,13 @@ def _make_api_request(function_name: str, params: dict, timeout: int = 30) -> di
|
||||||
f"Server error: status={response.status_code}, function={function_name}, "
|
f"Server error: status={response.status_code}, function={function_name}, "
|
||||||
f"body={response.text[:200]}"
|
f"body={response.text[:200]}"
|
||||||
)
|
)
|
||||||
response.raise_for_status()
|
try:
|
||||||
|
response.raise_for_status()
|
||||||
|
except requests.exceptions.HTTPError as exc:
|
||||||
|
raise ThirdPartyError(
|
||||||
|
f"HTTP error: status={response.status_code}, function={function_name}, "
|
||||||
|
f"body={response.text[:200]}"
|
||||||
|
) from exc
|
||||||
|
|
||||||
response_text = response.text
|
response_text = response.text
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -201,7 +201,7 @@ def route_to_vendor(method: str, *args, **kwargs):
|
||||||
|
|
||||||
try:
|
try:
|
||||||
return impl_func(*args, **kwargs)
|
return impl_func(*args, **kwargs)
|
||||||
except AlphaVantageError:
|
except (AlphaVantageError, ConnectionError, TimeoutError):
|
||||||
continue # Any AV error triggers fallback to next vendor
|
continue # Any AV error or connection/timeout triggers fallback to next vendor
|
||||||
|
|
||||||
raise RuntimeError(f"No available vendor for '{method}'")
|
raise RuntimeError(f"No available vendor for '{method}'")
|
||||||
|
|
@ -1,45 +1,83 @@
|
||||||
import os
|
import os
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
# Load .env so that TRADINGAGENTS_* variables are available before
|
||||||
|
# DEFAULT_CONFIG is evaluated. CWD is checked first, then the project
|
||||||
|
# root (two levels up from this file). load_dotenv never overwrites
|
||||||
|
# variables that are already present in the environment.
|
||||||
|
load_dotenv()
|
||||||
|
load_dotenv(Path(__file__).resolve().parent.parent / ".env")
|
||||||
|
|
||||||
|
|
||||||
|
def _env(key: str, default=None):
|
||||||
|
"""Read ``TRADINGAGENTS_<KEY>`` from the environment.
|
||||||
|
|
||||||
|
Returns *default* when the variable is unset **or** empty, so that
|
||||||
|
``TRADINGAGENTS_MID_THINK_LLM=`` in a ``.env`` file is treated the
|
||||||
|
same as not setting it at all (preserving the ``None`` semantics for
|
||||||
|
"fall back to the parent setting").
|
||||||
|
"""
|
||||||
|
val = os.getenv(f"TRADINGAGENTS_{key.upper()}")
|
||||||
|
if not val: # None or ""
|
||||||
|
return default
|
||||||
|
return val
|
||||||
|
|
||||||
|
|
||||||
|
def _env_int(key: str, default=None):
|
||||||
|
"""Like :func:`_env` but coerces the value to ``int``."""
|
||||||
|
val = _env(key)
|
||||||
|
if val is None:
|
||||||
|
return default
|
||||||
|
try:
|
||||||
|
return int(val)
|
||||||
|
except (ValueError, TypeError):
|
||||||
|
return default
|
||||||
|
|
||||||
|
|
||||||
DEFAULT_CONFIG = {
|
DEFAULT_CONFIG = {
|
||||||
"project_dir": os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
|
"project_dir": os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
|
||||||
"results_dir": os.getenv("TRADINGAGENTS_RESULTS_DIR", "./results"),
|
"results_dir": _env("RESULTS_DIR", "./results"),
|
||||||
"data_cache_dir": os.path.join(
|
"data_cache_dir": os.path.join(
|
||||||
os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
|
os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
|
||||||
"dataflows/data_cache",
|
"dataflows/data_cache",
|
||||||
),
|
),
|
||||||
# LLM settings
|
# LLM settings — all overridable via TRADINGAGENTS_<KEY> env vars
|
||||||
"mid_think_llm": "qwen3.5:27b", # falls back to quick_think_llm when None
|
"llm_provider": _env("LLM_PROVIDER", "openai"),
|
||||||
"quick_think_llm": "qwen3.5:27b",
|
"deep_think_llm": _env("DEEP_THINK_LLM", "gpt-5.2"),
|
||||||
|
"mid_think_llm": _env("MID_THINK_LLM"), # falls back to quick_think_llm when None
|
||||||
|
"quick_think_llm": _env("QUICK_THINK_LLM", "gpt-5-mini"),
|
||||||
|
"backend_url": _env("BACKEND_URL", "https://api.openai.com/v1"),
|
||||||
# Per-role provider overrides (fall back to llm_provider / backend_url when None)
|
# Per-role provider overrides (fall back to llm_provider / backend_url when None)
|
||||||
"deep_think_llm_provider": "openrouter",
|
"deep_think_llm_provider": _env("DEEP_THINK_LLM_PROVIDER"), # e.g. "google", "anthropic", "openrouter"
|
||||||
"deep_think_llm": "deepseek/deepseek-r1-0528",
|
"deep_think_backend_url": _env("DEEP_THINK_BACKEND_URL"), # override backend URL for deep-think model
|
||||||
"deep_think_backend_url": None, # uses OpenRouter's default URL
|
"mid_think_llm_provider": _env("MID_THINK_LLM_PROVIDER"), # e.g. "ollama"
|
||||||
"mid_think_llm_provider": "ollama", # falls back to ollama
|
"mid_think_backend_url": _env("MID_THINK_BACKEND_URL"), # override backend URL for mid-think model
|
||||||
"mid_think_backend_url": "http://192.168.50.76:11434", # falls back to backend_url (ollama host)
|
"quick_think_llm_provider": _env("QUICK_THINK_LLM_PROVIDER"), # e.g. "openai", "ollama"
|
||||||
"quick_think_llm_provider": "ollama", # falls back to ollama
|
"quick_think_backend_url": _env("QUICK_THINK_BACKEND_URL"), # override backend URL for quick-think model
|
||||||
"quick_think_backend_url": "http://192.168.50.76:11434", # falls back to backend_url (ollama host)
|
|
||||||
# Provider-specific thinking configuration (applies to all roles unless overridden)
|
# Provider-specific thinking configuration (applies to all roles unless overridden)
|
||||||
"google_thinking_level": None, # "high", "minimal", etc.
|
"google_thinking_level": _env("GOOGLE_THINKING_LEVEL"), # "high", "minimal", etc.
|
||||||
"openai_reasoning_effort": None, # "medium", "high", "low"
|
"openai_reasoning_effort": _env("OPENAI_REASONING_EFFORT"), # "medium", "high", "low"
|
||||||
# Per-role provider-specific thinking configuration
|
# Per-role provider-specific thinking configuration
|
||||||
"deep_think_google_thinking_level": None,
|
"deep_think_google_thinking_level": _env("DEEP_THINK_GOOGLE_THINKING_LEVEL"),
|
||||||
"deep_think_openai_reasoning_effort": None,
|
"deep_think_openai_reasoning_effort": _env("DEEP_THINK_OPENAI_REASONING_EFFORT"),
|
||||||
"mid_think_google_thinking_level": None,
|
"mid_think_google_thinking_level": _env("MID_THINK_GOOGLE_THINKING_LEVEL"),
|
||||||
"mid_think_openai_reasoning_effort": None,
|
"mid_think_openai_reasoning_effort": _env("MID_THINK_OPENAI_REASONING_EFFORT"),
|
||||||
"quick_think_google_thinking_level": None,
|
"quick_think_google_thinking_level": _env("QUICK_THINK_GOOGLE_THINKING_LEVEL"),
|
||||||
"quick_think_openai_reasoning_effort": None,
|
"quick_think_openai_reasoning_effort": _env("QUICK_THINK_OPENAI_REASONING_EFFORT"),
|
||||||
# Debate and discussion settings
|
# Debate and discussion settings
|
||||||
"max_debate_rounds": 1,
|
"max_debate_rounds": _env_int("MAX_DEBATE_ROUNDS", 1),
|
||||||
"max_risk_discuss_rounds": 1,
|
"max_risk_discuss_rounds": _env_int("MAX_RISK_DISCUSS_ROUNDS", 1),
|
||||||
"max_recur_limit": 100,
|
"max_recur_limit": _env_int("MAX_RECUR_LIMIT", 100),
|
||||||
# Data vendor configuration
|
# Data vendor configuration
|
||||||
# Category-level configuration (default for all tools in category)
|
# Category-level configuration (default for all tools in category)
|
||||||
"data_vendors": {
|
"data_vendors": {
|
||||||
"core_stock_apis": "yfinance", # Options: alpha_vantage, yfinance
|
"core_stock_apis": _env("VENDOR_CORE_STOCK_APIS", "yfinance"),
|
||||||
"technical_indicators": "yfinance", # Options: alpha_vantage, yfinance
|
"technical_indicators": _env("VENDOR_TECHNICAL_INDICATORS", "yfinance"),
|
||||||
"fundamental_data": "yfinance", # Options: alpha_vantage, yfinance
|
"fundamental_data": _env("VENDOR_FUNDAMENTAL_DATA", "yfinance"),
|
||||||
"news_data": "yfinance", # Options: alpha_vantage, yfinance
|
"news_data": _env("VENDOR_NEWS_DATA", "yfinance"),
|
||||||
"scanner_data": "alpha_vantage", # Options: alpha_vantage (primary), yfinance (fallback)
|
"scanner_data": _env("VENDOR_SCANNER_DATA", "yfinance"),
|
||||||
},
|
},
|
||||||
# Tool-level configuration (takes precedence over category-level)
|
# Tool-level configuration (takes precedence over category-level)
|
||||||
"tool_vendors": {
|
"tool_vendors": {
|
||||||
|
|
|
||||||
|
|
@ -139,9 +139,10 @@ class ScannerGraph:
|
||||||
}
|
}
|
||||||
|
|
||||||
if self.debug:
|
if self.debug:
|
||||||
trace = []
|
# stream() yields partial state updates; use invoke() for the
|
||||||
|
# full accumulated state and print chunks for debugging only.
|
||||||
for chunk in self.graph.stream(initial_state):
|
for chunk in self.graph.stream(initial_state):
|
||||||
trace.append(chunk)
|
print(f"[scanner debug] chunk keys: {list(chunk.keys())}")
|
||||||
return trace[-1] if trace else initial_state
|
# Fall through to invoke() for the correct accumulated result
|
||||||
|
|
||||||
return self.graph.invoke(initial_state)
|
return self.graph.invoke(initial_state)
|
||||||
|
|
|
||||||
|
|
@ -238,10 +238,10 @@ async def run_all_tickers(
|
||||||
List of TickerResult in completion order.
|
List of TickerResult in completion order.
|
||||||
"""
|
"""
|
||||||
semaphore = asyncio.Semaphore(max_concurrent)
|
semaphore = asyncio.Semaphore(max_concurrent)
|
||||||
loop = asyncio.get_event_loop()
|
|
||||||
|
|
||||||
async def _run_one(candidate: StockCandidate) -> TickerResult:
|
async def _run_one(candidate: StockCandidate) -> TickerResult:
|
||||||
async with semaphore:
|
async with semaphore:
|
||||||
|
loop = asyncio.get_running_loop()
|
||||||
# TradingAgentsGraph is synchronous — run it in a thread pool
|
# TradingAgentsGraph is synchronous — run it in a thread pool
|
||||||
return await loop.run_in_executor(
|
return await loop.run_in_executor(
|
||||||
None,
|
None,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue