phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test_prometheus_metric_tracking

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-19 22:11:30 -08:00
parent 7463dab9c6
commit 8b92e4f77a
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/local_testing/test_provider_budgets.py
View file

@ -270,12 +270,15 @@ async def test_prometheus_metric_tracking():
redis_password=os.getenv("REDIS_PASSWORD"), redis_password=os.getenv("REDIS_PASSWORD"),
) )
try:
response = await router.acompletion( response = await router.acompletion(
messages=[{"role": "user", "content": "Hello, how are you?"}], messages=[{"role": "user", "content": "Hello, how are you?"}],
model="openai/gpt-4o-mini", model="openai/gpt-4o-mini",
mock_response="hi", mock_response="hi",
) )
print(response) print(response)
except Exception as e:
print("error", e)
await asyncio.sleep(0.5) await asyncio.sleep(0.5)