fix test_call_with_end_user_over_budget

BerriAI · May 30, 2024 · 6f06315 · 6f06315
1 parent ec35393
commit 6f06315
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py
@@ -467,7 +467,7 @@ async def test():
         asyncio.run(test())
     except Exception as e:
         error_detail = e.message
-        assert "Budget has been exceeded" in error_detail
+        assert "Authentication Error, ExceededBudget:" in error_detail
         print(vars(e))
 
 
@@ -559,7 +559,7 @@ async def return_body():
         asyncio.run(test())
     except Exception as e:
         error_detail = e.message
-        assert "Budget has been exceeded" in error_detail
+        assert "Budget has been exceeded! Current" in error_detail
         print(vars(e))
 
 
@@ -648,7 +648,7 @@ async def test():
             error_detail = e.message
         else:
             error_detail = traceback.format_exc()
-        assert "Budget has been exceeded" in error_detail
+        assert "Authentication Error, ExceededBudget:" in error_detail
         print(vars(e))
 
 
@@ -726,7 +726,7 @@ async def test():
         asyncio.run(test())
     except Exception as e:
         error_detail = e.message
-        assert "Budget has been exceeded" in error_detail
+        assert "Authentication Error, ExceededBudget:" in error_detail
         print(vars(e))
 
 
@@ -823,7 +823,7 @@ async def test():
         asyncio.run(test())
     except Exception as e:
         error_detail = e.message
-        assert "Budget has been exceeded" in error_detail
+        assert "Authentication Error, ExceededBudget:" in error_detail
         print(vars(e))