BaranziniLab / KG_RAG

Empower Large Language Models (LLM) using Knowledge Graph based Retrieval-Augmented Generation (KG-RAG) for knowledge intensive tasks
Apache License 2.0
624 stars 84 forks source link

disease_nodes_db #33

Closed yiyangfelix closed 3 months ago

yiyangfelix commented 3 months ago

Hi,

I have a question about the vector_db_path in the config.yaml. What is the diseases_nodes_db here? Thank you

fjfd commented 3 months ago

it appears to be entity node data in knowledge graph!

karthiksoman commented 3 months ago

@yiyangfelix apologies for the late response! VECTOR_DB_PATH in the cofig.yaml file is the name of the chroma vector db that will be created when running

python -m kg_rag.run_setup

In the backend, it runs the script

create_vectordb.py

You can find that script here and note Line no. 6 there, where it uses the VECTOR_DB_PATH as the name to create the database.