Fix(aws): Use abort signal when available with SDK requests (#6540)

Co-authored-by: Peter Lisy <[email protected]>
langchain-ai · Aug 16, 2024 · c4cb3a0 · c4cb3a0
1 parent 928ca89
commit c4cb3a0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/libs/langchain-aws/src/chat_models.ts b/libs/langchain-aws/src/chat_models.ts
@@ -817,7 +817,9 @@ export class ChatBedrockConverse
       system: converseSystem,
       ...params,
     });
-    const response = await this.client.send(command);
+    const response = await this.client.send(command, {
+      abortSignal: options.signal,
+    });
     const { output, ...responseMetadata } = response;
     if (!output?.message) {
       throw new Error("No message found in Bedrock response.");
@@ -855,7 +857,9 @@ export class ChatBedrockConverse
       system: converseSystem,
       ...params,
     });
-    const response = await this.client.send(command);
+    const response = await this.client.send(command, {
+      abortSignal: options.signal,
+    });
     if (response.stream) {
       for await (const chunk of response.stream) {
         if (chunk.contentBlockStart) {