deepfish-ai 1.0.29 → 2.0.2

package/src/AgentRobot/BaseAgentRobot/utils/AIRequest.js

@@ -1,297 +0,0 @@
-const { OpenAI } = require('openai')
-const {
-  refreshGithubModelsTokenIfNeeded,
-  buildDefaultHeaders,
-} = require('./copilot.js')
-
-function normalizeAiRequestConfig(aiConfig = {}) {
-  const max_tokens = (!aiConfig.maxTokens || aiConfig.maxTokens === -1)
-    ? undefined
-    : aiConfig.maxTokens * 1024 // 兼容旧配置，按 1KB ~= 1024 tokens 估算
-
-  const requestConfig = {
-    model: aiConfig.model,
-    temperature: aiConfig.temperature,
-    stream: aiConfig.stream,
-    max_tokens,
-  }
-
-  // 仅 DeepSeek 兼容端注入扩展字段，避免其它网关因未知参数报错
-  if (aiConfig.type === 'deepseek') {
-    requestConfig.extra_body = { reasoning_split: true }
-  }
-
-  return requestConfig
-}
-
-function creatClient(aiConfig) {
-  if (aiConfig.type === 'github-models') {
-    if (!aiConfig.apiKey || !String(aiConfig.apiKey).trim()) {
-      throw new Error(
-        'GitHub Models requires apiKey. Please set a GitHub token with models:read permission in your current AI config.',
-      )
-    }
-  }
-  if (aiConfig.type === 'copilot') {
-    if (!aiConfig.apiKey || !String(aiConfig.apiKey).trim()) {
-      throw new Error(
-        'GitHub Copilot requires apiKey. Please set a GitHub Copilot access token in your current AI config.',
-      )
-    }
-  }
-
-  return new OpenAI({
-    baseURL: aiConfig.baseUrl,
-    apiKey: aiConfig.apiKey || '',
-    defaultHeaders: buildDefaultHeaders(aiConfig),
-  })
-}
-
-/**
- * Ai单轮问答
- * @param {*} openAiClient OpenAI客户端
- * @param {*} aiConfig {model, temperature, maxTokens, stream}
- * @param {*} systemDescription
- * @param {*} prompt
- * @param {*} temperature
- * @returns
- */
-async function think(
-  openAiClient,
-  aiConfig,
-  messages,
-  thinkBefore = () => {},
-  thinkAfter = () => {},
-  streamThinkOutput,
-  streamContentOutput,
-  streamToolCallsOutput,
-  streamEnd,
-) {
-  try {
-    await refreshGithubModelsTokenIfNeeded(aiConfig)
-    const requestClient = creatClient(aiConfig)
-    const requestConfig = normalizeAiRequestConfig(aiConfig)
-    const opt = {
-      messages: messages,
-      ...requestConfig,
-    }
-    await thinkBefore()
-    const extraHeaders = aiConfig.type === 'copilot' ? { 'x-initiator': 'user' } : undefined
-    const response = await requestClient.chat.completions.create(opt, extraHeaders ? { headers: extraHeaders } : undefined)
-    if (aiConfig.stream) {
-      const messageRes = await _streamToNonStream(
-        response,
-        streamThinkOutput,
-        streamContentOutput,
-        streamToolCallsOutput,
-        streamEnd,
-      )
-      await thinkAfter()
-      return clearThinkTag(messageRes.choices[0].message.content)
-    }
-    await thinkAfter()
-    return clearThinkTag(response.choices[0].message.content)
-  } catch (error) {
-    throw new Error(`AI response error: ${error.message}`)
-  }
-}
-/**
- * Ai携带工具请求
- * @param {*} openAiClient
- * @param {*} aiConfig {model, temperature, maxTokens}
- * @param {*} messages
- * @param {*} functionDescriptions
- * @returns
- */
-async function thinkByTool(
-  openAiClient,
-  aiConfig,
-  messages,
-  functionDescriptions,
-  thinkBefore = () => {},
-  thinkAfter = () => {},
-  streamThinkOutput,
-  streamContentOutput,
-  streamToolCallsOutput,
-  streamEnd,
-) {
-  try {
-    await refreshGithubModelsTokenIfNeeded(aiConfig)
-    const requestClient = creatClient(aiConfig)
-    const requestConfig = normalizeAiRequestConfig(aiConfig)
-    await thinkBefore()
-    const extraHeaders = aiConfig.type === 'copilot' ? { 'x-initiator': 'agent' } : undefined
-    const response = await requestClient.chat.completions.create(
-      {
-        messages: messages,
-        tools: functionDescriptions,
-        tool_choice: 'auto',
-        ...requestConfig,
-      },
-      extraHeaders ? { headers: extraHeaders } : undefined,
-    )
-    if (aiConfig.stream) {
-      const messageRes = await _streamToNonStream(
-        response,
-        streamThinkOutput,
-        streamContentOutput,
-        streamToolCallsOutput,
-        streamEnd,
-      )
-      await thinkAfter()
-      messageRes.choices[0].message.content = clearThinkTag(messageRes.choices[0].message.content)
-      return {
-        content: messageRes.choices[0].message.content,
-        tool_calls: messageRes.choices[0].message.tool_calls,
-        message: messageRes.choices[0].message,
-      }
-    }
-    await thinkAfter()
-    response.choices[0].message.content = clearThinkTag(response.choices[0].message.content)
-    return {
-      content: response.choices[0].message.content,
-      tool_calls: response.choices[0].message.tool_calls,
-      message: response.choices[0].message,
-    }
-  } catch (error) {
-    throw new Error(`AI response error: ${error.message}`)
-  }
-}
-
-// 清除字符串中<think></think>标签以及之间的内容
-function clearThinkTag(content) {
-  return content.replace(/<think>[\s\S]*?<\/think>/g, '')
-}
-
-// 流式输出结果转非流式输出
-async function _streamToNonStream(
-  stream,
-  streamThinkOutput = () => {},
-  streamContentOutput = () => {},
-  streamToolCallsOutput = () => {},
-  streamEnd = () => {},
-) {
-  // 初始化最终响应结构（对齐 OpenAI 非流式响应格式）
-  const finalResponse = {
-    id: '',
-    object: 'chat.completion',
-    created: Math.floor(Date.now() / 1000), // 生成时间戳
-    model: '',
-    choices: [
-      {
-        index: 0,
-        message: {
-          role: 'assistant',
-          content: '',
-          reasoning_content: '',
-          tool_calls: [], // 存储完整的工具调用列表
-        },
-        finish_reason: null,
-        logprobs: null,
-      },
-    ],
-    usage: {
-      prompt_tokens: 0,
-      completion_tokens: 0,
-      total_tokens: 0,
-    },
-  }
-
-  // 工具调用缓冲区：处理多工具调用 + 分段参数拼接
-  const toolCallBuffers = new Map() // key: tool_call_id, value: toolCall object
-  const toolCallIndexMap = new Map() // key: index, value: tool_call_id
-  try {
-    // 遍历所有流式数据块
-    for await (const chunk of stream) {
-      // 1. 填充全局信息（仅首次获取）
-      if (!finalResponse.id) {
-        finalResponse.id = chunk.id || `chatcmpl-${Date.now()}`
-      }
-      if (!finalResponse.model) {
-        finalResponse.model = chunk.model || 'deepseek-reasoner'
-      }
-      const choice = chunk.choices[0]
-      if (!choice) {
-        continue
-      }
-      const delta = choice.delta
-      if (!delta) {
-        continue
-      }
-      // 2. 处理普通文本内容
-      const reasoning_content = delta.reasoning_content
-      if (reasoning_content) {
-        if (finalResponse.choices[0].message.reasoning_content.length === 0) {
-          streamThinkOutput('[Thinking...]')
-        }
-        finalResponse.choices[0].message.reasoning_content += reasoning_content
-        // 流式输出
-        streamThinkOutput(reasoning_content)
-      }
-      const content = delta.content
-      if (content) {
-        if (finalResponse.choices[0].message.content.length === 0) {
-          streamContentOutput('\r\n')
-        }
-        finalResponse.choices[0].message.content += content
-        // 流式输出
-        streamContentOutput(content)
-      }
-      // 3. 处理工具调用（核心逻辑）
-      if (delta.tool_calls && delta.tool_calls.length > 0) {
-        delta.tool_calls.forEach((toolCallChunk) => {
-          const index = toolCallChunk.index
-          if (toolCallChunk.id) {
-            const id = toolCallChunk.id
-            toolCallIndexMap.set(index, id)
-            let toolCall = toolCallBuffers.get(id)
-            if (!toolCall) {
-              toolCall = {
-                id: id,
-                type: toolCallChunk.type || 'function',
-                function: {
-                  name: toolCallChunk.function.name,
-                  arguments: '',
-                },
-              }
-              toolCallBuffers.set(id, toolCall)
-            }
-          } else {
-            const id = toolCallIndexMap.get(index)
-            const toolCall = toolCallBuffers.get(id)
-            if (toolCall && toolCallChunk.function?.arguments) {
-              if (toolCall.function.arguments.length === 0) {
-                streamToolCallsOutput(`\r\n[ToolCalls...][${toolCall.function.name}]`)
-              }
-              toolCall.function.arguments += toolCallChunk.function.arguments
-              streamToolCallsOutput(toolCallChunk.function.arguments)
-            }
-          }
-        })
-      }
-
-      // 4. 处理结束标记
-      if (choice.finish_reason) {
-        finalResponse.choices[0].finish_reason = choice.finish_reason
-
-        // 工具调用结束：将缓冲区数据写入最终响应
-        if (choice.finish_reason === 'tool_calls' && toolCallBuffers.size > 0) {
-          finalResponse.choices[0].message.content = '' // 工具调用时 content 为 null
-          finalResponse.choices[0].message.tool_calls = Array.from(
-            toolCallBuffers.values(),
-          )
-        } else {
-          finalResponse.choices[0].message.tool_calls = undefined
-        }
-      }
-    }
-    streamEnd()
-    return finalResponse
-  } catch (error) {
-    console.error('流式数据转换失败：', error.message)
-    throw error // 抛出错误让上层处理
-  }
-}
-
-
-module.exports = { creatClient, think, thinkByTool }

