Vec 229 aerospike 1.0 (#2)

* feat: Update aerospike-vector-search client to 1.0.0

* fix: Remove unsupported options from ini_options

* rename metadatas to labels

---------

Co-authored-by: Jesse Schumacher <[email protected]>

Author: Jesse S

docs/docs/integrations/vectorstores/aerospike.ipynb

@@ -51,7 +51,7 @@
    },
    "outputs": [],
    "source": [
-    "!pip install --upgrade --quiet aerospike-vector-search==0.6.1 langchain-community sentence-transformers langchain"
+    "!pip install --upgrade --quiet aerospike-vector-search==1.0.0 langchain-community sentence-transformers langchain"
    ]
   },
   {
@@ -401,7 +401,7 @@
     "        vector_field=VECTOR_KEY,\n",
     "        vector_distance_metric=MODEL_DISTANCE_CALC,\n",
     "        dimensions=MODEL_DIM,\n",
-    "        index_meta_data={\n",
+    "        index_labels={\n",
     "            \"model\": \"miniLM-L6-v2\",\n",
     "            \"date\": \"05/04/2024\",\n",
     "            \"dim\": str(MODEL_DIM),\n",

libs/community/langchain_community/vectorstores/aerospike.py

@@ -155,7 +155,7 @@ def convert_distance_strategy(
     def add_texts(
         self,
         texts: Iterable[str],
-        metadatas: Optional[List[dict]] = None,
+        labels: Optional[List[dict[str, str]]] = None,
         ids: Optional[List[str]] = None,
         set_name: Optional[str] = None,
         embedding_chunk_size: int = 1000,
@@ -168,7 +168,7 @@ def add_texts(
 
         Args:
             texts: Iterable of strings to add to the vectorstore.
-            metadatas: Optional list of metadatas associated with the texts.
+            labels: Optional list of labels associated with the texts.
             ids: Optional list of ids to associate with the texts.
             set_name: Optional aerospike set name to add the texts to.
             batch_size: Batch size to use when adding the texts to the vectorstore.
@@ -197,24 +197,24 @@ def add_texts(
         ids = ids or [str(uuid.uuid4()) for _ in texts]
 
         # We need to shallow copy so that we can add the vector and text keys
-        if metadatas:
-            metadatas = [m.copy() for m in metadatas]
+        if labels:
+            labels = [m.copy() for m in labels]
         else:
-            metadatas = metadatas or [{} for _ in texts]
+            labels = labels or [{} for _ in texts]
 
         for i in range(0, len(texts), embedding_chunk_size):
             chunk_texts = texts[i : i + embedding_chunk_size]
             chunk_ids = ids[i : i + embedding_chunk_size]
-            chunk_metadatas = metadatas[i : i + embedding_chunk_size]
+            chunk_labels = labels[i : i + embedding_chunk_size]
             embeddings = self._embed_documents(chunk_texts)
 
             for metadata, embedding, text in zip(
-                chunk_metadatas, embeddings, chunk_texts
+                chunk_labels, embeddings, chunk_texts
             ):
                 metadata[self._vector_key] = embedding
                 metadata[self._text_key] = text
 
-            for id, metadata in zip(chunk_ids, chunk_metadatas):
+            for id, metadata in zip(chunk_ids, chunk_labels):
                 metadata[self._id_key] = id
                 self._client.upsert(
                     namespace=self._namespace,
@@ -545,7 +545,7 @@ def from_texts(
         cls,
         texts: List[str],
         embedding: Embeddings,
-        metadatas: Optional[List[dict]] = None,
+        labels: Optional[List[dict]] = None,
         client: Client = None,
         namespace: str = "test",
         index_name: Optional[str] = None,
@@ -589,7 +589,7 @@ def from_texts(
 
         aerospike.add_texts(
             texts,
-            metadatas=metadatas,
+            labels=labels,
             ids=ids,
             index_name=index_name,
             embedding_chunk_size=embeddings_chunk_size,

libs/community/pyproject.toml

@@ -54,9 +54,8 @@ select = [ "E", "F", "I", "T201",]
 omit = [ "tests/*",]
 
 [tool.pytest.ini_options]
-addopts = "--strict-markers --strict-config --durations=5 --snapshot-warn-unused -vv"
+addopts = "--strict-markers --strict-config --durations=5 -vv"
 markers = [ "requires: mark tests as requiring a specific library", "scheduled: mark tests to run in scheduled testing", "compile: mark placeholder test used to compile integration tests without running them",]
-asyncio_mode = "auto"
 
 [tool.poetry.group.test]
 optional = true

libs/community/tests/integration_tests/vectorstores/docker-compose/aerospike/.gitignore

@@ -0,0 +1 @@
+features.conf

libs/community/tests/integration_tests/vectorstores/docker-compose/aerospike/aerospike-proximus.yml renamed to libs/community/tests/integration_tests/vectorstores/docker-compose/aerospike/aerospike-vector-search.yml

@@ -6,12 +6,11 @@ cluster:
 # The Proximus service listening ports, TLS and network interface.
 service:
   ports:
-    5002: {}
-  # Uncomment for local debugging
-  advertised-listeners:
-    default:
-      address: 127.0.0.1
-      port: 5002
+    5002:
+      advertised-listeners:
+        default:
+          address: 127.0.0.1
+          port: 5002
 
 # Management API listening ports, TLS and network interface.
 manage:
@@ -24,13 +23,14 @@ interconnect:
     5001: {}
 
 # Target Aerospike cluster
-aerospike:
+storage:
   seeds:
     - aerospike:
         port: 3000
 
 # The logging properties.
 logging:
   enable-console-logging: true
+  file: "/etc/aerospike-vector-search/avs.log"
   levels:
-    metrics-ticker: off
+    metrics-ticker: info

libs/community/tests/integration_tests/vectorstores/docker-compose/aerospike/aerospike.conf

@@ -51,7 +51,7 @@ namespace test {
         }
 }
 
-namespace proximus-meta {
+namespace avs-meta {
         replication-factor 1
         nsup-period 100
 

libs/community/tests/integration_tests/vectorstores/docker-compose/aerospike/docker-compose.yml

@@ -10,14 +10,14 @@ services:
     command:
       - "--config-file"
       - "/opt/aerospike/etc/aerospike/aerospike.conf"
-  proximus:
-    image: aerospike/aerospike-proximus:0.4.0
+  avs:
+    image: aerospike/aerospike-vector-search:0.9.0
     ports:
       - "5002:5002"
     networks:
       - aerospike-test
     volumes:
-      - .:/etc/aerospike-proximus
+      - .:/etc/aerospike-vector-search
 
 networks:
   aerospike-test: {}