chat.py 248 B

12345678910111213141516
  1. from typing import List
  2. from pydantic import BaseModel
  3. class Message(BaseModel):
  4. role: str
  5. content: str
  6. class ChatCompletionRequest(BaseModel):
  7. messages: List[Message]
  8. user: str
  9. class NotifyRequest(BaseModel):
  10. message: str