topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement](chat) optimize functionCall result format (#549)

Browse Source
This commit is contained in:
lexluo09
2023-12-19 15:46:17 +08:00
committed by GitHub
parent d8930e8906
commit 1d9b6d6877
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OutputFormat.java
View File

@@ -1,7 +1,8 @@
package com.tencent.supersonic.chat.parser.sql.llm;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.tencent.supersonic.chat.parser.plugin.function.FunctionResp;
import com.tencent.supersonic.common.util.JsonUtil;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
@@ -114,15 +115,11 @@ public class OutputFormat {
public static FunctionResp functionCallParse(String llmOutput) {
try {
String[] findResult = llmOutput.split(PATTERN);
String result = findResult[0].trim();
Map<String, String> resultDict = JsonUtil.toMap(result, String.class, String.class);
log.info("result:{},resultDict:{}", result, resultDict);
String selection = resultDict.get("选择工具");
ObjectMapper objectMapper = new ObjectMapper();
JsonNode jsonNode = objectMapper.readTree(llmOutput);
String selectedTool = jsonNode.get("选择工具").asText();
FunctionResp resp = new FunctionResp();
resp.setToolSelection(selection);
resp.setToolSelection(selectedTool);
return resp;
} catch (Exception e) {
log.error("", e);