topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-14 05:43:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][chat]Incorporate Response into Context objects.

Browse Source
This commit is contained in:
jerryjzhang
2024-10-27 17:44:29 +08:00
parent 1842261dfe
commit 3e0f724e97
13 changed files with 36 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/RetrieveServiceImpl.java
View File

@@ -120,7 +120,7 @@ public class RetrieveServiceImpl implements RetrieveService {
return searchResults.stream().limit(RESULT_SIZE).collect(Collectors.toList());
}
private List<Long> getPossibleDataSets(QueryNLReq queryCtx, List<S2Term> originals,
private List<Long> getPossibleDataSets(QueryNLReq queryReq, List<S2Term> originals,
Set<Long> dataSetIds) {
if (CollectionUtils.isNotEmpty(dataSetIds)) {
return new ArrayList<>(dataSetIds);
@@ -128,8 +128,8 @@ public class RetrieveServiceImpl implements RetrieveService {
List<Long> possibleDataSets = NatureHelper.selectPossibleDataSets(originals);
if (possibleDataSets.isEmpty()) {
if (Objects.nonNull(queryCtx.getContextParseInfo())) {
possibleDataSets.add(queryCtx.getContextParseInfo().getDataSetId());
if (Objects.nonNull(queryReq.getContextParseInfo())) {
possibleDataSets.add(queryReq.getContextParseInfo().getDataSetId());
}
}