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