@@ -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