From 28b712b2ce41c5ac663d52384de07155a9bbdfba Mon Sep 17 00:00:00 2001 From: Dylan Snyder <114695692+dylan-apex@users.noreply.github.com> Date: Wed, 20 Aug 2025 17:13:26 -0500 Subject: [PATCH 1/3] fix: return invocation_id when using tools + output_schema --- src/google/adk/flows/llm_flows/_output_schema_processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/google/adk/flows/llm_flows/_output_schema_processor.py b/src/google/adk/flows/llm_flows/_output_schema_processor.py index 16638702c..88d2ce774 100644 --- a/src/google/adk/flows/llm_flows/_output_schema_processor.py +++ b/src/google/adk/flows/llm_flows/_output_schema_processor.py @@ -78,7 +78,7 @@ def create_final_model_response_event( from google.genai import types # Create a proper model response event - final_event = Event(author=invocation_context.agent.name) + final_event = Event(author=invocation_context.agent.name, invocation_id=invocation_context.invocation_id) final_event.content = types.Content( role='model', parts=[types.Part(text=json_response)] ) From bf4056141ad52b0875492c11f369f7baab8ffef2 Mon Sep 17 00:00:00 2001 From: Dylan Snyder <114695692+dylan-apex@users.noreply.github.com> Date: Wed, 20 Aug 2025 17:17:36 -0500 Subject: [PATCH 2/3] autoformat --- src/google/adk/flows/llm_flows/_output_schema_processor.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/google/adk/flows/llm_flows/_output_schema_processor.py b/src/google/adk/flows/llm_flows/_output_schema_processor.py index 88d2ce774..4a326179a 100644 --- a/src/google/adk/flows/llm_flows/_output_schema_processor.py +++ b/src/google/adk/flows/llm_flows/_output_schema_processor.py @@ -78,7 +78,10 @@ def create_final_model_response_event( from google.genai import types # Create a proper model response event - final_event = Event(author=invocation_context.agent.name, invocation_id=invocation_context.invocation_id) + final_event = Event( + author=invocation_context.agent.name, + invocation_id=invocation_context.invocation_id, + ) final_event.content = types.Content( role='model', parts=[types.Part(text=json_response)] ) From ae7a40448bded44b98fa5162829857b030e4c6e1 Mon Sep 17 00:00:00 2001 From: Dylan Snyder <114695692+dylan-apex@users.noreply.github.com> Date: Wed, 20 Aug 2025 17:24:11 -0500 Subject: [PATCH 3/3] add branch to Event and consolidate content into constructor --- .../adk/flows/llm_flows/_output_schema_processor.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/google/adk/flows/llm_flows/_output_schema_processor.py b/src/google/adk/flows/llm_flows/_output_schema_processor.py index 4a326179a..b9d73469f 100644 --- a/src/google/adk/flows/llm_flows/_output_schema_processor.py +++ b/src/google/adk/flows/llm_flows/_output_schema_processor.py @@ -81,10 +81,12 @@ def create_final_model_response_event( final_event = Event( author=invocation_context.agent.name, invocation_id=invocation_context.invocation_id, + branch=invocation_context.branch, + content=types.Content( + role='model', parts=[types.Part(text=json_response)] + ), ) - final_event.content = types.Content( - role='model', parts=[types.Part(text=json_response)] - ) + return final_event