84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
"""Portfolio Manager configuration.
|
|
|
|
Integrates with the existing ``tradingagents/default_config.py`` pattern,
|
|
reading all portfolio settings from ``TRADINGAGENTS_<KEY>`` env vars.
|
|
|
|
All env-var reading is delegated to the shared helpers in
|
|
``tradingagents.default_config`` — that module is the single entry point
|
|
for .env loading and the ``_env*`` helper functions.
|
|
|
|
Usage::
|
|
|
|
from tradingagents.portfolio.config import get_portfolio_config, validate_config
|
|
|
|
cfg = get_portfolio_config()
|
|
validate_config(cfg)
|
|
print(cfg["max_positions"]) # 15
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
# Importing default_config triggers its module-level load_dotenv() calls,
|
|
# which loads the .env file (CWD first, then project root) before we read
|
|
# any TRADINGAGENTS_* variables below. This is the single source of truth
|
|
# for .env loading — no separate load_dotenv() call needed here.
|
|
from tradingagents.default_config import _env, _env_float, _env_int
|
|
|
|
|
|
PORTFOLIO_CONFIG: dict = {
|
|
"supabase_connection_string": os.getenv("SUPABASE_CONNECTION_STRING", ""),
|
|
# PORTFOLIO_DATA_DIR takes precedence; falls back to TRADINGAGENTS_REPORTS_DIR,
|
|
# then to "reports" (relative to CWD) — same default as report_paths.REPORTS_ROOT.
|
|
"data_dir": os.getenv("PORTFOLIO_DATA_DIR") or _env("REPORTS_DIR", "reports"),
|
|
"max_positions": 15,
|
|
"max_position_pct": 0.15,
|
|
"max_sector_pct": 0.35,
|
|
"min_cash_pct": 0.05,
|
|
"default_budget": 100_000.0,
|
|
}
|
|
|
|
|
|
def get_portfolio_config() -> dict:
|
|
"""Return the merged portfolio config (defaults overridden by env vars).
|
|
|
|
Returns:
|
|
A dict with all portfolio configuration keys.
|
|
"""
|
|
cfg = dict(PORTFOLIO_CONFIG)
|
|
cfg["supabase_connection_string"] = os.getenv("SUPABASE_CONNECTION_STRING", cfg["supabase_connection_string"])
|
|
cfg["data_dir"] = os.getenv("PORTFOLIO_DATA_DIR") or _env("REPORTS_DIR", cfg["data_dir"])
|
|
cfg["max_positions"] = _env_int("PM_MAX_POSITIONS", cfg["max_positions"])
|
|
cfg["max_position_pct"] = _env_float("PM_MAX_POSITION_PCT", cfg["max_position_pct"])
|
|
cfg["max_sector_pct"] = _env_float("PM_MAX_SECTOR_PCT", cfg["max_sector_pct"])
|
|
cfg["min_cash_pct"] = _env_float("PM_MIN_CASH_PCT", cfg["min_cash_pct"])
|
|
cfg["default_budget"] = _env_float("PM_DEFAULT_BUDGET", cfg["default_budget"])
|
|
return cfg
|
|
|
|
|
|
def validate_config(cfg: dict) -> None:
|
|
"""Validate a portfolio config dict, raising ValueError on invalid values.
|
|
|
|
Args:
|
|
cfg: Config dict as returned by ``get_portfolio_config()``.
|
|
|
|
Raises:
|
|
ValueError: With a descriptive message on the first failed check.
|
|
"""
|
|
if cfg["max_positions"] < 1:
|
|
raise ValueError(f"max_positions must be >= 1, got {cfg['max_positions']}")
|
|
if not (0 < cfg["max_position_pct"] <= 1.0):
|
|
raise ValueError(f"max_position_pct must be in (0, 1], got {cfg['max_position_pct']}")
|
|
if not (0 < cfg["max_sector_pct"] <= 1.0):
|
|
raise ValueError(f"max_sector_pct must be in (0, 1], got {cfg['max_sector_pct']}")
|
|
if not (0 <= cfg["min_cash_pct"] < 1.0):
|
|
raise ValueError(f"min_cash_pct must be in [0, 1), got {cfg['min_cash_pct']}")
|
|
if cfg["default_budget"] <= 0:
|
|
raise ValueError(f"default_budget must be > 0, got {cfg['default_budget']}")
|
|
if cfg["min_cash_pct"] + cfg["max_position_pct"] > 1.0:
|
|
raise ValueError(
|
|
f"min_cash_pct ({cfg['min_cash_pct']}) + max_position_pct ({cfg['max_position_pct']}) "
|
|
f"must be <= 1.0"
|
|
)
|