From d78010c376c27b324df3388b7e6bcd599ab44822 Mon Sep 17 00:00:00 2001 From: Zhichang Yu Date: Fri, 21 Feb 2025 18:50:54 +0800 Subject: [PATCH] Fixed similarity on infinity (#5236) ### What problem does this PR solve? Fixed similarity on infinity ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- rag/utils/infinity_conn.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/rag/utils/infinity_conn.py b/rag/utils/infinity_conn.py index 66d5145d9..dcdae7e8b 100644 --- a/rag/utils/infinity_conn.py +++ b/rag/utils/infinity_conn.py @@ -344,6 +344,10 @@ class InfinityConnection(DocStoreConnection): for k, v in matchExpr.extra_options.items(): if not isinstance(v, str): matchExpr.extra_options[k] = str(v) + similarity = matchExpr.extra_options.get("similarity") + if similarity: + matchExpr.extra_options["threshold"] = similarity + del matchExpr.extra_options["similarity"] logger.debug(f"INFINITY search MatchDenseExpr: {json.dumps(matchExpr.__dict__)}") elif isinstance(matchExpr, FusionExpr): logger.debug(f"INFINITY search FusionExpr: {json.dumps(matchExpr.__dict__)}")