File tree Expand file tree Collapse file tree 2 files changed +3
-1
lines changed
src/google/adk/a2a/converters
tests/unittests/a2a/converters Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Original file line number Diff line number Diff 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 """
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments