Add update

amun-ai · Dec 6, 2024 · 8124cb7 · 8124cb7
1 parent 408d6e2
commit 8124cb7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hypha/vectors.py b/hypha/vectors.py
@@ -373,6 +373,7 @@ async def add_vectors(
         self,
         collection_name: str,
         vectors: List[Dict[str, Any]],
+        update: bool = False,
         embedding_models: Optional[Dict[str, str]] = None,
         s3_client=None,
         bucket=None,
@@ -385,6 +386,8 @@ async def add_vectors(
             if "id" in vector:
                 _id = vector["id"]
             else:
+                if update:
+                    raise ValueError("ID is required for update operation.")
                 _id = str(uuid.uuid4())
                 vector["id"] = _id
             ids.append(_id)