formatting

Author: justin-cechmanek

redisvl/utils/vectorize/text/azureopenai.py

@@ -5,9 +5,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
-
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 # ignore that openai isn't imported
 # mypy: disable-error-code="name-defined"
@@ -169,7 +168,7 @@ def _set_model_dims(self, model) -> int:
         stop=stop_after_attempt(6),
         retry=retry_if_not_exception_type(TypeError),
     )
-    @deprecated_argument('dtype')
+    @deprecated_argument("dtype")
     def embed_many(
         self,
         texts: List[str],
@@ -216,7 +215,7 @@ def embed_many(
         stop=stop_after_attempt(6),
         retry=retry_if_not_exception_type(TypeError),
     )
-    @deprecated_argument('dtype')
+    @deprecated_argument("dtype")
     def embed(
         self,
         text: str,
@@ -255,7 +254,7 @@ def embed(
         stop=stop_after_attempt(6),
         retry=retry_if_not_exception_type(TypeError),
     )
-    @deprecated_argument('dtype')
+    @deprecated_argument("dtype")
     async def aembed_many(
         self,
         texts: List[str],
@@ -304,7 +303,7 @@ async def aembed_many(
         stop=stop_after_attempt(6),
         retry=retry_if_not_exception_type(TypeError),
     )
-    @deprecated_argument('dtype')
+    @deprecated_argument("dtype")
     async def aembed(
         self,
         text: str,

redisvl/utils/vectorize/text/bedrock.py

@@ -6,8 +6,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 
 class BedrockTextVectorizer(BaseVectorizer):

redisvl/utils/vectorize/text/cohere.py

@@ -5,8 +5,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 # ignore that cohere isn't imported
 # mypy: disable-error-code="name-defined"
@@ -112,7 +112,7 @@ def _set_model_dims(self, model) -> int:
             raise ValueError(f"Error setting embedding model dimensions: {str(e)}")
         return len(embedding)
 
-    @deprecated_argument('dtype')
+    @deprecated_argument("dtype")
     def embed(
         self,
         text: str,
@@ -179,7 +179,7 @@ def embed(
         stop=stop_after_attempt(6),
         retry=retry_if_not_exception_type(TypeError),
     )
-    @deprecated_argument('dtype')
+    @deprecated_argument("dtype")
     def embed_many(
         self,
         texts: List[str],

redisvl/utils/vectorize/text/custom.py

@@ -2,8 +2,8 @@
 
 from pydantic.v1 import PrivateAttr
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 
 def _check_vector(result: list, method_name: str) -> None:

redisvl/utils/vectorize/text/huggingface.py

@@ -2,8 +2,8 @@
 
 from pydantic.v1 import PrivateAttr
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 
 class HFTextVectorizer(BaseVectorizer):

redisvl/utils/vectorize/text/mistral.py

@@ -5,8 +5,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 # ignore that mistralai isn't imported
 # mypy: disable-error-code="name-defined"

redisvl/utils/vectorize/text/openai.py

@@ -5,8 +5,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 # ignore that openai isn't imported
 # mypy: disable-error-code="name-defined"

redisvl/utils/vectorize/text/vertexai.py

@@ -5,8 +5,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 
 class VertexAITextVectorizer(BaseVectorizer):

redisvl/utils/vectorize/text/voyageai.py

@@ -5,8 +5,8 @@
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tenacity.retry import retry_if_not_exception_type
 
-from redisvl.utils.vectorize.base import BaseVectorizer
 from redisvl.utils.utils import deprecated_argument
+from redisvl.utils.vectorize.base import BaseVectorizer
 
 # ignore that voyageai isn't imported
 # mypy: disable-error-code="name-defined"

schemas/semantic_router.yaml

@@ -19,4 +19,5 @@ vectorizer:
   model: sentence-transformers/all-mpnet-base-v2
 routing_config:
   max_k: 2
-  aggregation_method: avg
+  aggregation_method: avg
+  distance_threshold: 0.3