Skip to content

Commit

Permalink
test ingestion
Browse files Browse the repository at this point in the history
  • Loading branch information
sanchitvj committed Mar 25, 2024
1 parent 77b4bb8 commit b84504d
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 10 deletions.
5 changes: 1 addition & 4 deletions projects/Basic-RAG/BasicRAG_ingest.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,9 @@
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"
dir_path = Path(__file__).parents[2] / "data/test/pdfs/new_papers"

retriever.ingest(dir_path)
# rag = BasicRAG(doc_chain="refine")
6 changes: 0 additions & 6 deletions src/config.ini
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,6 @@ n_gpu_layers_cpp : -1
std_out : True
base_dir : ${root:root_path}/models

[deeplake]
collection_name : arxiv
embedding_type : instructor-embedding
embedding_model : hkunlp/instructor-xl
store_path : ${data:data_path}/vectordb

[chroma]
host : localhost
port : 8000
Expand Down

0 comments on commit b84504d

Please sign in to comment.