from fastapi import APIRouter, HTTPException, Request
from service import ChatService
from utils import jingrow_api_verify_and_billing
from settings import settings
import json
from functools import wraps

router = APIRouter(prefix=settings.router_prefix)
service = ChatService()

def dynamic_billing_wrapper(func):
    """动态API扣费装饰器，使用模型名称作为API名称"""
    @wraps(func)
    async def wrapper(data: dict, request: Request):
        api_name = settings.default_api_name  # 使用settings中的默认API名称
        if "model" in data:
            api_name = data["model"]
        
        dynamic_decorator = jingrow_api_verify_and_billing(api_name=api_name)
        decorated_func = dynamic_decorator(func)
        return await decorated_func(**{"data": data, "request": request})
    
    return wrapper

@router.post(settings.chat_route)
@dynamic_billing_wrapper
async def chat_api(data: dict, request: Request):
    """
    通用文本聊天API，支持OpenAI和豆包等模型的请求格式
    
    Args:
        data: 包含以下字段的字典：
            - messages: 消息列表，每个消息包含 role 和 content（必需）
            - model: 选择使用的模型（可选，默认为配置的默认模型）
            - temperature: 温度参数（可选，默认为0.7）
            - top_p: top_p参数（可选，默认为0.9）
            - max_tokens: 最大生成token数（可选，默认为2048）
        request: FastAPI 请求对象
        
    Returns:
        AI生成的回复内容
    """
    if "messages" not in data:
        raise HTTPException(status_code=400, detail="缺少messages参数")
    
    try:
        if "model" in data:
            service.model = data["model"]
        if "temperature" in data:
            service.temperature = data["temperature"]
        if "top_p" in data:
            service.top_p = data["top_p"]
        if "max_tokens" in data:
            service.max_tokens = data["max_tokens"]
        
        result = await service.chat(data["messages"])
        return result
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))