flashrag-dev 0.1.4.dev20250222__tar.gz → 0.1.4.dev20250224__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 (61) hide show
  1. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/PKG-INFO +3 -1
  2. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/README.md +2 -0
  3. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/index_builder.py +2 -0
  4. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/retriever.py +2 -0
  5. flashrag_dev-0.1.4.dev20250224/flashrag/version.py +1 -0
  6. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag_dev.egg-info/PKG-INFO +3 -1
  7. flashrag_dev-0.1.4.dev20250222/flashrag/version.py +0 -1
  8. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/LICENSE +0 -0
  9. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/__init__.py +0 -0
  10. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/config/__init__.py +0 -0
  11. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/config/basic_config.yaml +0 -0
  12. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/config/config.py +0 -0
  13. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/dataset/__init__.py +0 -0
  14. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/dataset/dataset.py +0 -0
  15. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/dataset/utils.py +0 -0
  16. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/evaluator/__init__.py +0 -0
  17. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/evaluator/_bleu.py +0 -0
  18. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/evaluator/evaluator.py +0 -0
  19. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/evaluator/metrics.py +0 -0
  20. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/evaluator/utils.py +0 -0
  21. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/__init__.py +0 -0
  22. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/fid.py +0 -0
  23. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/generator.py +0 -0
  24. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/multimodal_generator.py +0 -0
  25. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/openai_generator.py +0 -0
  26. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/stop_word_criteria.py +0 -0
  27. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/generator/utils.py +0 -0
  28. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/judger/__init__.py +0 -0
  29. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/judger/judger.py +0 -0
  30. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/pipeline/__init__.py +0 -0
  31. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/pipeline/active_pipeline.py +0 -0
  32. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/pipeline/branching_pipeline.py +0 -0
  33. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/pipeline/mm_pipeline.py +0 -0
  34. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/pipeline/pipeline.py +0 -0
  35. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/pipeline/replug_utils.py +0 -0
  36. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/prompt/__init__.py +0 -0
  37. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/prompt/base_prompt.py +0 -0
  38. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/prompt/mm_prompt.py +0 -0
  39. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/prompt/selfask_examplars.py +0 -0
  40. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/prompt/trace_examplars.py +0 -0
  41. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/refiner/__init__.py +0 -0
  42. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/refiner/kg_refiner.py +0 -0
  43. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/refiner/llmlingua_compressor.py +0 -0
  44. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/refiner/refiner.py +0 -0
  45. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/refiner/selective_context_compressor.py +0 -0
  46. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/__init__.py +0 -0
  47. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/__main__.py +0 -0
  48. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/encoder.py +0 -0
  49. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/reranker.py +0 -0
  50. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/retriever/utils.py +0 -0
  51. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/utils/__init__.py +0 -0
  52. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/utils/constants.py +0 -0
  53. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/utils/pred_parse.py +0 -0
  54. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag/utils/utils.py +0 -0
  55. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag_dev.egg-info/SOURCES.txt +0 -0
  56. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag_dev.egg-info/dependency_links.txt +0 -0
  57. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag_dev.egg-info/requires.txt +0 -0
  58. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/flashrag_dev.egg-info/top_level.txt +0 -0
  59. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/pyproject.toml +0 -0
  60. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/setup.cfg +0 -0
  61. {flashrag_dev-0.1.4.dev20250222 → flashrag_dev-0.1.4.dev20250224}/setup.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flashrag_dev
3
- Version: 0.1.4.dev20250222
3
+ Version: 0.1.4.dev20250224
4
4
  Summary: A library for efficient Retrieval-Augmented Generation research
5
5
  Home-page: https://github.com/RUC-NLPIR/FlashRAG
6
6
  Author: Jiajie Jin, Yutao Zhu, Chenghao Zhang, Xinyu Yang, Zhicheng Dou
@@ -90,11 +90,13 @@ FlashRAG is still under development and there are many issues and room for impro
90
90
  - [x] Support OpenAI models
91
91
  - [x] Provdide instructions for each component
92
92
  - [x] Integrate sentence Transformers
93
+ - [x] Support multimodal RAG
93
94
  - [ ] Inlcude more RAG approaches
94
95
  - [ ] Add more evaluation metrics (e.g., Unieval, name-entity F1) and benchmarks (e.g., RGB benchmark)
95
96
  - [ ] Enhance code adaptability and readability
96
97
 
97
98
  ## :page_with_curl: Changelog
99
+ [25/02/24] We have added support for **multimodal RAG**, including **MLLMs like Llava, Qwen, InternVL**, and various **multimodal retrievers with Clip architecture**. More information can be found in our new version of arxiv article. Try it🔥!
98
100
 
99
101
  [25/01/21] Our technical paper [FlashRAG: A Python Toolkit for Efficient RAG Research](https://arxiv.org/abs/2405.13576) is honored to have been accepted to the Resource Track of the 2025 **ACM Web Conference (WWW 2025)**. Please Check it out!
100
102
 
@@ -73,11 +73,13 @@ FlashRAG is still under development and there are many issues and room for impro
73
73
  - [x] Support OpenAI models
74
74
  - [x] Provdide instructions for each component
75
75
  - [x] Integrate sentence Transformers
76
+ - [x] Support multimodal RAG
76
77
  - [ ] Inlcude more RAG approaches
77
78
  - [ ] Add more evaluation metrics (e.g., Unieval, name-entity F1) and benchmarks (e.g., RGB benchmark)
78
79
  - [ ] Enhance code adaptability and readability
79
80
 
80
81
  ## :page_with_curl: Changelog
82
+ [25/02/24] We have added support for **multimodal RAG**, including **MLLMs like Llava, Qwen, InternVL**, and various **multimodal retrievers with Clip architecture**. More information can be found in our new version of arxiv article. Try it🔥!
81
83
 
82
84
  [25/01/21] Our technical paper [FlashRAG: A Python Toolkit for Efficient RAG Research](https://arxiv.org/abs/2405.13576) is honored to have been accepted to the Resource Track of the 2025 **ACM Web Conference (WWW 2025)**. Please Check it out!
83
85
 
@@ -210,6 +210,8 @@ class Index_Builder:
210
210
  retriever = bm25s.BM25(corpus=corpus, backend="numba")
211
211
  retriever.index(corpus_tokens)
212
212
  retriever.save(self.save_dir, corpus=None)
213
+ tokenizer.save_vocab(self.save_dir)
214
+ tokenizer.save_stopwords(self.save_dir)
213
215
 
214
216
  print("Finish!")
215
217
 
@@ -255,6 +255,8 @@ class BM25Retriever(BaseTextRetriever):
255
255
  else:
256
256
  stemmer = Stemmer.Stemmer("english")
257
257
  self.tokenizer = bm25s.tokenization.Tokenizer(stopwords='en', stemmer=stemmer)
258
+ self.tokenizer.load_stopwords(self.index_path)
259
+ self.tokenizer.load_vocab(self.index_path)
258
260
 
259
261
  self.searcher.corpus = self.corpus
260
262
  self.searcher.backend = "numba"
@@ -0,0 +1 @@
1
+ __version__ = "0.1.4dev20250224"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flashrag-dev
3
- Version: 0.1.4.dev20250222
3
+ Version: 0.1.4.dev20250224
4
4
  Summary: A library for efficient Retrieval-Augmented Generation research
5
5
  Home-page: https://github.com/RUC-NLPIR/FlashRAG
6
6
  Author: Jiajie Jin, Yutao Zhu, Chenghao Zhang, Xinyu Yang, Zhicheng Dou
@@ -90,11 +90,13 @@ FlashRAG is still under development and there are many issues and room for impro
90
90
  - [x] Support OpenAI models
91
91
  - [x] Provdide instructions for each component
92
92
  - [x] Integrate sentence Transformers
93
+ - [x] Support multimodal RAG
93
94
  - [ ] Inlcude more RAG approaches
94
95
  - [ ] Add more evaluation metrics (e.g., Unieval, name-entity F1) and benchmarks (e.g., RGB benchmark)
95
96
  - [ ] Enhance code adaptability and readability
96
97
 
97
98
  ## :page_with_curl: Changelog
99
+ [25/02/24] We have added support for **multimodal RAG**, including **MLLMs like Llava, Qwen, InternVL**, and various **multimodal retrievers with Clip architecture**. More information can be found in our new version of arxiv article. Try it🔥!
98
100
 
99
101
  [25/01/21] Our technical paper [FlashRAG: A Python Toolkit for Efficient RAG Research](https://arxiv.org/abs/2405.13576) is honored to have been accepted to the Resource Track of the 2025 **ACM Web Conference (WWW 2025)**. Please Check it out!
100
102
 
@@ -1 +0,0 @@
1
- __version__ = "0.1.4dev20250222"