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

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

View File

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

View File

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

View File

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