diff --git a/schemas.py b/schemas.py index 5cad1e6..6f90d68 100644 --- a/schemas.py +++ b/schemas.py @@ -12,6 +12,6 @@ class OpenAIChatMessage(BaseModel): class OpenAIChatCompletionForm(BaseModel): stream: bool = True model: str - messages: List[dict] + messages: List[OpenAIChatMessage] model_config = ConfigDict(extra="allow")