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)