diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/s2ql/LLMRequestService.java b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/s2ql/LLMRequestService.java index 862f2270f..a930ef625 100644 --- a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/s2ql/LLMRequestService.java +++ b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/s2ql/LLMRequestService.java @@ -110,6 +110,10 @@ public class LLMRequestService { LLMReq llmReq = new LLMReq(); llmReq.setQueryText(queryText); + LLMReq.FilterCondition filterCondition = new LLMReq.FilterCondition(); + filterCondition.setTableName(modelIdToName.get(modelId)); + llmReq.setFilterCondition(filterCondition); + LLMReq.LLMSchema llmSchema = new LLMReq.LLMSchema(); llmSchema.setModelName(modelIdToName.get(modelId)); llmSchema.setDomainName(modelIdToName.get(modelId)); diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/query/llm/s2ql/LLMReq.java b/chat/core/src/main/java/com/tencent/supersonic/chat/query/llm/s2ql/LLMReq.java index cf036197a..ae4e6689c 100644 --- a/chat/core/src/main/java/com/tencent/supersonic/chat/query/llm/s2ql/LLMReq.java +++ b/chat/core/src/main/java/com/tencent/supersonic/chat/query/llm/s2ql/LLMReq.java @@ -8,6 +8,8 @@ public class LLMReq { private String queryText; + private FilterCondition filterCondition; + private LLMSchema schema; private List linking; @@ -35,4 +37,10 @@ public class LLMReq { private List fieldNameList; } + + @Data + public static class FilterCondition { + + private String tableName; + } }