Skip to content

Commit 76773e0

Browse files
committed
Fix test_fast_api.py mocks to match new run_async signature
1 parent 81a45a0 commit 76773e0

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/unittests/cli/test_fast_api.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ async def dummy_run_async(
125125
session_id,
126126
new_message,
127127
state_delta=None,
128+
request_state=None,
128129
run_config: Optional[RunConfig] = None,
129130
invocation_id: Optional[str] = None,
130131
):
@@ -1074,9 +1075,10 @@ async def run_async_with_artifact_delta(
10741075
invocation_id: Optional[str] = None,
10751076
new_message: Optional[types.Content] = None,
10761077
state_delta: Optional[dict[str, Any]] = None,
1078+
request_state: Optional[dict[str, Any]] = None,
10771079
run_config: Optional[RunConfig] = None,
10781080
):
1079-
del user_id, session_id, invocation_id, new_message, state_delta, run_config
1081+
del user_id, session_id, invocation_id, new_message, state_delta, request_state, run_config
10801082
yield Event(
10811083
author="dummy agent",
10821084
invocation_id="invocation_id",

0 commit comments

Comments
 (0)