flashrag-dev 0.1.4.dev20250617__tar.gz → 0.1.4.dev20250715__tar.gz

{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250715}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flashrag_dev
-Version: 0.1.4.dev20250617
+Version: 0.1.4.dev20250715
 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.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250715}/flashrag/retriever/encoder.py

@@ -24,7 +24,7 @@ class Encoder:
             Encodes a list of queries into embeddings.
     """
 
-    def __init__(self, model_name, model_path, pooling_method, max_length, use_fp16=True, instruction="", silent=False):
+    def __init__(self, model_name, model_path, pooling_method, max_length, use_fp16=True, instruction=None, silent=False):
         self.model_name = model_name
         self.model_path = model_path
         self.pooling_method = pooling_method

{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250715}/flashrag/retriever/retriever.py

@@ -305,6 +305,8 @@ class BM25Retriever(BaseTextRetriever):
             else:
                 results = load_docs(self.corpus, [hit.docid for hit in hits])
         elif self.backend == "bm25s":
+            import bm25s
+
             # query_tokens = self.tokenizer.tokenize([query], return_as="tuple", update_vocab=False)
             query_tokens = bm25s.tokenize([query])
             results, scores = self.searcher.retrieve(query_tokens, k=num)
@@ -328,6 +330,8 @@ class BM25Retriever(BaseTextRetriever):
                 results.append(item_result)
                 scores.append(item_score)
         elif self.backend == "bm25s":
+            import bm25s
+
             # query_tokens = self.tokenizer.tokenize(query, return_as="tuple", update_vocab=False)
             query_tokens = bm25s.tokenize(query)
             results, scores = self.searcher.retrieve(query_tokens, k=num)

flashrag_dev-0.1.4.dev20250715/flashrag/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.4dev20250715"

{flashrag_dev-0.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250715}/flashrag_dev.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flashrag-dev
-Version: 0.1.4.dev20250617
+Version: 0.1.4.dev20250715
 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.1.4.dev20250617 → flashrag_dev-0.1.4.dev20250715}/flashrag_dev.egg-info/requires.txt

@@ -13,7 +13,7 @@ tiktoken
 torch
 tqdm
 transformers>=4.40.0
-bm25s[core]==0.2.0
+bm25s[core]==0.2.1
 fschat
 streamlit
 chonkie<1.1.0,>=1.0.2
@@ -37,7 +37,7 @@ tiktoken
 torch
 tqdm
 transformers>=4.40.0
-bm25s[core]==0.2.0
+bm25s[core]==0.2.1
 fschat
 streamlit
 chonkie<1.1.0,>=1.0.2
@@ -61,7 +61,7 @@ tiktoken
 torch
 tqdm
 transformers>=4.40.0
-bm25s[core]==0.2.0
+bm25s[core]==0.2.1
 fschat
 streamlit
 chonkie<1.1.0,>=1.0.2

flashrag_dev-0.1.4.dev20250617/flashrag/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.4dev20250617"