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

[improvement][headless-chat] improvement sql generation ``` (#1498) (#1499)

Browse Source
This commit is contained in:
ylhh353
2024-08-01 20:04:11 +08:00
committed by GitHub
parent ac96f72b07
commit 5a18ad5229
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/OnePassSCSqlGenStrategy.java
View File

@@ -62,8 +62,10 @@ public class OnePassSCSqlGenStrategy extends SqlGenStrategy {
ChatLanguageModel chatLanguageModel = getChatLanguageModel(llmReq.getModelConfig()); ChatLanguageModel chatLanguageModel = getChatLanguageModel(llmReq.getModelConfig());
Response<AiMessage> response = chatLanguageModel.generate(prompt.toUserMessage()); Response<AiMessage> response = chatLanguageModel.generate(prompt.toUserMessage());
String sqlOutput = StringUtils.normalizeSpace(response.content().text()); String sqlOutput = StringUtils.normalizeSpace(response.content().text());
output2Prompt.put(sqlOutput, prompt); // replace ```
keyPipelineLog.info("OnePassSCSqlGenStrategy modelResp:\n{}", sqlOutput); String sqlOutputFormat = sqlOutput.replaceAll("(?s)```sql\\s*(.*?)\\s*```", "$1").trim();
output2Prompt.put(sqlOutputFormat, prompt);
keyPipelineLog.info("OnePassSCSqlGenStrategy modelResp:\n{}", sqlOutputFormat);
} }
); );