Skip to content

Commit 85434e2

Browse files
reniowoodcopybara-github
authored andcommitted
feat: Pass event id as a metadata when it is converted into a message
PiperOrigin-RevId: 860929058
1 parent 324796b commit 85434e2

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/google/adk/a2a/converters/event_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ def _get_context_metadata(
131131
_get_adk_metadata_key("session_id"): invocation_context.session.id,
132132
_get_adk_metadata_key("invocation_id"): event.invocation_id,
133133
_get_adk_metadata_key("author"): event.author,
134+
_get_adk_metadata_key("event_id"): event.id,
134135
}
135136

136137
# Add optional metadata fields if present
@@ -479,7 +480,6 @@ def _create_status_update_event(
479480
task_id: Optional task ID to use for generated events.
480481
context_id: Optional Context ID to use for generated events.
481482
482-
483483
Returns:
484484
A TaskStatusUpdateEvent with RUNNING state.
485485
"""

tests/unittests/a2a/converters/test_event_converter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def setup_method(self):
5656
self.mock_invocation_context.artifact_service = self.mock_artifact_service
5757

5858
self.mock_event = Mock(spec=Event)
59+
self.mock_event.id = None
5960
self.mock_event.invocation_id = "test-invocation-id"
6061
self.mock_event.author = "test-author"
6162
self.mock_event.branch = None
@@ -130,6 +131,7 @@ def test_get_context_metadata_success(self):
130131
f"{ADK_METADATA_KEY_PREFIX}session_id",
131132
f"{ADK_METADATA_KEY_PREFIX}invocation_id",
132133
f"{ADK_METADATA_KEY_PREFIX}author",
134+
f"{ADK_METADATA_KEY_PREFIX}event_id",
133135
]
134136

135137
for key in expected_keys:

0 commit comments

Comments
 (0)