"""
简单配置管理
"""
import os
from typing import Literal
from dotenv import load_dotenv

# 加载 .env 文件
load_dotenv()

# 模型配置
MODEL_CONFIG = {
    "gpt-3.5-turbo": {
        "input_cost_per_1k": 0.0005,
        "output_cost_per_1k": 0.0015,
        "max_tokens": 4096,
    },
    "gpt-4o-mini": {
        "input_cost_per_1k": 0.00015,
        "output_cost_per_1k": 0.0006,
        "max_tokens": 128000,
    },
    "gpt-4o": {
        "input_cost_per_1k": 0.005,
        "output_cost_per_1k": 0.015,
        "max_tokens": 128000,
    },
}

# 路由阈值
ROUTING_THRESHOLDS = {
    "simple": 100,   # < 100 tokens -> gpt-3.5-turbo
    "medium": 500,   # < 500 tokens -> gpt-4o-mini
    # >= 500 tokens -> gpt-4o
}

# API Key
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")