deepfish-ai 1.0.13 → 1.0.15

package/src/core/extension/InquirerExtension.js

@@ -0,0 +1,167 @@
+const { default: inquirer } = require('inquirer')
+
+// 判断问答
+async function askConfirm(name, message, defaultVal = true, opt = {}) {
+  const questions = [
+    {
+      type: 'confirm',
+      name,
+      message,
+      default: defaultVal,
+      ...opt
+    }
+  ]
+  const answers = await inquirer.prompt(questions)
+  return answers[name]
+}
+
+// 选择问答
+function askList(name, message, choices, defaultVal = 0, opt = {}) {
+  const questions = [
+    {
+      type: 'list',
+      name,
+      message,
+      choices,
+      default: defaultVal,
+      ...opt
+    }
+  ]
+  return inquirer.prompt(questions)
+}
+
+// 输入问答
+async function askInput(name, message, defaultVal = '', opt = {}) {
+  const questions = [
+    {
+      type: 'input',
+      name,
+      message,
+      default: defaultVal,
+      ...opt
+    },
+  ]
+  const answers = await inquirer.prompt(questions)
+  return answers[name]
+}
+
+// 输入数字
+async function askNumber(name, message, defaultVal = 0, opt = {}) {
+  const questions = [
+    {
+      type: 'number',
+      name,
+      message,
+      default: defaultVal,
+      ...opt
+    },
+  ]
+  const answers = await inquirer.prompt(questions)
+  return answers[name]
+}
+
+// 输入任何
+function askAny(questions) {
+  return inquirer.prompt(questions)
+}
+
+const descriptions = [
+  {
+    type: "function",
+    function: {
+      name: "askConfirm",
+      description: "用户交互：提示用户确认问题，返回布尔值。通过message参数显示问题文本，defaultVal参数设置默认值（默认为true），opt参数可传递inquirer的其他选项。",
+      parameters: {
+        type: "object",
+        properties: {
+          name: { type: "string", description: "问题的标识名称" },
+          message: { type: "string", description: "显示给用户的问题文本" },
+          defaultVal: { type: "boolean", description: "默认值，默认为true" },
+          opt: { type: "object", description: "inquirer的其他选项对象，可选" },
+        },
+        required: ["name", "message"],
+      },
+    },
+  },
+  {
+    type: "function",
+    function: {
+      name: "askList",
+      description: "用户交互：提示用户从列表中选择一项，返回选择项对象。通过choices参数传递选项数组，defaultVal参数设置默认选择的索引（默认为0），opt参数可传递inquirer的其他选项。",
+      parameters: {
+        type: "object",
+        properties: {
+          name: { type: "string", description: "问题的标识名称" },
+          message: { type: "string", description: "显示给用户的问题文本" },
+          choices: { type: "array", description: "选项数组" },
+          defaultVal: { type: "number", description: "默认选择索引，默认为0" },
+          opt: { type: "object", description: "inquirer的其他选项对象，可选" },
+        },
+        required: ["name", "message", "choices"],
+      },
+    },
+  },
+  {
+    type: "function",
+    function: {
+      name: "askInput",
+      description: "用户交互：提示用户输入文本，返回输入的字符串。通过message参数显示问题文本，defaultVal参数设置默认值（默认为空字符串），opt参数可传递inquirer的其他选项。",
+      parameters: {
+        type: "object",
+        properties: {
+          name: { type: "string", description: "问题的标识名称" },
+          message: { type: "string", description: "显示给用户的问题文本" },
+          defaultVal: { type: "string", description: "默认值，默认为空字符串" },
+          opt: { type: "object", description: "inquirer的其他选项对象，可选" },
+        },
+        required: ["name", "message"],
+      },
+    },
+  },
+  {
+    type: "function",
+    function: {
+      name: "askNumber",
+      description: "用户交互：提示用户输入数字，返回输入的数字。通过message参数显示问题文本，defaultVal参数设置默认值（默认为0），opt参数可传递inquirer的其他选项。",
+      parameters: {
+        type: "object",
+        properties: {
+          name: { type: "string", description: "问题的标识名称" },
+          message: { type: "string", description: "显示给用户的问题文本" },
+          defaultVal: { type: "number", description: "默认值，默认为0" },
+          opt: { type: "object", description: "inquirer的其他选项对象，可选" },
+        },
+        required: ["name", "message"],
+      },
+    },
+  },
+  {
+    type: "function",
+    function: {
+      name: "askAny",
+      description: "用户交互：提示用户根据自定义问题对象数组进行交互，返回答案对象。通过questions参数传递inquirer格式的问题数组，支持多种问题类型和复杂配置。",
+      parameters: {
+        type: "object",
+        properties: {
+          questions: { type: "array", description: "inquirer的问题对象数组" },
+        },
+        required: ["questions"],
+      },
+    },
+  },
+]
+
+const functions = {
+  askConfirm,
+  askList,
+  askInput,
+  askNumber,
+  askAny,
+}
+
+module.exports = {
+  name: 'InquirerExtension',
+  extensionDescription: "提供用户交互功能，支持确认、列表选择、文本输入、数字输入等多种交互方式",
+  descriptions,
+  functions,
+}

