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

feat: Pass event id as a metadata when it is converted into a message

Browse Source 
PiperOrigin-RevId: 860929058
This commit is contained in:
Jinhyuk Kim
2026-01-25 16:05:45 -08:00
committed by Copybara-Service
parent 324796b4fe
commit 85434e293f
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/a2a/converters/event_converter.py
+1 -1
View File
@@ -131,6 +131,7 @@ def _get_context_metadata(
_get_adk_metadata_key("session_id"): invocation_context.session.id, _get_adk_metadata_key("session_id"): invocation_context.session.id,
_get_adk_metadata_key("invocation_id"): event.invocation_id, _get_adk_metadata_key("invocation_id"): event.invocation_id,
_get_adk_metadata_key("author"): event.author, _get_adk_metadata_key("author"): event.author,
_get_adk_metadata_key("event_id"): event.id,
} }
# Add optional metadata fields if present # Add optional metadata fields if present
@@ -479,7 +480,6 @@ def _create_status_update_event(
task_id: Optional task ID to use for generated events. task_id: Optional task ID to use for generated events.
context_id: Optional Context ID to use for generated events. context_id: Optional Context ID to use for generated events.
Returns: Returns:
A TaskStatusUpdateEvent with RUNNING state. A TaskStatusUpdateEvent with RUNNING state.
""" """
tests/unittests/a2a/converters/test_event_converter.py
+2
View File
@@ -56,6 +56,7 @@ class TestEventConverter:
self.mock_invocation_context.artifact_service = self.mock_artifact_service self.mock_invocation_context.artifact_service = self.mock_artifact_service
self.mock_event = Mock(spec=Event) self.mock_event = Mock(spec=Event)
self.mock_event.id = None
self.mock_event.invocation_id = "test-invocation-id" self.mock_event.invocation_id = "test-invocation-id"
self.mock_event.author = "test-author" self.mock_event.author = "test-author"
self.mock_event.branch = None self.mock_event.branch = None
@@ -130,6 +131,7 @@ class TestEventConverter:
f"{ADK_METADATA_KEY_PREFIX}session_id", f"{ADK_METADATA_KEY_PREFIX}session_id",
f"{ADK_METADATA_KEY_PREFIX}invocation_id", f"{ADK_METADATA_KEY_PREFIX}invocation_id",
f"{ADK_METADATA_KEY_PREFIX}author", f"{ADK_METADATA_KEY_PREFIX}author",
f"{ADK_METADATA_KEY_PREFIX}event_id",
] ]
for key in expected_keys: for key in expected_keys: