test: Vector Store Implementation Integration Tests

src/langchain_google_spanner/__init__.py

@@ -17,6 +17,7 @@
     SecondaryIndex,
     SpannerVectorStore,
     TableColumn,
+    DistanceStrategy,
 )
 
 from .version import __version__
@@ -27,4 +28,5 @@
     "TableColumn",
     "SecondaryIndex",
     "QueryParameters",
+    "DistanceStrategy",
 ]

src/langchain_google_spanner/vector_store.py

@@ -672,12 +672,12 @@ def add_texts(
         if number_of_records == 0:
             return []
 
-        if ids is not None:
+        if ids is not None and len(ids) != number_of_records:
             raise ValueError(
                 f"size of list of IDs should be equals to number of documents. Expected: {number_of_records}  but found {len(ids)}"
             )
 
-        if metadatas is not None:
+        if metadatas is not None and len(metadatas) != number_of_records:
             raise ValueError(
                 f"size of list of metadatas should be equals to number of documents. Expected: {number_of_records}  but found {len(metadatas)}"
             )
@@ -785,6 +785,8 @@ def delete(
 
                 values.append(value)
 
+        delete_row_count = 0
+
         def delete_records(transaction):
             # ToDo: Debug why not working
             base_delete_statement = "DELETE FROM {} WHERE ".format(self._table_name)
@@ -798,7 +800,6 @@ def delete_records(transaction):
             sql_delete = base_delete_statement + where_clause
 
             # Iterate over the list of lists of values
-            delete_row_count = 0
             for value_tuple in values:
                 # Construct the params dictionary
                 values_tuple_param = (
@@ -817,7 +818,9 @@ def delete_records(transaction):
 
         self._database.run_in_transaction(delete_records)
 
-        return True
+        if delete_row_count > 0:
+            return True
+        return None
 
     def similarity_search_with_score_by_vector(
         self,