topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-13 21:17:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][chat]Add queryId to QueryNLReq.
Some checks failed
supersonic CentOS CI / build (21) (push) Has been cancelled
Details
supersonic mac CI / build (21) (push) Has been cancelled
Details
supersonic ubuntu CI / build (21) (push) Has been cancelled
Details
supersonic windows CI / build (21) (push) Has been cancelled
Details

Browse Source
This commit is contained in:
jerryjzhang
2025-02-05 17:24:48 +08:00
parent ef161fe1f2
commit cc66ebd684
3 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatQueryServiceImpl.java
View File

@@ -91,6 +91,7 @@ public class ChatQueryServiceImpl implements ChatQueryService {
Long queryId = chatParseReq.getQueryId(); Long queryId = chatParseReq.getQueryId();
if (Objects.isNull(queryId)) { if (Objects.isNull(queryId)) {
queryId = chatManageService.createChatQuery(chatParseReq); queryId = chatManageService.createChatQuery(chatParseReq);
chatParseReq.setQueryId(queryId);
} }
ParseContext parseContext = buildParseContext(chatParseReq, new ChatParseResp(queryId)); ParseContext parseContext = buildParseContext(chatParseReq, new ChatParseResp(queryId));

1
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/QueryNLReq.java
View File

@@ -18,6 +18,7 @@ import java.util.Set;
@Data @Data
public class QueryNLReq extends SemanticQueryReq implements Serializable { public class QueryNLReq extends SemanticQueryReq implements Serializable {
private Long queryId;
private String queryText; private String queryText;
private Set<Long> dataSetIds = Sets.newHashSet(); private Set<Long> dataSetIds = Sets.newHashSet();
private User user; private User user;

1
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/query/llm/s2sql/LLMReq.java
View File

@@ -18,6 +18,7 @@ import java.util.stream.Collectors;
@Data @Data
public class LLMReq { public class LLMReq {
private Long queryId;
private String queryText; private String queryText;
private LLMSchema schema; private LLMSchema schema;
private List<Term> terms; private List<Term> terms;