schemas.py 930 B

1234567891011121314151617181920212223
  1. from typing import List, Optional
  2. from pydantic import BaseModel, Field
  3. class QuestionRequest(BaseModel):
  4. """问题请求模型"""
  5. question: str = Field(..., description="用户提出的问题", min_length=1, max_length=1000)
  6. knowledgeType: str = Field(..., description="知识类型")
  7. class QueryResponse(BaseModel):
  8. """查询词响应模型"""
  9. task_id: int = Field(..., description="任务ID")
  10. queries: List[str] = Field(..., description="生成的查询词列表")
  11. original_question: str = Field(..., description="原始问题")
  12. total_count: int = Field(..., description="生成的查询词数量")
  13. status: int = Field(..., description="任务状态:0-待执行,1-执行中,2-成功,3-失败")
  14. class HealthResponse(BaseModel):
  15. """健康检查响应模型"""
  16. status: str = Field(..., description="服务状态")
  17. message: str = Field(..., description="状态消息")