deepfish-ai 1.0.8

package/src/core/ai-services/AiWorker/AiTools.js

@@ -0,0 +1,219 @@
+/**
+ * @Author: Roman 306863030@qq.com
+ * @Date: 2026-03-17 09:12:22
+ * @LastEditors: Roman 306863030@qq.com
+ * @LastEditTime: 2026-03-17 10:03:21
+ * @FilePath: \src\core\ai-services\AiWorker\AiTools.js
+ * @Description: 对话初始化、对话请求
+ * @
+ */
+const { OpenAI } = require('openai')
+const { AiAgentSystemPrompt } = require('./AiPrompt')
+const { streamOutput, streamLineBreak, objStrToObj } = require('../../utils')
+
+// 创建client
+function createOpenAiClient(aiConfig) {
+  return new OpenAI({
+    baseURL: aiConfig.baseUrl,
+    apiKey: aiConfig.apiKey || '',
+  })
+}
+
+// 获取初始的message
+function getInitialMessages(goal) {
+  return [
+    {
+      role: 'system',
+      content: AiAgentSystemPrompt,
+    },
+    {
+      role: 'user',
+      content: goal,
+    },
+  ]
+}
+
+/**
+ * Ai单轮问答
+ * @param {*} openAiClient OpenAI客户端
+ * @param {*} aiConfig {model, temperature, maxTokens, stream}
+ * @param {*} systemDescription
+ * @param {*} prompt
+ * @param {*} temperature
+ * @returns
+ */
+async function aiRequestSingle(
+  openAiClient,
+  aiConfig,
+  systemDescription,
+  prompt,
+) {
+  const messages = []
+  messages.push({
+    role: 'system',
+    content: systemDescription,
+  })
+  messages.push({
+    role: 'user',
+    content: prompt,
+  })
+  const response = await openAiClient.chat.completions.create({
+    messages: messages,
+    ...aiConfig,
+    stream: false
+  })
+  return response.choices[0].message.content
+}
+/**
+ * Ai携带工具请求
+ * @param {*} openAiClient
+ * @param {*} aiConfig {model, temperature, maxTokens}
+ * @param {*} messages
+ * @param {*} functionDescriptions
+ * @returns
+ */
+async function aiRequestByTools(
+  openAiClient,
+  aiConfig,
+  messages,
+  functionDescriptions,
+) {
+  const response = await openAiClient.chat.completions.create({
+    messages: messages,
+    tools: functionDescriptions,
+    tool_choice: 'auto',
+    ...aiConfig,
+  })
+  if (aiConfig.stream) {
+    const messageRes = await _streamToNonStream(response)
+    return {
+      content: messageRes.choices[0].message.content,
+      tool_calls: messageRes.choices[0].message.tool_calls,
+      message: messageRes.choices[0].message,
+    }
+  }
+  return {
+    content: response.choices[0].message.content,
+    tool_calls: response.choices[0].message.tool_calls,
+    message: response.choices[0].message,
+  }
+}
+
+// 流式输出结果转非流式输出
+async function _streamToNonStream(stream) {
+  // 初始化最终响应结构（对齐 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]
+      const delta = choice.delta
+      if (!delta) {
+        continue
+      }
+      // 2. 处理普通文本内容
+      const reasoning_content = delta.reasoning_content
+      if (reasoning_content) {
+        finalResponse.choices[0].message.reasoning_content += reasoning_content
+        // 流式输出
+        streamOutput(reasoning_content)
+      }
+      const content = delta.content
+      if (content) {
+        finalResponse.choices[0].message.content += content
+        // 流式输出
+        streamOutput(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) {
+              toolCall.function.arguments += 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
+        }
+      }
+    }
+    streamLineBreak()
+    return finalResponse
+  } catch (error) {
+    console.error('流式数据转换失败：', error.message)
+    throw error // 抛出错误让上层处理
+  }
+}
+
+module.exports = {
+  createOpenAiClient,
+  aiRequestSingle,
+  aiRequestByTools,
+  getInitialMessages,
+}

package/src/core/ai-services/AiWorker/index.js

@@ -0,0 +1,88 @@
+/**
+ * @Author: Roman 306863030@qq.com
+ * @Date: 2026-03-16 09:18:05
+ * @LastEditors: Roman 306863030@qq.com
+ * @LastEditTime: 2026-03-17 10:22:46
+ * @FilePath: \cmd\src\core\ai-services\AiWorker\index.js
+ * @Description: 工作流类
+ * @
+ */
+const { logInfo } = require('../../utils')
+const AiAgent = require('./AiAgent')
+const { getInitialMessages } = require('./AiTools')
+
+class AiWorker {
+  constructor(aiCli, client) {
+    this.aiCli = aiCli
+    this.client = client
+    this.aiRecorder = this.aiCli.aiRecorder
+    this.messages = []
+    this.aiAgent = new AiAgent(
+      this.client,
+      this.aiCli.config,
+      this.aiCli.aiConfig,
+      this.aiCli.extensionManager.extensions,
+    )
+  }
+
+  async main(goal) {
+    // 判断是否回复会话
+    const isRecover = await this.aiRecorder.recover()
+    if (isRecover) {
+      const { messages } = isRecover
+      // 判断是否已经完成
+      const lastMessage = messages[messages.length - 1]
+      if (lastMessage.role === 'assistant' && !lastMessage.tool_calls) {
+        // 说明已经执行完毕，直接返回
+        this.messages = messages
+        logInfo(lastMessage.content)
+        return
+      }
+      this.messages = messages
+      this.aiAgent.aiMessageManager.reLinkMsgs(this.messages)
+      this._recoverHistory(goal, this.messages)
+    } else {
+      if (!this.messages.length) {
+        this.messages = getInitialMessages(goal)
+        this.aiAgent.work(this.messages)
+      } else {
+        this.aiAgent.aiMessageManager.reLinkMsgs(this.messages)
+        this.aiAgent.aiMessageManager.addMsg({
+          role: 'user',
+          content: goal,
+        })
+        this.aiAgent.work(this.messages)
+      }
+    }
+    // this.aiRecorder.clear()
+  }
+
+  async _recoverHistory(goal, messages) {
+    logInfo('Recovering from previous conversation...')
+    let lastMessage = messages[messages.length - 1]
+    if (lastMessage.role === 'tool') {
+      // 删除最后一项
+      messages.pop()
+    }
+    lastMessage = messages[messages.length - 1]
+    if (lastMessage.role === 'assistant' && lastMessage.tool_calls) {
+      // 最后一项正在执行工具，则重新执行
+      await this.aiAgent.execTools(lastMessage.tool_calls)
+      this.aiAgent.work(this.messages)
+    } else if (lastMessage.role === 'assistant' && lastMessage.content) {
+      return lastMessage.content || ''
+    } else if (lastMessage.role === 'user') {
+      // 最后一项是用户输入，说明是新的一轮对话
+      this.aiAgent.work(this.messages)
+    } else if (lastMessage.role === 'system') {
+      this.aiAgent.aiMessageManager.addMsg({
+        role: 'user',
+        content: goal,
+      })
+      this.aiAgent.work(this.messages)
+    }
+    return ''
+  }
+}
+
+module.exports = AiWorker

package/src/core/extension/BaseExtension.js

@@ -0,0 +1,7 @@
+const descriptions = [] // openai能识别的描述
+const functions = {} // key为函数名称，value为方法体
+
+module.exports = {
+  descriptions,
+  functions,
+}