Merge back from 1b55b48

This commit is contained in:
gaoao-3 2025-01-03 19:21:45 +08:00
parent 694ac94419
commit 5ce08b7119
2 changed files with 27 additions and 30 deletions

View file

@ -593,23 +593,8 @@ export function supportGuoba () {
field: 'geminiKey', field: 'geminiKey',
label: 'API密钥', label: 'API密钥',
bottomHelpMessage: '前往https://makersuite.google.com/app/apikey获取', bottomHelpMessage: '前往https://makersuite.google.com/app/apikey获取',
component: 'GTags', component: 'InputPassword'
componentProps: {
placeholder: '请输入您的 API 密钥',
allowAdd: true,
allowDel: true,
showPrompt: true,
promptProps: {
content: '请填写您的 API 密钥',
placeholder: '请输入 API 密钥',
okText: '添加'
}, },
valueParser: (value) => {
if (!value) return [];
return value.split(',').filter(Boolean);
},
},
},
{ {
field: 'geminiModel', field: 'geminiModel',
label: '模型', label: '模型',

View file

@ -5,7 +5,7 @@ import common from '../../../../lib/common/common.js';
/** /**
* CodeExecutionTool - Gemini API代码执行工具类 * CodeExecutionTool - Gemini API代码执行工具类
* 支持多种编程语言的代码执行并提供详细的执行结果和分析 * 支持多种编程语言的代码执行提供代码生成执行结果和分析
* @extends {AbstractTool} * @extends {AbstractTool}
*/ */
export class CodeExecutionTool extends AbstractTool { export class CodeExecutionTool extends AbstractTool {
@ -32,7 +32,7 @@ export class CodeExecutionTool extends AbstractTool {
}; };
// 工具功能描述 // 工具功能描述
description = '使用 Gemini API 执行代码,支持多种编程语言,并返回执行结果和相关解释。'; description = '使用 Gemini API 执行代码,支持多种编程语言,提供代码生成、执行结果和分析。';
// API请求超时设置30秒 // API请求超时设置30秒
static TIMEOUT = 30000; static TIMEOUT = 30000;
@ -60,14 +60,18 @@ export class CodeExecutionTool extends AbstractTool {
console.debug(`[CodeExecutionTool] 执行结果:`, result); console.debug(`[CodeExecutionTool] 执行结果:`, result);
// 解构结果对象 // 解构结果对象
const { output, executionOutput, explanation, error } = result; const { output, executionOutput, programCode, explanation, error } = result;
const forwardMsg = []; const forwardMsg = [];
// 根据执行结果构建消息 // 根据执行结果构建消息
if (error) { if (error) {
forwardMsg.push(`执行出错:\n${error}`); forwardMsg.push(`执行出错:\n${error}`);
} else { } else {
// 添加程序实际输出(如果有) // 添加生成的程序代码(如果有)
if (programCode) {
forwardMsg.push(`生成的程序代码:\n${programCode}`);
}
// 添加程序输出(如果有)
if (executionOutput) { if (executionOutput) {
forwardMsg.push(`程序输出:\n${executionOutput}`); forwardMsg.push(`程序输出:\n${executionOutput}`);
} }
@ -178,6 +182,7 @@ export class CodeExecutionTool extends AbstractTool {
} }
prompt += `\n执行代码:\n\`\`\`${language}\n${code}\n\`\`\`\n`; prompt += `\n执行代码:\n\`\`\`${language}\n${code}\n\`\`\`\n`;
prompt += '请按照以下格式返回结果:\n'; prompt += '请按照以下格式返回结果:\n';
prompt += '生成的程序代码:\n```\n<生成或修改后的程序代码>\n```\n';
prompt += '执行输出:\n```\n<程序实际输出内容>\n```\n'; prompt += '执行输出:\n```\n<程序实际输出内容>\n```\n';
prompt += '执行结果(OUTCOME_OK)\n```\n<执行状态和结果说明>\n```\n'; prompt += '执行结果(OUTCOME_OK)\n```\n<执行状态和结果说明>\n```\n';
prompt += '如果执行出错,请使用:\n'; prompt += '如果执行出错,请使用:\n';
@ -189,7 +194,7 @@ export class CodeExecutionTool extends AbstractTool {
/** /**
* 处理Gemini API响应数据 * 处理Gemini API响应数据
* @param {Object} data - API响应数据 * @param {Object} data - API响应数据
* @returns {Object} 处理后的结果对象包含输出执行输出解释和错误信息 * @returns {Object} 处理后的结果对象
* @private * @private
*/ */
processGeminiResponse(data) { processGeminiResponse(data) {
@ -208,8 +213,15 @@ export class CodeExecutionTool extends AbstractTool {
let explanation = ''; let explanation = '';
let error = null; let error = null;
let executionOutput = ''; let executionOutput = '';
let programCode = '';
try { try {
// 提取程序代码
const codeMatch = response.match(/生成的程序代码:\s*```(?:\w*\n)?([\s\S]*?)```/);
if (codeMatch) {
programCode = codeMatch[1].trim();
}
// 提取执行输出 // 提取执行输出
const outputMatch = response.match(/执行输出:\s*```(?:\w*\n)?([\s\S]*?)```/); const outputMatch = response.match(/执行输出:\s*```(?:\w*\n)?([\s\S]*?)```/);
if (outputMatch) { if (outputMatch) {
@ -226,7 +238,7 @@ export class CodeExecutionTool extends AbstractTool {
if (errorMatch) { if (errorMatch) {
error = errorMatch[1].trim(); error = errorMatch[1].trim();
} else { } else {
output = response; // 使用原始响应作为输出 output = response;
} }
} }
@ -237,16 +249,16 @@ export class CodeExecutionTool extends AbstractTool {
} }
} catch (err) { } catch (err) {
console.error('[CodeExecutionTool] 响应解析失败:', err); console.error('[CodeExecutionTool] 响应解析失败:', err);
output = response; // 解析失败时返回原始响应 output = response;
} }
// 返回完整的结果对象 // 返回处理结果
return { return {
output, // 执行结果说明 output, // 执行结果说明
executionOutput, // 程序实际输出 executionOutput, // 程序实际输出
programCode, // 生成的程序代码
explanation, // 代码分析 explanation, // 代码分析
error, // 错误信息 error // 错误信息
executionTime: Date.now() // 执行时间戳
}; };
} }
} }