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 | |