handle kwargs in agentic decorators

packages/agentic-workflows/src/zeroshot_agentic_workflows/decorators.py

@@ -88,8 +88,17 @@ async def wrapper(self: Any, *args: Any, **kwargs: Any) -> AgentRunResult[Any]:
             if tools_ref:
                 instructions = tools_ref + "\n" + instructions
 
-            # Map parameters
-            all_args = args
+            # Map parameters — merge kwargs into positional order
+            if kwargs:
+                import inspect as _inspect
+
+                sig = _inspect.signature(fn)
+                params = [p for p in sig.parameters if p != "self"]
+                all_args = tuple(
+                    kwargs.get(p, args[i] if i < len(args) else None) for i, p in enumerate(params)
+                )
+            else:
+                all_args = args
             mapped = mapper.map_arguments(all_args)
             session = mapper.find_session(all_args)
 
@@ -183,8 +192,19 @@ async def wrapper(self: Any, *args: Any, **kwargs: Any) -> ConsensusRunResult[An
             if tools_ref:
                 instructions = tools_ref + "\n" + instructions
 
-            mapped = mapper.map_arguments(args)
-            session = mapper.find_session(args)
+            # Map parameters — merge kwargs into positional order
+            if kwargs:
+                import inspect as _inspect
+
+                sig = _inspect.signature(fn)
+                params = [p for p in sig.parameters if p != "self"]
+                all_args = tuple(
+                    kwargs.get(p, args[i] if i < len(args) else None) for i, p in enumerate(params)
+                )
+            else:
+                all_args = args
+            mapped = mapper.map_arguments(all_args)
+            session = mapper.find_session(all_args)
             agent_name = name or f"{self.__class__.__name__}:{fn.__name__}"
             effective_max_turns = max_turns or (8 if resolved_tools else 1)
 

packages/agentic-workflows/tests/unit/test_decorators.py

@@ -146,6 +146,28 @@ async def spy(config: AgentConfig[str], run_config: AgentRunConfig) -> AgentRunR
         assert "session" not in input_json
         assert input_json["input_text"] == "some text"
 
+    async def test_kwargs_are_mapped_to_input_json(self, prompts_dir: str) -> None:
+        """When the decorated method is called with keyword arguments, they
+        should still appear in the input JSON sent to the agent service."""
+        Cls = _make_workflow_class(prompts_dir)
+        service = AiAgentServiceLocal.get_instance()
+
+        captured: list[AgentRunConfig] = []
+        original = service.create_and_run
+
+        async def spy(config: AgentConfig[str], run_config: AgentRunConfig) -> AgentRunResult[str]:
+            captured.append(run_config)
+            return await original(config, run_config)
+
+        service.create_and_run = spy  # type: ignore[assignment]
+
+        wf = Cls(service)
+        await wf.test_method(input_text="kwarg value")
+
+        assert len(captured) == 1
+        input_json = json.loads(captured[0].input)
+        assert input_json["input_text"] == "kwarg value"
+
     async def test_passes_model_override(self, prompts_dir: str) -> None:
         Cls = _make_workflow_class(prompts_dir)
         service = AiAgentServiceLocal.get_instance()