@mybricks/plugin-ai 0.0.1 → 0.0.2

package/src/tools/coding-subagent-as-tool.ts

@@ -0,0 +1,119 @@
+import type { CodingManager } from '../agents/workspace/coding-manager'
+import { requestVibeCodingAgent } from '../agents/custom'
+
+const NAME = 'coding-subagent-as-tool'
+codingSubagentAsTool.toolName = NAME
+
+interface Config {
+  codingManager: CodingManager,
+  onStart: () => void,
+  onComplete: () => void,
+  onError: () => void,
+}
+
+function getBatchCodingPayload(codingManager: CodingManager, getUserMessage: () => string | any) {
+  const codings = [...codingManager.waitForCoding]
+  const userMessageContent = getUserMessage?.()?.content
+
+  const userText = Array.isArray(userMessageContent) ? userMessageContent?.find(m => m.type === 'text')?.text : userMessageContent
+  const message = `
+<用户需求>
+${userText}
+</用户需求>  
+# 批量组件代码开发任务
+
+## 当前页面结构
+${codingManager.getJsxById(codingManager.pageId)}
+
+## 任务说明
+需要按顺序和用户需求开发以下 ${codings.length} 个组件。
+
+## 待开发组件
+
+${codings
+  .map(
+    (coding, index) =>
+      `### 组件 ${index + 1}: ${coding.comId}
+**需求描述：**
+${coding.requirement}`
+  )
+  .join('\n\n')}
+`
+  return {
+    key: `vibe_coding_${codingManager.pageId}_${Math.random().toString(36).substring(2, 15)}`,
+    message,
+    attachments: codingManager.attachments,
+    pageId: codingManager.pageId,
+  }
+}
+
+function onDevelopModule(
+  p: { files: Array<{ fileName: string; content: string }> },
+  updateComponent: (comId: string, files: Array<{ fileName: string; content: string }>) => void
+) {
+  const { files } = p
+  const filesByUuid = files.reduce(
+    (acc, file) => {
+      const match = file.fileName.match(/^(.+)@([^.]+)(\..+)$/)
+      if (match) {
+        const [, name, uuid, ext] = match
+        if (!acc[uuid]) acc[uuid] = []
+        acc[uuid].push({ fileName: `${name}${ext}`, content: file.content })
+      }
+      return acc
+    },
+    {} as Record<string, Array<{ fileName: string; content: string }>>
+  )
+  Object.entries(filesByUuid).forEach(([uuid, componentFiles]) => {
+    updateComponent(uuid, componentFiles)
+  })
+}
+
+export default function codingSubagentAsTool(config: Config): any {
+  return {
+    name: NAME,
+    displayName: '代码开发',
+    description: `执行「AI区域开发」：当生成页面时添加了需要开发代码的AI组件时，调用智能组件助手批量开发/还原这些组件的代码。无需规划此工具，此工具会自行调用。`,
+    async execute({ getUserMessage }) {
+      const { codingManager, onStart, onComplete, onError } = config
+      if (!codingManager.waitForCoding.length) {
+        return {
+          llmContent: '当前没有需要开发的组件。',
+          displayContent: '当前没有需要开发的组件。',
+        }
+      }
+      onStart?.();
+      const payload = getBatchCodingPayload(codingManager, getUserMessage)
+      return new Promise<{ llmContent: string; displayContent: string }>((resolve, reject) => {
+        const onProgress = (status: string) => {
+          if (status === 'complete') {
+            onComplete?.();
+            resolve({ llmContent: '代码开发已完成。', displayContent: '代码开发已完成。' })
+          }
+          if (status === 'error') {
+            onError?.();
+            reject(new Error('开发出问题了'))
+          }
+        }
+        const promise = requestVibeCodingAgent(
+          {
+            ...payload,
+            onProgress,
+            onDevelopModule,
+            asTool: true,
+          },
+          { pageId: payload.pageId }
+        )
+        if (!promise) {
+          onError?.();
+          reject(new Error('此工具暂不支持调用'))
+          return
+        }
+        promise.catch((error) => {
+          onError?.();
+          reject(error);
+        })
+      })
+    },
+  }
+}