From ab3b8e0061c18ceb397ca81d23c4c1fa76d4604c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?PHA=CC=A3M=20MINH=20LONG?= <59755436+LongPML@users.noreply.github.com> Date: Thu, 12 Jun 2025 23:34:32 +0700 Subject: [PATCH] Fix OpenAI Reranker logic (#579) --- graphiti_core/cross_encoder/openai_reranker_client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphiti_core/cross_encoder/openai_reranker_client.py b/graphiti_core/cross_encoder/openai_reranker_client.py index 3e3d7566..8cc03f5c 100644 --- a/graphiti_core/cross_encoder/openai_reranker_client.py +++ b/graphiti_core/cross_encoder/openai_reranker_client.py @@ -106,7 +106,7 @@ class OpenAIRerankerClient(CrossEncoderClient): if len(top_logprobs) == 0: continue 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) else: scores.append(1 - norm_logprobs)