From 81ad331d92197ac77c81bbecef1043953c1cd003 Mon Sep 17 00:00:00 2001 From: alisalim17 Date: Wed, 1 May 2024 17:01:45 +0400 Subject: [PATCH] set default tool calls and function call --- litellm/utils.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 28dee68b59..c39f32b800 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -364,8 +364,6 @@ class ChatCompletionMessageToolCall(OpenAIObject): class Message(OpenAIObject): - tool_calls: Optional[List[ChatCompletionMessageToolCall]] - def __init__( self, content="default", @@ -378,13 +376,16 @@ class Message(OpenAIObject): super(Message, self).__init__(**params) self.content = content self.role = role + self.tool_calls = [] + self.function_call = None + if function_call is not None: self.function_call = FunctionCall(**function_call) if tool_calls is not None: - self.tool_calls = [] - for tool_call in tool_calls: - self.tool_calls.append(ChatCompletionMessageToolCall(**tool_call)) + self.tool_calls = [ + ChatCompletionMessageToolCall(**tool_call) for tool_call in tool_calls + ] if logprobs is not None: self._logprobs = ChoiceLogprobs(**logprobs) @@ -407,8 +408,7 @@ class Message(OpenAIObject): except: # if using pydantic v1 return self.dict() - - + class Delta(OpenAIObject): tool_calls: Optional[List[ChatCompletionDeltaToolCall]]