Skip to content

Commit 5ac446d

Browse files
wyf7107copybara-github
authored andcommitted
chore: add invocation_id to AgentRunRequest for resuming long running functions
PiperOrigin-RevId: 816362937
1 parent 2699ad7 commit 5ac446d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/google/adk/cli/adk_web_server.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,8 @@ class RunAgentRequest(common.BaseModel):
172172
new_message: types.Content
173173
streaming: bool = False
174174
state_delta: Optional[dict[str, Any]] = None
175+
# for resume long running functions
176+
invocation_id: Optional[str] = None
175177

176178

177179
class CreateSessionRequest(common.BaseModel):
@@ -1386,6 +1388,7 @@ async def event_generator():
13861388
new_message=req.new_message,
13871389
state_delta=req.state_delta,
13881390
run_config=RunConfig(streaming_mode=stream_mode),
1391+
invocation_id=req.invocation_id,
13891392
)
13901393
) as agen:
13911394
async for event in agen:

0 commit comments

Comments
 (0)