package/src/AgentRobot/BaseAgentRobot/utils/AIToolManager.js

@@ -1,131 +0,0 @@
-const path = require('path')
-const os = require('os')
-const fs = require('fs-extra')
-const dayjs = require('dayjs')
-const axios = require('axios')
-const echarts = require('echarts')
-const canvas = require('canvas')
-const cheerio = require('cheerio')
-const puppeteer = require('puppeteer')
-const lodash = require('lodash')
-const { AttachmentToolScanner } = require('./AttachmentToolScanner')
-
-class AIToolManager {
-  originalTools = null // 原装工具
-  attachTools = null // 附加工具, Agent后续安装的工具函数
-  functions = {} // key为函数名称，value为方法体
-  descriptions = [] // openai能识别的描述
-  tools = [] // 工具列表
-  toolCollection = null // 工具集合
-  clawSkillCollection = null // Claw技能集合
-  constructor(agentRobot) {
-    this.agentRobot = agentRobot
-    this.initTools(agentRobot.opt)
-  }
-  initTools(opt) {
-    this.originalTools = this._getOriginalTools() // 天赋技能
-    this.attachTools = opt.attachTools || [] // 附加工具, Agent后续安装的工具函数
-    const tools = [...this.originalTools, ...this.attachTools]
-    tools.forEach((tool) => {
-      this._addTool(tool)
-    })
-    Object.assign(this.functions, {
-      fs,
-      axios,
-      dayjs,
-      lodash,
-      canvas,
-      echarts,
-      cheerio,
-      puppeteer,
-    })
-
-    this.functions.agentRobot = this.agentRobot
-    this.functions.Tools = this.functions
-    // 兼容老版本
-    this.functions.aiCli = {
-      Tools: this.functions,
-    }
-    // 外部工具扫描
-    this.toolCollection = AttachmentToolScanner.getToolCollection(
-      this.agentRobot.workspace,
-      this.agentRobot.basespace,
-    ) // 加载工具集合
-    this.clawSkillCollection = AttachmentToolScanner.getClawSkillCollection(
-      this.agentRobot.basespace,
-    ) // 加载Claw技能集合
-    this.agentRobot.toolCollection = this.toolCollection
-    this.agentRobot.clawSkillCollection = this.clawSkillCollection
-  }
-
-  // 动态添加工具
-  addTool(toolName) {
-    let tool = this.tools.find((t) => t.name === toolName)
-    if (!tool) {
-      tool = this.toolCollection.find((t) => t.name === toolName)
-      if (tool) {
-        this._addTool(tool)
-      }
-    }
-  }
-
-  _addTool(tool) {
-    const platform = tool.platform || 'all'
-    if (platform === 'all' || platform === process.platform) {
-      tool.descriptions = tool.descriptions.map((item) => {
-        if (!item.type) {
-          item = {
-            type: 'function',
-            function: item,
-          }
-        }
-        return item
-      })
-      tool.isSystem = true
-      if (tool.name && !tool.isSystem) {
-        for (const funcName in tool.functions) {
-          if (!funcName.includes('_')) {
-            this.functions[`${tool.name}_${funcName}`] =
-              tool.functions[funcName]
-          } else {
-            this.functions[funcName] = tool.functions[funcName]
-          }
-        }
-        const descriptions = tool.descriptions.map((item) => {
-          if (
-            tool.name &&
-            item.function.name &&
-            !item.function.name.includes('_')
-          ) {
-            item.function.name = `${tool.name}_${item.function.name}`
-            return item
-          } else {
-            return item
-          }
-        })
-        this.descriptions.push(...descriptions)
-      } else {
-        Object.assign(this.functions, tool.functions)
-        this.descriptions.push(...tool.descriptions)
-      }
-      this.tools.push(tool)
-    }
-  }
-
-  // 获取原装工具
-  _getOriginalTools() {
-    // 自动扫描tools目录
-    const toolsPath = path.join(__dirname, '../tools')
-    const toolFiles = fs.readdirSync(toolsPath).filter((file) => {
-      return file.endsWith('.js') || file.endsWith('.cjs')
-    })
-    const tools = []
-    toolFiles.forEach((file) => {
-      const tool = require(path.join(toolsPath, file))
-      tools.push(tool)
-    })
-    return tools
-  }
-}
-
-module.exports = AIToolManager

