From f9bdc20be2be1f89e84f4a6c33a1e2fab0c5a5cd Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Fri, 23 Jan 2026 17:07:51 -0800
Subject: [PATCH] fixing tests

---
 .../playground/chat_ui/CodeSnippets.test.tsx  |  3 +-
 .../llm_calls/chat_completion.test.tsx        |  4 ++-
 .../llm_calls/responses_api.test.tsx          | 29 ++++++++++---------
 3 files changed, 20 insertions(+), 16 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/playground/chat_ui/CodeSnippets.test.tsx b/ui/litellm-dashboard/src/components/playground/chat_ui/CodeSnippets.test.tsx
index 56c5fd404e3..0113f7b2832 100644
--- a/ui/litellm-dashboard/src/components/playground/chat_ui/CodeSnippets.test.tsx
+++ b/ui/litellm-dashboard/src/components/playground/chat_ui/CodeSnippets.test.tsx
@@ -34,7 +34,8 @@ describe("CodeSnippets", () => {
     selectedTags: [],
     selectedVectorStores: [],
     selectedGuardrails: [],
-    selectedMCPTools: [],
+    selectedPolicies: [],
+    selectedMCPServers: [],
     selectedSdk: "openai" as const,
     selectedVoice: "alloy",
   };
diff --git a/ui/litellm-dashboard/src/components/playground/llm_calls/chat_completion.test.tsx b/ui/litellm-dashboard/src/components/playground/llm_calls/chat_completion.test.tsx
index dd23e0591ea..8786e022013 100644
--- a/ui/litellm-dashboard/src/components/playground/llm_calls/chat_completion.test.tsx
+++ b/ui/litellm-dashboard/src/components/playground/llm_calls/chat_completion.test.tsx
@@ -95,7 +95,8 @@ describe("chat_completion", () => {
       undefined, // traceId
       undefined, // vector_store_ids
       undefined, // guardrails
-      undefined, // selectedMCPTools
+      undefined, // policies
+      undefined, // selectedMCPServers
       undefined, // onImageGenerated
       undefined, // onSearchResults
       0.7, // temperature
@@ -157,6 +158,7 @@ describe("chat_completion", () => {
       undefined, // traceId
       undefined, // vector_store_ids
       undefined, // guardrails
+      undefined, // policies
       selectedMCPServers,
       undefined, // onImageGenerated
       undefined, // onSearchResults
diff --git a/ui/litellm-dashboard/src/components/playground/llm_calls/responses_api.test.tsx b/ui/litellm-dashboard/src/components/playground/llm_calls/responses_api.test.tsx
index 569c9878243..f14ac52b840 100644
--- a/ui/litellm-dashboard/src/components/playground/llm_calls/responses_api.test.tsx
+++ b/ui/litellm-dashboard/src/components/playground/llm_calls/responses_api.test.tsx
@@ -102,21 +102,22 @@ describe("responses_api", () => {
       mockUpdateTextUI,
       "gpt-4",
       "test-token",
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
+      undefined, // tags
+      undefined, // signal
+      undefined, // onReasoningContent
+      undefined, // onTimingData
+      undefined, // onUsageData
+      undefined, // traceId
+      undefined, // vector_store_ids
+      undefined, // guardrails
+      undefined, // policies
       selectedMCPServers,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
+      undefined, // previousResponseId
+      undefined, // onResponseId
+      undefined, // onMCPEvent
+      undefined, // codeInterpreterEnabled
+      undefined, // onCodeInterpreterResult
+      undefined, // customBaseUrl
       mcpServers,
       mcpServerToolRestrictions,
     );