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); } );