Merge 187d7bed32 into 4534cafd3f

application/vectorstore/lancedb.py

@@ -0,0 +1,31 @@
+from langchain_community.vectorstores import LanceDB
+from application.vectorstore.base import BaseVectorStore
+from application.core.settings import settings
+
+class LancedbStore(BaseVectorStore):
+
+    def __init__(self, path, embeddings_key, docs_init=None):
+        super().__init__()
+        self.path = path
+        self.embeddings_key = embeddings_key
+        self.docsearch = None  
+        
+        embeddings = self._get_embeddings(settings.EMBEDDINGS_NAME, self.embeddings_key)
+        
+        if docs_init:
+            self.docsearch = LanceDB.from_documents(docs_init, embeddings)
+        else:
+            print(f"No initial documents provided for LanceDB initialization at {path}.")
+    
+    
+    def search(self, *args, **kwargs):
+        return self.docsearch.similarity_search(*args, **kwargs)
+
+    def add_texts(self, *args, **kwargs):
+        return self.docsearch.add_texts(*args, **kwargs)
+
+    def save_local(self, *args, **kwargs):
+        pass
+
+    def delete_index(self, *args, **kwargs):
+        pass

application/vectorstore/vector_creator.py

@@ -2,6 +2,7 @@ from application.vectorstore.faiss import FaissStore
 from application.vectorstore.elasticsearch import ElasticsearchStore
 from application.vectorstore.mongodb import MongoDBVectorStore
 from application.vectorstore.qdrant import QdrantStore
+from application.vectorstore.lancedb import LancedbStore
 
 
 class VectorCreator:
@@ -10,6 +11,8 @@ class VectorCreator:
         "elasticsearch": ElasticsearchStore,
         "mongodb": MongoDBVectorStore,
         "qdrant": QdrantStore,
+        "lancedb": LancedbStore,
+
     }
 
     @classmethod