From 115c663c98e6ae7e7b3998b2d8b73099511a2241 Mon Sep 17 00:00:00 2001
From: Jason Guo <jasonguo@amazon.com>
Date: Sat, 5 Apr 2025 18:51:27 -0700
Subject: [PATCH] fix(chat): Fix read ECONNRESET

---
 packages/core/src/shared/clients/codewhispererChatClient.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/packages/core/src/shared/clients/codewhispererChatClient.ts b/packages/core/src/shared/clients/codewhispererChatClient.ts
index 93b7b3bceb4..075f51a8736 100644
--- a/packages/core/src/shared/clients/codewhispererChatClient.ts
+++ b/packages/core/src/shared/clients/codewhispererChatClient.ts
@@ -17,6 +17,10 @@ export async function createCodeWhispererChatStreamingClient(): Promise<CodeWhis
         endpoint: cwsprConfig.endpoint,
         token: { token: bearerToken },
         customUserAgent: getUserAgent(),
+        requestHandler: {
+            // TODO: use a real number
+            timeoutInMs: 600000, // 10 mins for now
+        },
         retryStrategy: new ConfiguredRetryStrategy(1, (attempt: number) => 500 + attempt ** 10),
     })
     return streamingClient