From 8b92e4f77a2d87df34998f837a2f87b24968d95d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 19 Nov 2024 22:11:30 -0800 Subject: [PATCH] fix test_prometheus_metric_tracking --- tests/local_testing/test_provider_budgets.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/tests/local_testing/test_provider_budgets.py b/tests/local_testing/test_provider_budgets.py index 0c1995d43..78c95246d 100644 --- a/tests/local_testing/test_provider_budgets.py +++ b/tests/local_testing/test_provider_budgets.py @@ -270,12 +270,15 @@ async def test_prometheus_metric_tracking(): redis_password=os.getenv("REDIS_PASSWORD"), ) - response = await router.acompletion( - messages=[{"role": "user", "content": "Hello, how are you?"}], - model="openai/gpt-4o-mini", - mock_response="hi", - ) - print(response) + try: + response = await router.acompletion( + messages=[{"role": "user", "content": "Hello, how are you?"}], + model="openai/gpt-4o-mini", + mock_response="hi", + ) + print(response) + except Exception as e: + print("error", e) await asyncio.sleep(0.5)