from typing import List, Optional
from pydantic import BaseModel, Field


class QuestionRequest(BaseModel):
    """问题请求模型"""
    question: str = Field(..., description="用户提出的问题", min_length=1, max_length=1000)
    knowledgeType: str = Field(..., description="知识类型")


class QueryResponse(BaseModel):
    """查询词响应模型"""
    task_id: int = Field(..., description="任务ID")
    queries: List[str] = Field(..., description="生成的查询词列表")
    original_question: str = Field(..., description="原始问题")
    total_count: int = Field(..., description="生成的查询词数量")
    status: int = Field(..., description="任务状态：0-待执行，1-执行中，2-成功，3-失败")


class HealthResponse(BaseModel):
    """健康检查响应模型"""
    status: str = Field(..., description="服务状态")
    message: str = Field(..., description="状态消息")