From 772d5bd3aeba3121358274bc31b7a003b843a3f8 Mon Sep 17 00:00:00 2001 From: lexluo09 <39718951+lexluo09@users.noreply.github.com> Date: Wed, 8 Nov 2023 12:07:34 +0800 Subject: [PATCH] (improvement)(chat) support filterCondition in requestLLM (#337) --- .../chat/parser/llm/s2ql/LLMRequestService.java | 4 ++++ .../tencent/supersonic/chat/query/llm/s2ql/LLMReq.java | 8 ++++++++ 2 files changed, 12 insertions(+) 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; + } }