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