scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(types/completion.py): fix completion configdict pydantic v1 error

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-14 16:35:23 -07:00
parent c503c471dc
commit 811406a2c6
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/types/completion.py
View file

@ -1,4 +1,4 @@
from typing import List, Optional, Union, Iterable
from typing import List, Optional, Union, Iterable, cast
from pydantic import ConfigDict, BaseModel, validator, VERSION
@ -15,7 +15,9 @@ def get_model_config() -> ConfigDict:
if is_pydantic_v2() >= 2:
model_config = ConfigDict(extra="allow", protected_namespaces=()) # type: ignore
else:
model_config = ConfigDict(extra="allow") # No protected_namespaces for v1
from pydantic import Extra
model_config = ConfigDict(extra=Extra.allow) # type: ignore
return model_config