flashrag-dev 0.1.3.dev20241104__tar.gz → 0.1.3.dev20241106__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 (57) hide show
  1. {flashrag-dev-0.1.3.dev20241104/flashrag_dev.egg-info → flashrag-dev-0.1.3.dev20241106}/PKG-INFO +1 -1
  2. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/generator/generator.py +1 -0
  3. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/pipeline/active_pipeline.py +1 -1
  4. flashrag-dev-0.1.3.dev20241106/flashrag/version.py +1 -0
  5. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106/flashrag_dev.egg-info}/PKG-INFO +1 -1
  6. flashrag-dev-0.1.3.dev20241104/flashrag/version.py +0 -1
  7. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/LICENSE +0 -0
  8. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/README.md +0 -0
  9. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/__init__.py +0 -0
  10. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/config/__init__.py +0 -0
  11. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/config/basic_config.yaml +0 -0
  12. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/config/config.py +0 -0
  13. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/dataset/__init__.py +0 -0
  14. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/dataset/dataset.py +0 -0
  15. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/dataset/utils.py +0 -0
  16. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/evaluator/__init__.py +0 -0
  17. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/evaluator/_bleu.py +0 -0
  18. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/evaluator/evaluator.py +0 -0
  19. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/evaluator/metrics.py +0 -0
  20. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/evaluator/utils.py +0 -0
  21. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/generator/__init__.py +0 -0
  22. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/generator/fid.py +0 -0
  23. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/generator/openai_generator.py +0 -0
  24. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/generator/stop_word_criteria.py +0 -0
  25. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/judger/__init__.py +0 -0
  26. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/judger/judger.py +0 -0
  27. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/pipeline/__init__.py +0 -0
  28. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/pipeline/branching_pipeline.py +0 -0
  29. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/pipeline/pipeline.py +0 -0
  30. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/pipeline/replug_utils.py +0 -0
  31. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/prompt/__init__.py +0 -0
  32. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/prompt/base_prompt.py +0 -0
  33. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/prompt/selfask_examplars.py +0 -0
  34. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/prompt/trace_examplars.py +0 -0
  35. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/refiner/__init__.py +0 -0
  36. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/refiner/kg_refiner.py +0 -0
  37. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/refiner/llmlingua_compressor.py +0 -0
  38. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/refiner/refiner.py +0 -0
  39. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/refiner/selective_context_compressor.py +0 -0
  40. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/__init__.py +0 -0
  41. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/__main__.py +0 -0
  42. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/encoder.py +0 -0
  43. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/index_builder.py +0 -0
  44. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/reranker.py +0 -0
  45. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/retriever.py +0 -0
  46. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/retriever/utils.py +0 -0
  47. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/utils/__init__.py +0 -0
  48. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/utils/constants.py +0 -0
  49. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/utils/pred_parse.py +0 -0
  50. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/utils/utils.py +0 -0
  51. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag_dev.egg-info/SOURCES.txt +0 -0
  52. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag_dev.egg-info/dependency_links.txt +0 -0
  53. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag_dev.egg-info/requires.txt +0 -0
  54. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag_dev.egg-info/top_level.txt +0 -0
  55. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/pyproject.toml +0 -0
  56. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/setup.cfg +0 -0
  57. {flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/setup.py +0 -0
{flashrag-dev-0.1.3.dev20241104/flashrag_dev.egg-info → flashrag-dev-0.1.3.dev20241106}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flashrag-dev
3
- Version: 0.1.3.dev20241104
3
+ Version: 0.1.3.dev20241106
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.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/generator/generator.py RENAMED
@@ -235,6 +235,7 @@ class VLLMGenerator(BaseGenerator):
235
235
  # fix for llama3
236
236
  if "stop" in generation_params:
237
237
  generation_params["stop"].append("<|eot_id|>")
238
+ generation_params['include_stop_str_in_output'] = True
238
239
  else:
239
240
  generation_params["stop"] = ["<|eot_id|>"]
240
241
 
{flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106}/flashrag/pipeline/active_pipeline.py RENAMED
@@ -932,7 +932,7 @@ class IRCOTPipeline(BasicPipeline):
932
932
  input_prompt = self.prompt_template.get_string(
933
933
  question=question, retrieval_result=retrieval_result, previous_gen=" ".join(thoughts)
934
934
  )
935
- new_thought = self.generator.generate(input_prompt)[0]
935
+ new_thought = self.generator.generate(input_prompt,stop=['.', '\n'])[0]
936
936
  thoughts.append(new_thought)
937
937
  iter_num += 1
938
938
  if "So the answer is:" in new_thought:
flashrag-dev-0.1.3.dev20241106/flashrag/version.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.1.3dev20241106"
{flashrag-dev-0.1.3.dev20241104 → flashrag-dev-0.1.3.dev20241106/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.3.dev20241104
3
+ Version: 0.1.3.dev20241106
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.3.dev20241104/flashrag/version.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.1.3dev20241104"