fix metadata and insert_vector

rizquuula · Nov 11, 2024 · c2af99f · c2af99f
1 parent 77f6050
commit c2af99f
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 9 deletions.
diff --git a/examples/chromadb.py b/examples/chromadb.py
@@ -18,7 +18,7 @@
     distance_function=DistanceFunction.L2,
 )
 
-new_v = ch.create_vector(v)
+new_v = ch.insert_vector(v)
 print("CREATE_VECTOR", new_v)
 
 new_v = ch.read_vector(new_v.id)

diff --git a/examples/pgvector.py b/examples/pgvector.py
@@ -19,7 +19,7 @@
     distance_function=DistanceFunction.L2,
 )
 
-new_v = pgv.create_vector(v)
+new_v = pgv.insert_vector(v)
 print("CREATE_VECTOR", new_v)
 
 new_v = pgv.read_vector(new_v.id)

diff --git a/examples/qdrant.py b/examples/qdrant.py
@@ -18,7 +18,7 @@
     distance_function=DistanceFunction.COSINE,
 )
 
-new_v = qv.create_vector(v)
+new_v = qv.insert_vector(v)
 print("CREATE_VECTOR", new_v)
 
 new_v = qv.read_vector(new_v.id)

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyvectordb"
-version = "0.1.3.1"
+version = "0.1.4"
 description = "Simple Wrapper for vector database in Python with minimal support for CRUD and retrieve."
 authors = ["M Razif Rizqullah <razifrizqullah@gmail.com>"]
 repository = "https://github.com/rizquuula/pyvectordb"

diff --git a/pyvectordb/chromadb/chromadb.py b/pyvectordb/chromadb/chromadb.py
@@ -66,11 +66,11 @@ def __init_collection(self) -> None:
                 metadata={"hnsw:space": distance_func.value}
             )
 
-    def create_vector(self, vector: Vector) -> Vector:
+    def insert_vector(self, vector: Vector) -> Vector:
         self.collection.add(
             ids=[vector.get_id()],
             embeddings=[vector.embedding],
-            metadatas=[{"metadata": vector.metadata}],
+            metadatas=[vector.metadata],
         )
         return vector
 

diff --git a/pyvectordb/driver.py b/pyvectordb/driver.py
@@ -9,7 +9,7 @@
 class VectorDB(ABC):
 
     @abstractmethod
-    def create_vector(self, vector: Vector) -> Vector:
+    def insert_vector(self, vector: Vector) -> Vector:
         ...
 
     @abstractmethod

diff --git a/pyvectordb/pgvector/pgvector.py b/pyvectordb/pgvector/pgvector.py
@@ -78,7 +78,7 @@ def __init_collection(self) -> None:
         self.conn.commit()
 
 
-    def create_vector(self, vector: Vector) -> Vector:
+    def insert_vector(self, vector: Vector) -> Vector:
         conn = self.conn
 
         v = self.__vector_orm(

diff --git a/pyvectordb/qdrant/qdrant.py b/pyvectordb/qdrant/qdrant.py
@@ -67,7 +67,7 @@ def __get_distance_function(self, distance: DistanceFunction) -> Distance:
 
         return distance_func
 
-    def create_vector(self, vector: Vector) -> Vector:
+    def insert_vector(self, vector: Vector) -> Vector:
         vector_id = vector.get_id()
 
         self.client.upsert(