@blocklet/pages-kit-agents 0.5.47 → 0.5.48

package/lib/types/utils/agent-utils.d.ts

@@ -1,3 +1,7 @@
+import { FSMemory, FSMemoryOptions } from '@aigne/agent-library/fs-memory/index.js';
 import { AnthropicChatModel } from '@aigne/anthropic';
 import { OpenAIChatModel } from '@aigne/openai';
 export declare function getModel(modelName?: string): OpenAIChatModel | AnthropicChatModel | undefined;
+export declare function getMemory({ projectId, ...rest }: Pick<FSMemoryOptions, 'autoUpdate'> & {
+    projectId?: string;
+}): FSMemory;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blocklet/pages-kit-agents",
-  "version": "0.5.47",
+  "version": "0.5.48",
   "description": "AIGNE based agents for pages kit",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/blocklet/pages-kit#readme",
@@ -32,22 +32,27 @@
     "@aigne/xai": "^0.2.7",
     "@types/js-yaml": "^4.0.9",
     "@types/yargs": "^17.0.33",
+    "fastq": "^1.19.1",
     "inquirer": "^12.5.2",
     "js-yaml": "^4.1.0",
     "open": "^10.1.2",
     "yargs": "^17.7.2",
     "zod": "^3.24.2",
-    "@blocklet/pages-kit-block-studio": "0.5.47"
+    "@blocklet/pages-kit-block-studio": "0.5.48"
+  },
+  "devDependencies": {
+    "copyfiles": "^2.4.1"
   },
   "scripts": {
     "start": "npx -y bun run cli.ts",
     "workflow": "npx -y bun run workflow-cli.ts",
     "lint": "eslint --ext .mjs,.js,.jsx,.ts,.tsx",
     "lint:fix": "npm run lint -- --fix",
-    "build": "run-p build:*",
+    "build": "run-s build:* copy-prompts",
     "build:cjs": "tsc --module commonjs --outDir lib/cjs",
     "build:esm": "tsc --module es2022 --outDir lib/esm",
     "build:types": "tsc --declaration --emitDeclarationOnly --outDir lib/types",
+    "copy-prompts": "copyfiles \"prompts/**/*\" lib/cjs/ && copyfiles \"prompts/**/*\" lib/esm/",
     "dev": "run-p 'build:* -- -w'",
     "clean": "rimraf lib"
   }

package/workflow-cli.ts

@@ -1,24 +1,20 @@
-import { AIGNE, ProcessMode, TeamAgent } from '@aigne/core';
+import { AIGNE } from '@aigne/core';
 import { logger as aigneLogger, LogLevel } from '@aigne/core/utils/logger.js';
 import yaml from 'js-yaml';
 import fs from 'node:fs/promises';
 import path from 'node:path';
 import { z } from 'zod';
 
-import { designerAgent } from './agents/template-designer.js';
-import { generatorAgentWithReflection, saveYamlAgent } from './agents/template-generator.js';
-import { readFileByPath } from './utils/file-utils.js';
+// import { OrchestratorAgent } from './agents/orchestrator/index.js';
 import { getModel } from './utils/index.js';
 import { logger } from './utils/logger.js';
-import { contentCompletenessEvaluatorAgent } from './workflow-agents/content-completeness-evaluator.js';
-import { contentCompletionWorkflowAgent } from './workflow-agents/content-completion-workflow-agent.js';
-import { parseDataSourcesAgent } from './workflow-agents/parse-datasources.js';
-import { UserFeedbackAgent } from './workflow-agents/user-feedback-agent.js';
+// import { siteHtmlBatchGeneratorAgent } from './workflow-agents/site-html-batch-generator-agent.js';
+import { siteMapGeneratorAgent } from './workflow-agents/site-map-generator.js';
 
 aigneLogger.level = LogLevel.DEBUG;
 
 const DEFAULT_LOCALE = 'zh';