package/src/core/extension/SystemExtension.js

@@ -0,0 +1,370 @@
+/**
+ * @Author: Roman 306863030@qq.com
+ * @Date: 2026-03-17 11:59:19
+ * @LastEditors: Roman 306863030@qq.com
+ * @LastEditTime: 2026-03-25 20:05:23
+ * @FilePath: \deepfish\src\core\extension\SystemExtension.js
+ * @Description: 默认扩展函数
+ * @
+ */
+const path = require('path')
+const { logError, logSuccess, logInfo, getConfigPath } = require('../utils/log')
+const iconv = require('iconv-lite') // 用于编码转换
+const { cloneDeep } = require('lodash')
+const { aiRequestSingle } = require('../ai-services/AiWorker/AiTools')
+const { spawnSync } = require('child_process')
+const { detectEncoding } = require('../utils/normal')
+
+// 执行系统命令
+// 执行系统命令（全平台兼容：Windows/PowerShell/CentOS）
+function executeCommand(command, timeout = -1) {
+  logSuccess(`Executing system command: ${command}; ${timeout > 0 ? `Timeout: ${timeout}ms` : 'No timeout limit'}`)
+  try {
+    const result = spawnSync(command, {
+      cwd: process.cwd(),
+      stdio: 'pipe',
+      shell: true,
+      encoding: 'buffer',
+      windowsHide: true,
+      argv0: 'deepfish-shell',
+      timeout: timeout > 0 ? timeout : undefined,
+    })
+    let targetEncoding = this.config?.encoding
+    if (!targetEncoding || targetEncoding === 'auto') {
+      targetEncoding = detectEncoding(result.stdout || result.stderr)
+    }
+    const stdout = iconv.decode(result.stdout, targetEncoding)
+    const stderr = iconv.decode(result.stderr, targetEncoding)
+    const code = result.status
+    if (stderr && !stderr.trim().startsWith('WARNING')) {
+      const error = new Error(`Command failed (code ${code}): ${stderr.trim()}`)
+      logError(`Execute error: ${error.message}`)
+      return `Execute error: ${error.message}`
+    }
+    logSuccess(`${stdout}\nCommand executed successfully`)
+    return stdout || 'Command executed successfully'
+  } catch (decodeError) {
+    logError(`Encoding convert error: ${decodeError.message}`)
+    return `Failed to parse command output: ${decodeError.message}`
+  }
+}
+
+// 请求ai服务
+async function requestAI(
+  systemDescription,
+  prompt,
+  temperature = this.aiConfig.temperature,
+) {
+  logSuccess(`Requesting AI`)
+  if (
+    typeof systemDescription === 'object' &&
+    systemDescription.systemDescription
+  ) {
+    prompt = systemDescription.prompt || prompt || ''
+    systemDescription = systemDescription.systemDescription || ''
+  }
+  try {
+    logInfo(`aiSystem: ${systemDescription}`)
+    logInfo(`aiPrompt: ${prompt}`)
+    let aiConfig = this.aiConfig
+    if (temperature !== aiConfig.temperature) {
+      aiConfig = cloneDeep(aiConfig)
+      aiConfig.temperature = temperature
+    }
+    const response = await aiRequestSingle(
+      this.aiService.client,
+      aiConfig,
+      systemDescription,
+      prompt,
+    )
+    logInfo(`aiResponse: ${response}`)
+    return response
+  } catch (error) {
+    logError(`Error executing AI function: ${error.message}`)
+    throw error
+  }
+}
+
+// 执行js代码
+async function executeJSCode(code) {
+  logSuccess('Executing JavaScript code: ')
+  logSuccess(code)
+
+  try {
+    const { functions } = this.extensionManager.extensions
+    const Func = new Function(
+      'Tools',
+      'require',
+      `return (async () => {
+          this.logMessages = []
+          const originalLog = console.log
+          const newLog = function () {
+            originalLog.apply(console, arguments)
+            this.logMessages.push(Array.from(arguments).join(' '))
+          }
+          console.log = newLog.bind(this)
+          ${code}
+          console.log = originalLog
+          return this.logMessages.join('\\n')
+        })()`,
+    )
+    const originalRequire = require
+    const newRequire = (modulePath) => {
+      if (modulePath.startsWith('./')) {
+        const resolvedPath = path.resolve('.', modulePath)
+        return originalRequire(resolvedPath)
+      }
+      return originalRequire(modulePath)
+    }
+
+    const result = await Func(functions, newRequire)
+
+    return result || ''
+  } catch (error) {
+    logError(`Error executing code: ${error.stack}`)
+    throw error
+  }
+}
+// 生成一个扩展函数文件 关键字：内置函数、扩展工具
+async function getExtensionFileRule(goal) {
+  const newGoal = `
+### 任务目标
+基于指定规则创建一个标准化的Node.js NPM项目，实现用户目标：${goal}，最终输出符合AI工作流调用规范的函数模块，并配套中英文说明文档。
+
+### 第一步：项目初始化
+1. 目录创建：新建目录，目录名称为「项目功能名称」，作为NPM项目根目录
+2. package.json配置：
+   - name字段值：@deepfish-ai/项目功能名称（替换「项目功能名称」为实际功能名）
+   - git仓库地址：固定为 https://github.com/qq306863030/deepfish-extensions.git
+   - author设置为"DeepFish AI",
+3. 主文件：项目入口文件必须命名为index.js
+4. 文档文件：项目根目录需新增2个文档文件：
+   - README_CN.md（中文说明文档）
+   - README.md（英文说明文档）
+
+### 第二步：index.js 完整开发规范
+#### 2.1 核心输出要求
+文件需输出四个核心字段，且代码逻辑清晰、可运行：
+- name：字符串类型，扩展的名称标识
+- extensionDescription：字符串类型，扩展功能的简要描述，说明该扩展提供的核心能力
+- descriptions：数组类型，每个元素为OpenAI可识别的函数描述对象
+- functions：对象类型，key为函数名称，value为函数方法体
+
+#### 2.2 开发强制规则
+1. 参数一致性：functions中函数的入参，必须与descriptions中对应函数声明的parameters完全一致
+2. 命名规范：
+   - 函数名称前缀：「领域用途+分隔符」（如systemFileManagement_）
+   - 函数描述开头：统一格式「领域用途+分隔符+功能描述」（如系统文件管理:重命名文件）
+3. 内置工具调用：函数内部可直接使用this.Tools下的内置方法，示例：
+   - this.Tools.requestAI(systemDescription, prompt, temperature)
+   - this.Tools.readFile(filePath)
+   - 其他文件处理类内置函数（运行时自动注入）
+4. 函数数量：至少包含1个可被AI工作流调用的函数
+
+#### 2.3 基础代码模板（必须遵循）
+const descriptions = []
+const functions = {}
+module.exports = {
+  name: '扩展名称',
+  extensionDescription: '扩展功能的简要描述',
+  descriptions,
+  functions,
+}
+
+#### 2.4 参考示例（可参考格式）
+const descriptions = [
+  {
+    name: 'systemFileManagement_renameFile',
+    description: '系统文件管理:重命名文件',
+    parameters: {
+      type: 'object',
+      properties: {
+        oldPath: { type: 'string', description: '旧文件路径' },
+        newPath: { type: 'string', description: '新文件路径' },
+      },
+    },
+  },
+]
+const functions = {
+  systemFileManagement_renameFile: (oldPath, newPath) => {
+    return this.Tools.rename(oldPath, newPath)
+  },
+}
+module.exports = {
+  name: 'systemFileManagement',
+  extensionDescription: '提供文件管理相关功能，包括文件重命名等操作',
+  descriptions,
+  functions,
+}
+
+### 第三步：README文档规范
+#### 3.1 通用要求
+- 两个文档需在标题下方包含「中英文切换标签」（如文档顶部标注「English | 中文」/「中文 | English」）
+- 结构保持一致，仅语言不同，核心模块顺序不可调整
+- 文件名称README_CN.md（中文）、README.md（英文）
+- 链接使用相对路径，如[中文](./README_CN.md)
+
+#### 3.2 核心模块
+1. 总体功能描述：
+   - 清晰说明当前NPM包的核心定位、整体功能价值、适用场景
+   - 语言简洁易懂，无需技术细节，聚焦「做什么」而非「怎么做」
+2. 快速开始：
+   - 明确说明安装步骤，顺序不可颠倒：
+     ① 先安装deepfish-ai全局库：npm install deepfish-ai -g
+     ② 再安装当前项目库：npm install @deepfish-ai/项目功能名称 -g
+3. 函数列表及功能描述：
+   - 列出当前项目中所有函数名称
+   - 对应说明每个函数的核心功能
+   - 无需编写各个函数的具体使用方法
+  `
+  return newGoal
+}
+
+// 获取ai的配置
+function getAiConfig() {
+  return cloneDeep(this.aiConfig)
+}
+
+// 获取ai的配置文件所在目录
+function getAiConfigPath() {
+  return getConfigPath()
+}
+
+// 加载skill
+function executeSkill(skillFilePath, subGoalPrompt = '') {
+  const skillContent = this.skillConfigManager.loadSkill(skillFilePath)
+  if (!subGoalPrompt) {
+    return skillContent
+  }
+  // 调用子工作流完成目标
+  return this.aiService.subSkillWorkflow(skillContent, subGoalPrompt)
+}
+
+const descriptions = [
+  {
+    type: 'function',
+    function: {
+      name: 'executeCommand',
+      description:
+        '执行系统命令，返回执行结果。适用于运行shell命令、系统工具等。command为需要执行的系统命令字符串，如"ls -l"; timeout为命令执行的超时时间，单位为毫秒，-1表示不限制超时时间, 默认值为-1。注意：如果执行多条命令，且需要保持会话，使用命令链的方式执行。命令执行失败时会抛出错误，成功时返回命令执行结果字符串或"System command executed successfully"。',
+      parameters: {
+        type: 'object',
+        properties: {
+          command: { type: 'string' },
+          timeout: { type: 'number' },
+        },
+        required: ['command'],
+      },
+    },
+  },
+  {
+    type: 'function',
+    function: {
+      name: 'requestAI',
+      description:
+        '请求AI服务处理简单任务，如随机生成一段话、翻译文本、数学计算、代码分析、知识检索等。通过systemDescription参数指定AI的行为和限制，prompt参数输入任务描述, temperature参数指定AI的温度（0-2之间的浮点数，默认为用户配置）。返回AI处理后的结果字符串，执行失败时会抛出错误。',
+      parameters: {
+        type: 'object',
+        properties: {
+          systemDescription: { type: 'string' },
+          prompt: { type: 'string' },
+          temperature: { type: 'number' },
+        },
+        required: ['systemDescription', 'prompt'],
+      },
+    },
+  },
+  {
+    type: 'function',
+    function: {
+      name: 'executeJSCode',
+      description:
+        '执行JavaScript代码，返回代码执行结果。代码中可通过Tools命名空间直接调用其他工具函数（如await Tools.createFile(),注意：不需要使用require引入）,Tools中引入了一些常用库可直接调用（Tools.fs="fs-extra", Tools.dayjs="dayjs", Tools.axios="axios", Tools.lodash="lodash"），支持引入自定义模块（需使用绝对路径）。注意：1.代码中不要使用__dirname获取当前目录，请使用path.resolve(".")来获取当前目录。2.执行失败时会抛出错误，成功时返回代码执行结果或空字符串。',
+      parameters: {
+        type: 'object',
+        properties: {
+          code: { type: 'string' },
+        },
+        required: ['code'],
+      },
+    },
+  },
+
+  {
+    type: 'function',
+    function: {
+      name: 'getExtensionFileRule',
+      description:
+        '如果用户需要为本程序ai工作流生成一个或多个工具函数作为一个工作流执行过程中调用的扩展工具，则需要先调用此函数获取生成扩展文件的规则;示例：生成一个扩展工具: 能够产生一个随机数的函数;',
+      parameters: {
+        type: 'object',
+        properties: {
+          goal: { type: 'string' },
+        },
+        required: ['goal'],
+      },
+    },
+  },
+  {
+    type: 'function',
+    function: {
+      name: 'getAiConfig',
+      description: '获取ai请求接口的配置参数',
+      parameters: {
+        type: 'object',
+        properties: {},
+        required: [],
+      },
+    },
+  },
+  {
+    type: 'function',
+    function: {
+      name: 'getAiConfigPath',
+      description: '获取ai请求接口配置的文件地址',
+      parameters: {
+        type: 'object',
+        properties: {},
+        required: [],
+      },
+    },
+  },
+  {
+    type: 'function',
+    function: {
+      name: 'executeSkill',
+      description:
+        '加载并执行指定路径的skill，以该skill提供的工具函数为基础，启动子工作流完成目标任务。skillFilePath为SKILL.md文件的绝对路径，subGoalPrompt为子工作流需要完成的目标描述，留空则仅加载skill不执行子任务。',
+      parameters: {
+        type: 'object',
+        properties: {
+          skillFilePath: {
+            type: 'string',
+            description: 'SKILL.md文件的绝对路径',
+          },
+          subGoalPrompt: {
+            type: 'string',
+            description: '子工作流需要完成的目标描述，留空则仅加载skill',
+          },
+        },
+        required: ['skillFilePath'],
+      },
+    },
+  },
+]
+const functions = {
+  executeCommand,
+  requestAI,
+  executeJSCode,
+  getExtensionFileRule,
+  getAiConfig,
+  getAiConfigPath,
+  executeSkill,
+}
+
+module.exports = {
+  name: 'SystemExtension',
+  extensionDescription: "提供系统命令执行、AI请求、JS代码执行、扩展文件生成规则、AI配置管理、Skill加载执行等核心系统功能",
+  descriptions,
+  functions,
+}

