Skip to content

Commit 95ec715

Browse files
committed
test(cli): update fast api mocks for request_state
1 parent 3d54fbf commit 95ec715

1 file changed

Lines changed: 20 additions & 2 deletions

File tree

tests/unittests/cli/test_fast_api.py

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1231,9 +1231,18 @@ async def run_async_capture(
12311231
invocation_id: Optional[str] = None,
12321232
new_message: Optional[types.Content] = None,
12331233
state_delta: Optional[dict[str, Any]] = None,
1234+
request_state: Optional[dict[str, Any]] = None,
12341235
run_config: Optional[RunConfig] = None,
12351236
):
1236-
del self, user_id, session_id, new_message, state_delta, run_config
1237+
del (
1238+
self,
1239+
user_id,
1240+
session_id,
1241+
new_message,
1242+
state_delta,
1243+
request_state,
1244+
run_config,
1245+
)
12371246
captured_invocation_id["invocation_id"] = invocation_id
12381247
yield _event_1()
12391248

@@ -1333,9 +1342,18 @@ async def run_async_with_artifact_delta(
13331342
invocation_id: Optional[str] = None,
13341343
new_message: Optional[types.Content] = None,
13351344
state_delta: Optional[dict[str, Any]] = None,
1345+
request_state: Optional[dict[str, Any]] = None,
13361346
run_config: Optional[RunConfig] = None,
13371347
):
1338-
del user_id, session_id, invocation_id, new_message, state_delta, run_config
1348+
del (
1349+
user_id,
1350+
session_id,
1351+
invocation_id,
1352+
new_message,
1353+
state_delta,
1354+
request_state,
1355+
run_config,
1356+
)
13391357
yield Event(
13401358
author="dummy agent",
13411359
invocation_id="invocation_id",

0 commit comments

Comments
 (0)