[improvement][headless-chat]Optimize text2sql prompt.

tencentmusic · Oct 18, 2024 · 482ccf3 · 482ccf3
1 parent c697f0d
commit 482ccf3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...rc/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java b/...rc/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java
@@ -46,7 +46,7 @@ public class OnePassSCSqlGenStrategy extends SqlGenStrategy {
             + "\n8.ALWAYS enclose alias created by `AS` command in underscores."
             + "\n9.ALWAYS translate alias created by `AS` command to the same language as the `#Question`."
             + "\n#Exemplars: {{exemplar}}"
-            + "\n#Question: Question:{{question}},Schema:{{schema}},SideInfo:{{information}}";
+            + "\n#Query: Question:{{question}},Schema:{{schema}},SideInfo:{{information}}";
 
     public OnePassSCSqlGenStrategy() {
         ChatAppManager.register(APP_KEY, ChatApp.builder().prompt(INSTRUCTION).name("语义SQL解析")
@@ -92,7 +92,8 @@ public LLMResp generate(LLMReq llmReq) {
         prompt2Exemplar.keySet().parallelStream().forEach(prompt -> {
             SemanticSql s2Sql = extractor.generateSemanticSql(prompt.toUserMessage().singleText());
             output2Prompt.put(s2Sql.getSql(), prompt);
-            keyPipelineLog.info("OnePassSCSqlGenStrategy modelReq:\n{} \nmodelResp:\n{}", prompt.text(), s2Sql);
+            keyPipelineLog.info("OnePassSCSqlGenStrategy modelReq:\n{} \nmodelResp:\n{}",
+                    prompt.text(), s2Sql);
         });
 
         // 4.format response.