diff --git a/src/a2a/server/tasks/task_manager.py b/src/a2a/server/tasks/task_manager.py index 27e67e1a..5474e155 100644 --- a/src/a2a/server/tasks/task_manager.py +++ b/src/a2a/server/tasks/task_manager.py @@ -107,7 +107,13 @@ async def save_task_event( ) if not self.task_id: self.task_id = task_id_from_event - if not self.context_id and self.context_id != event.contextId: + if self.context_id and self.context_id != event.contextId: + raise ServerError( + error=InvalidParamsError( + message=f"Context in event doesn't match TaskManager {self.context_id} : {event.contextId}" + ) + ) + if not self.context_id: self.context_id = event.contextId logger.debug(