flashrag-dev 0.3.0.dev20251106__tar.gz → 0.3.0.dev20251122__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.dev20251106/flashrag_dev.egg-info → flashrag_dev-0.3.0.dev20251122}/PKG-INFO +1 -1
  2. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/base_prompt.py +1 -1
  3. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/retriever.py +2 -2
  4. flashrag_dev-0.3.0.dev20251122/flashrag/version.py +1 -0
  5. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122/flashrag_dev.egg-info}/PKG-INFO +1 -1
  6. flashrag_dev-0.3.0.dev20251106/flashrag/version.py +0 -1
  7. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/LICENSE +0 -0
  8. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/README.md +0 -0
  9. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/__init__.py +0 -0
  10. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/config/__init__.py +0 -0
  11. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/config/basic_config.yaml +0 -0
  12. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/config/config.py +0 -0
  13. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/dataset/__init__.py +0 -0
  14. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/dataset/dataset.py +0 -0
  15. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/dataset/utils.py +0 -0
  16. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/evaluator/__init__.py +0 -0
  17. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/evaluator/_bleu.py +0 -0
  18. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/evaluator/evaluator.py +0 -0
  19. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/evaluator/metrics.py +0 -0
  20. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/evaluator/utils.py +0 -0
  21. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/__init__.py +0 -0
  22. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/fid.py +0 -0
  23. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/generator.py +0 -0
  24. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/multimodal_generator.py +0 -0
  25. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/openai_generator.py +0 -0
  26. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/stop_word_criteria.py +0 -0
  27. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/generator/utils.py +0 -0
  28. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/judger/__init__.py +0 -0
  29. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/judger/judger.py +0 -0
  30. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/ReaRAG_utils.py +0 -0
  31. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/__init__.py +0 -0
  32. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/active_pipeline.py +0 -0
  33. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/branching_pipeline.py +0 -0
  34. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/mm_pipeline.py +0 -0
  35. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/pipeline.py +0 -0
  36. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/reasoning_pipeline.py +0 -0
  37. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/pipeline/replug_utils.py +0 -0
  38. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/__init__.py +0 -0
  39. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/coRAG_prompt.py +0 -0
  40. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/mm_prompt.py +0 -0
  41. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/selfask_examplars.py +0 -0
  42. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/trace_examplars.py +0 -0
  43. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/refiner/__init__.py +0 -0
  44. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/refiner/kg_refiner.py +0 -0
  45. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/refiner/llmlingua_compressor.py +0 -0
  46. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/refiner/refiner.py +0 -0
  47. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/refiner/selective_context_compressor.py +0 -0
  48. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/__init__.py +0 -0
  49. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/__main__.py +0 -0
  50. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/encoder.py +0 -0
  51. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/index_builder.py +0 -0
  52. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/reranker.py +0 -0
  53. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/utils.py +0 -0
  54. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/utils/__init__.py +0 -0
  55. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/utils/constants.py +0 -0
  56. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/utils/pred_parse.py +0 -0
  57. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/utils/utils.py +0 -0
  58. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag_dev.egg-info/SOURCES.txt +0 -0
  59. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag_dev.egg-info/dependency_links.txt +0 -0
  60. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag_dev.egg-info/requires.txt +0 -0
  61. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag_dev.egg-info/top_level.txt +0 -0
  62. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/pyproject.toml +0 -0
  63. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/setup.cfg +0 -0
  64. {flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/setup.py +0 -0
{flashrag_dev-0.3.0.dev20251106/flashrag_dev.egg-info → flashrag_dev-0.3.0.dev20251122}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flashrag_dev
3
- Version: 0.3.0.dev20251106
3
+ Version: 0.3.0.dev20251122
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.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/prompt/base_prompt.py RENAMED
@@ -113,7 +113,7 @@ class PromptTemplate:
113
113
  return self.truncate_prompt(messages)
114
114
  if self.is_chat and self.enable_chat:
115
115
  if self.is_openai:
116
- self.truncate_prompt(messages)
116
+ return self.truncate_prompt(messages)
117
117
  else:
118
118
  prompt = self._get_tokenizer().apply_chat_template(
119
119
  messages, tokenize=False, add_generation_prompt=True
{flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122}/flashrag/retriever/retriever.py RENAMED
@@ -461,7 +461,7 @@ class DenseRetriever(BaseTextRetriever):
461
461
  scores = scores.tolist()
462
462
  idxs = idxs.tolist()
463
463
 
464
- flat_idxs = sum(idxs, [])
464
+ flat_idxs = [idx for sublist in idxs for idx in sublist]
465
465
  results = load_docs(self.corpus, flat_idxs)
466
466
  results = [results[i * num : (i + 1) * num] for i in range(len(idxs))]
467
467
 
@@ -569,7 +569,7 @@ class MultiModalRetriever(BaseRetriever):
569
569
  batch_scores = batch_scores.tolist()
570
570
  batch_idxs = batch_idxs.tolist()
571
571
 
572
- flat_idxs = sum(batch_idxs, [])
572
+ flat_idxs = flat_idxs = [idx for sublist in batch_idxs for idx in sublist]
573
573
  batch_results = load_docs(self.corpus, flat_idxs)
574
574
  batch_results = [batch_results[i * num : (i + 1) * num] for i in range(len(batch_idxs))]
575
575
 
flashrag_dev-0.3.0.dev20251122/flashrag/version.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.3.0dev20251122"
{flashrag_dev-0.3.0.dev20251106 → flashrag_dev-0.3.0.dev20251122/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.dev20251106
3
+ Version: 0.3.0.dev20251122
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.dev20251106/flashrag/version.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.3.0dev20251106"