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

This commit is contained in:
Krrish Dholakia 2024-03-26 09:47:16 -07:00
parent 2ecdd92619
commit 584d187e0e

View file

@ -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: