From 97c82225b128e495b8686ff5e2e2fed82ff51825 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 4 Jun 2024 07:55:53 -0700 Subject: [PATCH] test fix - test_acompletion_caching_on_router --- litellm/tests/test_router_caching.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_router_caching.py b/litellm/tests/test_router_caching.py index 4b459bab84..0f64a1bd47 100644 --- a/litellm/tests/test_router_caching.py +++ b/litellm/tests/test_router_caching.py @@ -115,7 +115,8 @@ async def test_acompletion_caching_on_router(): model="gpt-3.5-turbo", messages=messages, temperature=1 ) print(f"response1: {response1}") - await asyncio.sleep(1) # add cache is async, async sleep for cache to get set + await asyncio.sleep(5) # add cache is async, async sleep for cache to get set + response2 = await router.acompletion( model="gpt-3.5-turbo", messages=messages, temperature=1 )