flashrag-dev 0.1.4.dev20250107__tar.gz → 0.1.4.dev20250108__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.dev20250107 → flashrag_dev-0.1.4.dev20250108}/PKG-INFO +1 -1
  2. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/config/config.py +12 -3
  3. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/dataset/dataset.py +1 -0
  4. flashrag_dev-0.1.4.dev20250108/flashrag/version.py +1 -0
  5. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag_dev.egg-info/PKG-INFO +1 -1
  6. flashrag_dev-0.1.4.dev20250107/flashrag/version.py +0 -1
  7. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/LICENSE +0 -0
  8. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/README.md +0 -0
  9. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/__init__.py +0 -0
  10. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/config/__init__.py +0 -0
  11. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/config/basic_config.yaml +0 -0
  12. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/dataset/__init__.py +0 -0
  13. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/dataset/utils.py +0 -0
  14. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/evaluator/__init__.py +0 -0
  15. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/evaluator/_bleu.py +0 -0
  16. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/evaluator/evaluator.py +0 -0
  17. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/evaluator/metrics.py +0 -0
  18. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/evaluator/utils.py +0 -0
  19. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/__init__.py +0 -0
  20. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/fid.py +0 -0
  21. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/generator.py +0 -0
  22. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/multimodal_generator.py +0 -0
  23. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/openai_generator.py +0 -0
  24. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/stop_word_criteria.py +0 -0
  25. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/generator/utils.py +0 -0
  26. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/judger/__init__.py +0 -0
  27. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/judger/judger.py +0 -0
  28. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/pipeline/__init__.py +0 -0
  29. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/pipeline/active_pipeline.py +0 -0
  30. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/pipeline/branching_pipeline.py +0 -0
  31. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/pipeline/mm_pipeline.py +0 -0
  32. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/pipeline/pipeline.py +0 -0
  33. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/pipeline/replug_utils.py +0 -0
  34. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/prompt/__init__.py +0 -0
  35. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/prompt/base_prompt.py +0 -0
  36. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/prompt/mm_prompt.py +0 -0
  37. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/prompt/selfask_examplars.py +0 -0
  38. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/prompt/trace_examplars.py +0 -0
  39. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/refiner/__init__.py +0 -0
  40. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/refiner/kg_refiner.py +0 -0
  41. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/refiner/llmlingua_compressor.py +0 -0
  42. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/refiner/refiner.py +0 -0
  43. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/refiner/selective_context_compressor.py +0 -0
  44. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/__init__.py +0 -0
  45. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/__main__.py +0 -0
  46. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/encoder.py +0 -0
  47. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/index_builder.py +0 -0
  48. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/reranker.py +0 -0
  49. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/retriever.py +0 -0
  50. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/retriever/utils.py +0 -0
  51. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/utils/__init__.py +0 -0
  52. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/utils/constants.py +0 -0
  53. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/utils/pred_parse.py +0 -0
  54. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/utils/utils.py +0 -0
  55. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag_dev.egg-info/SOURCES.txt +0 -0
  56. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag_dev.egg-info/dependency_links.txt +0 -0
  57. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag_dev.egg-info/requires.txt +0 -0
  58. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag_dev.egg-info/top_level.txt +0 -0
  59. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/pyproject.toml +0 -0
  60. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/setup.cfg +0 -0
  61. {flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/setup.py +0 -0
{flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flashrag_dev
3
- Version: 0.1.4.dev20250107
3
+ Version: 0.1.4.dev20250108
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.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/config/config.py RENAMED
@@ -237,9 +237,14 @@ class Config:
237
237
 
238
238
  def _prepare_dir(self):
239
239
  save_note = self.final_config["save_note"]
240
+ save_dir = self.final_config['save_dir']
241
+ if not save_dir.endswith("/"):
242
+ save_dir += "/"
243
+
240
244
  current_time = datetime.datetime.now()
245
+
241
246
  self.final_config["save_dir"] = os.path.join(
242
- self.final_config["save_dir"],
247
+ save_dir,
243
248
  f"{self.final_config['dataset_name']}_{current_time.strftime('%Y_%m_%d_%H_%M')}_{save_note}",
244
249
  )
245
250
  os.makedirs(self.final_config["save_dir"], exist_ok=True)
@@ -251,8 +256,12 @@ class Config:
251
256
  def _set_seed(self):
252
257
  import torch
253
258
  import numpy as np
254
-
255
- seed = self.final_config["seed"]
259
+ seed = self.final_config['seed']
260
+ try:
261
+ seed = int(seed)
262
+ except:
263
+ seed = 2025
264
+ self.final_config['seed'] = seed
256
265
  random.seed(seed)
257
266
  np.random.seed(seed)
258
267
  torch.manual_seed(seed)
{flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/flashrag/dataset/dataset.py RENAMED
@@ -131,6 +131,7 @@ class Dataset:
131
131
  raise NotImplementedError
132
132
 
133
133
  if self.sample_num is not None:
134
+ self.sample_num = int(self.sample_num)
134
135
  if self.random_sample:
135
136
  print(f"Random sample {self.sample_num} items in test set.")
136
137
  data = random.sample(data, self.sample_num)
flashrag_dev-0.1.4.dev20250108/flashrag/version.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.1.4dev20250108"
{flashrag_dev-0.1.4.dev20250107 → flashrag_dev-0.1.4.dev20250108}/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.dev20250107
3
+ Version: 0.1.4.dev20250108
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.dev20250107/flashrag/version.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.1.4dev20250107"