fix SILICONFLOW rerank error (#2980)

### What problem does this PR solve?

#2977

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2024-10-23 10:12:39 +08:00
committed by GitHub
parent 91ea559f9e
commit 89d5b2414e
2 changed files with 24 additions and 2 deletions

View File

@ -344,6 +344,9 @@ class SILICONFLOWRerank(Base):
self.base_url, json=payload, headers=self.headers
).json()
rank = np.zeros(len(texts), dtype=float)
if "results" not in response:
return rank, 0
for d in response["results"]:
rank[d["index"]] = d["relevance_score"]
return (