From 5a18ad52294609dd16302c8d4083db98197de1ad Mon Sep 17 00:00:00 2001 From: ylhh353 <41932386+ylhh353@users.noreply.github.com> Date: Thu, 1 Aug 2024 20:04:11 +0800 Subject: [PATCH] [improvement][headless-chat] improvement sql generation ``` (#1498) (#1499) --- .../headless/chat/parser/llm/OnePassSCSqlGenStrategy.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java index 6a74c17e5..d013cf72c 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java @@ -62,8 +62,10 @@ public class OnePassSCSqlGenStrategy extends SqlGenStrategy { ChatLanguageModel chatLanguageModel = getChatLanguageModel(llmReq.getModelConfig()); Response response = chatLanguageModel.generate(prompt.toUserMessage()); String sqlOutput = StringUtils.normalizeSpace(response.content().text()); - output2Prompt.put(sqlOutput, prompt); - keyPipelineLog.info("OnePassSCSqlGenStrategy modelResp:\n{}", sqlOutput); + // replace ``` + String sqlOutputFormat = sqlOutput.replaceAll("(?s)```sql\\s*(.*?)\\s*```", "$1").trim(); + output2Prompt.put(sqlOutputFormat, prompt); + keyPipelineLog.info("OnePassSCSqlGenStrategy modelResp:\n{}", sqlOutputFormat); } );