Spaces:
Sleeping
Sleeping
File size: 4,432 Bytes
a2cbcac | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | import json
import os
from pathlib import Path
from typing import Dict, Any, Optional
from dotenv import load_dotenv
load_dotenv()
class Config:
def __init__(self):
self._config_data = self._load_config()
self._api_keys = self._load_api_keys()
def _load_config(self) -> Dict[str, Any]:
config_path = Path(__file__).parent / "config.json"
if not config_path.exists():
raise FileNotFoundError(f"config.json not found at {config_path}")
with open(config_path, 'r', encoding='utf-8') as f:
return json.load(f)
def _load_api_keys(self) -> Dict[str, Optional[str]]:
return {
'openai': os.getenv('OPENAI_API_KEY'),
'anthropic': os.getenv('ANTHROPIC_API_KEY'),
'finnhub': os.getenv('FINNHUB_API_KEY'),
'firecrawl': os.getenv('FIRECRAWL_API_KEY'),
'perplexity': os.getenv('PERPLEXITY_API_KEY')
}
# API Keys
@property
def api_keys(self) -> Dict[str, Optional[str]]:
return self._api_keys
def get_api_key(self, service: str) -> Optional[str]:
return self._api_keys.get(service)
# Trading Configuration
@property
def market_open_time(self) -> str:
return self._config_data['trading']['market_open_time']
@property
def trading_timezone(self) -> str:
return self._config_data['trading']['timezone']
# Portfolio Configuration
@property
def portfolio_historical_context_count(self) -> int:
return self._config_data['portfolio']['historical_context_count']
# News Configuration
@property
def news_significance_threshold(self) -> float:
return self._config_data['news']['significance_threshold']
@property
def news_moderate_threshold(self) -> float:
return self._config_data['news']['moderate_threshold']
@property
def news_max_per_minute(self) -> int:
return self._config_data['news']['max_news_per_minute']
@property
def news_sample_count(self) -> int:
return self._config_data['news']['sample_news_count']
@property
def news_enable_firecrawl(self) -> bool:
return self._config_data['news']['enable_firecrawl_extraction']
@property
def news_valid_sources(self) -> list:
return self._config_data['news']['valid_sources']
# Models Configuration
@property
def model_portfolio_manager(self) -> Dict[str, Any]:
"""Get portfolio manager model configuration."""
config_value = self._config_data['models']['portfolio_manager']
# Handle both legacy string format and new dict format
if isinstance(config_value, str):
return {"provider": "openai", "model": config_value, "temperature": 0.7}
return config_value
@property
def model_assess_significance(self) -> Dict[str, Any]:
"""Get assess significance model configuration."""
config_value = self._config_data['models']['assess_significance']
# Handle both legacy string format and new dict format
if isinstance(config_value, str):
return {"provider": "openai", "model": config_value, "temperature": 0.5}
return config_value
@property
def model_enhanced_summary(self) -> Dict[str, Any]:
"""Get enhanced summary model configuration."""
config_value = self._config_data['models']['enhanced_summary']
# Handle both legacy string format and new dict format
if isinstance(config_value, str):
return {"provider": "openai", "model": config_value, "temperature": 0.3}
return config_value
@property
def model_nlp_features(self) -> Dict[str, Any]:
"""Get NLP features model configuration."""
config_value = self._config_data['models']['nlp_features']
# Handle both legacy string format and new dict format
if isinstance(config_value, str):
return {"provider": "openai", "model": config_value, "temperature": 0.3}
return config_value
# Output Configuration
@property
def csv_output_path(self) -> str:
return self._config_data['output']['csv_path']
@property
def reports_output_path(self) -> str:
return self._config_data['output']['reports_path']
# Global config instance
config = Config() |