fix: resolve all test failures for TaskManager extraction

- Add TaskManager to requestStream test protocols (they need it for tasks API)
- Fix IIFE closures wrongly matched by sed (restore to no-arg)
- Add removeProgressHandler to TaskManagerHost interface
- Fix prepareOutboundRequest to propagate queue overflow errors
- Update requestTaskStore references to createRequestTaskStore
- Fix optional chaining to non-null assertions in test casts

430/430 core tests pass. Zero type errors.

Author: felixweinberger

packages/core/src/shared/protocol.ts

@@ -288,7 +288,8 @@ export abstract class Protocol<SendRequestT extends Request, SendNotificationT e
             taskManager.bind({
                 request: (request, resultSchema, options) => this.request(request as SendRequestT, resultSchema, options),
                 notification: (notification, options) => this.notification(notification as SendNotificationT, options),
-                reportError: (error) => this._onerror(error)
+                reportError: (error) => this._onerror(error),
+                removeProgressHandler: (token) => this._progressHandlers.delete(token)
             });
             this._registerTaskHandlers();
         }
@@ -787,7 +788,10 @@ export abstract class Protocol<SendRequestT extends Request, SendNotificationT e
             // Delegate to TaskManager for task param augmentation and queue routing
             if (this._taskManager) {
                 const responseHandler = this._responseHandlers.get(messageId)!;
-                const queued = this._taskManager.prepareOutboundRequest(jsonrpcRequest, options, messageId, responseHandler);
+                const queued = this._taskManager.prepareOutboundRequest(jsonrpcRequest, options, messageId, responseHandler, (error) => {
+                    this._cleanupTimeout(messageId);
+                    reject(error);
+                });
                 if (queued) {
                     // Request was queued for task side-channel delivery, don't send via transport
                     return;

packages/core/src/shared/taskManager.ts

@@ -99,6 +99,7 @@ export interface TaskManagerHost {
     request<T extends AnySchema>(request: Request, resultSchema: T, options?: RequestOptions): Promise<SchemaOutput<T>>;
     notification(notification: Notification, options?: NotificationOptions): Promise<void>;
     reportError(error: Error): void;
+    removeProgressHandler(token: number): void;
 }
 
 // ============================================================================
@@ -390,7 +391,8 @@ export class TaskManager {
         jsonrpcRequest: JSONRPCRequest,
         options: RequestOptions | undefined,
         messageId: number,
-        responseHandler: (response: JSONRPCResultResponse | Error) => void
+        responseHandler: (response: JSONRPCResultResponse | Error) => void,
+        onError: (error: unknown) => void
     ): boolean {
         const { task, relatedTask } = options ?? {};
 
@@ -423,7 +425,7 @@ export class TaskManager {
                 message: jsonrpcRequest,
                 timestamp: Date.now()
             }).catch(error => {
-                this._host?.reportError(new Error(`Failed to enqueue task message: ${error}`));
+                onError(error);
             });
 
             return true; // queued
@@ -706,6 +708,10 @@ export class TaskManager {
     }
 
     private _cleanupTaskProgressHandler(taskId: string): void {
-        this._taskProgressTokens.delete(taskId);
+        const progressToken = this._taskProgressTokens.get(taskId);
+        if (progressToken !== undefined) {
+            this._host?.removeProgressHandler(progressToken);
+            this._taskProgressTokens.delete(taskId);
+        }
     }
 }