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

(improvement)(headless)Adapt s2_exemplar file to new version of prompting strategy.

Browse Source
This commit is contained in:
jerryjzhang
2024-06-30 11:17:27 +08:00
parent 5bfef816c4
commit 1d0b288922
7 changed files with 320 additions and 507 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/ParserConfig.java
View File

@@ -15,10 +15,9 @@ public class ParserConfig extends ParameterConfig {
public static final Parameter PARSER_STRATEGY_TYPE =
new Parameter("s2.parser.s2sql.strategy", "ONE_PASS_SELF_CONSISTENCY",
"LLM解析生成S2SQL策略",
"ONE_PASS_SELF_CONSISTENCY: 通过投票方式一步生成sql"
+ "\nTWO_PASS_AUTO_COT_SELF_CONSISTENCY: 通过思维链且投票方式两步生成sql",
"list", "Parser相关配置", Lists.newArrayList(
"ONE_PASS_SELF_CONSISTENCY", "TWO_PASS_AUTO_COT_SELF_CONSISTENCY"));
"ONE_PASS_SELF_CONSISTENCY: 通过投票方式一步生成sql",
"list", "Parser相关配置",
Lists.newArrayList("ONE_PASS_SELF_CONSISTENCY"));
public static final Parameter PARSER_LINKING_VALUE_ENABLE =
new Parameter("s2.parser.linking.value.enable", "true",

4
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/query/llm/s2sql/LLMReq.java
View File

@@ -76,8 +76,8 @@ public class LLMReq {
}
public enum SqlGenType {
ONE_PASS_SELF_CONSISTENCY("1_pass_self_consistency"),
TWO_PASS_AUTO_COT_SELF_CONSISTENCY("2_pass_auto_cot_self_consistency");
ONE_PASS_SELF_CONSISTENCY("1_pass_self_consistency");
private String name;