@mybricks/plugin-ai 0.0.1 → 0.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mybricks/plugin-ai",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "main": "src/index.tsx",
   "files": [
     "src",
@@ -20,8 +20,13 @@
   "dependencies": {
     "@ant-design/icons": "^6.1.0",
     "@mybricks/rxai": "0.0.2",
+    "antd": "^4.24.16",
     "classnames": "^2.5.1",
     "jsonrepair": "^3.13.1",
-    "marked": "^17.0.0"
+    "markdown-it": "^14.1.0",
+    "node-forge": "^1.3.2"
+  },
+  "devDependencies": {
+    "@types/markdown-it": "^14.1.2"
   }
 }

package/src/agents/app.ts

@@ -1,22 +1,52 @@
 import { context } from './../context';
 import { MYBRICKS_TOOLS } from "./../tools"
 
-window.plugin_ai_context = context
+import { WorkSpace } from './workspace/workspace'
+import { FocusOutlineInfoManager, FocusInfo } from './workspace/outline-focus'
+import { getAgentConfigs } from './utils/config'
 
 export const requestGeneratePageAgent = (pageId: string, pageTitle: string, params: any) => {
 
-  const prompts = context.prompts
+  // 从 agents 配置中获取提示词配置，优先使用参数传入的，否则使用 context 中的
+  const agents = params.agents || context.agents;
+  const agentConfig = getAgentConfigs(agents, 'page');
+
+  const focusInfo: FocusInfo = {
+    pageId,
+    comId: undefined,
+    title: pageTitle,
+    type: 'page'
+  };
+
+  const outlineInfoManager = new FocusOutlineInfoManager({
+    api: context.api,
+    focusInfo
+  })
+
+  const componentIdToTitleMap = outlineInfoManager.getComponentIdToTitleMap(pageId);
+
+  const workspace = new WorkSpace({ currentFocus: focusInfo } as any, {
+    getAllPageInfo() {
+      return []
+    },
+    getComponentDoc(namespace: string) {
+      return (context.api?.global?.api?.getComEditorPrompts ?? context.api?.uiCom?.api?.getComEditorPrompts)?.(namespace)
+    }
+  } as any, outlineInfoManager)
 
   params?.onProgress?.('start')
 
   context.rxai.requestAI({
     ...params,
+    extension: {
+      mentions: [focusInfo]
+    },
     message: params?.message,
     key: pageId,
     emits: {
       write: () => { },
       complete: () => {
-        // params?.onProgress?.("complete");
+        params?.onProgress?.("complete");
       },
       error: () => {
         params?.onProgress?.("error");
@@ -25,35 +55,117 @@ export const requestGeneratePageAgent = (pageId: string, pageTitle: string, para
         // params?.onProgress?.("complete");
       },
     },
-    planList: ["generate-page-prd-and-require-component", "generate-page"],
+    planList: [`${MYBRICKS_TOOLS.AnalyzeRequirementAndComponents.toolName} -mode generate`, MYBRICKS_TOOLS.GenerateUiContent.toolName, MYBRICKS_TOOLS.BuildProcess.toolName],
     tools: [
-      MYBRICKS_TOOLS.GetComponentsDocAndPrd({
-        allowComponents: context.api?.global?.api?.getAllComDefPrompts?.(),
-        examples: prompts.prdExamplesPrompts,
-        canvasWidth: prompts.canvasWidth,
-        queryComponentsDocsByNamespaces: (namespaces) => {
-          return namespaces.reduce((acc, cur) => {
-            return acc + '\n' + context.api?.uiCom?.api?.getComEditorPrompts?.(cur.namespace)
-          }, '')
-        }
+      MYBRICKS_TOOLS.AnalyzeRequirementAndComponents({
+        allowComponents: context.designer?.getAllComDefPrompts?.() || "",
+        ...agentConfig?.getToolParams(MYBRICKS_TOOLS.AnalyzeRequirementAndComponents.toolName),
+        onComponentDocOpen: (namespace) => {
+          workspace.openComponentDoc(namespace)
+        },
+        appendPrompt: agentConfig?.attentions,
+        shouldUseExpert: true,
+        deviceType: context.deviceType,
       }),
-      MYBRICKS_TOOLS.GeneratePage({
-        getFocusRootComponentDoc: () => context.api?.page?.api?.getPageContainerPrompts?.(pageId) as string,
+      MYBRICKS_TOOLS.GenerateUiContent({
+        getRootComponentDoc: () => context.api?.page?.api?.getPageContainerPrompts?.(pageId) as string,
         getTargetId: () => pageId as string,
-        appendPrompt: prompts.systemAppendPrompts,
-        examples: prompts.generatePageActionExamplesPrompts,
+        getRootIdByPageId(pageId: string) {
+          return outlineInfoManager.getPageMetaInfo(pageId)?.rootId
+        },
+        componentIdToTitleMap,
+        appendPrompt: agentConfig?.attentions,
+        ...agentConfig?.getToolParams(MYBRICKS_TOOLS.GenerateUiContent.toolName),
         onActions: (actions, status) => {
-          context.api?.page?.api?.updatePage?.(pageId, actions, status)
-
-          if (status === 'complete') {
-            params?.onProgress?.("complete");
-          }
+          return context.designer?.updatePage?.(pageId, actions, status)
         },
         onClearPage: () => {
           context.api?.page?.api?.clearPageContent?.(pageId)
         }
       }),
+      MYBRICKS_TOOLS.BuildProcess({
+        getPageId: () => focusInfo.pageId,
+        getPageOutlineInfo: () => {
+          workspace.openDocument(focusInfo.pageId!);
+          return context.api?.page?.api?.getOutlineInfo(focusInfo.pageId)
+        },
+        getComponentOutlineInfo: () => {
+          const { type, comId } = focusInfo
+          if (type === "uiCom") {
+            return {
+              type,
+              outlineInfo: context.api?.uiCom?.api?.getOutlineInfo(comId)
+            }
+          } else if (type === "logicCom") {
+            return {
+              type,
+              outlineInfo: context.api?.logicCom?.api?.getOutlineInfo(comId)
+            }
+          }
+        },
+        getAllPageInfo() {
+          return context.api?.global?.api?.getAllPageInfo()
+        },
+        createDiagram: (...args: any) => {
+          // console.log("[createDiagram - args]", args)
+          return context.designer?.createDiagram?.(...args)
+        },
+        updateDiagram: (...args: any) => {
+          // console.log("[updateDiagram - args]", args)
+          return context.designer?.updateDiagram?.(...args)
+        },
+        getDiagramInfo: (...args: any) => {
+          if (!args[0]) {
+            if (focusInfo.diagramId) {
+              return {
+                id: focusInfo.diagramId
+              }
+            }
+
+            return null
+          }
+          return context.designer?.getDiagramInfo?.(...args)
+        },
+        updatePage: (...args: any) => {
+          // console.log("[updatePage]", args)
+          return context.designer?.updatePage?.(focusInfo.pageId, ...args)
+        },
+        updateCom: (...args: any) => {
+          // console.log("[updateCom]", args)
+          return context.designer?.updateLogicCom?.(...args)
+        },
+      }),,
     ],
+    // presetMessages: () => [
+    //   {
+    //     role: 'user',
+    //     content: workspace.getComponentsDocs()
+    //   }
+    // ],
+    presetMessages: () => {
+      const projectStruct = `# 工作空间 \n 当前为空项目`;;
+      const componentsDocs = workspace.getComponentsDocs();
+      const hasComponentsDocs = workspace.hasComponentsDocs();
+      
+      // 合并内容
+      let projectInfo = projectStruct;
+      if (hasComponentsDocs) {
+        projectInfo = `${projectStruct}\n\n${componentsDocs}`;
+      }
+      
+      return [
+        {
+          role: 'user',
+          content: `<当前项目信息>\n${projectInfo}\n</当前项目信息>`
+        },
+        {
+          role: 'assistant',
+          content: hasComponentsDocs 
+            ? '收到，谢谢你提供的项目信息和组件文档，我会根据这些信息完成任务～'
+            : '收到，谢谢你提供的项目信息，我会根据这些信息完成任务～'
+        },
+      ]
+    }
     // presetMessages: [
     //   {
     //     role: 'user',
@@ -71,7 +183,7 @@ export const requestGeneratePageAgent = (pageId: string, pageTitle: string, para
   });
 }
 
-async function createCanvasByAICanvas(canvasId: string, aiCanvas: any) {
+async function createCanvasByAICanvas(canvasId: string, aiCanvas: any, originParams: any[]) {
 
   function sleep(ms: number) {
     return new Promise(resolve => setTimeout(resolve, ms));
@@ -80,46 +192,46 @@ async function createCanvasByAICanvas(canvasId: string, aiCanvas: any) {
   // TOOD，之前不去掉设计器会有报错
   await sleep(1000)
 
-  const pageArray = [];
+  const pageArray: Array<{ page: any; pageRef: { id: string; onProgress: Function } }> = [];
 
   for (let index = 0; index < aiCanvas.pages.length; index++) {
     const page = aiCanvas.pages[index];
 
-    const pageRef = await context.api.page?.api?.createPage?.(canvasId, page.title, {
-      type: "normal",
-      title: page.title,
-      template: {
-        namespace: 'mybricks.harmony.systemPage',
-        deletable: false,
-        asRoot: true,
-        data: {
-          useTabBar: false,
-        },
-      },
-      inputs: [
-        {
-          id: "open",
-          title: "打开",
-          schema: {
-            type: "object",
-          },
-        },
-      ],
-    })
-    pageArray.push({ page, pageRef })
+    const pageRef = await context.designer?.createPage?.(canvasId, page.title, context.createTemplates?.page?.({ title: page.title }))
+    if (pageRef) {
+      pageArray.push({ page, pageRef })
+    }
   }
 
-  pageArray.forEach(async ({ page, pageRef }) => {
+  pageArray.forEach(async ({ page, pageRef }, index) => {
+    const otherPage = pageArray.slice(0, index).concat(pageArray.slice(index + 1))
     await requestGeneratePageAgent(pageRef.id, page.title, {
-      message: `标题：${page.title}
-<需求>
+      message: `帮忙实现项目「${aiCanvas.title}」的其中一个页面，页面为${page.title}。
+<可供参考的需求>
 ${page.prd}
-</需求>
+</可供参考的需求>
 
 <样式风格>
 ${aiCanvas.style}
 </样式风格>
+
+<正在实现的页面列表>
+${(pageArray ?? []).map(p => `- ${p.page.title}`).join('\n')}
+</正在实现的页面列表>
+
+<事件流程>
+当前项目有以下${otherPage.length}张页面可以跳转
+${otherPage.reduce((pre: string, { page, pageRef }: any) => {
+return pre + `- ${page.title}（${pageRef.id}）\n`
+}, "")}
+
+基于页面的名称、事实，分析页面间的关联性、组件事件中可能的页面跳转流程，例如某个按钮的点击事件跳转一个正确的页面。
+
+注意：
+1. 除页面跳转外，禁止一切其它形式的事件流程和变量。
+</事件流程>
 `,
+      attachments: originParams?.attachments ?? [],
       onProgress: pageRef.onProgress,
       id: pageRef.id,
     })
@@ -128,9 +240,22 @@ ${aiCanvas.style}
 
 export const requestGenerateCanvasAgent = (params: any) => {
   return new Promise((resolve, reject) => {
-    params?.onProgress?.('start')
+    params?.onProgress?.('start');
 
-    context.rxai.requestAI({
+
+    const createTargetContainer = async () => {
+      if (!context.isMutiCanvas) {
+        return { id: '_root_' };
+      }
+      return await context.designer?.createCanvas?.() || { id: '_root_' };
+    };
+
+    const originParams = {
+      message: params?.message,
+      attachments: params?.attachments ?? [],
+    };
+
+    (params.rxai || context.rxai).requestAI({
       ...params,
       message: params?.message,
       key: params.id,
@@ -147,25 +272,28 @@ export const requestGenerateCanvasAgent = (params: any) => {
           params?.onProgress?.("complete");
         },
       },
-      planList: ["analyze-and-expand-prd"],
+      planList: [MYBRICKS_TOOLS.AnalyzeAndExpandPrd.toolName],
       tools: [
         MYBRICKS_TOOLS.AnalyzeAndExpandPrd({
-          onProjectCreate: (projectJson) => {
+          onProjectCreate: async (projectJson) => {
             if (!projectJson || !projectJson.title) {
-              return reject('不合法的项目文件')
+              return resolve("complete");
+              // return reject('不合法的项目文件')
             }
             let canvasId
             try {
-              const { id, title } = context.api.page?.api?.createCanvas?.()
-              canvasId = id
+              const canvas = await createTargetContainer()
+              canvasId = canvas?.id
             } catch (error) {
-              reject(error)
+              return reject(error)
             }
             resolve('complete')
             if (canvasId) {
-              createCanvasByAICanvas(canvasId, projectJson);
+              params?.onProgress?.('ing');
+              createCanvasByAICanvas(canvasId, projectJson, originParams);
             }
-          }
+          },
+          deviceType: context.deviceType
         })
       ],
     });