AutoRAG 0.0.4__tar.gz → 0.0.5__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (257) hide show
  1. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/AutoRAG.egg-info/PKG-INFO +72 -79
  2. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/AutoRAG.egg-info/SOURCES.txt +61 -2
  3. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/AutoRAG.egg-info/requires.txt +1 -0
  4. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/PKG-INFO +72 -79
  5. AutoRAG-0.0.5/README.md +233 -0
  6. AutoRAG-0.0.5/autorag/VERSION +1 -0
  7. AutoRAG-0.0.5/autorag/data/qacreation/simple.py +89 -0
  8. AutoRAG-0.0.5/autorag/data/utils/llamaindex.py +64 -0
  9. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/tart/tart.py +6 -2
  10. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/upr.py +3 -0
  11. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/vectordb.py +6 -4
  12. AutoRAG-0.0.5/docs/source/_static/data_creation.png +0 -0
  13. AutoRAG-0.0.5/docs/source/_static/data_folder.png +0 -0
  14. AutoRAG-0.0.5/docs/source/_static/node_folder.png +0 -0
  15. AutoRAG-0.0.5/docs/source/_static/node_line_folder.png +0 -0
  16. AutoRAG-0.0.5/docs/source/_static/node_line_summary.png +0 -0
  17. AutoRAG-0.0.5/docs/source/_static/node_lines.png +0 -0
  18. AutoRAG-0.0.5/docs/source/_static/node_summary.png +0 -0
  19. AutoRAG-0.0.5/docs/source/_static/project_folder_example.png +0 -0
  20. AutoRAG-0.0.5/docs/source/_static/project_folders.png +0 -0
  21. AutoRAG-0.0.5/docs/source/_static/resources_folder.png +0 -0
  22. AutoRAG-0.0.5/docs/source/_static/roadmap/RAG_paradigms.png +0 -0
  23. AutoRAG-0.0.5/docs/source/_static/roadmap/advanced_RAG.png +0 -0
  24. AutoRAG-0.0.5/docs/source/_static/roadmap/cycle.png +0 -0
  25. AutoRAG-0.0.5/docs/source/_static/roadmap/merger.png +0 -0
  26. AutoRAG-0.0.5/docs/source/_static/roadmap/node_line_modular.png +0 -0
  27. AutoRAG-0.0.5/docs/source/_static/roadmap/policy.png +0 -0
  28. AutoRAG-0.0.5/docs/source/_static/trial_folder.png +0 -0
  29. AutoRAG-0.0.5/docs/source/_static/trial_json.png +0 -0
  30. AutoRAG-0.0.5/docs/source/_static/trial_summary.png +0 -0
  31. AutoRAG-0.0.5/docs/source/data_creation/data_format.md +143 -0
  32. AutoRAG-0.0.5/docs/source/data_creation/tutorial.md +103 -0
  33. AutoRAG-0.0.5/docs/source/deploy/api_endpoint.md +46 -0
  34. AutoRAG-0.0.5/docs/source/index.rst +125 -0
  35. AutoRAG-0.0.5/docs/source/install.md +87 -0
  36. AutoRAG-0.0.5/docs/source/local_model.md +150 -0
  37. AutoRAG-0.0.5/docs/source/nodes/generator/generator.md +47 -0
  38. AutoRAG-0.0.5/docs/source/nodes/generator/llama_index_llm.md +23 -0
  39. AutoRAG-0.0.5/docs/source/nodes/index.md +14 -0
  40. AutoRAG-0.0.5/docs/source/nodes/passage_compressor/passage_compressor.md +48 -0
  41. AutoRAG-0.0.5/docs/source/nodes/passage_compressor/tree_summarize.md +25 -0
  42. AutoRAG-0.0.5/docs/source/nodes/passage_reranker/monot5.md +43 -0
  43. AutoRAG-0.0.5/docs/source/nodes/passage_reranker/passage_reranker.md +48 -0
  44. AutoRAG-0.0.5/docs/source/nodes/passage_reranker/tart.md +14 -0
  45. AutoRAG-0.0.5/docs/source/nodes/passage_reranker/upr.md +32 -0
  46. AutoRAG-0.0.5/docs/source/nodes/prompt_maker/fstring.md +16 -0
  47. AutoRAG-0.0.5/docs/source/nodes/prompt_maker/prompt_maker.md +53 -0
  48. AutoRAG-0.0.5/docs/source/nodes/query_expansion/hyde.md +22 -0
  49. AutoRAG-0.0.5/docs/source/nodes/query_expansion/query_decompose.md +22 -0
  50. AutoRAG-0.0.5/docs/source/nodes/query_expansion/query_expansion.md +68 -0
  51. AutoRAG-0.0.5/docs/source/nodes/retrieval/bm25.md +12 -0
  52. AutoRAG-0.0.5/docs/source/nodes/retrieval/hybrid_cc.md +35 -0
  53. AutoRAG-0.0.5/docs/source/nodes/retrieval/hybrid_rrf.md +31 -0
  54. AutoRAG-0.0.5/docs/source/nodes/retrieval/retrieval.md +59 -0
  55. AutoRAG-0.0.5/docs/source/nodes/retrieval/vectordb.md +26 -0
  56. AutoRAG-0.0.5/docs/source/optimization/custom_config.md +110 -0
  57. AutoRAG-0.0.5/docs/source/optimization/folder_structure.md +128 -0
  58. AutoRAG-0.0.5/docs/source/optimization/optimization.md +117 -0
  59. AutoRAG-0.0.5/docs/source/roadmap/modular_rag.md +180 -0
  60. AutoRAG-0.0.5/docs/source/structure.md +54 -0
  61. AutoRAG-0.0.5/docs/source/troubleshooting.md +96 -0
  62. AutoRAG-0.0.5/docs/source/tutorial.md +157 -0
  63. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/requirements.txt +1 -0
  64. AutoRAG-0.0.5/sample_config/compact_local.yaml +54 -0
  65. AutoRAG-0.0.5/sample_config/compact_openai.yaml +52 -0
  66. AutoRAG-0.0.5/sample_config/full.yaml +78 -0
  67. AutoRAG-0.0.5/sample_config/simple_local.yaml +27 -0
  68. AutoRAG-0.0.5/sample_config/simple_openai.yaml +26 -0
  69. AutoRAG-0.0.5/tests/autorag/data/qacreation/test_simple.py +54 -0
  70. AutoRAG-0.0.5/tests/conftest.py +0 -0
  71. AutoRAG-0.0.5/tests/resources/data_creation/raw_dir/sample1.txt +30 -0
  72. AutoRAG-0.0.5/tests/resources/data_creation/raw_dir/sample2.txt +27 -0
  73. AutoRAG-0.0.5/tests/resources/data_creation/raw_dir/sample3.txt +31 -0
  74. AutoRAG-0.0.5/tests/resources/full.yaml +78 -0
  75. AutoRAG-0.0.5/tests/resources/result_project/resources/chroma/6595d304-5270-4d9f-a267-c191366804ce/link_lists.bin +0 -0
  76. AutoRAG-0.0.4/README.md +0 -241
  77. AutoRAG-0.0.4/autorag/VERSION +0 -1
  78. AutoRAG-0.0.4/docs/source/data_creation/tutorial.md +0 -1
  79. AutoRAG-0.0.4/docs/source/index.rst +0 -45
  80. AutoRAG-0.0.4/docs/source/install.md +0 -1
  81. AutoRAG-0.0.4/docs/source/modules/index.md +0 -1
  82. AutoRAG-0.0.4/docs/source/optimization/optimization.md +0 -3
  83. AutoRAG-0.0.4/docs/source/tutorial.md +0 -1
  84. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/.github/dependabot.yml +0 -0
  85. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/.github/workflows/sphinx.yml +0 -0
  86. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/.github/workflows/test.yml +0 -0
  87. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/.gitignore +0 -0
  88. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/AutoRAG.egg-info/dependency_links.txt +0 -0
  89. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/AutoRAG.egg-info/entry_points.txt +0 -0
  90. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/AutoRAG.egg-info/top_level.txt +0 -0
  91. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/LICENSE +0 -0
  92. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/__init__.py +0 -0
  93. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/data/__init__.py +0 -0
  94. {AutoRAG-0.0.4/autorag/nodes → AutoRAG-0.0.5/autorag/data/qacreation}/__init__.py +0 -0
  95. {AutoRAG-0.0.4/autorag/nodes/passagereranker/tart → AutoRAG-0.0.5/autorag/data/utils}/__init__.py +0 -0
  96. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/deploy.py +0 -0
  97. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/__init__.py +0 -0
  98. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/generation.py +0 -0
  99. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/metric/__init__.py +0 -0
  100. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/metric/generation.py +0 -0
  101. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/metric/retrieval.py +0 -0
  102. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/metric/retrieval_contents.py +0 -0
  103. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/retrieval.py +0 -0
  104. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluate/retrieval_contents.py +0 -0
  105. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/evaluator.py +0 -0
  106. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/node_line.py +0 -0
  107. /AutoRAG-0.0.4/tests/conftest.py → /AutoRAG-0.0.5/autorag/nodes/__init__.py +0 -0
  108. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/generator/__init__.py +0 -0
  109. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/generator/base.py +0 -0
  110. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/generator/llama_index_llm.py +0 -0
  111. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/generator/run.py +0 -0
  112. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagecompressor/__init__.py +0 -0
  113. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagecompressor/base.py +0 -0
  114. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagecompressor/run.py +0 -0
  115. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagecompressor/tree_summarize.py +0 -0
  116. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/__init__.py +0 -0
  117. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/base.py +0 -0
  118. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/monot5.py +0 -0
  119. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/run.py +0 -0
  120. /AutoRAG-0.0.4/tests/resources/result_project/resources/chroma/6595d304-5270-4d9f-a267-c191366804ce/link_lists.bin → /AutoRAG-0.0.5/autorag/nodes/passagereranker/tart/__init__.py +0 -0
  121. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/tart/modeling_enc_t5.py +0 -0
  122. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/passagereranker/tart/tokenization_enc_t5.py +0 -0
  123. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/promptmaker/__init__.py +0 -0
  124. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/promptmaker/base.py +0 -0
  125. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/promptmaker/fstring.py +0 -0
  126. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/promptmaker/run.py +0 -0
  127. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/queryexpansion/__init__.py +0 -0
  128. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/queryexpansion/base.py +0 -0
  129. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/queryexpansion/hyde.py +0 -0
  130. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/queryexpansion/query_decompose.py +0 -0
  131. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/queryexpansion/run.py +0 -0
  132. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/__init__.py +0 -0
  133. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/base.py +0 -0
  134. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/bm25.py +0 -0
  135. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/hybrid_cc.py +0 -0
  136. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/hybrid_rrf.py +0 -0
  137. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/nodes/retrieval/run.py +0 -0
  138. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/schema/__init__.py +0 -0
  139. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/schema/module.py +0 -0
  140. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/schema/node.py +0 -0
  141. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/strategy.py +0 -0
  142. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/support.py +0 -0
  143. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/utils/__init__.py +0 -0
  144. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/utils/preprocess.py +0 -0
  145. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/autorag/utils/util.py +0 -0
  146. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/dev_requirements.txt +0 -0
  147. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/Makefile +0 -0
  148. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/make.bat +0 -0
  149. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/requirements.txt +0 -0
  150. {AutoRAG-0.0.4/docs/source/resources → AutoRAG-0.0.5/docs/source/_static}/samsung_sundae.jpeg +0 -0
  151. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.data.rst +0 -0
  152. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.evaluate.metric.rst +0 -0
  153. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.evaluate.rst +0 -0
  154. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.generator.rst +0 -0
  155. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.passagecompressor.rst +0 -0
  156. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.passagereranker.rst +0 -0
  157. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.passagereranker.tart.rst +0 -0
  158. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.promptmaker.rst +0 -0
  159. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.queryexpansion.rst +0 -0
  160. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.retrieval.rst +0 -0
  161. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.nodes.rst +0 -0
  162. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.rst +0 -0
  163. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.schema.rst +0 -0
  164. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/autorag.utils.rst +0 -0
  165. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/api_spec/modules.rst +0 -0
  166. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/docs/source/conf.py +0 -0
  167. /AutoRAG-0.0.4/tests/resources/full.yaml → /AutoRAG-0.0.5/docs/source/optimization/sample_full_config.yaml +0 -0
  168. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/pyproject.toml +0 -0
  169. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/sample_dataset/README.md +0 -0
  170. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/sample_dataset/eli5/load_eli5_dataset.py +0 -0
  171. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/sample_dataset/msmarco/load_msmarco_dataset.py +0 -0
  172. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/sample_dataset/triviaqa/load_triviaqa_dataset.py +0 -0
  173. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/setup.cfg +0 -0
  174. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/evaluate/metric/test_generation_metric.py +0 -0
  175. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/evaluate/metric/test_retrieval_contents_metric.py +0 -0
  176. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/evaluate/metric/test_retrieval_metric.py +0 -0
  177. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/evaluate/test_generation_evaluate.py +0 -0
  178. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/evaluate/test_retrieval_contents_evaluate.py +0 -0
  179. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/evaluate/test_retrieval_evaluate.py +0 -0
  180. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/generator/test_llama_index_llm.py +0 -0
  181. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/generator/test_run_generator_node.py +0 -0
  182. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagecompressor/test_run_passage_compressor_node.py +0 -0
  183. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagecompressor/test_tree_summarize.py +0 -0
  184. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagereranker/test_monot5.py +0 -0
  185. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagereranker/test_passage_reranker_base.py +0 -0
  186. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagereranker/test_passage_reranker_run.py +0 -0
  187. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagereranker/test_tart.py +0 -0
  188. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/passagereranker/test_upr.py +0 -0
  189. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/promptmaker/test_fstring.py +0 -0
  190. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/promptmaker/test_prompt_maker_run.py +0 -0
  191. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/queryexpansion/test_hyde.py +0 -0
  192. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/queryexpansion/test_query_decompose.py +0 -0
  193. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/queryexpansion/test_query_expansion_base.py +0 -0
  194. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/queryexpansion/test_query_expansion_run.py +0 -0
  195. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/retrieval/test_bm25.py +0 -0
  196. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/retrieval/test_hybrid_cc.py +0 -0
  197. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/retrieval/test_hybrid_rrf.py +0 -0
  198. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/retrieval/test_retrieval_base.py +0 -0
  199. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/retrieval/test_run_retrieval_node.py +0 -0
  200. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/nodes/retrieval/test_vectordb.py +0 -0
  201. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/schema/test_module_schema.py +0 -0
  202. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/schema/test_node_schema.py +0 -0
  203. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/test_deploy.py +0 -0
  204. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/test_evaluator.py +0 -0
  205. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/test_strategy.py +0 -0
  206. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/test_support.py +0 -0
  207. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/utils/test_preprocess.py +0 -0
  208. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/autorag/utils/test_util.py +0 -0
  209. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/corpus_data_sample.parquet +0 -0
  210. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/qa_data_sample.parquet +0 -0
  211. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/qa_test_data_sample.parquet +0 -0
  212. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/config.yaml +0 -0
  213. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/0.parquet +0 -0
  214. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/1.parquet +0 -0
  215. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/2.parquet +0 -0
  216. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/3.parquet +0 -0
  217. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/4.parquet +0 -0
  218. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/5.parquet +0 -0
  219. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/best_5.parquet +0 -0
  220. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/generator/summary.csv +0 -0
  221. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/prompt_maker/0.parquet +0 -0
  222. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/prompt_maker/1.parquet +0 -0
  223. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/prompt_maker/best_0.parquet +0 -0
  224. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/prompt_maker/summary.csv +0 -0
  225. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/post_retrieve_node_line/summary.csv +0 -0
  226. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/pre_retrieve_node_line/query_expansion/0.parquet +0 -0
  227. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/pre_retrieve_node_line/query_expansion/1.parquet +0 -0
  228. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/pre_retrieve_node_line/query_expansion/2.parquet +0 -0
  229. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/pre_retrieve_node_line/query_expansion/best_0.parquet +0 -0
  230. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/pre_retrieve_node_line/query_expansion/summary.csv +0 -0
  231. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/pre_retrieve_node_line/summary.csv +0 -0
  232. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_compressor/0.parquet +0 -0
  233. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_compressor/best_0.parquet +0 -0
  234. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_compressor/summary.csv +0 -0
  235. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_reranker/0.parquet +0 -0
  236. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_reranker/1.parquet +0 -0
  237. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_reranker/best_0.parquet +0 -0
  238. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/passage_reranker/summary.csv +0 -0
  239. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/retrieval/0.parquet +0 -0
  240. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/retrieval/1.parquet +0 -0
  241. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/retrieval/best_0.parquet +0 -0
  242. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/retrieval/summary.csv +0 -0
  243. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/retrieve_node_line/summary.csv +0 -0
  244. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/0/summary.csv +0 -0
  245. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/data/corpus.parquet +0 -0
  246. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/data/qa.parquet +0 -0
  247. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/resources/bm25.pkl +0 -0
  248. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/resources/chroma/6595d304-5270-4d9f-a267-c191366804ce/data_level0.bin +0 -0
  249. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/resources/chroma/6595d304-5270-4d9f-a267-c191366804ce/header.bin +0 -0
  250. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/resources/chroma/6595d304-5270-4d9f-a267-c191366804ce/length.bin +0 -0
  251. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/resources/chroma/chroma.sqlite3 +0 -0
  252. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/result_project/trial.json +0 -0
  253. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/sample_project/data/corpus.parquet +0 -0
  254. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/sample_project/data/qa.parquet +0 -0
  255. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/sample_project/resources/bm25.pkl +0 -0
  256. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/simple.yaml +0 -0
  257. {AutoRAG-0.0.4 → AutoRAG-0.0.5}/tests/resources/test_bm25_retrieval.pkl +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: AutoRAG
