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
    thinking: Optional[str] = None
    searching: Optional[str] = None

# 帖子请求
class CircleRequest(BaseModel):
    id: str  # 帖子的主键


# 兴趣圈提示词模版
class CirclePromptConfig(BaseModel):
    appName: str          # 包名，作为唯一标识
    name: str             # 兴趣圈名称
    role: str             # AI 扮演的角色
    style: str            # 回复风格描述
    keywords: List[str] = []   # 推荐使用的关键词
    forbidden: List[str] = []  # 禁止使用的词语
    extra_instruction: str    # 是给模型更细化的行为指引，可以包含在最终提示词中。

# Ai的请求对象
class ChatRequest(BaseModel):
    messages: List[ChatMessage]
    model: Optional[str] = config.MODEL_NAME
    stream: Optional[bool] = False
    source: Optional[str] = None  # source=app 时走第三方 token 认证
    token: Optional[str] = None   # App 端传入的第三方 token


# Ai的返回对象
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
    type: str = "answer"  # "thinking"=AI思考开过车delta | "searching"=搜索状态/关键词 | "answer" = 正式回答 delta（现有逻辑）