diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatQueryServiceImpl.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatQueryServiceImpl.java index dc4722441..2910c1cd2 100644 --- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatQueryServiceImpl.java +++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatQueryServiceImpl.java @@ -48,7 +48,7 @@ import com.tencent.supersonic.headless.chat.query.QueryManager; import com.tencent.supersonic.headless.chat.query.SemanticQuery; import com.tencent.supersonic.headless.chat.query.llm.s2sql.LLMSqlQuery; import com.tencent.supersonic.headless.server.facade.service.ChatLayerService; -import com.tencent.supersonic.headless.server.facade.service.RetrieveService; +import com.tencent.supersonic.headless.server.web.service.RetrieveService; import com.tencent.supersonic.headless.server.facade.service.SemanticLayerService; import lombok.extern.slf4j.Slf4j; import net.sf.jsqlparser.expression.Expression; diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/rest/ChatQueryApiController.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/rest/ChatQueryApiController.java index ca7cf8128..c63b36c3d 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/rest/ChatQueryApiController.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/rest/ChatQueryApiController.java @@ -7,7 +7,6 @@ import com.tencent.supersonic.headless.api.pojo.request.QueryNLReq; import com.tencent.supersonic.headless.api.pojo.request.QuerySqlReq; import com.tencent.supersonic.headless.api.pojo.response.ParseResp; import com.tencent.supersonic.headless.server.facade.service.ChatLayerService; -import com.tencent.supersonic.headless.server.facade.service.RetrieveService; import com.tencent.supersonic.headless.server.facade.service.SemanticLayerService; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; @@ -26,9 +25,6 @@ public class ChatQueryApiController { @Autowired private ChatLayerService chatLayerService; - @Autowired - private RetrieveService retrieveService; - @Autowired private SemanticLayerService semanticLayerService; @@ -37,7 +33,7 @@ public class ChatQueryApiController { HttpServletRequest request, HttpServletResponse response) throws Exception { queryNLReq.setUser(UserHolder.findUser(request, response)); - return retrieveService.retrieve(queryNLReq); + return chatLayerService.retrieve(queryNLReq); } @PostMapping("/chat/map") diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/ChatLayerService.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/ChatLayerService.java index 3a6b3df97..63b348999 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/ChatLayerService.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/ChatLayerService.java @@ -8,6 +8,9 @@ import com.tencent.supersonic.headless.api.pojo.request.QuerySqlReq; import com.tencent.supersonic.headless.api.pojo.response.MapInfoResp; import com.tencent.supersonic.headless.api.pojo.response.MapResp; import com.tencent.supersonic.headless.api.pojo.response.ParseResp; +import com.tencent.supersonic.headless.api.pojo.response.SearchResult; + +import java.util.List; /***dd * SemanticLayerService for query and search @@ -20,6 +23,8 @@ public interface ChatLayerService { MapInfoResp map(QueryMapReq queryMapReq); + List retrieve(QueryNLReq queryNLReq); + void correct(QuerySqlReq querySqlReq, User user); SqlEvaluation validate(QuerySqlReq querySqlReq, User user); diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/S2ChatLayerService.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/S2ChatLayerService.java index 2036171da..c4444d5d5 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/S2ChatLayerService.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/S2ChatLayerService.java @@ -21,12 +21,14 @@ import com.tencent.supersonic.headless.api.pojo.response.DataSetResp; import com.tencent.supersonic.headless.api.pojo.response.MapInfoResp; import com.tencent.supersonic.headless.api.pojo.response.MapResp; import com.tencent.supersonic.headless.api.pojo.response.ParseResp; +import com.tencent.supersonic.headless.api.pojo.response.SearchResult; import com.tencent.supersonic.headless.chat.ChatQueryContext; import com.tencent.supersonic.headless.chat.corrector.GrammarCorrector; import com.tencent.supersonic.headless.chat.corrector.SchemaCorrector; import com.tencent.supersonic.headless.chat.knowledge.builder.BaseWordBuilder; import com.tencent.supersonic.headless.chat.query.SemanticQuery; import com.tencent.supersonic.headless.server.facade.service.ChatLayerService; +import com.tencent.supersonic.headless.server.web.service.RetrieveService; import com.tencent.supersonic.headless.server.pojo.MetaFilter; import com.tencent.supersonic.headless.server.utils.ChatWorkflowEngine; import com.tencent.supersonic.headless.server.utils.ComponentFactory; @@ -56,6 +58,8 @@ public class S2ChatLayerService implements ChatLayerService { @Autowired private DataSetService dataSetService; @Autowired + private RetrieveService retrieveService; + @Autowired private ChatWorkflowEngine chatWorkflowEngine; @Override @@ -127,6 +131,11 @@ public class S2ChatLayerService implements ChatLayerService { return semanticParseInfo.getSqlEvaluation(); } + @Override + public List retrieve(QueryNLReq queryNLReq) { + return retrieveService.retrieve(queryNLReq); + } + private SemanticParseInfo correctSqlReq(QuerySqlReq querySqlReq, User user) { ChatQueryContext queryCtx = new ChatQueryContext(); SemanticSchema semanticSchema = schemaService.getSemanticSchema(); diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/task/FlightServerInitTask.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/task/FlightServerInitTask.java index 4585d88ee..a08a48650 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/task/FlightServerInitTask.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/task/FlightServerInitTask.java @@ -1,6 +1,6 @@ package com.tencent.supersonic.headless.server.task; -import com.tencent.supersonic.headless.server.facade.service.FlightService; +import com.tencent.supersonic.headless.server.web.service.FlightService; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import lombok.extern.slf4j.Slf4j; diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/FlightService.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/FlightService.java similarity index 83% rename from headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/FlightService.java rename to headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/FlightService.java index 57d5b0efe..ecaeeb336 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/FlightService.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/FlightService.java @@ -1,4 +1,4 @@ -package com.tencent.supersonic.headless.server.facade.service; +package com.tencent.supersonic.headless.server.web.service; import java.util.concurrent.ExecutorService; import org.apache.arrow.flight.sql.FlightSqlProducer; diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/RetrieveService.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/RetrieveService.java similarity index 80% rename from headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/RetrieveService.java rename to headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/RetrieveService.java index e2c50ac14..46cfe1ddb 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/RetrieveService.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/RetrieveService.java @@ -1,4 +1,4 @@ -package com.tencent.supersonic.headless.server.facade.service; +package com.tencent.supersonic.headless.server.web.service; import com.tencent.supersonic.headless.api.pojo.request.QueryNLReq; import com.tencent.supersonic.headless.api.pojo.response.SearchResult; diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/FlightServiceImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/FlightServiceImpl.java similarity index 99% rename from headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/FlightServiceImpl.java rename to headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/FlightServiceImpl.java index ec42b7f06..901a2faf6 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/FlightServiceImpl.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/FlightServiceImpl.java @@ -1,4 +1,4 @@ -package com.tencent.supersonic.headless.server.facade.service.impl; +package com.tencent.supersonic.headless.server.web.service.impl; import static com.google.protobuf.Any.pack; import static com.google.protobuf.ByteString.copyFrom; @@ -20,7 +20,7 @@ import com.tencent.supersonic.headless.api.pojo.Param; import com.tencent.supersonic.headless.api.pojo.request.QuerySqlReq; import com.tencent.supersonic.headless.api.pojo.request.SemanticQueryReq; import com.tencent.supersonic.headless.api.pojo.response.SemanticQueryResp; -import com.tencent.supersonic.headless.server.facade.service.FlightService; +import com.tencent.supersonic.headless.server.web.service.FlightService; import com.tencent.supersonic.headless.server.facade.service.SemanticLayerService; import com.tencent.supersonic.headless.server.utils.FlightUtils; import java.nio.charset.StandardCharsets; diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/RetrieveServiceImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/RetrieveServiceImpl.java similarity index 99% rename from headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/RetrieveServiceImpl.java rename to headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/RetrieveServiceImpl.java index 64617f417..334040d4f 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/RetrieveServiceImpl.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/RetrieveServiceImpl.java @@ -1,4 +1,4 @@ -package com.tencent.supersonic.headless.server.facade.service.impl; +package com.tencent.supersonic.headless.server.web.service.impl; import com.google.common.collect.Lists; import com.tencent.supersonic.auth.api.authentication.pojo.User; @@ -21,7 +21,7 @@ import com.tencent.supersonic.headless.chat.knowledge.helper.NatureHelper; import com.tencent.supersonic.headless.chat.mapper.MatchText; import com.tencent.supersonic.headless.chat.mapper.ModelWithSemanticType; import com.tencent.supersonic.headless.chat.mapper.SearchMatchStrategy; -import com.tencent.supersonic.headless.server.facade.service.RetrieveService; +import com.tencent.supersonic.headless.server.web.service.RetrieveService; import com.tencent.supersonic.headless.server.web.service.DataSetService; import com.tencent.supersonic.headless.server.web.service.SchemaService; import lombok.extern.slf4j.Slf4j;