Skip to content

Commit c19b28a

Browse files
committed
Fix test_fast_api.py mocks to match new run_async signature
1 parent d87db52 commit c19b28a

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
@@ -126,6 +126,7 @@ async def dummy_run_async(
126126
session_id,
127127
new_message,
128128
state_delta=None,
129+
request_state=None,
129130
run_config: Optional[RunConfig] = None,
130131
invocation_id: Optional[str] = None,
131132
):
@@ -1267,9 +1268,10 @@ async def run_async_with_artifact_delta(
12671268
invocation_id: Optional[str] = None,
12681269
new_message: Optional[types.Content] = None,
12691270
state_delta: Optional[dict[str, Any]] = None,
1271+
request_state: Optional[dict[str, Any]] = None,
12701272
run_config: Optional[RunConfig] = None,
12711273
):
1272-
del user_id, session_id, invocation_id, new_message, state_delta, run_config
1274+
del user_id, session_id, invocation_id, new_message, state_delta, request_state, run_config
12731275
yield Event(
12741276
author="dummy agent",
12751277
invocation_id="invocation_id",

0 commit comments

Comments
 (0)