diff --git a/projects/Basic-RAG/BasicRAG_ingest.py b/projects/Basic-RAG/BasicRAG_ingest.py new file mode 100644 index 0000000..b80e6ef --- /dev/null +++ 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")