From e1a33a22b41405a82354473f37f2f5d8a70316d0 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 12 Mar 2025 20:46:29 -0700
Subject: [PATCH] fix(columns.tsx): add llm credentials column on model table

make it easy to know if a model is using a reusable credential
---
 .../components/model_dashboard/columns.tsx    | 23 +++++++++++++++++++
 .../src/components/model_dashboard/types.ts   |  3 ++-
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/ui/litellm-dashboard/src/components/model_dashboard/columns.tsx b/ui/litellm-dashboard/src/components/model_dashboard/columns.tsx
index 61048fa3b5..78043a574b 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard/columns.tsx
+++ b/ui/litellm-dashboard/src/components/model_dashboard/columns.tsx
@@ -188,6 +188,29 @@ export const columns = (
       );
     },
   },
+  {
+    header: "Credentials",
+    accessorKey: "litellm_credential_name",
+    cell: ({ row }) => {
+      const model = row.original;
+      return model.litellm_params.litellm_credential_name ? (
+        <div className="overflow-hidden">
+          <Tooltip title={model.litellm_params.litellm_credential_name}>
+            <Button
+              size="xs"
+              variant="light"
+              className="font-mono text-blue-500 bg-blue-50 hover:bg-blue-100 text-xs font-normal px-2 py-0.5 text-left overflow-hidden truncate max-w-[200px]"
+              onClick={() => setSelectedTeamId(model.model_info.team_id)}
+            >
+              {model.litellm_params.litellm_credential_name.slice(0, 7)}...
+            </Button>
+          </Tooltip>
+        </div>
+      ) : (
+        "-"
+      );
+    },
+  },
   {
     header: "Status",
     accessorKey: "model_info.db_model",
diff --git a/ui/litellm-dashboard/src/components/model_dashboard/types.ts b/ui/litellm-dashboard/src/components/model_dashboard/types.ts
index 04ecc1f009..67802c456b 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard/types.ts
+++ b/ui/litellm-dashboard/src/components/model_dashboard/types.ts
@@ -13,6 +13,7 @@ export interface LiteLLMParams {
   input_cost_per_token?: number;
   output_cost_per_token?: number;
   custom_llm_provider?: string;
+  litellm_credential_name?: string;
   [key: string]: any;
 }
 
@@ -41,4 +42,4 @@ export interface ModelDashboardProps {
   setModelData: (data: any) => void;
   premiumUser: boolean;
   teams: any[];
-} 
\ No newline at end of file
+}