mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 11:43:54 +00:00
fix(utils.py): handle none in tool call for mistral tool calling
This commit is contained in:
parent
6727b009a1
commit
0eb67e50a1
1 changed files with 20 additions and 4 deletions
|
@ -225,9 +225,25 @@ class ChatCompletionDeltaToolCall(OpenAIObject):
|
||||||
|
|
||||||
|
|
||||||
class ChatCompletionMessageToolCall(OpenAIObject):
|
class ChatCompletionMessageToolCall(OpenAIObject):
|
||||||
id: str
|
def __init__(
|
||||||
function: Function
|
self,
|
||||||
type: str
|
function: Function,
|
||||||
|
id: Optional[str] = None,
|
||||||
|
type: Optional[str] = None,
|
||||||
|
**params,
|
||||||
|
):
|
||||||
|
super(ChatCompletionMessageToolCall, self).__init__(**params)
|
||||||
|
self.function = function
|
||||||
|
|
||||||
|
if id is not None:
|
||||||
|
self.id = id
|
||||||
|
else:
|
||||||
|
self.id = f"{uuid.uuid4()}"
|
||||||
|
|
||||||
|
if type is not None:
|
||||||
|
self.type = type
|
||||||
|
else:
|
||||||
|
self.type = "function"
|
||||||
|
|
||||||
|
|
||||||
class Message(OpenAIObject):
|
class Message(OpenAIObject):
|
||||||
|
@ -6232,7 +6248,7 @@ def convert_to_model_response_object(
|
||||||
|
|
||||||
return model_response_object
|
return model_response_object
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise Exception(f"Invalid response object {e}")
|
raise Exception(f"Invalid response object {traceback.format_exc()}")
|
||||||
|
|
||||||
|
|
||||||
def acreate(*args, **kwargs): ## Thin client to handle the acreate langchain call
|
def acreate(*args, **kwargs): ## Thin client to handle the acreate langchain call
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue