fix mistyped args

redis · Sep 5, 2023 · f1d940b · f1d940b
1 parent 632720c
commit f1d940b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/redisvl/vectorize/text/openai.py b/redisvl/vectorize/text/openai.py
@@ -80,7 +80,7 @@ def embed_many(
                 perform before vectorization. Defaults to None.
             batch_size (int, optional): Batch size of texts to use when creating
                 embeddings. Defaults to 10.
-            as_buffer (Optional[float], optional): Whether to convert the raw embedding
+            as_buffer (bool, optional): Whether to convert the raw embedding
                 to a byte string. Defaults to False.
 
         Returns:
@@ -197,7 +197,7 @@ async def aembed(
             text (str): Chunk of text to embed.
             preprocess (Optional[Callable], optional): Optional preprocessing callable to
                 perform before vectorization. Defaults to None.
-            as_buffer (float, optional): Whether to convert the raw embedding
+            as_buffer (bool, optional): Whether to convert the raw embedding
                 to a byte string. Defaults to False.
 
         Returns:

diff --git a/redisvl/vectorize/text/vertexai.py b/redisvl/vectorize/text/vertexai.py
@@ -71,8 +71,8 @@ def embed_many(
         self,
         texts: List[str],
         preprocess: Optional[Callable] = None,
-        batch_size: Optional[int] = 10,
-        as_buffer: Optional[float] = False,
+        batch_size: int = 10,
+        as_buffer: bool = False,
     ) -> List[List[float]]:
         """Embed many chunks of texts using the VertexAI API.
 
@@ -82,7 +82,7 @@ def embed_many(
                 perform before vectorization. Defaults to None.
             batch_size (int, optional): Batch size of texts to use when creating
                 embeddings. Defaults to 10.
-            as_buffer (Optional[float], optional): Whether to convert the raw embedding
+            as_buffer (bool, optional): Whether to convert the raw embedding
                 to a byte string. Defaults to False.
 
         Returns:
@@ -113,15 +113,15 @@ def embed(
         self,
         text: str,
         preprocess: Optional[Callable] = None,
-        as_buffer: Optional[float] = False,
+        as_buffer: bool = False,
     ) -> List[float]:
         """Embed a chunk of text using the VertexAI API.
 
         Args:
             text (str): Chunk of text to embed.
             preprocess (Optional[Callable], optional): Optional preprocessing callable to
                 perform before vectorization. Defaults to None.
-            as_buffer (Optional[float], optional): Whether to convert the raw embedding
+            as_buffer (bool, optional): Whether to convert the raw embedding
                 to a byte string. Defaults to False.
 
         Returns: