# DeepSeek 模型配置
import os

# 价格配置 (CNY per 1M tokens)
TOKEN_PRICE_INPUT = 400.0 / 1000000   # 4元 / 1M tokens
TOKEN_PRICE_OUTPUT = 1600.0 / 1000000 # 16元 / 1M tokens

# 最低余额要求
MIN_BALANCE = 0.01

# API 配置 (优先从环境变量获取)
def get_config():
    return {
        'api_url': os.getenv('DEEPSEEK_API_URL', 'https://api.deepseek.com'),
        'api_key': os.getenv('DEEPSEEK_API_KEY')
    }