Skip to content

Commit 406cc4d

Browse files
author
Paweł Kędzia
committed
Update OpenAIChatModel to inherit from BaseModelOptions instead of pydantic.BaseModel and adjust imports accordingly.
1 parent 17e7837 commit 406cc4d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llm_router_lib/data_models/openai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
from pydantic import BaseModel
21
from typing import List, Dict, Any
32

43
from llm_router_api.base.constants_base import DEFAULT_EP_LANGUAGE
4+
from llm_router_lib.data_models.base_model import BaseModelOptions
55

66

7-
class OpenAIChatModel(BaseModel):
7+
class OpenAIChatModel(BaseModelOptions):
88
model: str
99
messages: List[Dict[str, Any]]
1010

0 commit comments

Comments
 (0)