3
- Version: 0.0.4
3
+ Version: 0.0.5
4
4
  Summary: Automatically Evaluate RAG pipelines with your own data. Find optimal structure for new RAG product.
5
5
  Author-email: Marker-Inc <vkehfdl1@gmail.com>
6
6
  License: Apache License
@@ -241,64 +241,70 @@ Requires-Dist: fastapi
241
241
  Requires-Dist: uvicorn
242
242
  Requires-Dist: torch
243
243
  Requires-Dist: sentencepiece
244
+ Requires-Dist: guidance
244
245
 
245
246
  # AutoRAG
246
247
 
247
- RAG AutoML tool for automatically find optimal RAG pipeline for your date.
248
+ RAG AutoML tool for automatically finds an optimal RAG pipeline for your data.
248
249
 
249
- [![Release Notes](https://img.shields.io/github/release/Marker-Inc-Korea/AutoRAG)](https://github.com/Marker-Inc-Korea/AutoRAG/releases)
250
- [![Downloads](https://static.pepy.tech/badge/AutoRAG)](https://pepy.tech/project/AutoRAG)
250
+ Explore our 📖 [Document](https://marker-inc-korea.github.io/AutoRAG/)!!
251
251
 
252
- [Document](https://marker-inc-korea.github.io/AutoRAG/)
252
+ Plus, join our 📞 [Discord](https://discord.gg/P4DYXfmSAs) Community.
253
+ # 📑 Index
254
+
255
+ - [Introduction](#introduction)
256
+ - [Quick Install](#-quick-install)
257
+ - [Index](#-index)
258
+ - [Strengths](#-strengths)
259
+ - [QuickStart](#-quickstart)
260
+ - [1. Prepare your evaluation data](#1-prepare-your-evaluation-data)
261
+ - [2. Evaluate your data to various RAG modules](#2-evaluate-your-data-to-various-rag-modules)
262
+ - [3. Use a found optimal RAG pipeline](#3-use-a-found-optimal-rag-pipeline)
263
+ - [4. Share your RAG pipeline](#4-share-your-rag-pipeline)
264
+ - [+ Config yaml file](#-create-your-own-config-yaml-file)
265
+ - [Supporting RAG modules](#supporting-nodes--modules)
266
+ - [Roadmap](#roadmap)
267
+ - [Contribution](#contribution)
253
268
 
254
269
  # Introduction
255
270
 
256
- There are numerous RAG pipelines and modules out there,
257
- but you don’t know what pipeline is great for “your own data” and "your own use-case."
258
- Making and evaluating all RAG modules is very time-consuming and hard to do.
259
- But without it, you will never know which RAG pipeline is the best for your own use-case.
271
+ There are numerous RAG pipelines and modules out there,
272
+ but you don’t know what pipeline is great for “your own data” and "your own use-case."
273
+ Making and evaluating all RAG modules is very time-consuming and hard to do.
274
+ But without it, you will never know which RAG pipeline is the best for your own use-case.
260
275
 
261
- AutoRAG is a tool for finding optimal RAG pipeline for “your data.”
262
- You can evaluate various RAG modules automatically with your own evaluation data,
263
- and find the best RAG pipeline for your own use-case.
276
+ AutoRAG is a tool for finding optimal RAG pipeline for “your data.”
277
+ You can evaluate various RAG modules automatically with your own evaluation data,
278
+ and find the best RAG pipeline for your own use-case.
264
279
 
265
- AutoRAG supports a simple way to evaluate numerous RAG module combinations.
266
- Try now and find the best RAG pipeline for your own use-case.
280
+ AutoRAG supports a simple way to evaluate numerous RAG module combinations.
281
+ Try now and find the best RAG pipeline for your own use-case.
267
282
 
268
- # Quick Install
283
+ # ⚡ Quick Install
269
284
 
270
285
  ```bash
271
286
  pip install AutoRAG
272
287
  ```
273
288
 
274
- # Index
275
289
 
276
- - [Introduction](#introduction)
277
- - [Quick Install](#quick-install)
278
- - [Index](#index)
279
- - [Strengths](#strengths)
280
- - [QuickStart](#quickstart)
281
- - [Prepare your evaluation data](#prepare-your-evaluation-data)
282
- - [Evaluate your data to various RAG modules](#evaluate-your-data-to-various-rag-modules)
283
- - [Use a found optimal RAG pipeline](#use-a-found-optimal-rag-pipeline)
284
- - [Share your RAG pipeline](#share-your-rag-pipeline)
285
- - [Config yaml file](#config-yaml-file)
286
- - [Supporting RAG modules](#supporting-rag-modules)
287
- - [Roadmap](#roadmap)
288
- - [Contribution](#contribution)
289
-
290
- # Strengths
290
+ # 💪 Strengths
291
291
 
292
- - Find your RAG baseline: Benchmark RAG pipelines with few lines of code. You can quickly get a high-performance RAG
292
+ ### **1. Find your RAG baseline**
293
+ Benchmark RAG pipelines with few lines of code. You can quickly get a high-performance RAG
293
294
  pipeline just for your data. Don’t waste time dealing with complex RAG modules and academic paper. Focus on your data.
294
- - Analyze where is wrong: Sometimes it is hard to keep tracking where is the major problem within your RAG pipeline.
295
+ ### **2. Analyze where is wrong**
296
+ Sometimes it is hard to keep tracking where is the major problem within your RAG pipeline.
295
297
  AutoRAG gives you the data of it, so you can analyze and focus where is the major problem and where you to focus on.
296
- - Quick Starter Pack for your new RAG product: Get the most effective RAG workflow among many pipelines, and start from
298
+ ### **3. Quick Starter Pack for your new RAG product**
299
+ Get the most effective RAG workflow among many pipelines, and start from
297
300
  there. Don’t start at toy-project level, start from advanced level.
298
- - Share your experiment to others: It's really easy to share your experiment to others. Share your config yaml file and
301
+ ### **4. Share your experiment to others**
302
+ It's really easy to share your experiment to others. Share your config yaml file and
299
303
  summary csv files. Plus, check out others result and adapt to your use-case.
300
304
 
301
- # QuickStart
305
+ # ⚡ QuickStart
306
+
307
+ ### 1. Prepare your evaluation data
302
308
 
303
309
  For evaluation, you need to prepare just three files.
304
310
 
@@ -306,18 +312,19 @@ For evaluation, you need to prepare just three files.
306
312
  - Corpus dataset file (corpus.parquet)
307
313
  - Config yaml file (config.yaml)
308
314
 
309
- ### Prepare your evaluation data
310
315
 
311
316
  There is a template for your evaluation data for using AutoRAG.
312
- Check out the evaluation data rule at [here]().
313
- Plus, you can get example datasets for testing AutoRAG at [here](./sample_dataset).
314
317
 
315
- ### Evaluate your data to various RAG modules
318
+ - Check out how to make evaluation data at [here](https://marker-inc-korea.github.io/AutoRAG/data_creation/tutorial.html).
319
+ - Check out the evaluation data rule at [here](https://marker-inc-korea.github.io/AutoRAG/data_creation/data_format.html).
320
+ - Plus, you can get example datasets for testing AutoRAG at [here](./sample_dataset).
316
321
 
317
- You can get various config yaml files at [here]().
322
+ ### 2. Evaluate your data to various RAG modules
323
+
324
+ You can get various config yaml files at [here](./sample_config).
318
325
  We highly recommend using pre-made config yaml files for starter.
319
326
 
320
- If you want to make your own config yaml files, check out the [Config yaml file](#config-yaml-file) section.
327
+ If you want to make your own config yaml files, check out the [Config yaml file](#-create-your-own-config-yaml-file) section.
321
328
 
322
329
  You can evaluate your RAG pipeline with just a few lines of code.
323
330
 
@@ -338,9 +345,9 @@ Once it is done, you can see several files and folders created at your current d
338
345
  At the trial folder named to numbers (like 0),
339
346
  you can check `summary.csv` file that summarizes the evaluation results and the best RAG pipeline for your data.
340
347
 
341
- For more details, you can check out how the folder structure looks like at [here]().
348
+ For more details, you can check out how the folder structure looks like at [here](https://marker-inc-korea.github.io/AutoRAG/optimization/folder_structure.html).
342
349
 
343
- ### Use a found optimal RAG pipeline
350
+ ### 3. Use a found optimal RAG pipeline
344
351
 
345
352
  You can use a found optimal RAG pipeline right away.
346
353
  It needs just a few lines of code, and you are ready to use!
@@ -358,7 +365,7 @@ runner.run('your question')
358
365
 
359
366
  Or, you can run this pipeline as api server.
360
367
  You can use python code or CLI command.
361
- Check out API endpoint at [here]().
368
+ Check out API endpoint at [here](https://marker-inc-korea.github.io/AutoRAG/deploy/api_endpoint.html).
362
369
 
363
370
  ```python
364
371
  from autorag.deploy import Runner
@@ -373,7 +380,7 @@ You can run api server with CLI command.
373
380
  autorag run_api --config_path your/path/to/pipeline.yaml --host 0.0.0.0 --port 8000
374
381
  ```
375
382
 
376
- ### Share your RAG pipeline
383
+ ### 4. Share your RAG pipeline
377
384
 
378
385
  You can use your RAG pipeline from extracted pipeline yaml file.
379
386
  This extracted pipeline is great for sharing your RAG pipeline to others.
@@ -386,12 +393,13 @@ from autorag.deploy import extract_best_config
386
393
 
387
394
  pipeline_dict = extract_best_config(trial_path='your/path/to/trial_folder', output_path='your/path/to/pipeline.yaml')
388
395
  ```
389
-
390
- ### Config yaml file
396
+ ### ➕ Create your own Config yaml file
391
397
 
392
398
  You can build your own evaluation process with config yaml file.
393
- You can check detailed explaination how to configure each module and node at [here]().
399
+ You can check detailed explanation how to configure each module and node at [here](https://marker-inc-korea.github.io/AutoRAG/nodes/index.html#).
400
+
394
401
  There is a simple yaml file example.
402
+
395
403
  It evaluates two retrieval modules which are BM25 and Vector Retriever, and three reranking modules.
396
404
  Lastly, it generates prompt and makes generation with two other LLM models and three temperatures.
397
405
 
@@ -434,39 +442,22 @@ node_lines:
434
442
 
435
443
  ```
436
444
 
437
- # Supporting RAG modules
438
-
439
- ### Query Expansion
440
-
441
- - Query Decompose
442
- - HyDE
443
-
444
- ### Retrieval
445
-
446
- - BM25
447
- - Vector Retriever (choose your own embedding model)
448
- - Hybrid with rrf (reciprocal rank fusion)
449
- - Hybrid with cc (convex combination)
450
445
 
451
- ### Reranker
446
+ # ❗Supporting Nodes & modules
452
447
 
453
- - UPR
454
- - TART
455
- - MonoT5
456
448
 
457
- ### Passage Compressor
449
+ | Nodes | Modules |
450
+ |:-----------------------------------------------------------------------------------------------------------------:|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
451
+ | [Query_Expansion](https://marker-inc-korea.github.io/AutoRAG/nodes/query_expansion/query_expansion.html) | [Query_Decompose](https://marker-inc-korea.github.io/AutoRAG/nodes/query_expansion/query_decompose.html)<br/>[HyDE](https://marker-inc-korea.github.io/AutoRAG/nodes/query_expansion/hyde.html) |
452
+ | [Retrieval](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/retrieval.html) | [BM25](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/bm25.html)<br/>[VectorDB (choose embedding model)](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/vectordb.html)<br/>[Hybrid with rrf (reciprocal rank fusion)](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/hybrid_rrf.html)<br/>[Hybrid with cc (convex combination)](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/hybrid_cc.html) |
453
+ | [Passage_Reranker](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/passage_reranker.html) | [UPR](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/upr.html)<br/>[Tart](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/tart.html)<br/>[MonoT5](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/monot5.html) |
454
+ | [Passage_Compressor](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_compressor/passage_compressor.html) | [Tree Summarize](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_compressor/tree_summarize.html) |
455
+ | [Prompt Maker](https://marker-inc-korea.github.io/AutoRAG/nodes/prompt_maker/prompt_maker.html) | [Default Prompt Maker (f-string)](https://marker-inc-korea.github.io/AutoRAG/nodes/prompt_maker/fstring.html) |
456
+ | [Generator](https://marker-inc-korea.github.io/AutoRAG/nodes/generator/generator.html) | [llama_index llm](https://marker-inc-korea.github.io/AutoRAG/nodes/generator/llama_index_llm.html) |
458
457
 
459
- - Tree Summarize
460
458
 
461
- ### Prompt Maker
462
459
 
463
- - Default Prompt Maker (f-string)
464
-
465
- ### Generator
466
-
467
- - Llama Index LLM (choose your own model)
468
-
469
- # Roadmap
460
+ # 🛣Roadmap
470
461
 
471
462
  - [ ] Policy Module for modular RAG pipeline
472
463
  - [ ] Visualize evaluation result
@@ -480,6 +471,8 @@ node_lines:
480
471
 
481
472
  # Contribution
482
473
 
483
- We are developing AutoRAG as open-source. Feel free to contribute to this project.
484
- If you want to contribute to this project, please check out the [Contribution Guide](CONTRIBUTING.md).
485
- Plus, check out our detailed documentation at [here]().
474
+ We are developing AutoRAG as open-source.
475
+
476
+ So this project welcomes contributions and suggestions. Feel free to contribute to this project.
477
+
478
+ Plus, check out our detailed documentation at [here](https://marker-inc-korea.github.io/AutoRAG/index.html).
@@ -21,6 +21,10 @@ autorag/node_line.py
21
21
  autorag/strategy.py
22
22
  autorag/support.py
23
23
  autorag/data/__init__.py
24
+ autorag/data/qacreation/__init__.py
25
+ autorag/data/qacreation/simple.py
26
+ autorag/data/utils/__init__.py
27
+ autorag/data/utils/llamaindex.py
24
28
  autorag/evaluate/__init__.py
25
29
  autorag/evaluate/generation.py
26
30
  autorag/evaluate/retrieval.py
@@ -75,7 +79,30 @@ docs/requirements.txt
75
79
  docs/source/conf.py
76
80
  docs/source/index.rst
77
81
  docs/source/install.md
82
+ docs/source/local_model.md
83
+ docs/source/structure.md
84
+ docs/source/troubleshooting.md
78
85
  docs/source/tutorial.md
86
+ docs/source/_static/data_creation.png
87
+ docs/source/_static/data_folder.png
88
+ docs/source/_static/node_folder.png
89
+ docs/source/_static/node_line_folder.png
90
+ docs/source/_static/node_line_summary.png
91
+ docs/source/_static/node_lines.png
92
+ docs/source/_static/node_summary.png
93
+ docs/source/_static/project_folder_example.png
94
+ docs/source/_static/project_folders.png
95
+ docs/source/_static/resources_folder.png
96
+ docs/source/_static/samsung_sundae.jpeg
97
+ docs/source/_static/trial_folder.png
98
+ docs/source/_static/trial_json.png
99
+ docs/source/_static/trial_summary.png
100
+ docs/source/_static/roadmap/RAG_paradigms.png
101
+ docs/source/_static/roadmap/advanced_RAG.png
102
+ docs/source/_static/roadmap/cycle.png
103
+ docs/source/_static/roadmap/merger.png
104
+ docs/source/_static/roadmap/node_line_modular.png
105
+ docs/source/_static/roadmap/policy.png
79
106
  docs/source/api_spec/autorag.data.rst
80
107
  docs/source/api_spec/autorag.evaluate.metric.rst
81
108
  docs/source/api_spec/autorag.evaluate.rst
@@ -91,10 +118,38 @@ docs/source/api_spec/autorag.rst
91
118
  docs/source/api_spec/autorag.schema.rst
92
119
  docs/source/api_spec/autorag.utils.rst
93
120
  docs/source/api_spec/modules.rst
121
+ docs/source/data_creation/data_format.md
94
122
  docs/source/data_creation/tutorial.md
95
- docs/source/modules/index.md
123
+ docs/source/deploy/api_endpoint.md
124
+ docs/source/nodes/index.md
125
+ docs/source/nodes/generator/generator.md
126
+ docs/source/nodes/generator/llama_index_llm.md
127
+ docs/source/nodes/passage_compressor/passage_compressor.md
128
+ docs/source/nodes/passage_compressor/tree_summarize.md
129
+ docs/source/nodes/passage_reranker/monot5.md
130
+ docs/source/nodes/passage_reranker/passage_reranker.md
131
+ docs/source/nodes/passage_reranker/tart.md
132
+ docs/source/nodes/passage_reranker/upr.md
133
+ docs/source/nodes/prompt_maker/fstring.md
134
+ docs/source/nodes/prompt_maker/prompt_maker.md
135
+ docs/source/nodes/query_expansion/hyde.md
136
+ docs/source/nodes/query_expansion/query_decompose.md
137
+ docs/source/nodes/query_expansion/query_expansion.md
138
+ docs/source/nodes/retrieval/bm25.md
139
+ docs/source/nodes/retrieval/hybrid_cc.md
140
+ docs/source/nodes/retrieval/hybrid_rrf.md
141
+ docs/source/nodes/retrieval/retrieval.md
142
+ docs/source/nodes/retrieval/vectordb.md
143
+ docs/source/optimization/custom_config.md
144
+ docs/source/optimization/folder_structure.md
96
145
  docs/source/optimization/optimization.md
97
- docs/source/resources/samsung_sundae.jpeg
146
+ docs/source/optimization/sample_full_config.yaml
147
+ docs/source/roadmap/modular_rag.md
148
+ sample_config/compact_local.yaml
149
+ sample_config/compact_openai.yaml
150
+ sample_config/full.yaml
151
+ sample_config/simple_local.yaml
152
+ sample_config/simple_openai.yaml
98
153
  sample_dataset/README.md
99
154
  sample_dataset/eli5/load_eli5_dataset.py
100
155
  sample_dataset/msmarco/load_msmarco_dataset.py
@@ -104,6 +159,7 @@ tests/autorag/test_deploy.py
104
159
  tests/autorag/test_evaluator.py
105
160
  tests/autorag/test_strategy.py
106
161
  tests/autorag/test_support.py
162
+ tests/autorag/data/qacreation/test_simple.py
107
163
  tests/autorag/evaluate/test_generation_evaluate.py
108
164
  tests/autorag/evaluate/test_retrieval_contents_evaluate.py
109
165
  tests/autorag/evaluate/test_retrieval_evaluate.py
@@ -141,6 +197,9 @@ tests/resources/qa_data_sample.parquet
141
197
  tests/resources/qa_test_data_sample.parquet
142
198
  tests/resources/simple.yaml
143
199
  tests/resources/test_bm25_retrieval.pkl
200
+ tests/resources/data_creation/raw_dir/sample1.txt
201
+ tests/resources/data_creation/raw_dir/sample2.txt
202
+ tests/resources/data_creation/raw_dir/sample3.txt
144
203
  tests/resources/result_project/trial.json
145
204
  tests/resources/result_project/0/config.yaml
146
205
  tests/resources/result_project/0/summary.csv
@@ -20,3 +20,4 @@ fastapi
20
20
  uvicorn
21
21
  torch
22
22
  sentencepiece
23
+ guidance
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: AutoRAG
3
- Version: 0.0.4
3
+ Version: 0.0.5
4
4
  Summary: Automatically Evaluate RAG pipelines with your own data. Find optimal structure for new RAG product.
5
5
  Author-email: Marker-Inc <vkehfdl1@gmail.com>
6
6
  License: Apache License
@@ -241,64 +241,70 @@ Requires-Dist: fastapi
241
241
  Requires-Dist: uvicorn
242
242
  Requires-Dist: torch
243
243
  Requires-Dist: sentencepiece
244
+ Requires-Dist: guidance
244
245
 
245
246
  # AutoRAG
246
247
 
247
- RAG AutoML tool for automatically find optimal RAG pipeline for your date.
248
+ RAG AutoML tool for automatically finds an optimal RAG pipeline for your data.
248
249
 
249
- [![Release Notes](https://img.shields.io/github/release/Marker-Inc-Korea/AutoRAG)](https://github.com/Marker-Inc-Korea/AutoRAG/releases)
250
- [![Downloads](https://static.pepy.tech/badge/AutoRAG)](https://pepy.tech/project/AutoRAG)
250
+ Explore our 📖 [Document](https://marker-inc-korea.github.io/AutoRAG/)!!
251
251
 
252
- [Document](https://marker-inc-korea.github.io/AutoRAG/)
252
+ Plus, join our 📞 [Discord](https://discord.gg/P4DYXfmSAs) Community.
253
+ # 📑 Index
254
+
255
+ - [Introduction](#introduction)
256
+ - [Quick Install](#-quick-install)
257
+ - [Index](#-index)
258
+ - [Strengths](#-strengths)
259
+ - [QuickStart](#-quickstart)
260
+ - [1. Prepare your evaluation data](#1-prepare-your-evaluation-data)
261
+ - [2. Evaluate your data to various RAG modules](#2-evaluate-your-data-to-various-rag-modules)
262
+ - [3. Use a found optimal RAG pipeline](#3-use-a-found-optimal-rag-pipeline)
263
+ - [4. Share your RAG pipeline](#4-share-your-rag-pipeline)
264
+ - [+ Config yaml file](#-create-your-own-config-yaml-file)
265
+ - [Supporting RAG modules](#supporting-nodes--modules)
266
+ - [Roadmap](#roadmap)
267
+ - [Contribution](#contribution)
253
268
 
254
269
  # Introduction
255
270
 
256
- There are numerous RAG pipelines and modules out there,
257
- but you don’t know what pipeline is great for “your own data” and "your own use-case."
258
- Making and evaluating all RAG modules is very time-consuming and hard to do.
259
- But without it, you will never know which RAG pipeline is the best for your own use-case.
271
+ There are numerous RAG pipelines and modules out there,
272
+ but you don’t know what pipeline is great for “your own data” and "your own use-case."
273
+ Making and evaluating all RAG modules is very time-consuming and hard to do.
274
+ But without it, you will never know which RAG pipeline is the best for your own use-case.
260
275
 
261
- AutoRAG is a tool for finding optimal RAG pipeline for “your data.”
262
- You can evaluate various RAG modules automatically with your own evaluation data,
263
- and find the best RAG pipeline for your own use-case.
276
+ AutoRAG is a tool for finding optimal RAG pipeline for “your data.”
277
+ You can evaluate various RAG modules automatically with your own evaluation data,
278
+ and find the best RAG pipeline for your own use-case.
264
279
 
265
- AutoRAG supports a simple way to evaluate numerous RAG module combinations.
266
- Try now and find the best RAG pipeline for your own use-case.
280
+ AutoRAG supports a simple way to evaluate numerous RAG module combinations.
281
+ Try now and find the best RAG pipeline for your own use-case.
267
282
 
268
- # Quick Install
283
+ # ⚡ Quick Install
269
284
 
270
285
  ```bash
271
286
  pip install AutoRAG
272
287
  ```
273
288
 
274
- # Index
275
289
 
276
- - [Introduction](#introduction)
277
- - [Quick Install](#quick-install)
278
- - [Index](#index)
279
- - [Strengths](#strengths)
280
- - [QuickStart](#quickstart)
281
- - [Prepare your evaluation data](#prepare-your-evaluation-data)
282
- - [Evaluate your data to various RAG modules](#evaluate-your-data-to-various-rag-modules)
283
- - [Use a found optimal RAG pipeline](#use-a-found-optimal-rag-pipeline)
284
- - [Share your RAG pipeline](#share-your-rag-pipeline)
285
- - [Config yaml file](#config-yaml-file)
286
- - [Supporting RAG modules](#supporting-rag-modules)
287
- - [Roadmap](#roadmap)
288
- - [Contribution](#contribution)
289
-
290
- # Strengths
290
+ # 💪 Strengths
291
291
 
292
- - Find your RAG baseline: Benchmark RAG pipelines with few lines of code. You can quickly get a high-performance RAG
292
+ ### **1. Find your RAG baseline**
293
+ Benchmark RAG pipelines with few lines of code. You can quickly get a high-performance RAG
293
294
  pipeline just for your data. Don’t waste time dealing with complex RAG modules and academic paper. Focus on your data.
294
- - Analyze where is wrong: Sometimes it is hard to keep tracking where is the major problem within your RAG pipeline.
295
+ ### **2. Analyze where is wrong**
296
+ Sometimes it is hard to keep tracking where is the major problem within your RAG pipeline.
295
297
  AutoRAG gives you the data of it, so you can analyze and focus where is the major problem and where you to focus on.
296
- - Quick Starter Pack for your new RAG product: Get the most effective RAG workflow among many pipelines, and start from
298
+ ### **3. Quick Starter Pack for your new RAG product**
299
+ Get the most effective RAG workflow among many pipelines, and start from
297
300
  there. Don’t start at toy-project level, start from advanced level.
298
- - Share your experiment to others: It's really easy to share your experiment to others. Share your config yaml file and
301
+ ### **4. Share your experiment to others**
302
+ It's really easy to share your experiment to others. Share your config yaml file and
299
303
  summary csv files. Plus, check out others result and adapt to your use-case.
300
304
 
301
- # QuickStart
305
+ # ⚡ QuickStart
306
+
307
+ ### 1. Prepare your evaluation data
302
308
 
303
309
  For evaluation, you need to prepare just three files.
304
310
 
@@ -306,18 +312,19 @@ For evaluation, you need to prepare just three files.
306
312
  - Corpus dataset file (corpus.parquet)
307
313
  - Config yaml file (config.yaml)
308
314
 
309
- ### Prepare your evaluation data
310
315
 
311
316
  There is a template for your evaluation data for using AutoRAG.
312
- Check out the evaluation data rule at [here]().
313
- Plus, you can get example datasets for testing AutoRAG at [here](./sample_dataset).
314
317
 
315
- ### Evaluate your data to various RAG modules
318
+ - Check out how to make evaluation data at [here](https://marker-inc-korea.github.io/AutoRAG/data_creation/tutorial.html).
319
+ - Check out the evaluation data rule at [here](https://marker-inc-korea.github.io/AutoRAG/data_creation/data_format.html).
320
+ - Plus, you can get example datasets for testing AutoRAG at [here](./sample_dataset).
316
321
 
317
- You can get various config yaml files at [here]().
322
+ ### 2. Evaluate your data to various RAG modules
323
+
324
+ You can get various config yaml files at [here](./sample_config).
318
325
  We highly recommend using pre-made config yaml files for starter.
319
326
 
320
- If you want to make your own config yaml files, check out the [Config yaml file](#config-yaml-file) section.
327
+ If you want to make your own config yaml files, check out the [Config yaml file](#-create-your-own-config-yaml-file) section.
321
328
 
322
329
  You can evaluate your RAG pipeline with just a few lines of code.
323
330
 
@@ -338,9 +345,9 @@ Once it is done, you can see several files and folders created at your current d
338
345
  At the trial folder named to numbers (like 0),
339
346
  you can check `summary.csv` file that summarizes the evaluation results and the best RAG pipeline for your data.
340
347
 
341
- For more details, you can check out how the folder structure looks like at [here]().
348
+ For more details, you can check out how the folder structure looks like at [here](https://marker-inc-korea.github.io/AutoRAG/optimization/folder_structure.html).
342
349
 
343
- ### Use a found optimal RAG pipeline
350
+ ### 3. Use a found optimal RAG pipeline
344
351
 
345
352
  You can use a found optimal RAG pipeline right away.
346
353
  It needs just a few lines of code, and you are ready to use!
@@ -358,7 +365,7 @@ runner.run('your question')
358
365
 
359
366
  Or, you can run this pipeline as api server.
360
367
  You can use python code or CLI command.
361
- Check out API endpoint at [here]().
368
+ Check out API endpoint at [here](https://marker-inc-korea.github.io/AutoRAG/deploy/api_endpoint.html).
362
369
 
363
370
  ```python
364
371
  from autorag.deploy import Runner
@@ -373,7 +380,7 @@ You can run api server with CLI command.
373
380
  autorag run_api --config_path your/path/to/pipeline.yaml --host 0.0.0.0 --port 8000
374
381
  ```
375
382
 
376
- ### Share your RAG pipeline
383
+ ### 4. Share your RAG pipeline
377
384
 
378
385
  You can use your RAG pipeline from extracted pipeline yaml file.
379
386
  This extracted pipeline is great for sharing your RAG pipeline to others.
@@ -386,12 +393,13 @@ from autorag.deploy import extract_best_config
386
393
 
387
394
  pipeline_dict = extract_best_config(trial_path='your/path/to/trial_folder', output_path='your/path/to/pipeline.yaml')
388
395
  ```
389
-
390
- ### Config yaml file
396
+ ### ➕ Create your own Config yaml file
391
397
 
392
398
  You can build your own evaluation process with config yaml file.
393
- You can check detailed explaination how to configure each module and node at [here]().
399
+ You can check detailed explanation how to configure each module and node at [here](https://marker-inc-korea.github.io/AutoRAG/nodes/index.html#).
400
+
394
401
  There is a simple yaml file example.
402
+
395
403
  It evaluates two retrieval modules which are BM25 and Vector Retriever, and three reranking modules.
396
404
  Lastly, it generates prompt and makes generation with two other LLM models and three temperatures.
397
405
 
@@ -434,39 +442,22 @@ node_lines:
434
442
 
435
443
  ```
436
444
 
437
- # Supporting RAG modules
438
-
439
- ### Query Expansion
440
-
441
- - Query Decompose
442
- - HyDE
443
-
444
- ### Retrieval
445
-
446
- - BM25
447
- - Vector Retriever (choose your own embedding model)
448
- - Hybrid with rrf (reciprocal rank fusion)
449
- - Hybrid with cc (convex combination)
450
445
 
451
- ### Reranker
446
+ # ❗Supporting Nodes & modules
452
447
 
453
- - UPR
454
- - TART
455
- - MonoT5
456
448
 
457
- ### Passage Compressor
449
+ | Nodes | Modules |
450
+ |:-----------------------------------------------------------------------------------------------------------------:|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
451
+ | [Query_Expansion](https://marker-inc-korea.github.io/AutoRAG/nodes/query_expansion/query_expansion.html) | [Query_Decompose](https://marker-inc-korea.github.io/AutoRAG/nodes/query_expansion/query_decompose.html)<br/>[HyDE](https://marker-inc-korea.github.io/AutoRAG/nodes/query_expansion/hyde.html) |
452
+ | [Retrieval](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/retrieval.html) | [BM25](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/bm25.html)<br/>[VectorDB (choose embedding model)](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/vectordb.html)<br/>[Hybrid with rrf (reciprocal rank fusion)](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/hybrid_rrf.html)<br/>[Hybrid with cc (convex combination)](https://marker-inc-korea.github.io/AutoRAG/nodes/retrieval/hybrid_cc.html) |
453
+ | [Passage_Reranker](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/passage_reranker.html) | [UPR](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/upr.html)<br/>[Tart](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/tart.html)<br/>[MonoT5](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_reranker/monot5.html) |
454
+ | [Passage_Compressor](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_compressor/passage_compressor.html) | [Tree Summarize](https://marker-inc-korea.github.io/AutoRAG/nodes/passage_compressor/tree_summarize.html) |
455
+ | [Prompt Maker](https://marker-inc-korea.github.io/AutoRAG/nodes/prompt_maker/prompt_maker.html) | [Default Prompt Maker (f-string)](https://marker-inc-korea.github.io/AutoRAG/nodes/prompt_maker/fstring.html) |
456
+ | [Generator](https://marker-inc-korea.github.io/AutoRAG/nodes/generator/generator.html) | [llama_index llm](https://marker-inc-korea.github.io/AutoRAG/nodes/generator/llama_index_llm.html) |
458
457
 
459
- - Tree Summarize
460
458
 
461
- ### Prompt Maker
462
459
 
463
- - Default Prompt Maker (f-string)
464
-
465
- ### Generator
466
-
467
- - Llama Index LLM (choose your own model)
468
-
469
- # Roadmap
460
+ # 🛣Roadmap
470
461
 
471
462
  - [ ] Policy Module for modular RAG pipeline
472
463
  - [ ] Visualize evaluation result
@@ -480,6 +471,8 @@ node_lines:
480
471
 
481
472
  # Contribution
482
473
 
483
- We are developing AutoRAG as open-source. Feel free to contribute to this project.
484
- If you want to contribute to this project, please check out the [Contribution Guide](CONTRIBUTING.md).
485
- Plus, check out our detailed documentation at [here]().
474
+ We are developing AutoRAG as open-source.
475
+
476
+ So this project welcomes contributions and suggestions. Feel free to contribute to this project.
477
+
478
+ Plus, check out our detailed documentation at [here](https://marker-inc-korea.github.io/AutoRAG/index.html).