Fix: #720 - Preserve sparse_embeddings in data_points_to_batch_update…

…_records function
langchain-ai · Jan 30, 2025 · d8b5683 · d8b5683
1 parent af08759
commit d8b5683
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/libs/vertexai/langchain_google_vertexai/vectorstores/_utils.py b/libs/vertexai/langchain_google_vertexai/vectorstores/_utils.py
@@ -170,6 +170,15 @@ def data_points_to_batch_update_records(
             ],
         }
 
+        if (
+            hasattr(data_point, "sparse_embedding")
+            and data_point.sparse_embedding is not None
+        ):
+            record["sparse_embedding"] = {
+                "values": [value for value in data_point.sparse_embedding.values],
+                "dimensions": [dim for dim in data_point.sparse_embedding.dimensions],
+            }
+
         records.append(record)
 
     return records