phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fix linting issues

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-09 22:55:18 +05:30
parent ba7026639d
commit befbcbb7ac
2 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/_types.py
View file

@ -16,6 +16,13 @@ class LiteLLMBase(BaseModel):
# if using pydantic v1 # if using pydantic v1
return self.dict() return self.dict()
def fields_set(self):
try:
return self.model_fields_set # noqa
except:
# if using pydantic v1
return self.__fields_set__
######### Request Class Definition ###### ######### Request Class Definition ######
class ProxyChatCompletionRequest(LiteLLMBase): class ProxyChatCompletionRequest(LiteLLMBase):

2
litellm/proxy/db/dynamo_db.py
View file

@ -128,7 +128,7 @@ class DynamoDBWrapper(CustomDB):
actions: List = [] actions: List = []
for field in data_obj.model_fields_set: for field in data_obj.fields_set():
field_value = getattr(data_obj, field) field_value = getattr(data_obj, field)
# Convert datetime object to ISO8601 string # Convert datetime object to ISO8601 string