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

fix(utils.py): check if message is pydantic object or dict before dereferencing

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-26 09:47:16 -07:00
parent 2ecdd92619
commit 584d187e0e
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/utils.py
View file

@ -354,6 +354,9 @@ class Choices(OpenAIObject):
if message is None: if message is None:
self.message = Message(content=None) self.message = Message(content=None)
else: else:
if isinstance(message, Message):
self.message = message
elif isinstance(message, dict):
self.message = Message(**message) self.message = Message(**message)
if logprobs is not None: if logprobs is not None:
self.logprobs = logprobs self.logprobs = logprobs