From 228a0bd6f7c0ea035c88e7b2f5e2aecf82d79afc Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 Aug 2024 12:11:49 -0700 Subject: [PATCH] fix qdrant semantic caching test --- litellm/tests/test_caching.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py index 80284a79f5..7df759add3 100644 --- a/litellm/tests/test_caching.py +++ b/litellm/tests/test_caching.py @@ -1761,6 +1761,7 @@ async def test_qdrant_semantic_cache_acompletion(): "content": f"write a one sentence poem about: {random_number}", } ], + mock_response="hello", max_tokens=20, ) print(f"Response1: {response1}") @@ -1810,6 +1811,7 @@ async def test_qdrant_semantic_cache_acompletion_stream(): max_tokens=40, temperature=1, stream=True, + mock_response="hi", ) async for chunk in response1: response_1_id = chunk.id