diff --git a/litellm/caching/qdrant_semantic_cache.py b/litellm/caching/qdrant_semantic_cache.py index 167a28791d5..f12f8bce32e 100644 --- a/litellm/caching/qdrant_semantic_cache.py +++ b/litellm/caching/qdrant_semantic_cache.py @@ -178,7 +178,9 @@ class QdrantSemanticCache(BaseCache): from litellm._uuid import uuid # get the prompt - messages = kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + messages = ( + kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + ) if messages is None: print_verbose("No messages provided for semantic caching") return @@ -223,7 +225,9 @@ class QdrantSemanticCache(BaseCache): print_verbose(f"sync qdrant semantic-cache get_cache, kwargs: {kwargs}") # get the messages - messages = kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + messages = ( + kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + ) if messages is None: print_verbose("No messages provided for semantic lookup") return @@ -295,7 +299,9 @@ class QdrantSemanticCache(BaseCache): print_verbose(f"async qdrant semantic-cache set_cache, kwargs: {kwargs}") # get the prompt - messages = kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + messages = ( + kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + ) if messages is None: print_verbose("No messages provided for semantic caching") return @@ -357,7 +363,9 @@ class QdrantSemanticCache(BaseCache): from litellm.proxy.proxy_server import llm_model_list, llm_router # get the messages - messages = kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + messages = ( + kwargs.get("messages") if "messages" in kwargs else kwargs.get("message") + ) if messages is None: print_verbose("No messages provided for semantic lookup") return