Note
The current repository version is for test usage and it's under update.
- Download the .zip file of the indexed knowledge graph here.
- snomed_all_dataset_nodoc_hitsnomed.zip: The indices of full knowledge graph embedded by hitsnomed
- snomed_all_dataset_nodoc_minilml12v2.zip: The indices of full knowledge graph embedded by minilml12v2
- snomed_partial_dataset_nodoc_minilml6v2.zip: The indices of partial knowledge graph embedded by minilml6v2
- Download the LLM and embedding model (recommendations: mistral-small and hitsnomed ). Place them in a new folder and check the paths in the rag_util.py
$ LLM = {} and
EMBED_MODEL = {}
-
Unzip the file and put it into the "index" folder. (If there is no "index" folder, create a new one under the root directory.)
-
Run the rag_playground.ipynb for test. This notebook contains several working scenarios for one sample input. For testing file input, run rag_extraction.py.