sdk.js•2.98 kB
function tryParseJson(input, fallback) {
try {
return JSON.parse(input);
}
catch (e) {
return fallback;
}
}
function parseToolXml(xmlContent) {
const toolNameRegex = /<tool_name>\s*(.*?)\s*<\/tool_name>/;
const toolNameMatch = xmlContent.match(toolNameRegex);
const toolName = toolNameMatch ? toolNameMatch[1] : null;
const argumentsRegex = /<arguments>\s*([\s\S]*?)\s*<\/arguments>/;
const argumentsMatch = xmlContent.match(argumentsRegex);
const jsonResult = {
type: 'function',
function: {
name: toolName,
arguments: argumentsMatch[1],
},
};
return jsonResult;
}
export function extractAndParseToolsToJson(text) {
let tools = extractAndParseToolsXmlToJson(text);
if (!tools || !tools.length) {
tools = extractAndParseToolsTextToJson(text);
}
return tools;
}
export function extractAndParseToolsXmlToJson(text) {
const regex = /<use_tool>\s*([\s\S]*?)\s*<\/use_tool>/g;
const match = text.match(regex);
if (match && match.length ) {
const results = match.map(text => parseToolXml(text.trim())).filter(Boolean);
return results;
}
// console.log("未找到 <use_tool> 标签或其内容。");
return [];
}
export function extractAndParseToolsTextToJson(text) {
// 极简主义的返回结果
if (text.includes('use_tool') && text.includes('<tool_name>') && text.includes('<arguments>')) {
// 补充上</arguments>避免没有返回
if (text.endsWith('}')) {
text += '</arguments>';
}
return [parseToolXml(text)].filter(Boolean);
}
// 超极简主义
const lines = text.split('\n');
if (lines.length === 2) {
const [toolName, args_json_str] = lines;
// 验证是否可以正常解析为json
const args_json = tryParseJson(args_json_str);
if (!args_json) {
return [];
}
const func = {
type: 'function',
function: {
name: toolName,
arguments: args_json_str,
},
};
return [func];
}
if (text.includes('<arguments>')) {
const index = text.indexOf('<arguments>');
const tool_name = text.substring(0, index).trim();
const right_text = text.substring(index).trim();
const args_xml_str = right_text.includes('</arguments>') ? right_text : right_text + '</arguments>';
const args_json_str = args_xml_str.match(/<arguments>([\s\S]*?)<\/arguments>/)[1];
// 验证是否可以正常解析为json
const args_json = tryParseJson(args_json_str);
if (!args_json) {
return [];
}
const func = {
type: 'function',
function: {
name: tool_name,
arguments: args_json_str,
},
};
return [func];
}
return [];
}