From a8473042d19e844af63ddbb8e4af001ce7356bbc Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 18 Mar 2024 18:28:13 -0700
Subject: [PATCH] (ui) show users their spend when max budget not set

---
 .../src/components/view_user_spend.tsx        | 22 +++----------------
 1 file changed, 3 insertions(+), 19 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_user_spend.tsx b/ui/litellm-dashboard/src/components/view_user_spend.tsx
index a2c61667d..125f0c314 100644
--- a/ui/litellm-dashboard/src/components/view_user_spend.tsx
+++ b/ui/litellm-dashboard/src/components/view_user_spend.tsx
@@ -37,31 +37,15 @@ const ViewUserSpend: React.FC<ViewUserSpendProps> = ({ userID, userSpendData, us
     const [spend, setSpend] = useState(userSpendData?.spend);
     const [maxBudget, setMaxBudget] = useState(userSpendData?.max_budget || null);
 
-    useEffect(() => {
-        const fetchData = async () => {
-            if (userRole === "Admin") {
-                try {
-                    const data = await spendUsersCall(accessToken, "litellm-proxy-budget");
-                    console.log("Result from callSpendUsers:", data);
-                    const total_budget = data[0]
-                    setSpend(total_budget?.spend);
-                    setMaxBudget(total_budget?.max_budget || null);
-                } catch (error) {
-                    console.error("Failed to get spend for user", error);
-                }
-            }
-        };
-
-        fetchData();
-    }, [userRole, accessToken, userID]);
-
     const displayMaxBudget = maxBudget !== null ? `$${maxBudget} limit` : "No limit";
+    const roundedSpend = spend !== undefined ? spend.toFixed(4) : null;
+
 
     return (
         <>
       <Card className="mx-auto mb-4">
         <Metric>
-          ${spend}
+          ${roundedSpend}
         </Metric>
         <Title>
             / {displayMaxBudget}