shudao-main/shudao-chat-py/models/report.py

"""
报告相关数据模型
"""
from pydantic import BaseModel, Field
from typing import Optional


class ReportCompleteFlowRequest(BaseModel):
    """完整报告生成流程请求"""
    user_question: str = Field(..., description="用户问题")
    window_size: int = Field(default=5, description="窗口大小")
    n_results: int = Field(default=5, description="结果数量")
    ai_conversation_id: Optional[int] = Field(default=None, description="AI对话ID")
    is_network_search_enabled: bool = Field(default=False, description="是否启用联网搜索")
    enable_online_model: bool = Field(default=False, description="是否启用在线模型")


class UpdateAIMessageRequest(BaseModel):
    """更新AI消息请求"""
    ai_message_id: int = Field(..., description="AI消息ID")
    content: str = Field(..., description="消息内容")


class StopSSERequest(BaseModel):
    """停止SSE请求"""
    ai_conversation_id: int = Field(..., description="AI对话ID")


class StreamChatRequest(BaseModel):
    """流式聊天请求（用于降级）"""
    message: str = Field(..., description="用户消息")
    ai_conversation_id: Optional[int] = Field(default=None, description="AI对话ID")
    business_type: int = Field(default=0, description="业务类型")