gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix OpenAI Reranker logic (#579)

Browse source
This commit is contained in:
PHẠM MINH LONG 2025-06-12 23:34:32 +07:00 • committed by GitHub
parent 3377ebce9c
commit ab3b8e0061
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
graphiti_core/cross_encoder/openai_reranker_client.py
View file

@ -106,7 +106,7 @@ class OpenAIRerankerClient(CrossEncoderClient):
if len(top_logprobs) == 0: if len(top_logprobs) == 0:
continue continue
norm_logprobs = np.exp(top_logprobs[0].logprob) norm_logprobs = np.exp(top_logprobs[0].logprob)
if bool(top_logprobs[0].token): if top_logprobs[0].token.strip().split(" ")[0].lower() == "true":
scores.append(norm_logprobs) scores.append(norm_logprobs)
else: else:
scores.append(1 - norm_logprobs) scores.append(1 - norm_logprobs)