fix: do not consider interpolated tool calls as inputs

Author: evilsocket

nerve/models.py

@@ -183,23 +183,26 @@ def get_inputs(self) -> dict[str, t.Any]:
         Get the input names for the agent with their default values if set.
         """
         input_names = set()
+        tools_names = [t.name if isinstance(t, Tool) else t.__name__ for t in self.tools]
 
         # from the system prompt
         if self.agent:
             for input_name in self._get_inputs_from_string(self.agent):
-                input_names.add(input_name)
+                if input_name not in tools_names:
+                    input_names.add(input_name)
 
         # from the task prompt
         if self.task:
             for input_name in self._get_inputs_from_string(self.task):
-                input_names.add(input_name)
+                if input_name not in tools_names:
+                    input_names.add(input_name)
 
         # from the tools
         for tool in self.tools:
             if isinstance(tool, Tool) and tool.tool:
                 arg_names = [arg.name for arg in tool.arguments]
                 for input_name in self._get_inputs_from_string(tool.tool):
-                    if input_name not in arg_names:
+                    if input_name not in arg_names and input_name not in tools_names:
                         input_names.add(input_name)
 
         if not self.task:

nerve/models_test.py

@@ -119,3 +119,21 @@ def test_get_inputs_includes_task_when_task_not_set(self) -> None:
 
         self.assertIn("task", inputs)
         self.assertIn("input1", inputs)
+
+    def test_get_inputs_ignores_interpolated_tool_calls(self) -> None:
+        """Test that get_inputs doesn't extract variables from interpolated tool calls"""
+        config = Configuration(
+            agent="I am an agent",
+            task="Complete the task using {{tool_name(param='value')}} and {{input1}}",
+            tools=[
+                Tool(
+                    name="tool_name",
+                    description="A test tool",
+                )
+            ],
+        )
+        inputs = config.get_inputs()
+
+        self.assertIn("input1", inputs)
+        self.assertNotIn("tool_name", inputs)  # Should be excluded as it's a tool
+        self.assertNotIn("param", inputs)  # Should be excluded as it's a parameter to a tool call