Improve the performance of the chat page retrieve function

This commit is contained in:
luowanghaoyun
2025-08-12 11:21:11 +08:00
parent 1f6d217b26
commit c2ce3a75b7

View File

@@ -1,5 +1,6 @@
package com.tencent.supersonic.headless.server.service.impl;
import com.google.common.collect.Lists;
import com.tencent.supersonic.common.pojo.User;
import com.tencent.supersonic.common.pojo.enums.DictWordType;
import com.tencent.supersonic.headless.api.pojo.SchemaElement;
@@ -79,7 +80,7 @@ public class RetrieveServiceImpl implements RetrieveService {
Set<Long> dataSetIds = queryNLReq.getDataSetIds();
ChatQueryContext chatQueryContext = new ChatQueryContext(queryNLReq);
chatQueryContext.setModelIdToDataSetIds(dataSetService.getModelIdToDataSetIds());
chatQueryContext.setModelIdToDataSetIds(modelIdToDataSetIds);
Map<MatchText, List<HanlpMapResult>> regTextMap =
searchMatchStrategy.match(chatQueryContext, originals, dataSetIds);