From 11779b260e4f6ef6fdd7e7add735baef4fd10e3b Mon Sep 17 00:00:00 2001
From: tin-berri <tin@berri.ai>
Date: Mon, 26 Oct 2026 20:58:27 -0400
Subject: [PATCH] fix(ui): include endpoint-qualified models in Decisions
 Playground (#48504)

---
 .../playground/llm_calls/system_one.test.ts   | 21 ++++++++++++++++++-
 .../playground/llm_calls/system_one.ts        | 12 ++++++++---
 .../src/lib/decisionModels.ts                 |  2 +-
 3 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.test.ts
index 90e9482c89c..6fb1f8c8b8d 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.test.ts
@@ -207,18 +207,37 @@ describe("fetchDecisionModels", () => {
     vi.unstubAllGlobals();
   });
 
-  it("lists the key's decision models from the proxy that answers the request, sorted by name", async () => {
+  it("lists the key's models with decision modes or endpoints from the request proxy, sorted by name", async () => {
     mockFetch.mockResolvedValueOnce(
       modelGroupInfo([
         { model_group: "pplx-decider", mode: "evaluation" },
         { model_group: "gpt-5.5", mode: "chat" },
         { model_group: "jev-latest", mode: "evaluation" },
         { model_group: "unknown-mode", mode: null },
+        { model_group: "native-decision", mode: "decisions" },
+        {
+          model_group: "chat-decisions",
+          mode: "chat",
+          supported_endpoints: ["/v1/chat/completions", "/v1/decisions"],
+        },
+        { model_group: "chat-systemone", mode: "chat", supported_endpoints: ["/v1/systemone"] },
+        { model_group: "custom-decision", mode: "custom", supported_endpoints: ["/v1/decisions"] },
+        { model_group: "endpoint-only", supported_endpoints: ["/v1/decisions"] },
+        { model_group: "chat-only", mode: "chat", supported_endpoints: ["/v1/chat/completions"] },
+        { model_group: "no-endpoints", mode: "chat", supported_endpoints: [] },
+        { model_group: "null-endpoints", mode: "chat", supported_endpoints: null },
+        { model_group: "missing-mode" },
+        { model_group: "custom-mode", mode: "custom" },
       ]),
     );
 
     await expect(fetchDecisionModels("virtual-key", "https://custom.example.com/")).resolves.toEqual([
+      "chat-decisions",
+      "chat-systemone",
+      "custom-decision",
+      "endpoint-only",
       "jev-latest",
+      "native-decision",
       "pplx-decider",
     ]);
     expect(mockFetch).toHaveBeenCalledWith(
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.ts b/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.ts
index a4030029903..035a05729ba 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/playground/llm_calls/system_one.ts
@@ -2,7 +2,7 @@ import { z } from "zod";
 import { getGatewayBaseUrl, getGlobalLitellmHeaderName, getProxyBaseUrl } from "@/components/networking";
 import { withRequiredHeaders } from "@/components/llm_calls/request_headers";
 import { createApiClient } from "@/lib/http/client";
-import { isDecisionMode } from "@/lib/decisionModels";
+import { isDecisionEntry } from "@/lib/decisionModels";
 import {
   systemOneResponseSchema,
   type PlaygroundRequest,
@@ -26,7 +26,13 @@ export type PlaygroundDecisionRequest =
   | { endpoint: "/v1/decisions"; payload: OpenAIDecisionsRequest };
 
 const modelGroupInfoSchema = z.object({
-  data: z.array(z.object({ model_group: z.string(), mode: z.string().nullish() })),
+  data: z.array(
+    z.object({
+      model_group: z.string(),
+      mode: z.string().nullish(),
+      supported_endpoints: z.array(z.string()).nullish(),
+    }),
+  ),
 });
 
 function proxyClient(customBaseUrl?: string, getDefaultBaseUrl: () => string = getProxyBaseUrl) {
@@ -56,7 +62,7 @@ export async function fetchDecisionModels(
   });
   return modelGroupInfoSchema
     .parse(body)
-    .data.filter((group) => isDecisionMode(group.mode))
+    .data.filter(isDecisionEntry)
     .map((group) => group.model_group)
     .sort((a, b) => a.localeCompare(b));
 }
diff --git a/ui/litellm-dashboard/src/lib/decisionModels.ts b/ui/litellm-dashboard/src/lib/decisionModels.ts
index 1808b13c62b..15c632fe055 100644
--- a/ui/litellm-dashboard/src/lib/decisionModels.ts
+++ b/ui/litellm-dashboard/src/lib/decisionModels.ts
@@ -38,7 +38,7 @@ export const displayMode = (mode: string | null | undefined): string | undefined
 export const expandModeFilterValues = (modes: readonly string[]): readonly string[] =>
   modes.flatMap((mode) => (mode === DECISIONS_MODE ? DECISION_MODE_VALUES : [mode]));
 
-const isDecisionEntry = (entry: CostMapEntry): boolean =>
+export const isDecisionEntry = (entry: Pick<CostMapEntry, "mode" | "supported_endpoints">): boolean =>
   isDecisionMode(entry.mode) || (entry.supported_endpoints ?? []).some((endpoint) => DECISION_ENDPOINTS.has(endpoint));
 
 const parsedEntries = (costMap: unknown): ReadonlyArray<readonly [string, CostMapEntry]> => {