diff --git a/redisvl/vectorize/text/openai.py b/redisvl/vectorize/text/openai.py index df96e80f..1e01b077 100644 --- 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 index 00779235..191bc75d 100644 --- 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,7 +113,7 @@ 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. @@ -121,7 +121,7 @@ def embed( 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: