From 5ac446d947522a322e552af481efbccceedf5573 Mon Sep 17 00:00:00 2001
From: Yifan Wang <wanyif@google.com>
Date: Tue, 7 Oct 2025 13:49:10 -0700
Subject: [PATCH] chore: add invocation_id to AgentRunRequest for resuming long
 running functions

PiperOrigin-RevId: 816362937
---
 src/google/adk/cli/adk_web_server.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/google/adk/cli/adk_web_server.py b/src/google/adk/cli/adk_web_server.py
index d503e762..98c77f5d 100644
--- a/src/google/adk/cli/adk_web_server.py
+++ b/src/google/adk/cli/adk_web_server.py
@@ -172,6 +172,8 @@ class RunAgentRequest(common.BaseModel):
   new_message: types.Content
   streaming: bool = False
   state_delta: Optional[dict[str, Any]] = None
+  # for resume long running functions
+  invocation_id: Optional[str] = None
 
 
 class CreateSessionRequest(common.BaseModel):
@@ -1386,6 +1388,7 @@ class AdkWebServer:
                   new_message=req.new_message,
                   state_delta=req.state_delta,
                   run_config=RunConfig(streaming_mode=stream_mode),
+                  invocation_id=req.invocation_id,
               )
           ) as agen:
             async for event in agen: