fix: address review feedback

- Update collect-user-info-task example to use single-arg getTask/getTaskResult
  signature matching the updated TaskRequestHandler type
- Clean up _taskToolMap entries after getTaskResult to prevent unbounded growth

Author: felixweinberger

src/examples/server/simpleStreamableHttp.ts

@@ -379,11 +379,11 @@ const getServer = () => {
 
                 return { task };
             },
-            async getTask(_args, { taskId, taskStore: getTaskStore }) {
-                return await getTaskStore.getTask(taskId);
+            async getTask({ taskId, taskStore }) {
+                return await taskStore.getTask(taskId);
             },
-            async getTaskResult(_args, { taskId, taskStore: getResultTaskStore }) {
-                const result = await getResultTaskStore.getTaskResult(taskId);
+            async getTaskResult({ taskId, taskStore }) {
+                const result = await taskStore.getTaskResult(taskId);
                 return result as CallToolResult;
             }
         }

src/server/mcp.ts

@@ -98,10 +98,16 @@ export class McpServer {
             getTaskResult: async (taskId: string, extra: RequestHandlerExtra<ServerRequest, ServerNotification>) => {
                 const taskStore = extra.taskStore!;
                 const handler = this._getTaskHandler(taskId);
-                if (handler) {
-                    return await handler.getTaskResult({ ...extra, taskId, taskStore });
+                try {
+                    if (handler) {
+                        return await handler.getTaskResult({ ...extra, taskId, taskStore });
+                    }
+                    return await taskStore.getTaskResult(taskId);
+                } finally {
+                    // Once the result has been retrieved the task is complete;
+                    // drop the taskId → toolName mapping to avoid unbounded growth.
+                    this._taskToolMap.delete(taskId);
                 }
-                return await taskStore.getTaskResult(taskId);
             }
         };
         this.server = new Server(serverInfo, { ...options, taskHandlerHooks });