From 811406a2c69bdc41eb21f731906328cd2fc2a4a9 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 14 May 2024 16:35:23 -0700 Subject: [PATCH] fix(types/completion.py): fix completion configdict pydantic v1 error --- litellm/types/completion.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/types/completion.py b/litellm/types/completion.py index e3d605c915e..87a7629dafe 100644 --- a/litellm/types/completion.py +++ b/litellm/types/completion.py @@ -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