adds new required dtype when calling array_to_buffer

redis-developer · Oct 3, 2024 · a53fc13 · a53fc13
1 parent 100567f
commit a53fc13
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python-recipes/RAG/01_redisvl.ipynb b/python-recipes/RAG/01_redisvl.ipynb
@@ -439,7 +439,7 @@
         "        'chunk_id': i,\n",
         "        'content': chunk.page_content,\n",
         "        # For HASH -- must convert embeddings to bytes\n",
-        "        'text_embedding': array_to_buffer(embeddings[i])\n",
+        "        'text_embedding': array_to_buffer(embeddings[i], dtype='float32')\n",
         "    } for i, chunk in enumerate(chunks)\n",
         "]\n",
         "\n",

diff --git a/python-recipes/RAG/04_advanced_redisvl.ipynb b/python-recipes/RAG/04_advanced_redisvl.ipynb
@@ -574,7 +574,7 @@
         "        'chunk_id': f'{i}',\n",
         "        'proposition': proposition,\n",
         "        # For HASH -- must convert embeddings to bytes\n",
-        "        'text_embedding': array_to_buffer(prop_embeddings[i])\n",
+        "        'text_embedding': array_to_buffer(prop_embeddings[i], dtype=\"float32\")\n",
         "    } for i, proposition in enumerate(propositions)\n",
         "]\n",
         "\n",