propagate abortSignal (#18477)

packages/cli/src/ui/hooks/atCommandProcessor.test.ts

@@ -1291,7 +1291,9 @@ describe('handleAtCommand', () => {
         signal: abortController.signal,
       });
 
-      expect(readResource).toHaveBeenCalledWith(resourceUri);
+      expect(readResource).toHaveBeenCalledWith(resourceUri, {
+        signal: abortController.signal,
+      });
       const processedParts = Array.isArray(result.processedQuery)
         ? result.processedQuery
         : [];

packages/cli/src/ui/hooks/atCommandProcessor.ts

@@ -371,6 +371,7 @@ function constructInitialQuery(
 async function readMcpResources(
   resourceParts: AtCommandPart[],
   config: Config,
+  signal: AbortSignal,
 ): Promise<{
   parts: PartUnion[];
   displays: IndividualToolCallDisplay[];
@@ -396,7 +397,7 @@ async function readMcpResources(
           `MCP client for server '${resource.serverName}' is not available or not connected.`,
         );
       }
-      const response = await client.readResource(resource.uri);
+      const response = await client.readResource(resource.uri, { signal });
       const resourceParts = convertResourceContentsToParts(response);
       return {
         success: true,
@@ -665,7 +666,7 @@ export async function handleAtCommand({
   }
 
   const [mcpResult, fileResult] = await Promise.all([
-    readMcpResources(resourceParts, config),
+    readMcpResources(resourceParts, config, signal),
     readLocalFiles(resolvedFiles, config, signal, userMessageTimestamp),
   ]);
 

packages/core/src/tools/mcp-client.ts

@@ -286,7 +286,10 @@ export class McpClient {
     this.resourceRegistry.setResourcesForServer(this.serverName, resources);
   }
 
-  async readResource(uri: string): Promise<ReadResourceResult> {
+  async readResource(
+    uri: string,
+    options?: { signal?: AbortSignal },
+  ): Promise<ReadResourceResult> {
     this.assertConnected();
     return this.client!.request(
       {
@@ -294,6 +297,7 @@ export class McpClient {
         params: { uri },
       },
       ReadResourceResultSchema,
+      options,
     );
   }