Skip to content

Commit 740557c

Browse files
wyf7107copybara-github
authored andcommitted
fix: hot reload agents for adk web
Co-authored-by: Yifan Wang <wanyif@google.com> PiperOrigin-RevId: 908496612
1 parent 4e81af6 commit 740557c

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

src/google/adk/cli/adk_web_server.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)