encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

chore: add invocation_id to AgentRunRequest for resuming long running functions

Browse Source 
PiperOrigin-RevId: 816362937
This commit is contained in:
Yifan Wang
2025-10-07 13:49:50 -07:00
committed by Copybara-Service
parent 2699ad7aff
commit 5ac446d947
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/cli/adk_web_server.py
+3
View File
@@ -172,6 +172,8 @@ class RunAgentRequest(common.BaseModel):
new_message: types.Content new_message: types.Content
streaming: bool = False streaming: bool = False
state_delta: Optional[dict[str, Any]] = None state_delta: Optional[dict[str, Any]] = None
# for resume long running functions
invocation_id: Optional[str] = None
class CreateSessionRequest(common.BaseModel): class CreateSessionRequest(common.BaseModel):
@@ -1386,6 +1388,7 @@ class AdkWebServer:
new_message=req.new_message, new_message=req.new_message,
state_delta=req.state_delta, state_delta=req.state_delta,
run_config=RunConfig(streaming_mode=stream_mode), run_config=RunConfig(streaming_mode=stream_mode),
invocation_id=req.invocation_id,
) )
) as agen: ) as agen:
async for event in agen: async for event in agen: