flashrag-dev 0.1.4.dev20250617__tar.gz → 0.1.4.dev20250623__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 (62) hide show
  1. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/PKG-INFO +1 -1
  2. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/retriever.py +4 -0
  3. flashrag_dev-0.1.4.dev20250623/flashrag/version.py +1 -0
  4. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/PKG-INFO +1 -1
  5. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/requires.txt +3 -3
  6. flashrag_dev-0.1.4.dev20250617/flashrag/version.py +0 -1
  7. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/LICENSE +0 -0
  8. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/README.md +0 -0
  9. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/__init__.py +0 -0
  10. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/config/__init__.py +0 -0
  11. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/config/basic_config.yaml +0 -0
  12. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/config/config.py +0 -0
  13. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/dataset/__init__.py +0 -0
  14. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/dataset/dataset.py +0 -0
  15. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/dataset/utils.py +0 -0
  16. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/evaluator/__init__.py +0 -0
  17. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/evaluator/_bleu.py +0 -0
  18. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/evaluator/evaluator.py +0 -0
  19. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/evaluator/metrics.py +0 -0
  20. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/evaluator/utils.py +0 -0
  21. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/__init__.py +0 -0
  22. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/fid.py +0 -0
  23. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/generator.py +0 -0
  24. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/multimodal_generator.py +0 -0
  25. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/openai_generator.py +0 -0
  26. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/stop_word_criteria.py +0 -0
  27. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/generator/utils.py +0 -0
  28. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/judger/__init__.py +0 -0
  29. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/judger/judger.py +0 -0
  30. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/__init__.py +0 -0
  31. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/active_pipeline.py +0 -0
  32. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/branching_pipeline.py +0 -0
  33. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/mm_pipeline.py +0 -0
  34. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/pipeline.py +0 -0
  35. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/reasoning_pipeline.py +0 -0
  36. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/pipeline/replug_utils.py +0 -0
  37. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/prompt/__init__.py +0 -0
  38. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/prompt/base_prompt.py +0 -0
  39. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/prompt/mm_prompt.py +0 -0
  40. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/prompt/selfask_examplars.py +0 -0
  41. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/prompt/trace_examplars.py +0 -0
  42. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/refiner/__init__.py +0 -0
  43. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/refiner/kg_refiner.py +0 -0
  44. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/refiner/llmlingua_compressor.py +0 -0
  45. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/refiner/refiner.py +0 -0
  46. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/refiner/selective_context_compressor.py +0 -0
  47. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/__init__.py +0 -0
  48. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/__main__.py +0 -0
  49. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/encoder.py +0 -0
  50. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/index_builder.py +0 -0
  51. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/reranker.py +0 -0
  52. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/utils.py +0 -0
  53. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/utils/__init__.py +0 -0
  54. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/utils/constants.py +0 -0
  55. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/utils/pred_parse.py +0 -0
  56. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/utils/utils.py +0 -0
  57. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/SOURCES.txt +0 -0
  58. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/dependency_links.txt +0 -0
  59. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/top_level.txt +0 -0
  60. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/pyproject.toml +0 -0
  61. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/setup.cfg +0 -0
  62. {flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/setup.py +0 -0
{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flashrag_dev
3
- Version: 0.1.4.dev20250617
3
+ Version: 0.1.4.dev20250623
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
{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag/retriever/retriever.py RENAMED
@@ -305,6 +305,8 @@ class BM25Retriever(BaseTextRetriever):
305
305
  else:
306
306
  results = load_docs(self.corpus, [hit.docid for hit in hits])
307
307
  elif self.backend == "bm25s":
308
+ import bm25s
309
+
308
310
  # query_tokens = self.tokenizer.tokenize([query], return_as="tuple", update_vocab=False)
309
311
  query_tokens = bm25s.tokenize([query])
310
312
  results, scores = self.searcher.retrieve(query_tokens, k=num)
@@ -328,6 +330,8 @@ class BM25Retriever(BaseTextRetriever):
328
330
  results.append(item_result)
329
331
  scores.append(item_score)
330
332
  elif self.backend == "bm25s":
333
+ import bm25s
334
+
331
335
  # query_tokens = self.tokenizer.tokenize(query, return_as="tuple", update_vocab=False)
332
336
  query_tokens = bm25s.tokenize(query)
333
337
  results, scores = self.searcher.retrieve(query_tokens, k=num)
flashrag_dev-0.1.4.dev20250623/flashrag/version.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.1.4dev20250623"
{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flashrag-dev
3
- Version: 0.1.4.dev20250617
3
+ Version: 0.1.4.dev20250623
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
{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250623}/flashrag_dev.egg-info/requires.txt RENAMED
@@ -13,7 +13,7 @@ tiktoken
13
13
  torch
14
14
  tqdm
15
15
  transformers>=4.40.0
16
- bm25s[core]==0.2.0
16
+ bm25s[core]==0.2.1
17
17
  fschat
18
18
  streamlit
19
19
  chonkie<1.1.0,>=1.0.2
@@ -37,7 +37,7 @@ tiktoken
37
37
  torch
38
38
  tqdm
39
39
  transformers>=4.40.0
40
- bm25s[core]==0.2.0
40
+ bm25s[core]==0.2.1
41
41
  fschat
42
42
  streamlit
43
43
  chonkie<1.1.0,>=1.0.2
@@ -61,7 +61,7 @@ tiktoken
61
61
  torch
62
62
  tqdm
63
63
  transformers>=4.40.0
64
- bm25s[core]==0.2.0
64
+ bm25s[core]==0.2.1
65
65
  fschat
66
66
  streamlit
67
67
  chonkie<1.1.0,>=1.0.2
flashrag_dev-0.1.4.dev20250617/flashrag/version.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.1.4dev20250617"