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