BasicRAG ingest cookbook

arjbingly · Mar 25, 2024 · ee884c7 · ee884c7
1 parent 682d708
commit ee884c7
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/projects/Basic-RAG/BasicRAG_ingest.py b/projects/Basic-RAG/BasicRAG_ingest.py
@@ -0,0 +1,14 @@
+from pathlib import Path
+
+from grag.components.multivec_retriever import Retriever
+from grag.components.vectordb.deeplake_client import DeepLakeClient
+
+# from grag.rag.basic_rag import BasicRAG
+
+client = DeepLakeClient(collection_name="test")
+retriever = Retriever(vectordb=client)
+
+dir_path = Path(__file__).parent / 'some_dir'
+
+retriever.ingest(dir_path)
+# rag = BasicRAG(doc_chain="refine")