package/src/core/extension/TestExtension.js

@@ -0,0 +1,67 @@
+// 执行测试任务
+function executeTestTask(subGoalPrompt = "") {
+  // 调用子工作流完成目标
+  return this.aiService.subTestWorkflow(subGoalPrompt)
+}
+
+// 生成测试任务
+function generateTestTaskRule(goal) {
+  return `
+用户测试任务：${goal}。
+请在当前目录下生成一个Markdown格式的程序功能测试说明文档，文档标题固定为“xxx测试任务”（其中“xxx”替换为具体的程序功能测试任务名称，需与我提供的程序功能测试目标完全一致）。生成文档前，请先仔细研读我给出的程序功能测试任务目标，明确程序待测试的功能模块、核心测试需求、测试范围及测试重点，精准理解每个功能的预期表现，确保完全掌握测试任务要求后再进行文档撰写。
+
+文档核心内容需围绕程序功能测试任务目标，针对程序的每个待测试功能，制定**可逐条执行、无歧义、覆盖全面**的功能测试用例，每个测试用例需明确对应程序功能模块、具体测试场景、详细测试步骤（步骤需具体可操作，可直接供测试人员执行，明确操作路径、输入数据、操作顺序，避免模糊表述）；同时，为每个测试用例对应明确、可验证的测试期望结果，期望结果需与测试步骤一一对应，清晰界定程序功能正常运行的标准，明确功能执行后应呈现的具体效果、输出结果或状态，无模糊不清的表述。
+
+补充要求：1. 测试用例需按程序功能模块分类、按测试优先级排序，每个测试用例标注唯一编号，便于区分、执行和追溯；2. 文档格式需符合Markdown规范，标题层级清晰（文档标题为一级标题，功能模块分组用二级标题，测试用例用三级标题或有序列表），排版整洁，便于阅读、编辑和后续维护；3. 严格贴合程序功能测试任务目标，聚焦程序功能的正确性、完整性，不添加无关内容，不遗漏核心功能测试场景，不规避边界场景，确保测试用例的针对性、完整性和有效性，能充分验证程序各功能是否符合预期。
+`}
+
+const descriptions = [
+  {
+    type: "function",
+    function: {
+      name: "generateTestTaskRule",
+      description:
+        "根据用户提供的程序功能测试目标，生成一份标准化的Markdown格式功能测试说明文档的规则提示词。返回的提示词将指导AI生成包含完整测试用例（测试步骤、期望结果、用例编号）的测试文档。适用于用户需要对程序进行功能测试等场景。",
+      parameters: {
+        type: "object",
+        properties: {
+          goal: {
+            type: "string",
+            description: "程序功能测试目标描述，需明确待测试的功能模块、测试范围和测试重点",
+          },
+        },
+        required: ["goal"],
+      },
+    },
+  },
+  {
+    type: "function",
+    function: {
+      name: "executeTestTask",
+      description:
+        "执行测试任务，启动一个专用的AI子工作流来完成指定的某一项测试目标。子工作流将根据测试任务描述，自动执行测试步骤并验证测试结果。适用于需要AI自动化执行功能测试的场景。",
+      parameters: {
+        type: "object",
+        properties: {
+          subGoalPrompt: {
+            type: "string",
+            description: "测试子工作流需要完成的目标描述，描述具体的测试任务内容",
+          },
+        },
+        required: ["subGoalPrompt"],
+      },
+    },
+  },
+];
+
+const functions = {
+  executeTestTask,
+  generateTestTaskRule,
+};
+
+module.exports = {
+  name: 'TestExtension',
+  extensionDescription: "提供程序功能测试任务的生成和执行功能，支持自动化测试工作流",
+  descriptions,
+  functions,
+};

package/src/core/utils/log.js

@@ -15,6 +15,14 @@ function logError(message) {
   log(message, '#ed7f7f')
 }
 
+function logWarning(message) {
+  log(message, '#f2c94c')
+}
+
+function logDisabled(message) {
+  log(message, '#999999')
+}
+
 function writeLine(msg1, msg2 = '', color = 'blue') {
   if (color === 'blue') {
     process.stdout.write('\r' + chalk.hex('#6dd2ea')(msg1) + ' ' + msg2)
@@ -134,6 +142,8 @@ module.exports = {
   logInfo,
   logSuccess,
   logError,
+  logWarning,
+  logDisabled,
   loading,
   writeLine,
   streamOutput,