File size: 440 Bytes
f71cd77 25262d5 f71cd77 25262d5 f71cd77 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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
|