From 4e38031a65b68b2c33de3ee40b5a522abab9cb06 Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Wed, 4 Jun 2025 14:03:36 -0700
Subject: [PATCH] Increase timeout for waiting for chat completion (#334)

---
 e2e-tests/helpers/test_helper.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/e2e-tests/helpers/test_helper.ts b/e2e-tests/helpers/test_helper.ts
index df8ff2b..e1817b2 100644
--- a/e2e-tests/helpers/test_helper.ts
+++ b/e2e-tests/helpers/test_helper.ts
@@ -12,6 +12,7 @@ export const Timeout = {
   // Why make this a constant? In some platforms, perhaps locally,
   // we may want to shorten this.
   LONG: os.platform() === "win32" ? 60_000 : 30_000,
+  MEDIUM: os.platform() === "win32" ? 30_000 : 15_000,
 };
 
 class PageObject {
@@ -178,7 +179,9 @@ class PageObject {
   }
 
   async waitForChatCompletion() {
-    await expect(this.getRetryButton()).toBeVisible();
+    await expect(this.getRetryButton()).toBeVisible({
+      timeout: Timeout.MEDIUM,
+    });
   }
 
   async clickRetry() {