TradingAgents/backend/check_setup.py

75 lines
2.0 KiB
Python

#!/usr/bin/env python3
"""
Check if the environment is properly set up for TradingAgents
"""
import os
import sys
print("🔍 TradingAgents Environment Check")
print("=" * 50)
# Check Python version
print(f"✓ Python version: {sys.version.split()[0]}")
# Check required packages
required_packages = [
'fastapi',
'uvicorn',
'pydantic',
'openai',
'langchain',
'langchain_openai',
'langgraph'
]
print("\nChecking required packages:")
missing_packages = []
for package in required_packages:
try:
__import__(package.replace('-', '_'))
print(f"{package} is installed")
except ImportError:
print(f"{package} is NOT installed")
missing_packages.append(package)
# Check environment variables
print("\nChecking environment variables:")
env_vars = {
'OPENAI_API_KEY': 'Required for AI agents',
'FINNHUB_API_KEY': 'Required for market data',
'REDDIT_CLIENT_ID': 'Optional for sentiment analysis',
'REDDIT_CLIENT_SECRET': 'Optional for sentiment analysis'
}
missing_required = []
for var, description in env_vars.items():
value = os.getenv(var)
if value:
print(f"{var} is set ({description})")
else:
if 'Required' in description:
print(f"{var} is NOT set ({description})")
missing_required.append(var)
else:
print(f"{var} is not set ({description})")
# Summary
print("\n" + "=" * 50)
if missing_packages:
print("❌ Missing packages:")
for pkg in missing_packages:
print(f" - {pkg}")
print("\nInstall with: pip install -r requirements.txt")
if missing_required:
print("❌ Missing required environment variables:")
for var in missing_required:
print(f" - {var}")
print("\nAdd these to your .env file")
if not missing_packages and not missing_required:
print("✅ Environment is properly configured!")
print("\nYou can now run:")
print(" uv run python3 run_api.py")
else:
print("\n⚠️ Fix the issues above before running the API")