From f95458dad8ad5ad8709e711c2be45ccff324d695 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 5 Mar 2024 18:10:43 -0800 Subject: [PATCH] fix(utils.py): handle dict object for chatcompletionmessagetoolcall --- litellm/utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 33f4f1c3d..68dc137af 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -227,13 +227,16 @@ class ChatCompletionDeltaToolCall(OpenAIObject): class ChatCompletionMessageToolCall(OpenAIObject): def __init__( self, - function: Function, + function: Union[Dict, Function], id: Optional[str] = None, type: Optional[str] = None, **params, ): super(ChatCompletionMessageToolCall, self).__init__(**params) - self.function = function + if isinstance(function, Dict): + self.function = Function(**function) + else: + self.function = function if id is not None: self.id = id