Skip to content

Commit 038a129

Browse files
committed
Fix test_fast_api.py mocks to match new run_async signature
1 parent e901e7c commit 038a129

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/unittests/cli/test_fast_api.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ async def dummy_run_async(
129129
session_id,
130130
new_message,
131131
state_delta=None,
132+
request_state=None,
132133
run_config: Optional[RunConfig] = None,
133134
invocation_id: Optional[str] = None,
134135
):
@@ -977,9 +978,10 @@ async def run_async_with_artifact_delta(
977978
invocation_id: Optional[str] = None,
978979
new_message: Optional[types.Content] = None,
979980
state_delta: Optional[dict[str, Any]] = None,
981+
request_state: Optional[dict[str, Any]] = None,
980982
run_config: Optional[RunConfig] = None,
981983
):
982-
del user_id, session_id, invocation_id, new_message, state_delta, run_config
984+
del user_id, session_id, invocation_id, new_message, state_delta, request_state, run_config
983985
yield Event(
984986
author="dummy agent",
985987
invocation_id="invocation_id",

0 commit comments

Comments
 (0)