-const DEFAULT_MODEL_NAME = 'gpt-4o';
+const DEFAULT_MODEL_NAME = 'gpt-4o-mini';
 const DEFAULT_PROJECT_ID = 'DEFAULT_PROJECT_ID';
 
 const ConfigSchema = z.object({
@@ -26,6 +22,7 @@ const ConfigSchema = z.object({
   locale: z.string(),
   modelName: z.string(),
   projectId: z.string(),
+  question: z.string(),
 });
 
 type WorkflowConfig = z.infer<typeof ConfigSchema>;
@@ -47,11 +44,13 @@ async function runWorkflow({
   locale,
   modelName,
   projectId,
+  question,
 }: {
   inputInfo: string;
   locale: string;
   modelName: string;
   projectId: string;
+  question: string;
 }) {
   logger.info('inputInfo', inputInfo);
   logger.info('locale', locale);
@@ -63,36 +62,58 @@ async function runWorkflow({
     model,
   });
 
-  const teamAgent = TeamAgent.from({
-    name: 'generate web site',
-    inputSchema: z.object({
-      rawData: z.string(),
-      locale: z.string(),
-      projectId: z.string(),
-    }),
-    mode: ProcessMode.sequential,
-    skills: [
-      parseDataSourcesAgent,
-      contentCompletenessEvaluatorAgent,
-      contentCompletionWorkflowAgent,
-      new UserFeedbackAgent({
-        name: 'template designer',
-        executor: designerAgent,
-        isApproved: () => false,
-        getShowMessage: ({ $message, ascii }) => `🤖：${$message}\n\n${ascii}`,
-      }),
-      generatorAgentWithReflection,
-      saveYamlAgent,
-    ],
-  });
-
-  const result = await engine.invoke(teamAgent, {
-    rawData: inputInfo,
-    locale,
-    projectId,
-  });
+  // const agent = OrchestratorAgent.from({
+  //   skills: [siteMapGeneratorAgent, siteHtmlBatchGeneratorAgent],
+  //   maxIterations: 3,
+  //   tasksConcurrency: 1,
+  //   chainStepResults: true,
+  // });
+
+  // const result = await engine.invoke(
+  //   agent,
+  //   `请根据用户的提问，分析用户希望创建的站点的主题，生成这个主题相关的 SiteMap。
+  //    然后遍历这些页面，生成每个页面的 HTML 内容。
+
+  //    用户的问题是：${question}
+  //    用户希望创建的站点的语言是：${locale}
+  //    用户希望创建的站点的项目ID是：${projectId}
+  //   `
+  // );
+
+  // const teamAgent = TeamAgent.from({
+  //   name: 'generate web site',
+  //   inputSchema: z.object({
+  //     question: z.string(),
+  //     locale: z.string(),
+  //     projectId: z.string(),
+  //   }),
+  //   mode: ProcessMode.sequential,
+  //   skills: [siteMapGeneratorAgent],
+  // });
+
+  const stream = await engine.invoke(
+    siteMapGeneratorAgent,
+    {
+      question,
+      locale,
+      projectId,
+    },
+    {
+      streaming: true,
+    }
+  );
+
+  let response = '';
+  process.stdout.write('\n');
+  for await (const chunk of stream) {
+    if (chunk?.delta?.text?.$message) {
+      response += chunk.delta.text.$message;
+      process.stdout.write(chunk.delta.text.$message);
+    }
+  }
+  process.stdout.write('\n\n');
 
-  logger.info('result', result);
+  logger.info('result', response);
 }
 
 // 主函数
@@ -100,14 +121,13 @@ async function main() {
   try {
     logger.info('🔧 读取 workflow-config.yaml 配置...');
     const config = await loadConfig();
-    logger.info('📚 读取原始信息...');
-    const inputInfo = await readFileByPath(config.inputInfoPath);
 
     await runWorkflow({
-      inputInfo,
+      inputInfo: '',
       locale: config.locale || DEFAULT_LOCALE,
       modelName: config.modelName || DEFAULT_MODEL_NAME,
       projectId: config.projectId || DEFAULT_PROJECT_ID,
+      question: config.question,
     });
   } catch (error) {
     logger.error(`工作流执行失败: ${error instanceof Error ? error.message : String(error)}`);

package/lib/cjs/agents/page-structure-validator.d.ts

@@ -1,30 +0,0 @@
-import { AIAgent } from '@aigne/core';
-export interface ValidatePageStructureInput {
-    context: string;
-    structurePlan: Array<{
-        sectionName: string;
-        sectionContentPlan: string;
-    }>;
-    expectedSections: string;
-    outputSchema: string;
-}
-export declare const pageStructureValidatorAgent: AIAgent<{
-    context: string;
-    question: string;
-    outputSchema: string;
-    expectedSections: string;
-    structurePlan: {
-        sectionName: string;
-        sectionContentPlan: string;
-    }[];
-}, {
-    validationFeedback: {
-        isValid: boolean;
-        missingSections: string[];
-        duplicateContent: {
-            description: string;
-            sections: string[];
-            score: number;
-        }[];
-    };
-}>;

package/lib/cjs/agents/page-structure-validator.js

@@ -1,124 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.pageStructureValidatorAgent = void 0;
-const core_1 = require("@aigne/core");
-const zod_1 = require("zod");
-const logger_js_1 = require("../utils/logger.js");
-/**
- * 验证页面结构规划
- * 检查是否包含所有必要section，以及是否存在信息重复
- */
-const prompt = `您是一名页面结构规划审核专家，擅长分析内容规划的完整性和有效性。
-
-您的职责
-1. 检查页面结构规划是否涵盖了所有必要的section
-2. 分析各section内容规划之间是否存在内容重复
-3. 确保内容规划与section描述相符
-4. 提出需要改进的具体方向
-
-## 输入上下文:
-{{context}}
-
-## 用户指令:
-{{question}}
-
-## 初始结构规划
-{{initialStructurePlan}}
-
-## 页面结构规划
-{{structurePlan}}
-
-## 预期包含的Sections
-{{expectedSections}}
-
-## 输出模式（包含section描述）
-{{outputSchema}}
-
-请按照以下流程工作：
-1. 缺失项检查：检查是否所有预期 section 都在结构规划中出现，若缺失，请列出具体缺失项
-2. 重复项检查（请严格遵循以下判断标准）：
-    仅当同时满足以下所有条件时，才应将两个 section 判定为重复：
-
-    - 两个 section 中的信息核心完全一致，所讲述的功能点在语义上为同一项；
-    - 内容的细节程度相近，未体现出明显的层级差异（如简要 vs. 深度展开）；
-    - 两者目的用途相同（例如都是介绍某个功能，而不是一个概览一个详细讲解）；
-    - 信息之间缺乏新增认知价值或应用角度。
-
-    否则，即使关键词或主题类似，也不应判定为重复。
-
-   - 以下情况 **不构成重复**，请避免误判：
-     2.1 Section A 是简要罗列某些信息，Section B 对该信息进行详细展开；
-     2.2 两个 section 提及了相同的信息，但该信息在两个 section 中都不是核心内容；
-     2.3 两个 section 使用了相同的信息或关键词，但服务于不同的核心功能或描述重点。     
-
-      【重复示例】
-      - Section A: 简要展示 5 个工作流特性（并发、顺序、路由等）
-      - Section B: 逐一讲解这 5 个工作流的结构与代码
-      → 非重复（简要 vs. 详细）
-
-      - Section A: 展示并发、顺序、路由工作流的用法
-      - Section B: 以不同方式再次展示这 3 个相同工作流及其用途
-      → 可判定为重复（内容本质一致）
-
-    特别注意：
-    - 如 FeatureList 是特性的简要展示，而 FeatureDetailSection 是详细介绍该特性的实际应用场景，这种搭配为常见的“概览 + 深度讲解”模式，**属于结构合理、信息递进，不应视为重复**。
-
-
-3. 检查上下文中主要的信息在规划中是否有缺失
-4. 检查section规划是否与section描述（如果有）相符
-5. 如果 section 中包含列表数据，检查描述中是否指定具体的数量，指定的数量是否尽可能包含了上下文中所有相关数据。
-6. 如果初始结构规划中包含的section，请根据初始结构规划中的内容规划，保持初始结构规划中的内容规划。
-
-请注意，您的职责是审核和提供反馈，而不是重写或修改结构规划。`;
-logger_js_1.logger.info('page structure validator prompt', prompt);
-exports.pageStructureValidatorAgent = core_1.AIAgent.from({
-    name: 'Page Structure Validator',
-    subscribeTopic: 'validation_request',
-    publishTopic: (output) => {
-        logger_js_1.logger.info('page structure validator feedback', JSON.stringify(output.validationFeedback));
-        const isInvalid = output.validationFeedback.missingSections?.length > 0 ||
-            (output.validationFeedback.duplicateContent?.length > 0 &&
-                output.validationFeedback.duplicateContent.some((item) => item.score > 75));
-        return isInvalid ? 'revision_request' : core_1.UserOutputTopic;
-    },
-    instructions: prompt,
-    inputSchema: zod_1.z.object({
-        structurePlan: zod_1.z.array(zod_1.z.object({
-            sectionName: zod_1.z.string(),
-            sectionContentPlan: zod_1.z.string(),
-        })),
-        expectedSections: zod_1.z.string(),
-        outputSchema: zod_1.z.string(),
-        context: zod_1.z.string(),
-        question: zod_1.z.string(),
-    }),
-    outputSchema: zod_1.z.object({
-        validationFeedback: zod_1.z.object({
-            isValid: zod_1.z.boolean().describe(`
-            当出现下面两种场景的时候返回 false：
-            1. 存在严重重复的内容，duplicateContent 中存在重复内容得分大于 75 的场景
-            2. 缺失了重要的 section，missingSections 中存在缺失的 section
-
-            请特别注意：
-            - 仅当重复评分超过 75 且符合所有重复标准时，才可将 isValid 设为 false。
-
-            其他场景都返回 true
-            `),
-            missingSections: zod_1.z.array(zod_1.z.string()).describe('缺失的section列表'),
-            duplicateContent: zod_1.z
-                .array(zod_1.z.object({
-                sections: zod_1.z.array(zod_1.z.string()).describe('存在内容重复的section列表'),
-                description: zod_1.z.string().describe('内容重复的描述'),
-                score: zod_1.z.number().describe(`
-                【重复判断参考评分】
-                0~30：结构或信息逻辑相近，但内容展开和聚焦点完全不同
-                31~60：存在信息重合，但表达方式和内容深度有明显差异
-                61~75：较大信息重叠但结构差异仍存在（注意此时仍不算“严重重复”）
-                76~100：表达几乎一致，属于严重重复内容
-              `),
-            }))
-                .describe('存在内容重复的section组'),
-        }),
-    }),
-    includeInputInOutput: true,
-});

package/lib/esm/agents/page-structure-validator.d.ts

@@ -1,30 +0,0 @@
-import { AIAgent } from '@aigne/core';
-export interface ValidatePageStructureInput {
-    context: string;
-    structurePlan: Array<{
-        sectionName: string;
-        sectionContentPlan: string;
-    }>;
-    expectedSections: string;
-    outputSchema: string;
-}
-export declare const pageStructureValidatorAgent: AIAgent<{
-    context: string;
-    question: string;
-    outputSchema: string;
-    expectedSections: string;
-    structurePlan: {
-        sectionName: string;
-        sectionContentPlan: string;
-    }[];
-}, {
-    validationFeedback: {
-        isValid: boolean;
-        missingSections: string[];
-        duplicateContent: {
-            description: string;
-            sections: string[];
-            score: number;
-        }[];
-    };
-}>;

package/lib/esm/agents/page-structure-validator.js

@@ -1,121 +0,0 @@
-import { AIAgent, UserOutputTopic } from '@aigne/core';
-import { z } from 'zod';
-import { logger } from '../utils/logger.js';
-/**
- * 验证页面结构规划
- * 检查是否包含所有必要section，以及是否存在信息重复
- */
-const prompt = `您是一名页面结构规划审核专家，擅长分析内容规划的完整性和有效性。
-
-您的职责
-1. 检查页面结构规划是否涵盖了所有必要的section
-2. 分析各section内容规划之间是否存在内容重复
-3. 确保内容规划与section描述相符
-4. 提出需要改进的具体方向
-
-## 输入上下文:
-{{context}}
-
-## 用户指令:
-{{question}}
-
-## 初始结构规划
-{{initialStructurePlan}}
-
-## 页面结构规划
-{{structurePlan}}
-
-## 预期包含的Sections
-{{expectedSections}}
-
-## 输出模式（包含section描述）
-{{outputSchema}}
-
-请按照以下流程工作：
-1. 缺失项检查：检查是否所有预期 section 都在结构规划中出现，若缺失，请列出具体缺失项
-2. 重复项检查（请严格遵循以下判断标准）：
-    仅当同时满足以下所有条件时，才应将两个 section 判定为重复：
-
-    - 两个 section 中的信息核心完全一致，所讲述的功能点在语义上为同一项；
-    - 内容的细节程度相近，未体现出明显的层级差异（如简要 vs. 深度展开）；
-    - 两者目的用途相同（例如都是介绍某个功能，而不是一个概览一个详细讲解）；
-    - 信息之间缺乏新增认知价值或应用角度。
-
-    否则，即使关键词或主题类似，也不应判定为重复。
-
-   - 以下情况 **不构成重复**，请避免误判：
-     2.1 Section A 是简要罗列某些信息，Section B 对该信息进行详细展开；
-     2.2 两个 section 提及了相同的信息，但该信息在两个 section 中都不是核心内容；
-     2.3 两个 section 使用了相同的信息或关键词，但服务于不同的核心功能或描述重点。     
-
-      【重复示例】
-      - Section A: 简要展示 5 个工作流特性（并发、顺序、路由等）
-      - Section B: 逐一讲解这 5 个工作流的结构与代码
-      → 非重复（简要 vs. 详细）
-
-      - Section A: 展示并发、顺序、路由工作流的用法
-      - Section B: 以不同方式再次展示这 3 个相同工作流及其用途
-      → 可判定为重复（内容本质一致）
-
-    特别注意：
-    - 如 FeatureList 是特性的简要展示，而 FeatureDetailSection 是详细介绍该特性的实际应用场景，这种搭配为常见的“概览 + 深度讲解”模式，**属于结构合理、信息递进，不应视为重复**。
-
-
-3. 检查上下文中主要的信息在规划中是否有缺失
-4. 检查section规划是否与section描述（如果有）相符
-5. 如果 section 中包含列表数据，检查描述中是否指定具体的数量，指定的数量是否尽可能包含了上下文中所有相关数据。
-6. 如果初始结构规划中包含的section，请根据初始结构规划中的内容规划，保持初始结构规划中的内容规划。
-
-请注意，您的职责是审核和提供反馈，而不是重写或修改结构规划。`;
-logger.info('page structure validator prompt', prompt);
-export const pageStructureValidatorAgent = AIAgent.from({
-    name: 'Page Structure Validator',
-    subscribeTopic: 'validation_request',
-    publishTopic: (output) => {
-        logger.info('page structure validator feedback', JSON.stringify(output.validationFeedback));
-        const isInvalid = output.validationFeedback.missingSections?.length > 0 ||
-            (output.validationFeedback.duplicateContent?.length > 0 &&
-                output.validationFeedback.duplicateContent.some((item) => item.score > 75));
-        return isInvalid ? 'revision_request' : UserOutputTopic;
-    },
-    instructions: prompt,
-    inputSchema: z.object({
-        structurePlan: z.array(z.object({
-            sectionName: z.string(),
-            sectionContentPlan: z.string(),
-        })),
-        expectedSections: z.string(),
-        outputSchema: z.string(),
-        context: z.string(),
-        question: z.string(),
-    }),
-    outputSchema: z.object({
-        validationFeedback: z.object({
-            isValid: z.boolean().describe(`
-            当出现下面两种场景的时候返回 false：
-            1. 存在严重重复的内容，duplicateContent 中存在重复内容得分大于 75 的场景
-            2. 缺失了重要的 section，missingSections 中存在缺失的 section
-
-            请特别注意：
-            - 仅当重复评分超过 75 且符合所有重复标准时，才可将 isValid 设为 false。
-
-            其他场景都返回 true
-            `),
-            missingSections: z.array(z.string()).describe('缺失的section列表'),
-            duplicateContent: z
-                .array(z.object({
-                sections: z.array(z.string()).describe('存在内容重复的section列表'),
-                description: z.string().describe('内容重复的描述'),
-                score: z.number().describe(`
-                【重复判断参考评分】
-                0~30：结构或信息逻辑相近，但内容展开和聚焦点完全不同
-                31~60：存在信息重合，但表达方式和内容深度有明显差异
-                61~75：较大信息重叠但结构差异仍存在（注意此时仍不算“严重重复”）
-                76~100：表达几乎一致，属于严重重复内容
-              `),
-            }))
-                .describe('存在内容重复的section组'),
-        }),
-    }),
-    includeInputInOutput: true,
-});

package/lib/types/agents/page-structure-validator.d.ts

@@ -1,30 +0,0 @@
-import { AIAgent } from '@aigne/core';
-export interface ValidatePageStructureInput {
-    context: string;
-    structurePlan: Array<{
-        sectionName: string;
-        sectionContentPlan: string;
-    }>;
-    expectedSections: string;
-    outputSchema: string;
-}
-export declare const pageStructureValidatorAgent: AIAgent<{
-    context: string;
-    question: string;
-    outputSchema: string;
-    expectedSections: string;
-    structurePlan: {
-        sectionName: string;
-        sectionContentPlan: string;
-    }[];
-}, {
-    validationFeedback: {
-        isValid: boolean;
-        missingSections: string[];
-        duplicateContent: {
-            description: string;
-            sections: string[];
-            score: number;
-        }[];
-    };
-}>;