From 1f2bf08136cdbd0db85f3fe180ba1e5404c9aa3e Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Fri, 5 Dec 2025 17:55:07 -0800
Subject: [PATCH] Fix flaky ui test

---
 .../src/components/add_model/add_model_tab.test.tsx       | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/add_model/add_model_tab.test.tsx b/ui/litellm-dashboard/src/components/add_model/add_model_tab.test.tsx
index 201bbb3c18..5f165bb7f8 100644
--- a/ui/litellm-dashboard/src/components/add_model/add_model_tab.test.tsx
+++ b/ui/litellm-dashboard/src/components/add_model/add_model_tab.test.tsx
@@ -42,7 +42,11 @@ const createQueryClient = () =>
     defaultOptions: {
       queries: {
         retry: false,
-        gcTime: 0,
+        staleTime: Infinity,
+        gcTime: Infinity,
+        refetchOnWindowFocus: false,
+        refetchOnReconnect: false,
+        refetchOnMount: false,
       },
     },
   });
@@ -140,7 +144,7 @@ describe("Add Model Tab", () => {
     );
 
     expect(await screen.findByRole("tab", { name: "Add Model" })).toBeInTheDocument();
-  });
+  }, 10000); // This test is flaky, adding a timeout until we find a better solution
 
   it("should display both Add Model and Add Auto Router tabs", async () => {
     const props = createTestProps();