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