OwusuBlessing
21 lines
289 B
Python
21 lines
289 B
Python
# api/models/requests.py
|
|
from pydantic import BaseModel
|
|
from typing import List
|
|
|
|
class BaseRequest(BaseModel):
|
|
pass
|
|
|
|
class ChatMessage(BaseModel):
|
|
role: str # "human" or "ai"
|
|
content: str
|
|
|
|
|
|
class ChatRequest(BaseModel):
|
|
query: str
|
|
history: List[ChatMessage] = []
|
|
|
|
|
|
|
|
|
|
|