From 584d187e0e0992c13f81c8d097bde74f2c71852f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 26 Mar 2024 09:47:16 -0700 Subject: [PATCH] fix(utils.py): check if message is pydantic object or dict before dereferencing --- litellm/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index f672fe5c5db..74005a3c5f5 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -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: