chore: support params on input filters besides just input

Author: ibuildthecloud

pkg/runner/input.go

@@ -15,13 +15,31 @@ func (r *Runner) handleInput(callCtx engine.Context, monitor Monitor, env []stri
 	}
 
 	for _, inputToolRef := range inputToolRefs {
-		inputData, err := json.Marshal(map[string]any{
-			"input": input,
-		})
-		if err != nil {
-			return "", fmt.Errorf("failed to marshal input: %w", err)
+		var (
+			rawInput  bool
+			toolInput string
+		)
+
+		inputTool := callCtx.Program.ToolSet[inputToolRef.ToolID]
+		if inputTool.Arguments != nil && len(inputTool.Arguments.Properties) > 0 {
+			if _, hasInput := inputTool.Arguments.Properties["input"]; !hasInput || len(inputTool.Arguments.Properties) > 1 {
+				rawInput = true
+			}
 		}
-		res, err := r.subCall(callCtx.Ctx, callCtx, monitor, env, inputToolRef.ToolID, string(inputData), "", engine.InputToolCategory)
+
+		if rawInput {
+			toolInput = input
+		} else {
+			inputData, err := json.Marshal(map[string]any{
+				"input": input,
+			})
+			if err != nil {
+				return "", fmt.Errorf("failed to marshal input: %w", err)
+			}
+			toolInput = string(inputData)
+		}
+
+		res, err := r.subCall(callCtx.Ctx, callCtx, monitor, env, inputToolRef.ToolID, toolInput, "", engine.InputToolCategory)
 		if err != nil {
 			return "", err
 		}