Skip to content

Commit 06699ba

Browse files
committed
Remove context_id support as this is currently being discussed
1 parent 0ba9d74 commit 06699ba

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

src/a2a/server/tasks/task_manager.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,6 @@ async def save_task_event(
130130
task.history = [task.status.message]
131131
else:
132132
task.history.append(task.status.message)
133-
if event.contextId:
134-
task.contextId = event.contextId
135133
if event.metadata:
136134
if not task.metadata:
137135
task.metadata = {}

tests/server/tasks/test_task_manager.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -128,18 +128,17 @@ async def test_save_task_event_artifact_update(
128128
mock_task_store.save.assert_called_once_with(updated_task)
129129

130130
@pytest.mark.asyncio
131-
async def test_save_task_event_metadata_contextid_update(
131+
async def test_save_task_event_metadata_update(
132132
task_manager: TaskManager, mock_task_store: AsyncMock
133133
) -> None:
134-
"""Test saving an metadata and context update for an existing task."""
134+
"""Test saving an updated metadata for an existing task."""
135135
initial_task = Task(**MINIMAL_TASK)
136136
mock_task_store.get.return_value = initial_task
137137
new_metadata = {"meta_key_test": "meta_value_test"}
138-
new_context_id = 'new_context_id'
139138

140139
event = TaskStatusUpdateEvent(
141140
taskId=MINIMAL_TASK['id'],
142-
contextId=new_context_id,
141+
contextId=MINIMAL_TASK['contextId'],
143142
metadata=new_metadata,
144143
status=TaskStatus(state=TaskState.working),
145144
final=False,
@@ -148,7 +147,6 @@ async def test_save_task_event_metadata_contextid_update(
148147

149148
updated_task = mock_task_store.save.call_args.args[0]
150149
assert updated_task.metadata == new_metadata
151-
assert updated_task.contextId == new_context_id
152150

153151

154152
@pytest.mark.asyncio

0 commit comments

Comments
 (0)