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