forked from phoenix/litellm-mirror
fix(utils.py): check if message is pydantic object or dict before dereferencing
This commit is contained in:
parent
2ecdd92619
commit
584d187e0e
1 changed files with 4 additions and 1 deletions
|
@ -354,7 +354,10 @@ class Choices(OpenAIObject):
|
|||
if message is None:
|
||||
self.message = Message(content=None)
|
||||
else:
|
||||
self.message = Message(**message)
|
||||
if isinstance(message, Message):
|
||||
self.message = message
|
||||
elif isinstance(message, dict):
|
||||
self.message = Message(**message)
|
||||
if logprobs is not None:
|
||||
self.logprobs = logprobs
|
||||
if enhancements is not None:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue