topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2026-04-19 04:44:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][chat]Merge RuleTool and LLMTool.#1766 (#1768)

Browse Source
This commit is contained in:
Jun Zhang
2024-10-10 14:53:54 +08:00
committed by GitHub
parent fe3f4c36b5
commit 85552fb73a
14 changed files with 44 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
launchers/standalone/src/test/java/com/tencent/supersonic/evaluation/Text2SQLEval.java
View File

@@ -148,11 +148,11 @@ public class Text2SQLEval extends BaseTest {
return agent;
}
private static LLMParserTool getLLMQueryTool() {
LLMParserTool llmParserTool = new LLMParserTool();
llmParserTool.setType(AgentToolType.NL2SQL_LLM);
llmParserTool.setDataSetIds(Lists.newArrayList(-1L));
private static DatasetTool getLLMQueryTool() {
DatasetTool datasetTool = new DatasetTool();
datasetTool.setType(AgentToolType.DATASET);
datasetTool.setDataSetIds(Lists.newArrayList(-1L));
return llmParserTool;
return datasetTool;
}
}

1
launchers/standalone/src/test/java/com/tencent/supersonic/util/DataUtils.java
View File

@@ -42,6 +42,7 @@ public class DataUtils {
chatParseReq.setQueryText(query);
chatParseReq.setChatId(id);
chatParseReq.setUser(user_test);
chatParseReq.setDisableLLM(true);
return chatParseReq;
}