16 lines
284 B
Python
16 lines
284 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] = []
|