Skip to content

Commit 664381a

Browse files
committed
fix: adjust **kwargs in multiagent primitives
# Conflicts: # tests/strands/multiagent/test_base.py
1 parent a135262 commit 664381a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/strands/multiagent/test_base.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,12 @@ class TestMultiAgent(MultiAgentBase):
153153
def __init__(self):
154154
self.invoke_async_called = False
155155
self.received_task = None
156-
self.received_invocation_state = None
156+
self.received_kwargs = None
157157

158-
async def invoke_async(self, task, invocation_state=None):
158+
async def invoke_async(self, task, invocation_state, **kwargs):
159159
self.invoke_async_called = True
160160
self.received_task = task
161+
self.received_kwargs = kwargs
161162
self.received_invocation_state = invocation_state
162163
return MultiAgentResult(
163164
status=Status.COMPLETED, results={"test": NodeResult(result=Exception("test"), status=Status.COMPLETED)}

0 commit comments

Comments
 (0)