topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(Headless) Add chat-api to text2sql and query data (#809)

Browse Source 
Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-03-12 22:33:42 +08:00
committed by GitHub
parent f93bee81cb
commit 46910fbfcf
1 changed files with 53 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/api/ChatQueryApiController.java Normal file
View File

@@ -0,0 +1,53 @@
package com.tencent.supersonic.headless.server.rest.api;
import com.tencent.supersonic.auth.api.authentication.utils.UserHolder;
import com.tencent.supersonic.headless.api.pojo.request.ExecuteQueryReq;
import com.tencent.supersonic.headless.api.pojo.request.QueryReq;
import com.tencent.supersonic.headless.server.service.ChatQueryService;
import com.tencent.supersonic.headless.server.service.SearchService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@RestController
@RequestMapping("/api/semantic/query")
@Slf4j
public class ChatQueryApiController {
@Autowired
private ChatQueryService chatQueryService;
@Autowired
private SearchService searchService;
@PostMapping("/chat/search")
public Object search(@RequestBody QueryReq queryReq,
HttpServletRequest request,
HttpServletResponse response) throws Exception {
queryReq.setUser(UserHolder.findUser(request, response));
return searchService.search(queryReq);
}
@PostMapping("/chat/parse")
public Object parse(@RequestBody QueryReq queryReq,
HttpServletRequest request,
HttpServletResponse response) throws Exception {
queryReq.setUser(UserHolder.findUser(request, response));
return chatQueryService.performParsing(queryReq);
}
@PostMapping("/chat/execute")
public Object execute(@RequestBody ExecuteQueryReq executeQueryReq,
HttpServletRequest request,
HttpServletResponse response) throws Exception {
executeQueryReq.setUser(UserHolder.findUser(request, response));
return chatQueryService.performExecution(executeQueryReq);
}
}