package/src/AgentRobot/BaseAgentRobot/utils/AgentTree.js

@@ -1,66 +0,0 @@
-const path = require('path')
-const fs = require('fs-extra')
-
-class AgentTree {
-  content = null
-  constructor(agentRobot) {
-    this.agentRobot = agentRobot
-    this.agentTreeFilePath = path.join(
-      this.agentRobot.agentSpace,
-      'agentTree.json',
-    )
-    this.parentAgentTree = this.agentRobot.parent?.agentTree
-    this.rootAgentTree = this.agentRobot.root?.agentTree
-  }
-  init() {
-    if (!this.rootAgentTree && !fs.pathExistsSync(this.agentTreeFilePath)) {
-      fs.writeJsonSync(
-        this.agentTreeFilePath,
-        { agentId: this.agentRobot.id, children: [] },
-        { spaces: 2 },
-      )
-    }
-    if (!this.rootAgentTree) {
-      this.content = fs.readJsonSync(this.agentTreeFilePath)
-    } else {
-      if (this.parentAgentTree && this.parentAgentTree.content.children) {
-        this.content = this.parentAgentTree.content.children.find(
-          (child) => child.agentId === this.agentRobot.id,
-        )
-        if (!this.content) {
-          this.content = {
-            agentId: this.agentRobot.id,
-            children: [],
-            type: this.agentRobot.type,
-          }
-          this.parentAgentTree.content.children.push(this.content)
-          fs.writeJsonSync(this.agentTreeFilePath, this.rootAgentTree.content, {
-            spaces: 2,
-          })
-        }
-      }
-    }
-  }
-
-  clear() {
-    if (this.rootAgentTree) {
-      if (this.parentAgentTree && this.parentAgentTree.content.children) {
-        const currentNode = this.parentAgentTree.content.children.findIndex(
-          (child) => child.agentId === this.agentRobot.id,
-        )
-        if (currentNode !== -1) {
-          this.parentAgentTree.content.children.splice(currentNode, 1)
-          fs.writeJsonSync(this.agentTreeFilePath, this.rootAgentTree.content, {
-            spaces: 2,
-          })
-        }
-      }
-      // 清除子memory
-      this.agentRobot.memoryFilePath &&
-        fs.removeSync(this.agentRobot.memoryFilePath)
-    }
-  }
-}
-
-module.exports = AgentTree
-