From 6f063156be5457e57dbd7ba356e72c267e8e4b16 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 29 May 2024 20:48:08 -0700 Subject: [PATCH] fix test_call_with_end_user_over_budget --- litellm/tests/test_key_generate_prisma.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index d937beab454..375a1c85d8f 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -467,7 +467,7 @@ def test_call_with_user_over_budget(prisma_client): 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 @@ def test_call_with_end_user_over_budget(prisma_client): 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 @@ def test_call_with_proxy_over_budget(prisma_client): 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 @@ def test_call_with_user_over_budget_stream(prisma_client): 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 @@ def test_call_with_proxy_over_budget_stream(prisma_client): 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))