From 6b6e54e95f085ce4ef0a86a19a2699505209a48c Mon Sep 17 00:00:00 2001 From: LXW <1264174498@qq.com> Date: Tue, 27 Feb 2024 21:42:40 +0800 Subject: [PATCH] (improvement)(chat) Fix modelId to viewId in QueryReq (#773) Co-authored-by: jolunoluo --- .../tencent/supersonic/chat/api/pojo/request/QueryReq.java | 2 +- .../chat/server/service/impl/SearchServiceImpl.java | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/QueryReq.java b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/QueryReq.java index 1d457610a..25adb87a5 100644 --- a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/QueryReq.java +++ b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/QueryReq.java @@ -7,7 +7,7 @@ import lombok.Data; public class QueryReq { private String queryText; private Integer chatId; - private Long modelId; + private Long viewId; private User user; private QueryFilters queryFilters; private boolean saveAnswer = true; diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/SearchServiceImpl.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/SearchServiceImpl.java index 2ccf64e4a..5518ab7be 100644 --- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/SearchServiceImpl.java +++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/SearchServiceImpl.java @@ -97,7 +97,7 @@ public class SearchServiceImpl implements SearchService { List originals = knowledgeService.getTerms(queryText); log.info("hanlp parse result: {}", originals); MapperHelper mapperHelper = ContextUtils.getBean(MapperHelper.class); - Set detectViewIds = mapperHelper.getViewIds(queryReq.getModelId(), agentService.getAgent(agentId)); + Set detectViewIds = mapperHelper.getViewIds(queryReq.getViewId(), agentService.getAgent(agentId)); QueryContext queryContext = new QueryContext(); BeanUtils.copyProperties(queryReq, queryContext); @@ -123,7 +123,7 @@ public class SearchServiceImpl implements SearchService { Set searchResults = new LinkedHashSet(); ViewInfoStat modelStat = NatureHelper.getViewStat(originals); - List possibleModels = getPossibleModels(queryReq, originals, modelStat, queryReq.getModelId()); + List possibleModels = getPossibleModels(queryReq, originals, modelStat, queryReq.getViewId()); // 5.1 priority dimension metric boolean existMetricAndDimension = searchMetricAndDimension(new HashSet<>(possibleModels), modelToName,