import os

DEFAULT_CONFIG = {
    "project_dir": "/home/jaskirat/coding/TradingAgents",
    # "results_dir": "/data/coding/TradingAgents/results",
    "results_dir": "/home/jaskirat/coding/TradingAgents/results",
    "data_dir": "/data",
    "data_cache_dir": '/home/jaskirat/coding/TradingAgents/dataflows/data_cache',
    # LLM settings
    "llm_provider": "ollama",
    # "deep_think_llm": "hf.co/unsloth/Qwen3-30B-A3B-Instruct-2507-GGUF:Q5_K_XL",
    # "quick_think_llm": "hf.co/unsloth/Qwen3-30B-A3B-Instruct-2507-GGUF:Q5_K_XL",
    # "deep_think_llm": "hf.co/unsloth/gpt-oss-20b-GGUF:F16",
    # "quick_think_llm": "hf.co/unsloth/gpt-oss-20b-GGUF:F16",
    # "deep_think_llm": "qwen3:30b",
    # "quick_think_llm": "qwen3:30b",
    # "deep_think_llm": "gpt-oss",
    # "quick_think_llm": "gpt-oss",
    # "temperature": 0.1,
    # "deep_think_llm": "gpt-5-mini",
    # "quick_think_llm": "gpt-5-mini",
    "deep_think_llm": "glm-4.7",
    "quick_think_llm": "glm-4.7",
    "temperature": 0.7,
    "top_p": 0.95,
    "backend_url": "http://localhost:8080/v1",
    # "backend_url": "https://api.openai.com/v1",
    # "backend_url": "https://api.z.ai/api/paas/v4",
    # "backend_embedding_url": "https://api.openai.com/v1",
    "backend_embedding_url": "http://localhost:8081/v1",
    # "backend_embedding_url":"https://api.z.ai/api/v1",
    # "backend_url": "http://localhost:11434/v1",
    "api_key": os.environ.get("OPENAI_API_KEY", "your-api-key-here"),
    "embedding_api_key": os.environ.get("OPENAI_API_KEY", "your-api-key-here"),
    # Debate and discussion settings
    "max_debate_rounds": 5,
    "max_risk_discuss_rounds": 5,
    "max_recur_limit": 100,
    # Data vendor configuration
    # Category-level configuration (default for all tools in category)
    "data_vendors": {
        "core_stock_apis": "yfinance",       # Options: yfinance, alpha_vantage, local
        "technical_indicators": "yfinance",  # Options: yfinance, alpha_vantage, local
        "fundamental_data": "alpha_vantage", # Options: openai, alpha_vantage, local
        "news_data": "local_news",        # Options: openai, alpha_vantage, google, local_news
    },
    # Tool-level configuration (takes precedence over category-level)
    "tool_vendors": {
        # Example: "get_stock_data": "alpha_vantage",  # Override category default
        # Example: "get_news": "openai",               # Override category default
        "get_stock_data": "yfinance",
        "get_indicators": "yfinance",
        "get_fundamentals": "alpha_vantage",
        "get_balance_sheet": "yfinance",
        "get_cashflow": "yfinance",
        "get_income_statement": "yfinance",
        "get_news": "local_news",
        "get_global_news": "openai",
        # "get_insider_sentiment": "na",
        "get_insider_transactions": "yfinance",
    },
}

# docker run --rm -it --gpus all   -v ~/.cache/llama.cpp:/root/.cache/llama.cpp   -p 8080:8080   ghcr.io/ggml-org/llama.cpp:server-cuda   -hf unsloth/GLM-4.7-Flash-GGUF:Q4_K_M   --port 8080   --host 0.0.0.0   -ngl 99   --jinja   --ctx-size 32768 --min-p 0.01 --embeddings --pooling cls --ubatch-size 8192