From 42a5d73e97b4845aab666c29ac922d6dbc75f6f9 Mon Sep 17 00:00:00 2001 From: Luke Hinds Date: Tue, 24 Jun 2025 19:25:05 +0100 Subject: [PATCH] Fixes Short Circuit clause for context ID If #235 is an issue, this should fix. If its as intended, apologies for the noise and feel free to close Closes: #235 --- src/a2a/server/tasks/task_manager.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/a2a/server/tasks/task_manager.py b/src/a2a/server/tasks/task_manager.py index ca42b69b..790146b5 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(