├── .gitignore ├── LICENSE ├── README.md ├── examples ├── documents │ ├── Gullivers-travels-A-Voyage-to-Lilliput.txt │ └── 西游记.txt ├── graphfiles │ ├── Gullivers-travels-communities.csv │ ├── Gullivers-travels-entities.csv │ ├── Gullivers-travels-relationships.csv │ ├── Gullivers-travels.html │ ├── Gullivers-travels.json │ └── 西游记.html ├── quick_start_index.py └── quick_start_search.py ├── imgs └── rag-lab.png ├── papaer.md ├── raglab ├── __init__.py ├── chunk │ ├── __init__.py │ └── regex_text_chunker.py ├── embeddings │ ├── __init__.py │ ├── azure_openai.py │ ├── base.py │ └── openai.py ├── graphrag │ ├── __init__.py │ ├── data_contracts │ │ ├── __init__.py │ │ ├── graph.py │ │ └── type.py │ ├── prompt │ │ ├── __init__.py │ │ ├── chunk_graph_extraction.py │ │ ├── community.py │ │ ├── disambiguation.py │ │ ├── expert.py │ │ ├── language.py │ │ └── search.py │ ├── prompt_functions │ │ ├── __init__.py │ │ ├── chunk_graph_extraction.py │ │ ├── community.py │ │ ├── disambiguation.py │ │ ├── expert.py │ │ └── language.py │ ├── search_functions │ │ ├── __init__.py │ │ └── example.py │ ├── utils │ │ ├── __init__.py │ │ ├── communities.py │ │ ├── dataclass_utils.py │ │ ├── graph_file_loader.py │ │ ├── graph_network_x_utils.py │ │ └── graph_utils.py │ └── visual │ │ ├── __init__.py │ │ ├── echart.py │ │ └── network_x.py ├── llms │ ├── __init__.py │ ├── azure_openai.py │ ├── base.py │ ├── huggingface_llm.py │ └── openai.py └── utils │ ├── __init__.py │ ├── json_paser.py │ └── parallel_utils.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/README.md -------------------------------------------------------------------------------- /examples/documents/Gullivers-travels-A-Voyage-to-Lilliput.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/documents/Gullivers-travels-A-Voyage-to-Lilliput.txt -------------------------------------------------------------------------------- /examples/documents/西游记.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/documents/西游记.txt -------------------------------------------------------------------------------- /examples/graphfiles/Gullivers-travels-communities.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/graphfiles/Gullivers-travels-communities.csv -------------------------------------------------------------------------------- /examples/graphfiles/Gullivers-travels-entities.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/graphfiles/Gullivers-travels-entities.csv -------------------------------------------------------------------------------- /examples/graphfiles/Gullivers-travels-relationships.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/graphfiles/Gullivers-travels-relationships.csv -------------------------------------------------------------------------------- /examples/graphfiles/Gullivers-travels.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/graphfiles/Gullivers-travels.html -------------------------------------------------------------------------------- /examples/graphfiles/Gullivers-travels.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/graphfiles/Gullivers-travels.json -------------------------------------------------------------------------------- /examples/graphfiles/西游记.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/graphfiles/西游记.html -------------------------------------------------------------------------------- /examples/quick_start_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/quick_start_index.py -------------------------------------------------------------------------------- /examples/quick_start_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/examples/quick_start_search.py -------------------------------------------------------------------------------- /imgs/rag-lab.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/imgs/rag-lab.png -------------------------------------------------------------------------------- /papaer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/papaer.md -------------------------------------------------------------------------------- /raglab/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/__init__.py -------------------------------------------------------------------------------- /raglab/chunk/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/chunk/__init__.py -------------------------------------------------------------------------------- /raglab/chunk/regex_text_chunker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/chunk/regex_text_chunker.py -------------------------------------------------------------------------------- /raglab/embeddings/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/embeddings/__init__.py -------------------------------------------------------------------------------- /raglab/embeddings/azure_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/embeddings/azure_openai.py -------------------------------------------------------------------------------- /raglab/embeddings/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/embeddings/base.py -------------------------------------------------------------------------------- /raglab/embeddings/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/embeddings/openai.py -------------------------------------------------------------------------------- /raglab/graphrag/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/__init__.py -------------------------------------------------------------------------------- /raglab/graphrag/data_contracts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/data_contracts/__init__.py -------------------------------------------------------------------------------- /raglab/graphrag/data_contracts/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/data_contracts/graph.py -------------------------------------------------------------------------------- /raglab/graphrag/data_contracts/type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/data_contracts/type.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /raglab/graphrag/prompt/chunk_graph_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt/chunk_graph_extraction.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt/community.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt/community.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt/disambiguation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt/disambiguation.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt/expert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt/expert.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt/language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt/language.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt/search.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt_functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /raglab/graphrag/prompt_functions/chunk_graph_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt_functions/chunk_graph_extraction.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt_functions/community.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt_functions/community.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt_functions/disambiguation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt_functions/disambiguation.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt_functions/expert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt_functions/expert.py -------------------------------------------------------------------------------- /raglab/graphrag/prompt_functions/language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/prompt_functions/language.py -------------------------------------------------------------------------------- /raglab/graphrag/search_functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/search_functions/__init__.py -------------------------------------------------------------------------------- /raglab/graphrag/search_functions/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/search_functions/example.py -------------------------------------------------------------------------------- /raglab/graphrag/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /raglab/graphrag/utils/communities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/utils/communities.py -------------------------------------------------------------------------------- /raglab/graphrag/utils/dataclass_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/utils/dataclass_utils.py -------------------------------------------------------------------------------- /raglab/graphrag/utils/graph_file_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/utils/graph_file_loader.py -------------------------------------------------------------------------------- /raglab/graphrag/utils/graph_network_x_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/utils/graph_network_x_utils.py -------------------------------------------------------------------------------- /raglab/graphrag/utils/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/utils/graph_utils.py -------------------------------------------------------------------------------- /raglab/graphrag/visual/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/visual/__init__.py -------------------------------------------------------------------------------- /raglab/graphrag/visual/echart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/visual/echart.py -------------------------------------------------------------------------------- /raglab/graphrag/visual/network_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/graphrag/visual/network_x.py -------------------------------------------------------------------------------- /raglab/llms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/llms/__init__.py -------------------------------------------------------------------------------- /raglab/llms/azure_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/llms/azure_openai.py -------------------------------------------------------------------------------- /raglab/llms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/llms/base.py -------------------------------------------------------------------------------- /raglab/llms/huggingface_llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/llms/huggingface_llm.py -------------------------------------------------------------------------------- /raglab/llms/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/llms/openai.py -------------------------------------------------------------------------------- /raglab/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/utils/__init__.py -------------------------------------------------------------------------------- /raglab/utils/json_paser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/utils/json_paser.py -------------------------------------------------------------------------------- /raglab/utils/parallel_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/raglab/utils/parallel_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyShopAI/rag-lab/HEAD/setup.py --------------------------------------------------------------------------------