Pass tools_to_call_from to TransformersModel._prepare_completion_kwargs (#1782)

albertvillanova · web-flow · commit 983bb71c3d57 · 2025-09-23T15:46:09.000+02:00
diff --git a/src/smolagents/models.py b/src/smolagents/models.py
@@ -926,6 +926,8 @@ def _prepare_completion_args(
         completion_kwargs = self._prepare_completion_kwargs(
             messages=messages,
             stop_sequences=stop_sequences,
+            tools_to_call_from=tools_to_call_from,
+            tool_choice=None,
             **kwargs,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -926,6 +926,8 @@ def _prepare_completion_args(`
`926`	`926`	`completion_kwargs = self._prepare_completion_kwargs(`
`927`	`927`	`messages=messages,`
`928`	`928`	`stop_sequences=stop_sequences,`
	`929`	`+ tools_to_call_from=tools_to_call_from,`
	`930`	`+ tool_choice=None,`
`929`	`931`	`**kwargs,`
`930`	`932`	`)`
`931`	`933`