From ab919004a2183e772474c9535dca7bccbcf6cf4b Mon Sep 17 00:00:00 2001 From: Tasha Upchurch Date: Sat, 23 Mar 2024 00:12:24 -0400 Subject: [PATCH] Update utils.py fix for constructed from dict choices.message being a dict still instead of Message class. --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 99e225556..c0abbd714 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -354,7 +354,7 @@ class Choices(OpenAIObject): if message is None: self.message = Message(content=None) else: - self.message = message + self.message = Message(**message) if logprobs is not None: self.logprobs = logprobs if enhancements is not None: