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 OpenRouterChoice(BaseModel): | |
message: Message | |
class OpenRouterResponse(BaseModel): | |
choices: List[OpenRouterChoice] | |
class OpenRouterModel(BaseModel): | |
id: str | |
name: str | |
description: Optional[str] = None | |