target migration

load redis dummy data
mantiumai · Jul 13, 2023 · 9a3d93f · 9a3d93f
1 parent c6fe339
commit 9a3d93f
Show file tree

Hide file tree

Showing 3 changed files with 1,575 additions and 2 deletions.
diff --git a/chirps/base_app/management/commands/load_redis_data.py b/chirps/base_app/management/commands/load_redis_data.py
@@ -26,7 +26,7 @@ def add_arguments(self, parser):
     def write_docs(pipe: redis.client.Pipeline, documents: list[dict], vector_field_name: str) -> None:
         """Write the documents to the redis database."""
         for doc in documents:
-            embedding = np.array(doc['embedding'], dtype=np.float32).tobytes()
+            embedding = np.array(doc['embeddings'], dtype=np.float32).tobytes()
             metadata = {vector_field_name: embedding}
             pipe.hset(doc['id'], mapping=metadata)  # type: ignore