(improvement)(chat) Fix modelId to viewId in QueryReq (#773)

Co-authored-by: jolunoluo
tencentmusic · Feb 27, 2024 · 6b6e54e · 6b6e54e
1 parent 26260c7
commit 6b6e54e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -7,7 +7,7 @@
 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/...rver/src/main/java/com/tencent/supersonic/chat/server/service/impl/SearchServiceImpl.java b/...rver/src/main/java/com/tencent/supersonic/chat/server/service/impl/SearchServiceImpl.java
@@ -97,7 +97,7 @@ public List<SearchResult> search(QueryReq queryReq) {
         List<S2Term> originals = knowledgeService.getTerms(queryText);
         log.info("hanlp parse result: {}", originals);
         MapperHelper mapperHelper = ContextUtils.getBean(MapperHelper.class);
-        Set<Long> detectViewIds = mapperHelper.getViewIds(queryReq.getModelId(), agentService.getAgent(agentId));
+        Set<Long> detectViewIds = mapperHelper.getViewIds(queryReq.getViewId(), agentService.getAgent(agentId));
 
         QueryContext queryContext = new QueryContext();
         BeanUtils.copyProperties(queryReq, queryContext);
@@ -123,7 +123,7 @@ public List<SearchResult> search(QueryReq queryReq) {
         Set<SearchResult> searchResults = new LinkedHashSet();
         ViewInfoStat modelStat = NatureHelper.getViewStat(originals);
 
-        List<Long> possibleModels = getPossibleModels(queryReq, originals, modelStat, queryReq.getModelId());
+        List<Long> possibleModels = getPossibleModels(queryReq, originals, modelStat, queryReq.getViewId());
 
         // 5.1 priority dimension metric
         boolean existMetricAndDimension = searchMetricAndDimension(new HashSet<>(possibleModels), modelToName,