chat-ai-api/app/schemas/chat.py

from pydantic import BaseModel
from typing import List, Optional, Dict, Any
from datetime import datetime
from ..core.ark_client import config


class ChatMessage(BaseModel):
    role: str  # "user" | "assistant" | "system"
    content: str
    timestamp: Optional[datetime] = None
    response_id: Optional[str] = None


class ChatRequest(BaseModel):
    messages: List[ChatMessage]
    model: Optional[str] = config.MODEL_NAME
    temperature: Optional[float] = config.TEMPERATURE
    max_tokens: Optional[int] = config.MAX_TOKENS
    stream: Optional[bool] = False
    source: Optional[str] = None  # source=app 时走第三方 token 认证
    token: Optional[str] = None   # App 端传入的第三方 token


class ChatResponse(BaseModel):
    message: ChatMessage
    model: str
    usage: Optional[Dict[str, Any]] = None
    response_id: Optional[str] = None


class StreamResponse(BaseModel):
    content: str
    finished: bool
    model: str
    timestamp: datetime