File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -227,9 +227,13 @@ async def run_async(
227227 )
228228
229229 last_content = None
230+ invocation_context = tool_context ._invocation_context
230231 async with Aclosing (
231232 runner .run_async (
232- user_id = session .user_id , session_id = session .id , new_message = content
233+ user_id = session .user_id ,
234+ session_id = session .id ,
235+ new_message = content ,
236+ run_config = invocation_context .run_config ,
233237 )
234238 ) as agen :
235239 async for event in agen :
@@ -269,9 +273,13 @@ async def run_async_with_events(
269273 )
270274
271275 # Yield events from the sub-agent as they are generated
276+ invocation_context = tool_context ._invocation_context
272277 async with Aclosing (
273278 runner .run_async (
274- user_id = session .user_id , session_id = session .id , new_message = content
279+ user_id = session .user_id ,
280+ session_id = session .id ,
281+ new_message = content ,
282+ run_config = invocation_context .run_config ,
275283 )
276284 ) as agen :
277285 async for event in agen :
You can’t perform that action at this time.
0 commit comments