"""
数据模型定义
"""
from pydantic import BaseModel
from typing import Dict, List, Optional, Any


class ChatMessage(BaseModel):
    role: str
    content: str
    images: Optional[List[str]] = None
    files: Optional[List[str]] = None


class ChatRequest(BaseModel):
    model: str = "qwen-plus"
    messages: List[Dict[str, Any]]
    stream: bool = True
    temperature: Optional[float] = 0.7
    max_tokens: Optional[int] = 2000


class ModelInfo(BaseModel):
    id: str
    name: str
    description: str
    maxTokens: int
    provider: str