flashrag-dev 0.3.0.dev20260821__tar.gz → 0.3.0.dev20260913__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 (64) hide show
  1. {flashrag_dev-0.3.0.dev20260821/flashrag_dev.egg-info → flashrag_dev-0.3.0.dev20260913}/PKG-INFO +1 -1
  2. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/generator.py +1 -0
  3. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/index_builder.py +1 -1
  4. flashrag_dev-0.3.0.dev20260913/flashrag/version.py +1 -0
  5. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913/flashrag_dev.egg-info}/PKG-INFO +1 -1
  6. flashrag_dev-0.3.0.dev20260821/flashrag/version.py +0 -1
  7. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/LICENSE +0 -0
  8. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/README.md +0 -0
  9. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/__init__.py +0 -0
  10. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/config/__init__.py +0 -0
  11. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/config/basic_config.yaml +0 -0
  12. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/config/config.py +0 -0
  13. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/dataset/__init__.py +0 -0
  14. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/dataset/dataset.py +0 -0
  15. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/dataset/utils.py +0 -0
  16. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/evaluator/__init__.py +0 -0
  17. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/evaluator/_bleu.py +0 -0
  18. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/evaluator/evaluator.py +0 -0
  19. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/evaluator/metrics.py +0 -0
  20. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/evaluator/utils.py +0 -0
  21. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/__init__.py +0 -0
  22. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/fid.py +0 -0
  23. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/multimodal_generator.py +0 -0
  24. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/openai_generator.py +0 -0
  25. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/stop_word_criteria.py +0 -0
  26. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/utils.py +0 -0
  27. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/judger/__init__.py +0 -0
  28. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/judger/judger.py +0 -0
  29. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/ReaRAG_utils.py +0 -0
  30. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/__init__.py +0 -0
  31. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/active_pipeline.py +0 -0
  32. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/branching_pipeline.py +0 -0
  33. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/mm_pipeline.py +0 -0
  34. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/pipeline.py +0 -0
  35. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/reasoning_pipeline.py +0 -0
  36. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/pipeline/replug_utils.py +0 -0
  37. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/prompt/__init__.py +0 -0
  38. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/prompt/base_prompt.py +0 -0
  39. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/prompt/coRAG_prompt.py +0 -0
  40. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/prompt/mm_prompt.py +0 -0
  41. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/prompt/selfask_examplars.py +0 -0
  42. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/prompt/trace_examplars.py +0 -0
  43. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/refiner/__init__.py +0 -0
  44. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/refiner/kg_refiner.py +0 -0
  45. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/refiner/llmlingua_compressor.py +0 -0
  46. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/refiner/refiner.py +0 -0
  47. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/refiner/selective_context_compressor.py +0 -0
  48. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/__init__.py +0 -0
  49. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/__main__.py +0 -0
  50. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/encoder.py +0 -0
  51. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/reranker.py +0 -0
  52. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/retriever.py +0 -0
  53. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/utils.py +0 -0
  54. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/utils/__init__.py +0 -0
  55. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/utils/constants.py +0 -0
  56. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/utils/pred_parse.py +0 -0
  57. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/utils/utils.py +0 -0
  58. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag_dev.egg-info/SOURCES.txt +0 -0
  59. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag_dev.egg-info/dependency_links.txt +0 -0
  60. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag_dev.egg-info/requires.txt +0 -0
  61. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag_dev.egg-info/top_level.txt +0 -0
  62. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/pyproject.toml +0 -0
  63. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/setup.cfg +0 -0
  64. {flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/setup.py +0 -0
{flashrag_dev-0.3.0.dev20260821/flashrag_dev.egg-info → flashrag_dev-0.3.0.dev20260913}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flashrag_dev
3
- Version: 0.3.0.dev20260821
3
+ Version: 0.3.0.dev20260913
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.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/generator/generator.py RENAMED
@@ -269,6 +269,7 @@ class VLLMGenerator(BaseGenerator):
269
269
  if return_scores:
270
270
  scores = []
271
271
  for output in outputs:
272
+ output_scores = []
272
273
  for single_output in output.outputs:
273
274
  if single_output.logprobs:
274
275
  token_probs = [np.exp(list(score_dict.values())[0].logprob)
{flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913}/flashrag/retriever/index_builder.py RENAMED
@@ -150,7 +150,6 @@ class Index_Builder:
150
150
  else:
151
151
  assert False, "Invalid bm25 backend!"
152
152
  elif self.retrieval_method == "splade":
153
- from seismic import SeismicIndex
154
153
  self.build_seismic_index()
155
154
  else:
156
155
  self.build_dense_index()
@@ -176,6 +175,7 @@ class Index_Builder:
176
175
  --batched_indexing 10000 # seismic batch
177
176
  --nknn 32
178
177
  """
178
+ from seismic import SeismicIndex
179
179
 
180
180
  if self.pooling_method != 'max':
181
181
  print(
flashrag_dev-0.3.0.dev20260913/flashrag/version.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.3.0dev20260913"
{flashrag_dev-0.3.0.dev20260821 → flashrag_dev-0.3.0.dev20260913/flashrag_dev.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flashrag_dev
3
- Version: 0.3.0.dev20260821
3
+ Version: 0.3.0.dev20260913
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.3.0.dev20260821/flashrag/version.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.3.0dev20260821"