Skip to content

Commit 5a82254

Browse files
committed
test(cli): update fast api mocks for request_state
1 parent 0cd807b commit 5a82254

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
@@ -1357,9 +1357,18 @@ async def run_async_capture(
13571357
invocation_id: Optional[str] = None,
13581358
new_message: Optional[types.Content] = None,
13591359
state_delta: Optional[dict[str, Any]] = None,
1360+
request_state: Optional[dict[str, Any]] = None,
13601361
run_config: Optional[RunConfig] = None,
13611362
):
1362-
del self, user_id, session_id, new_message, state_delta, run_config
1363+
del (
1364+
self,
1365+
user_id,
1366+
session_id,
1367+
new_message,
1368+
state_delta,
1369+
request_state,
1370+
run_config,
1371+
)
13631372
captured_invocation_id["invocation_id"] = invocation_id
13641373
yield _event_1()
13651374

@@ -1459,9 +1468,18 @@ async def run_async_with_artifact_delta(
14591468
invocation_id: Optional[str] = None,
14601469
new_message: Optional[types.Content] = None,
14611470
state_delta: Optional[dict[str, Any]] = None,
1471+
request_state: Optional[dict[str, Any]] = None,
14621472
run_config: Optional[RunConfig] = None,
14631473
):
1464-
del user_id, session_id, invocation_id, new_message, state_delta, run_config
1474+
del (
1475+
user_id,
1476+
session_id,
1477+
invocation_id,
1478+
new_message,
1479+
state_delta,
1480+
request_state,
1481+
run_config,
1482+
)
14651483
yield Event(
14661484
author="dummy agent",
14671485
invocation_id="invocation_id",

0 commit comments

Comments
 (0)