diff --git a/src/smolagents/models.py b/src/smolagents/models.py index ffe967b..927acf0 100644 --- a/src/smolagents/models.py +++ b/src/smolagents/models.py @@ -78,7 +78,7 @@ class ChatMessageToolCall: type: str @classmethod - def from_hf_api(cls, tool_call, raw) -> "ChatMessageToolCall": + def from_hf_api(cls, tool_call) -> "ChatMessageToolCall": return cls( function=ChatMessageToolCallDefinition.from_hf_api(tool_call.function), id=tool_call.id,