podcaster / models.py
marks
Implemented pydantic
f71cd77
raw
history blame
506 Bytes
from pydantic import BaseModel, Field
from typing import List, Optional
class Message(BaseModel):
role: str
content: str
class OpenRouterRequest(BaseModel):
model: str
messages: List[Message]
class Choice(BaseModel):
message: Message
index: int = 0
finish_reason: Optional[str] = None
class OpenRouterResponse(BaseModel):
id: str
choices: List[Choice]
model: str
class OpenRouterModel(BaseModel):
id: str
name: str
description: Optional[str] = None