@@ -1889,29 +1889,31 @@ def _set_telemetry_context_if_needed(runner: Runner):
18891889 _is_visual_builder .set (True )
18901890 else :
18911891 _is_visual_builder .set (False )
1892+
18921893 @app .post ("/run" , response_model_exclude_none = True )
18931894 async def run_agent (req : RunAgentRequest ) -> list [Event ]:
18941895 try :
18951896 runner = await self .get_runner_async (req .app_name )
18961897 except ValueError as e :
18971898 raise HTTPException (status_code = 404 , detail = str (e )) from e
1898-
1899+
18991900 _set_telemetry_context_if_needed (runner )
19001901 try :
19011902 async with Aclosing (
1902- runner .run_async (
1903- user_id = req .user_id ,
1904- session_id = req .session_id ,
1905- new_message = req .new_message ,
1906- state_delta = req .state_delta ,
1907- invocation_id = req .invocation_id ,
1903+ runner .run_async (
1904+ user_id = req .user_id ,
1905+ session_id = req .session_id ,
1906+ new_message = req .new_message ,
1907+ state_delta = req .state_delta ,
1908+ invocation_id = req .invocation_id ,
19081909 )
1909- ) as agen :
1910- events = [event async for event in agen ]
1910+ ) as agen :
1911+ events = [event async for event in agen ]
19111912 except SessionNotFoundError as e :
19121913 raise HTTPException (status_code = 404 , detail = str (e )) from e
1913-
1914- return events
1914+
1915+ return events
1916+
19151917 @app .post ("/run_sse" )
19161918 async def run_agent_sse (req : RunAgentRequest ) -> StreamingResponse :
19171919 stream_mode = StreamingMode .SSE if req .streaming else StreamingMode .NONE
0 commit comments