21 lines
531 B
Python
21 lines
531 B
Python
from pydantic import BaseModel, Field, HttpUrl
|
|
from typing import Literal, List
|
|
|
|
class ChatMessage(BaseModel):
|
|
role: Literal["user", "assistant", "system"]
|
|
content: str
|
|
|
|
class ChatRequest(BaseModel):
|
|
model: str = Field(default="llama2")
|
|
messages: List[ChatMessage] = Field(default_factory=list)
|
|
|
|
class AuthCallbackRequest(BaseModel):
|
|
code: str
|
|
code_verifier: str
|
|
redirect_uri: str
|
|
|
|
class AuthConfigResponse(BaseModel):
|
|
issuer: HttpUrl
|
|
authorization_endpoint: HttpUrl
|
|
token_endpoint: HttpUrl
|