Skip to content

Commit 6a40cba

Browse files
committed
test(cli): update fast api mocks for request_state
1 parent 858ed87 commit 6a40cba

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
@@ -1038,9 +1038,18 @@ async def run_async_capture(
10381038
invocation_id: Optional[str] = None,
10391039
new_message: Optional[types.Content] = None,
10401040
state_delta: Optional[dict[str, Any]] = None,
1041+
request_state: Optional[dict[str, Any]] = None,
10411042
run_config: Optional[RunConfig] = None,
10421043
):
1043-
del self, user_id, session_id, new_message, state_delta, run_config
1044+
del (
1045+
self,
1046+
user_id,
1047+
session_id,
1048+
new_message,
1049+
state_delta,
1050+
request_state,
1051+
run_config,
1052+
)
10441053
captured_invocation_id["invocation_id"] = invocation_id
10451054
yield _event_1()
10461055

@@ -1140,9 +1149,18 @@ async def run_async_with_artifact_delta(
11401149
invocation_id: Optional[str] = None,
11411150
new_message: Optional[types.Content] = None,
11421151
state_delta: Optional[dict[str, Any]] = None,
1152+
request_state: Optional[dict[str, Any]] = None,
11431153
run_config: Optional[RunConfig] = None,
11441154
):
1145-
del user_id, session_id, invocation_id, new_message, state_delta, run_config
1155+
del (
1156+
user_id,
1157+
session_id,
1158+
invocation_id,
1159+
new_message,
1160+
state_delta,
1161+
request_state,
1162+
run_config,
1163+
)
11461164
yield Event(
11471165
author="dummy agent",
11481166
invocation_id="invocation_id",

0 commit comments

Comments
 (0)