fix redis

run-llama · Oct 2, 2024 · 3bf7a95 · 3bf7a95
1 parent ea582d5
commit 3bf7a95
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...ons/vector_stores/llama-index-vector-stores-redis/llama_index/vector_stores/redis/base.py b/...ons/vector_stores/llama-index-vector-stores-redis/llama_index/vector_stores/redis/base.py
@@ -4,6 +4,7 @@
 """
 
 import logging
+import numpy as np
 from typing import Any, Dict, List, Optional
 
 import fsspec
@@ -250,7 +251,7 @@ def add(self, nodes: List[BaseNode], **add_kwargs: Any) -> List[str]:
                 NODE_ID_FIELD_NAME: node.node_id,
                 DOC_ID_FIELD_NAME: node.ref_doc_id,
                 TEXT_FIELD_NAME: node.get_content(metadata_mode=MetadataMode.NONE),
-                VECTOR_FIELD_NAME: array_to_buffer(embedding),
+                VECTOR_FIELD_NAME: array_to_buffer(embedding, dtype=np.float32),
             }
             # parse and append metadata
             additional_metadata = node_to_metadata_dict(

diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-redis/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-redis/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-redis"
 readme = "README.md"
-version = "0.3.2"
+version = "0.3.3"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"