Format, lint

Author: abrookins

redisvl/extensions/llmcache/semantic.py

@@ -180,7 +180,7 @@ async def _get_async_index(self) -> AsyncSearchIndex:
                 schema=self._index.schema,
                 redis_client=self.redis_kwargs["redis_client"],
                 redis_url=self.redis_kwargs["redis_url"],
-                **self.redis_kwargs["connection_kwargs"]
+                **self.redis_kwargs["connection_kwargs"],
             )
         return self._aindex
 

redisvl/index/index.py

@@ -2,6 +2,7 @@
 import atexit
 import json
 import threading
+import warnings
 from functools import wraps
 from typing import (
     TYPE_CHECKING,
@@ -16,7 +17,7 @@
     Union,
     cast,
 )
-import warnings
+
 from redisvl.utils.utils import deprecated_function
 
 if TYPE_CHECKING:

redisvl/redis/connection.py

@@ -261,15 +261,17 @@ def get_async_redis_connection(url: Optional[str] = None, **kwargs) -> AsyncRedi
         # fallback to env var REDIS_URL
         return AsyncRedis.from_url(get_address_from_env(), **kwargs)
 
-    @deprecated_function("sync_to_async_redis", "Please use an async Redis client instead.")
+    @deprecated_function(
+        "sync_to_async_redis", "Please use an async Redis client instead."
+    )
     @staticmethod
     def sync_to_async_redis(redis_client: Redis) -> AsyncRedis:
         # pick the right connection class
         connection_class: Type[AbstractConnection] = (
             AsyncSSLConnection
             if redis_client.connection_pool.connection_class == SSLConnection
             else AsyncConnection
-        ) # type: ignore
+        )  # type: ignore
         # make async client
         return AsyncRedis.from_pool(  # type: ignore
             AsyncConnectionPool(

redisvl/utils/utils.py

@@ -97,10 +97,13 @@ def deprecated_function(name: Optional[str] = None, replacement: Optional[str] =
     When the wrapped function is called, the decorator will log a deprecation
     warning.
     """
+
     def decorator(func):
         fn_name = name or func.__name__
-        warning_message = f"Function {fn_name} is deprecated and will be " \
+        warning_message = (
+            f"Function {fn_name} is deprecated and will be "
             "removed in the next major release. "
+        )
         if replacement:
             warning_message += replacement
 

tests/unit/test_utils.py

@@ -239,7 +239,6 @@ async def test_func(dtype=None, vectorizer=None):
         assert result == 1
 
 
-
 class TestDeprecatedFunction:
     def test_deprecated_function_warning(self):
         @deprecated_function("new_func", "Use new_func2")