Update src/ModelContextProtocol.Core/Server/InMemoryMcpTaskStore.cs

eiriktsarpalis · halter73 · web-flow · commit f72af0de229e · 2026-01-26T12:37:40.000+02:00
Co-authored-by: Stephen Halter &lt;halter73@gmail.com&gt;
diff --git a/src/ModelContextProtocol.Core/Server/InMemoryMcpTaskStore.cs b/src/ModelContextProtocol.Core/Server/InMemoryMcpTaskStore.cs
@@ -216,9 +216,9 @@ public Task<JsonElement> GetTaskResultAsync(string taskId, string? sessionId = n
         }
 
         // Enforce session isolation
-        if (sessionId != null && entry.SessionId != sessionId)
+        if (sessionId != entry.SessionId)
         {
-            throw new InvalidOperationException($"Task not found: {taskId}");
+            throw new InvalidOperationException($"Invalid sessionId: {sessionId} provided for {taskId}");
         }
 
         if (entry.StoredResult is not { } storedResult)

Original file line number	Diff line number	Diff line change
`@@ -216,9 +216,9 @@ public Task<JsonElement> GetTaskResultAsync(string taskId, string? sessionId = n`
`216`	`216`	`}`
`217`	`217`
`218`	`218`	`// Enforce session isolation`
`219`		`- if (sessionId != null && entry.SessionId != sessionId)`
	`219`	`+ if (sessionId != entry.SessionId)`
`220`	`220`	`{`
`221`		`- throw new InvalidOperationException($"Task not found: {taskId}");`
	`221`	`+ throw new InvalidOperationException($"Invalid sessionId: {sessionId} provided for {taskId}");`
`222`	`222`	`}`
`223`	`223`
`224`	`224`	`if (entry.StoredResult is not { } storedResult)`