ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki 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
Show Stats 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"),
)
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)