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

15
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"),
) )
response = await router.acompletion( try:
messages=[{"role": "user", "content": "Hello, how are you?"}], response = await router.acompletion(
model="openai/gpt-4o-mini", messages=[{"role": "user", "content": "Hello, how are you?"}],
mock_response="hi", model="openai/gpt-4o-mini",
) mock_response="hi",
print(response) )
print(response)
except Exception as e:
print("error", e)
await asyncio.sleep(0.5) await asyncio.sleep(0.5)