Add update

amun-ai · Dec 6, 2024 · 491017d · 491017d
1 parent 8124cb7
commit 491017d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/hypha/artifact.py b/hypha/artifact.py
@@ -1997,6 +1997,7 @@ async def add_vectors(
         self,
         artifact_id: str,
         vectors: list,
+        update: bool = False,
         embedding_models: Optional[Dict[str, str]] = None,
         context: dict = None,
     ):
@@ -2024,6 +2025,7 @@ async def add_vectors(
                     await self._vector_engine.add_vectors(
                         f"{artifact.workspace}/{artifact.alias}",
                         vectors,
+                        update=update,
                         embedding_models=embedding_models
                         or artifact.config.get("embedding_models"),
                         s3_client=s3_client,