From 491017defaa55d8195fe672cdc93374b7243e07a Mon Sep 17 00:00:00 2001 From: Wei Ouyang Date: Fri, 6 Dec 2024 09:28:23 -0800 Subject: [PATCH] Add update --- hypha/artifact.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/hypha/artifact.py b/hypha/artifact.py index aadea68a..3690a695 100644 --- 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,