Skip to content

Commit d56dd08

Browse files
GWealecopybara-github
authored andcommitted
fix: Enable saving input blobs as artifacts in ADK web server for issue #2176
The `run_config` for `run_agent` now includes `save_input_blobs_as_artifacts=True`. PiperOrigin-RevId: 804589111
1 parent 8452d2b commit d56dd08

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/google/adk/cli/adk_web_server.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1031,6 +1031,7 @@ async def run_agent(req: RunAgentRequest) -> list[Event]:
10311031
session_id=req.session_id,
10321032
new_message=req.new_message,
10331033
state_delta=req.state_delta,
1034+
run_config=RunConfig(save_input_blobs_as_artifacts=True),
10341035
)
10351036
) as agen:
10361037
events = [event async for event in agen]
@@ -1060,7 +1061,10 @@ async def event_generator():
10601061
session_id=req.session_id,
10611062
new_message=req.new_message,
10621063
state_delta=req.state_delta,
1063-
run_config=RunConfig(streaming_mode=stream_mode),
1064+
run_config=RunConfig(
1065+
streaming_mode=stream_mode,
1066+
save_input_blobs_as_artifacts=True,
1067+
),
10641068
)
10651069
) as agen:
10661070
async for event in agen:

0 commit comments

Comments
 (0)