refactor: add kwargs to multiagent interfaces (strands-agents#454)

Author: mkmeral

src/strands/multiagent/base.py

@@ -6,7 +6,7 @@
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Union
+from typing import Any, Union
 
 from ..agent import AgentResult
 from ..types.content import ContentBlock
@@ -76,11 +76,11 @@ class MultiAgentBase(ABC):
     """
 
     @abstractmethod
-    async def execute_async(self, task: str | list[ContentBlock]) -> MultiAgentResult:
+    async def execute_async(self, task: str | list[ContentBlock], **kwargs: Any) -> MultiAgentResult:
         """Execute task asynchronously."""
         raise NotImplementedError("execute_async not implemented")
 
     @abstractmethod
-    def execute(self, task: str | list[ContentBlock]) -> MultiAgentResult:
+    def execute(self, task: str | list[ContentBlock], **kwargs: Any) -> MultiAgentResult:
         """Execute task synchronously."""
         raise NotImplementedError("execute not implemented")

src/strands/multiagent/graph.py

@@ -254,7 +254,7 @@ def __init__(self, nodes: dict[str, GraphNode], edges: set[GraphEdge], entry_poi
         self.state = GraphState()
         self.tracer = get_tracer()
 
-    def execute(self, task: str | list[ContentBlock]) -> GraphResult:
+    def execute(self, task: str | list[ContentBlock], **kwargs: Any) -> GraphResult:
         """Execute task synchronously."""
 
         def execute() -> GraphResult:
@@ -264,7 +264,7 @@ def execute() -> GraphResult:
             future = executor.submit(execute)
             return future.result()
 
-    async def execute_async(self, task: str | list[ContentBlock]) -> GraphResult:
+    async def execute_async(self, task: str | list[ContentBlock], **kwargs: Any) -> GraphResult:
         """Execute the graph asynchronously."""
         logger.debug("task=<%s> | starting graph execution", task)