scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-02 22:12:44 -07:00
parent 3da8ce6a75
commit 67433a04a2
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_router.py
View file

@ -1865,7 +1865,10 @@ async def test_router_model_usage(mock_response):
)
await asyncio.sleep(3)
initial_usage = await llm_router.get_model_group_usage(model_group=model)
initial_usage_tuple = await llm_router.get_model_group_usage(
model_group=model
)
initial_usage = initial_usage_tuple[0]
# completion call - 10 tokens
_ = await llm_router.acompletion(
@ -1873,7 +1876,10 @@ async def test_router_model_usage(mock_response):
)
await asyncio.sleep(3)
updated_usage = await llm_router.get_model_group_usage(model_group=model)
updated_usage_tuple = await llm_router.get_model_group_usage(
model_group=model
)
updated_usage = updated_usage_tuple[0]
assert updated_usage == initial_usage + 10 # type: ignore
break