File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -1889,6 +1889,7 @@ def _set_telemetry_context_if_needed(runner: Runner):
18891889
18901890 @app .post ("/run" , response_model_exclude_none = True )
18911891 async def run_agent (req : RunAgentRequest ) -> list [Event ]:
1892+ self .current_app_name_ref .value = req .app_name
18921893 runner = await self .get_runner_async (req .app_name )
18931894 _set_telemetry_context_if_needed (runner )
18941895 try :
@@ -1910,6 +1911,7 @@ async def run_agent(req: RunAgentRequest) -> list[Event]:
19101911
19111912 @app .post ("/run_sse" )
19121913 async def run_agent_sse (req : RunAgentRequest ) -> StreamingResponse :
1914+ self .current_app_name_ref .value = req .app_name
19131915 stream_mode = StreamingMode .SSE if req .streaming else StreamingMode .NONE
19141916 runner = await self .get_runner_async (req .app_name )
19151917 _set_telemetry_context_if_needed (runner )
@@ -2089,6 +2091,7 @@ async def run_agent_live(
20892091 return
20902092
20912093 await websocket .accept ()
2094+ self .current_app_name_ref .value = req .app_name
20922095 runner_for_context = await self .get_runner_async (app_name )
20932096 _set_telemetry_context_if_needed (runner_for_context )
20942097
You can’t perform that action at this time.
0 commit comments