recordings

llama_stack/providers/inline/agents/meta_reference/agent_instance.py

@@ -790,37 +790,11 @@ class ChatAgent(ShieldRunnerMixin):
             if tool_name_to_def.get(tool_def.name, None):
                 raise ValueError(f"Tool {tool_def.name} already exists")
 
-            # Build JSON Schema from tool parameters
-            properties = {}
-            required = []
-
-            for param in tool_def.parameters:
-                param_schema = {
-                    "type": param.parameter_type,
-                    "description": param.description,
-                }
-                if param.default is not None:
-                    param_schema["default"] = param.default
-                if param.items is not None:
-                    param_schema["items"] = param.items
-                if param.title is not None:
-                    param_schema["title"] = param.title
-
-                properties[param.name] = param_schema
-
-                if param.required:
-                    required.append(param.name)
-
-            input_schema = {
-                "type": "object",
-                "properties": properties,
-                "required": required,
-            }
-
+            # Use input_schema from ToolDef directly
             tool_name_to_def[tool_def.name] = ToolDefinition(
                 tool_name=tool_def.name,
                 description=tool_def.description,
-                input_schema=input_schema,
+                input_schema=tool_def.input_schema,
             )
         for toolgroup_name_with_maybe_tool_name in agent_config_toolgroups:
             toolgroup_name, input_tool_name = self._parse_toolgroup_name(toolgroup_name_with_maybe_tool_name)