flashrag-dev 0.3.0.dev20250818__tar.gz → 0.3.0.dev20250923__tar.gz

{flashrag_dev-0.3.0.dev20250818 → flashrag_dev-0.3.0.dev20250923}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flashrag_dev
-Version: 0.3.0.dev20250818
+Version: 0.3.0.dev20250923
 Summary: A library for efficient Retrieval-Augmented Generation research
 Home-page: https://github.com/RUC-NLPIR/FlashRAG
 Author: Jiajie Jin, Yutao Zhu, Chenghao Zhang, Xinyu Yang, Zhicheng Dou

{flashrag_dev-0.3.0.dev20250818 → flashrag_dev-0.3.0.dev20250923}/flashrag/generator/generator.py

@@ -261,13 +261,25 @@ class VLLMGenerator(BaseGenerator):
         if return_raw_output:
             base_output = outputs
         else:
-            generated_texts = [output.outputs[0].text for output in outputs]
+            generated_texts = [
+                [c.text for c in output.outputs] if len(output.outputs) > 1 else output.outputs[0].text
+                for output in outputs
+            ]
             base_output = generated_texts
         if return_scores:
             scores = []
             for output in outputs:
-                logprobs = output.outputs[0].logprobs
-                scores.append([np.exp(list(score_dict.values())[0].logprob) for score_dict in logprobs])
+                for single_output in output.outputs:
+                    if single_output.logprobs:
+                        token_probs = [np.exp(list(score_dict.values())[0].logprob) 
+                                      for score_dict in single_output.logprobs]
+                        output_scores.append(token_probs)
+                    else:
+                        output_scores.append([])
+                if len(output_scores) == 1:
+                    scores.append(output_scores[0])
+                else:
+                    scores.append(output_scores)
             return base_output, scores
         else:
             return base_output

{flashrag_dev-0.3.0.dev20250818 → flashrag_dev-0.3.0.dev20250923}/flashrag/retriever/index_builder.py

@@ -12,7 +12,6 @@ import subprocess
 import argparse
 import datasets
 import torch
-from seismic import SeismicIndex
 from tqdm import tqdm
 from flashrag.retriever.utils import load_model, load_corpus, pooling, set_default_instruction, judge_zh
 from transformers import AutoTokenizer, AutoModelForMaskedLM
@@ -151,6 +150,7 @@ class Index_Builder:
             else:
                 assert False, "Invalid bm25 backend!"
         elif self.retrieval_method == "splade":
+            from seismic import SeismicIndex
             self.build_seismic_index()
         else:
             self.build_dense_index()

flashrag_dev-0.3.0.dev20250923/flashrag/version.py

@@ -0,0 +1 @@
+__version__ = "0.3.0dev20250923"

{flashrag_dev-0.3.0.dev20250818 → flashrag_dev-0.3.0.dev20250923}/flashrag_dev.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flashrag-dev
-Version: 0.3.0.dev20250818
+Version: 0.3.0.dev20250923
 Summary: A library for efficient Retrieval-Augmented Generation research
 Home-page: https://github.com/RUC-NLPIR/FlashRAG
 Author: Jiajie Jin, Yutao Zhu, Chenghao Zhang, Xinyu Yang, Zhicheng Dou

flashrag_dev-0.3.0.dev20250818/flashrag/version.py

@@ -1 +0,0 @@
-__version__ = "0.3.0dev20250818"