schemas.py 584 B

1234567891011121314151617181920212223242526
  1. """数据模型"""
  2. from pydantic import BaseModel
  3. from typing import Optional, Dict, Any
  4. from datetime import datetime
  5. class ChatMessage(BaseModel):
  6. """聊天消息"""
  7. role: str # "user" 或 "assistant"
  8. content: str
  9. timestamp: datetime = datetime.now()
  10. class ChatRequest(BaseModel):
  11. """聊天请求"""
  12. message: str
  13. conversation_id: Optional[str] = None
  14. context: Optional[Dict[str, Any]] = None # 学习上下文
  15. class ChatResponse(BaseModel):
  16. """聊天响应"""
  17. reply: str
  18. conversation_id: str
  19. agent_name: str = "编程导师"