File size: 348 Bytes
eabc9cd
 
 
 
 
4617b5f
 
f26cbb9
 
1
2
3
4
5
6
7
8
9
10
from pydantic import BaseModel, Field


class ChatsModel(BaseModel):
    prompt: str = Field(..., example="Hello, I'm a chatbot")
    histories: list[dict[str, str]] | None = Field(
        None, example=[{"user": "Hello", "assistant": "Hi"}])
    openai_key: str | None = Field(
        None, descriptions="Open AI API key", example="OPENAI_KEY")