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

(improvement)(chat) support filterCondition in requestLLM (#337)

Browse Source
This commit is contained in:
lexluo09
2023-11-08 12:07:34 +08:00
committed by GitHub
parent d6681ead60
commit 772d5bd3ae
2 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/s2ql/LLMRequestService.java
View File

@@ -110,6 +110,10 @@ public class LLMRequestService {
LLMReq llmReq = new LLMReq(); LLMReq llmReq = new LLMReq();
llmReq.setQueryText(queryText); llmReq.setQueryText(queryText);
LLMReq.FilterCondition filterCondition = new LLMReq.FilterCondition();
filterCondition.setTableName(modelIdToName.get(modelId));
llmReq.setFilterCondition(filterCondition);
LLMReq.LLMSchema llmSchema = new LLMReq.LLMSchema(); LLMReq.LLMSchema llmSchema = new LLMReq.LLMSchema();
llmSchema.setModelName(modelIdToName.get(modelId)); llmSchema.setModelName(modelIdToName.get(modelId));
llmSchema.setDomainName(modelIdToName.get(modelId)); llmSchema.setDomainName(modelIdToName.get(modelId));

8
chat/core/src/main/java/com/tencent/supersonic/chat/query/llm/s2ql/LLMReq.java
View File

@@ -8,6 +8,8 @@ public class LLMReq {
private String queryText; private String queryText;
private FilterCondition filterCondition;
private LLMSchema schema; private LLMSchema schema;
private List<ElementValue> linking; private List<ElementValue> linking;
@@ -35,4 +37,10 @@ public class LLMReq {
private List<String> fieldNameList; private List<String> fieldNameList;
} }
@Data
public static class FilterCondition {
private String tableName;
}
} }