from pydantic import BaseModel
from typing import List, Optional, Dict, Any

class ChatRequest(BaseModel):
    message: str
    session_id: Optional[str] = None
    history: Optional[List[List[str]]] = []
    persona_key: Optional[str] = None  # 新增：接收前端传来的persona

class ChatResponse(BaseModel):
    reply: str
    session_id: str
    status_info: Dict[str, Any]
    history: List[List[str]]