pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor:Improve VoyageRerank not texts handling (#9539)

Browse Source 
### What problem does this PR solve?

Improve VoyageRerank not texts handling

### Type of change

- [x] Refactoring
This commit is contained in:
Stephen Hu
2025-08-19 10:31:04 +08:00
committed by GitHub
parent b5b8032a56
commit a0d630365c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rag/llm/rerank_model.py
View File

@ -482,9 +482,10 @@ class VoyageRerank(Base):
self.model_name = model_name self.model_name = model_name
def similarity(self, query: str, texts: list): def similarity(self, query: str, texts: list):
rank = np.zeros(len(texts), dtype=float)
if not texts: if not texts:
return rank, 0 return np.array([]), 0
rank = np.zeros(len(texts), dtype=float)
res = self.client.rerank(query=query, documents=texts, model=self.model_name, top_k=len(texts)) res = self.client.rerank(query=query, documents=texts, model=self.model_name, top_k=len(texts))
try: try:
for r in res.results: for r in res.results: