embedding email object

2023-09-12 15:28:59 +08:00
parent ddc103af38
commit 8ac6065332
10 changed files with 155 additions and 42 deletions
@@ -6,11 +6,11 @@ import os


 class ChromaVectorStore(VectorBase):
-    def __init__(self, db_url, model_name: str) -> None:
-        super().__init__(db_url, model_name)
+    def __init__(self, model_name: str) -> None:
+        super().__init__(model_name)

        logging.info(
-            "will init chroma vector store, db={}, model={}".format(db_url, model_name)
+            "will init chroma vector store, model={}".format(model_name)
        )

        directory = os.path.join(