Skip to content

capture before_run modifications and fix session history sync in runners.py #6147

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
llalitkumarrr wants to merge 4 commits into
base: main
Choose a base branch
from
+12 −1
Open

capture before_run modifications and fix session history sync in runners.py #6147

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
98d12be
fix(runners): support message modification in workflows via before_ru…
llalitkumarrr Jun 17, 2026
07a8d21
Refactor line breaks for improved readability
llalitkumarrr Jun 17, 2026
6ea267c
Update runners.py
llalitkumarrr Jun 17, 2026
23f68bc
Merge branch 'main' into main
rohityan Jun 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 12 additions & 1 deletion src/google/adk/runners.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -509,7 +509,18 @@ async def _run_node_async(
yield user_event

# Run before_run callbacks
await ic.plugin_manager.run_before_run_callback(invocation_context=ic)
corrected_user_message = await ic.plugin_manager.run_before_run_callback(
invocation_context=ic
)
if corrected_user_message is not None:
if isinstance(corrected_user_message, types.Content):
node_input = corrected_user_message
ic.user_content = corrected_user_message
if hasattr(ic.session, 'events') and len(ic.session.events) > 0:
ic.session.events.pop()
user_event = await self._append_user_event(ic, corrected_user_message)
if yield_user_message and user_event:
yield user_event

# 3. Start root node in background
from .agents.base_agent import BaseAgent
Expand Down
Loading