├── .gitignore ├── LICENSE ├── README.md ├── README_zh.md ├── assert └── img │ └── overview.png ├── dataset └── cinderella │ └── cinderella_1 │ ├── cinderella.txt │ ├── corpus.jsonl │ └── qas.jsonl ├── main_openai.py ├── main_vllm.py ├── requirements.txt ├── script ├── chunk_doc_corpus.py └── eval_qa.py └── src └── comorag ├── ComoRAG.py ├── __init__.py ├── embedding_model ├── BGEEmbedding.py ├── OpenAI.py ├── __init__.py └── base.py ├── embedding_store.py ├── information_extraction ├── __init__.py ├── openie_openai.py └── openie_vllm_offline.py ├── llm ├── __init__.py ├── base.py ├── openai_gpt.py └── vllm_offline.py ├── prompts ├── __init__.py ├── dspy_prompts │ └── filter_llama3.3-70B-Instruct.json ├── filter_default_prompt.py ├── linking.py ├── prompt_template_manager.py └── templates │ ├── README.md │ ├── __init__.py │ ├── agent_probe.py │ ├── memory_fusion.py │ ├── ner.py │ ├── ner_query.py │ ├── node_fusion.py │ ├── rag_qa_mc.py │ ├── rag_qa_mc_memory.py │ ├── rag_qa_narrativeqa.py │ └── triple_extraction.py ├── rerank.py └── utils ├── __init__.py ├── agents.py ├── cluster_utils.py ├── config_utils.py ├── embed_utils.py ├── llm_utils.py ├── logging_utils.py ├── memory_utils.py ├── misc_utils.py ├── qa_utils.py ├── summarization_utils.py ├── timeline_utils.py └── typing_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/README.md -------------------------------------------------------------------------------- /README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/README_zh.md -------------------------------------------------------------------------------- /assert/img/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/assert/img/overview.png -------------------------------------------------------------------------------- /dataset/cinderella/cinderella_1/cinderella.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/dataset/cinderella/cinderella_1/cinderella.txt -------------------------------------------------------------------------------- /dataset/cinderella/cinderella_1/corpus.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/dataset/cinderella/cinderella_1/corpus.jsonl -------------------------------------------------------------------------------- /dataset/cinderella/cinderella_1/qas.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/dataset/cinderella/cinderella_1/qas.jsonl -------------------------------------------------------------------------------- /main_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/main_openai.py -------------------------------------------------------------------------------- /main_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/main_vllm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/chunk_doc_corpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/script/chunk_doc_corpus.py -------------------------------------------------------------------------------- /script/eval_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/script/eval_qa.py -------------------------------------------------------------------------------- /src/comorag/ComoRAG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/ComoRAG.py -------------------------------------------------------------------------------- /src/comorag/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/__init__.py -------------------------------------------------------------------------------- /src/comorag/embedding_model/BGEEmbedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/embedding_model/BGEEmbedding.py -------------------------------------------------------------------------------- /src/comorag/embedding_model/OpenAI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/embedding_model/OpenAI.py -------------------------------------------------------------------------------- /src/comorag/embedding_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/embedding_model/__init__.py -------------------------------------------------------------------------------- /src/comorag/embedding_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/embedding_model/base.py -------------------------------------------------------------------------------- /src/comorag/embedding_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/embedding_store.py -------------------------------------------------------------------------------- /src/comorag/information_extraction/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/information_extraction/__init__.py -------------------------------------------------------------------------------- /src/comorag/information_extraction/openie_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/information_extraction/openie_openai.py -------------------------------------------------------------------------------- /src/comorag/information_extraction/openie_vllm_offline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/information_extraction/openie_vllm_offline.py -------------------------------------------------------------------------------- /src/comorag/llm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/llm/__init__.py -------------------------------------------------------------------------------- /src/comorag/llm/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/llm/base.py -------------------------------------------------------------------------------- /src/comorag/llm/openai_gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/llm/openai_gpt.py -------------------------------------------------------------------------------- /src/comorag/llm/vllm_offline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/llm/vllm_offline.py -------------------------------------------------------------------------------- /src/comorag/prompts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/__init__.py -------------------------------------------------------------------------------- /src/comorag/prompts/dspy_prompts/filter_llama3.3-70B-Instruct.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/dspy_prompts/filter_llama3.3-70B-Instruct.json -------------------------------------------------------------------------------- /src/comorag/prompts/filter_default_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/filter_default_prompt.py -------------------------------------------------------------------------------- /src/comorag/prompts/linking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/linking.py -------------------------------------------------------------------------------- /src/comorag/prompts/prompt_template_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/prompt_template_manager.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/README.md -------------------------------------------------------------------------------- /src/comorag/prompts/templates/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/comorag/prompts/templates/agent_probe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/agent_probe.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/memory_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/memory_fusion.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/ner.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/ner_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/ner_query.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/node_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/node_fusion.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/rag_qa_mc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/rag_qa_mc.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/rag_qa_mc_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/rag_qa_mc_memory.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/rag_qa_narrativeqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/rag_qa_narrativeqa.py -------------------------------------------------------------------------------- /src/comorag/prompts/templates/triple_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/prompts/templates/triple_extraction.py -------------------------------------------------------------------------------- /src/comorag/rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/rerank.py -------------------------------------------------------------------------------- /src/comorag/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/comorag/utils/agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/agents.py -------------------------------------------------------------------------------- /src/comorag/utils/cluster_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/cluster_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/config_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/embed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/embed_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/llm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/llm_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/logging_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/memory_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/memory_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/misc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/misc_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/qa_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/qa_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/summarization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/summarization_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/timeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/timeline_utils.py -------------------------------------------------------------------------------- /src/comorag/utils/typing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EternityJune25/ComoRAG/HEAD/src/comorag/utils/typing_utils.py --------------------------------------------------------------------------------