pipelines/schemas.py
Timothy J. Baek 91f58c52e5 feat: valves
2024-05-27 19:04:35 -07:00

24 lines
455 B
Python

from typing import List
from pydantic import BaseModel, ConfigDict
class OpenAIChatMessage(BaseModel):
role: str
content: str
model_config = ConfigDict(extra="allow")
class OpenAIChatCompletionForm(BaseModel):
stream: bool = True
model: str
messages: List[OpenAIChatMessage]
model_config = ConfigDict(extra="allow")
class ValveForm(BaseModel):
model: str
body: dict
model_config = ConfigDict(extra="allow")