From 5757f4dd0556c6bb771c3025122e6aa77e78f9fd Mon Sep 17 00:00:00 2001 From: jerryjzhang Date: Fri, 3 Jan 2025 22:43:13 +0800 Subject: [PATCH] [improvement][chat]Introduce parameter to control # of semantic fields should be sent to the LLM. --- .../supersonic/headless/chat/mapper/AllFieldMapper.java | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/AllFieldMapper.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/AllFieldMapper.java index ad3533feb..bfffaff52 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/AllFieldMapper.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/AllFieldMapper.java @@ -28,12 +28,13 @@ public class AllFieldMapper extends BaseMapper { schemaElements.addAll(entry.getValue().getDimensions()); schemaElements.addAll(entry.getValue().getMetrics()); + List allMatches = Lists.newArrayList(); for (SchemaElement schemaElement : schemaElements) { - chatQueryContext.getMapInfo().getMatchedElements(entry.getKey()) - .add(SchemaElementMatch.builder().word(schemaElement.getName()) - .element(schemaElement).detectWord(schemaElement.getName()) - .similarity(1.0).build()); + allMatches.add(SchemaElementMatch.builder().word(schemaElement.getName()) + .element(schemaElement).detectWord(schemaElement.getName()) + .similarity(0.1).build()); } + chatQueryContext.getMapInfo().setMatchedElements(entry.getKey(), allMatches); } }