@blocklet/pages-kit-agents 0.5.14 → 0.5.16

package/.env.local.example

@@ -5,4 +5,7 @@ DEBUG=aigne:*
 OPENAI_API_KEY="" # Your OpenAI API key
 GEMINI_API_KEY="" 
 CLAUDE_API_KEY=""
+XAI_API_KEY=""
+DEEPSEEK_API_KEY=""
+
 OPENAI_BASE_URL=https://openai.arcblock.io/v1

package/cli.ts

@@ -7,11 +7,21 @@ import yargs from 'yargs';
 import { hideBin } from 'yargs/helpers';
 import { z } from 'zod';
 
+import { contentReviewerAgent } from './agents/content-reviwer.js';
 import generatePageContentMultiAgent, { type GeneratePageContentInput } from './agents/multi-agent-page-writer.js';
 import { sampleHtmlGeneratorAgent } from './agents/sample-html-generator.js';
-import { generateHtmlFromTemplate, getModel, readFile, saveHtmlContent, updateFileList } from './utils/index.js';
+import {
+  generateHtmlFromTemplate,
+  getModel,
+  readFile,
+  saveHtmlContent,
+  updateFileList,
+  saveJsonContent,
+} from './utils/index.js';
 import { logger } from './utils/logger.js';
 
+const REVIEW_MODEL_NAME = 'gemini-2.5-pro-preview-05-06';
+
 // 定义配置文件的结构
 const ConfigSchema = z.object({
   models: z.array(
@@ -23,6 +33,7 @@ const ConfigSchema = z.object({
   outputSettings: z.object({
     useTemplate: z.boolean().default(true),
     openBrowserAfterComplete: z.boolean().default(true),
+    locale: z.string().default('zh'),
   }),
   logging: z.object({
     showProgress: z.boolean().default(true),
@@ -42,6 +53,7 @@ const defaultConfig: Config = {
   outputSettings: {
     useTemplate: true,
     openBrowserAfterComplete: true,
+    locale: 'zh',
   },
   logging: {
     showProgress: true,
@@ -126,19 +138,55 @@ const generatePageContent = async (
   }
 
   if (showProgress) {
-    logger.info(`[${modelName}] Step 1/2: Generating content structure...`);
+    logger.info(`[${modelName}] Step 1/3: Generating content structure...`);
   }
 
-  const result = await generatePageContentMultiAgent(input);
+  let result = null;
+  let retryCount = 0;
+  const maxRetries = 2; // 最多重试2次，加上初始尝试共3次
+
+  while (retryCount <= maxRetries) {
+    try {
+      // eslint-disable-next-line no-await-in-loop
+      result = await generatePageContentMultiAgent(input);
+
+      if (result) {
+        break;
+      } else {
+        throw new Error('Empty result from generatePageContentMultiAgent');
+      }
+    } catch (error) {
+      retryCount++;
+      if (retryCount > maxRetries) {
+        throw new Error(
+          `Failed to generate page content after ${maxRetries} retries: ${error instanceof Error ? error.message : String(error)}`
+        );
+      }
+
+      if (showProgress) {
+        logger.warn(`[${modelName}] Content generation failed, retrying (${retryCount}/${maxRetries})...`);
+      }
+
+      // 增加短暂延迟后重试
+      // eslint-disable-next-line no-await-in-loop
+      await new Promise((resolve) => {
+        setTimeout(resolve, 1000);
+      });
+    }
+  }
 
   if (!result) {
     throw new Error('Failed to generate page content');
   }
 
   if (showProgress) {
-    logger.info(`[${modelName}] Step 2/2: Creating HTML output...`);
+    logger.info(`[${modelName}] Step 2/3: Creating HTML output...`);
   }
 
+  const model = getModel(REVIEW_MODEL_NAME);
+  const engine = new AIGNE({
+    model,
+  });
   if (outputUseTemplate) {
     await generateHtmlFromTemplate({
       jsonData: result,
@@ -146,11 +194,6 @@ const generatePageContent = async (
       outputFileName: modelName,
     });
   } else {
-    const model = getModel('gemini-2.5-pro-preview-05-06');
-    const engine = new AIGNE({
-      model,
-    });
-
     const htmlResult = await engine.invoke(sampleHtmlGeneratorAgent, {
       jsonData: JSON.stringify(result),
     });
@@ -163,6 +206,26 @@ const generatePageContent = async (
     });
   }
 
+  if (showProgress) {
+    logger.info(`[${modelName}] Step 3/3: Reviewing content...`);
+  }
+  const reviewResult = await engine.invoke(contentReviewerAgent, {
+    dataSource: input.context,
+    content: JSON.stringify(result),
+    question: '请对输入的文案进行严格的打分',
+    locale: input.locale,
+  });
+
+  // 保存评审结果到JSON文件
+  await saveJsonContent({
+    jsonContent: reviewResult,
+    outputFolder: runTime,
+    outputFileName: modelName,
+  });
+  logger.info(
+    `save review result to ${path.join(process.cwd(), 'sample-output', runTime, `${modelName}-review.json`)}`
+  );
+
   if (showProgress) {
     logger.info(`[${modelName}] HTML output completed`);
   }
@@ -198,6 +261,8 @@ async function main() {
     },
   };
 
+  logger.info('effective config: ', effectiveConfig);
+
   // 如果指定了单个模型，则只运行该模型
   const modelsToRun = args.model
     ? [{ name: String(args.model), enabled: true }]
@@ -227,7 +292,7 @@ ${imagesContext}`;
 尽可能多的包含 context 中相关的信息。
 `;
 
-  const locale = 'zh';
+  const locale = effectiveConfig.outputSettings.locale || 'zh';
 
   const input = {
     context,
@@ -336,9 +401,14 @@ ${imagesContext}`;
     }
   }
 
+  logger.info('All models completed');
+
+  logger.info('Updating file list...');
   // 更新文件列表
   await updateFileList();
 
+  logger.info('All tasks completed');
+
   // 如果配置启用了浏览器自动打开，则打开index.html
   if (effectiveConfig.outputSettings.openBrowserAfterComplete) {
     const indexPath = path.join(process.cwd(), 'sample-output', 'index.html');

package/lib/cjs/agents/content-reviwer.d.ts

@@ -0,0 +1,19 @@
+import { AIAgent } from '@aigne/core';
+export declare const contentReviewerAgent: AIAgent<{
+    question: string;
+    locale: string;
+    dataSource: string;
+    content: string;
+}, {
+    result: {
+        total_score: number;
+        level: string;
+        suggestions: string[];
+        dimension_scores: {
+            name: string;
+            score: number;
+            maxScore: number;
+            comment: string;
+        }[];
+    };
+}>;

package/lib/cjs/agents/content-reviwer.js

@@ -0,0 +1,154 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.contentReviewerAgent = void 0;
+const core_1 = require("@aigne/core");
+const zod_1 = require("zod");
+const prompt = `
+你是一个专业的网页文案评估专家，具备丰富的多行业文案分析经验。现在你将对一段网站文案进行全面评估，这些文案以结构化 JSON 格式提供，每一个字段对应网页中的一部分内容，主要字段包括：
+
+- title: 页面主标题
+- description: 页面描述
+- sectionsData: 页面内容部分，由多个模块（如 heroSection、featuresSection、faqSection 等）组成，每个模块有自己的标题、描述、详细内容等字段
+
+目标是：帮助创作者发现文案中的逻辑、表达、结构或营销短板，从而优化出更具影响力和传播力的内容。
+
+** 不要轻易给出高分**，除非你真的觉得提供的文案十分优秀。
+
+你可以参考提供原始数据来了解产品信息，帮助你判断文案中的信息是否准确、完整。
+
+---
+
+### 打分区间说明（满分 100 分）
+
+- **95-100 卓越级**：极具说服力与感染力，逻辑、节奏、表现力、SEO 全维度优秀，能直接用于投放或上线。
+- **90-94 优秀级**：表达清晰有力，有一定记忆点，结构严谨，整体完成度高，稍作优化即可达到专业发布水平。
+- **85-89 合格上乘**：表达准确、结构清楚，但较中规中矩，缺乏差异化或语言张力，建议提升亮点与传播力。
+- **80-84 可用但平庸**：内容基本清晰，但语言或结构平淡，缺乏吸引力或说服力，有明显可优化之处。
+- **70-79 结构与逻辑需优化**：存在信息遗漏、表达冗长或语言生硬等问题，可能影响理解或转化。
+- **<70 不达标**：表达混乱、价值不清、逻辑缺失，建议重写或从受众视角重新构思。
+
+---
+
+### 评估维度（共 7 项）
+
+** 每个维度的分数不能超过最大分数 **
+
+1. **信息完整性（15分）**  
+   - 是否明确传达核心价值或主张？
+   - 是否覆盖了“是什么、能做什么、为什么值得关注”？
+
+2. **结构与逻辑清晰度（15分）**  
+   - sectionsData 中的数据表示页面中的 Section ，页面结构是否完整清晰？
+   - 结构是否符合读者的阅读预期（如金字塔结构、FAB 模式等）？
+
+3. **语言表达质量（15分）**  
+   - 表达是否简练、自然、口语/书面语风格合适？
+   - 是否避免术语堆砌、空话套话？
+
+4. **吸引力与情绪感染力（15分）**  
+   - 是否有记忆点或引发兴趣？
+   - 是否有利益点、情绪驱动或视觉画面感？
+
+5. **品牌调性与目标受众匹配度（15分）**  
+   - 是否符合品牌语气？
+   - 是否贴合目标受众的心理与语境？
+
+6. **差异化与独特性（10分）**  
+   - 是否展现出区别于竞品的亮点或立场？
+   - 是否传递出“为什么选我”？
+
+7. **SEO 友好度（15分）**  
+   - 是否包含明确关键词？
+   - 是否具备良好的标题、副标题与语义层级？
+   - 是否利于搜索引擎抓取和用户点击？
+
+---
+
+### 输出格式要求：
+
+1. 每个维度请给出【分数（x/总分）】+ 简洁点评；
+2. 总结总体得分（满分 100）；
+3. 提供 3-5 条具体、可操作的优化建议，不能泛泛而谈；
+4. 如内容已接近满分，也需说明“进一步打磨”的方向。
+
+### 📦 输出格式要求（结构化）
+
+请使用以下结构输出评分结果：
+** 下面的 json 只是示例，只参考输出格式，不能影响评估结果**
+只返回 json 不要包含其他描述信息。
+
+\`\`\`json
+{
+  "total_score": xx,
+  "level": "xx",
+  "dimension_scores": [
+    {
+      "name": "信息完整性",
+      "score": xx,
+      "maxScore": xx,
+      "comment": "xxxxxx"
+    },
+    {
+      "name": "结构与逻辑清晰度",
+      "score": xx,
+      "maxScore": xx,
+      "comment": "xx"
+    },
+    {
+      "name": "语言表达质量",
+      "score": xx,
+      "maxScore": xx,
+      "comment": "xx"
+    },
+    ...
+  ],
+  "suggestions": [
+    "xxxx",
+    "xxxxxxx",
+    "xxxxxx。",
+    "xxxxxxxxxxx。"
+  ]
+}
+\`\`\`
+---
+
+### 📌 适用场景举例：
+
+- SEO 内容型页面（如博客、产品 landing page、产品介绍页）
+
+### 原始数据
+{{dataSource}}
+
+
+### 输入的营销文案
+{{ content }}
+
+### 其他要求
+{{ question }}
+
+### 语言
+请使用{{ locale }}语言输出
+  `;
+exports.contentReviewerAgent = core_1.AIAgent.from({
+    name: 'Content Reviewer',
+    instructions: prompt,
+    inputSchema: zod_1.z.object({
+        dataSource: zod_1.z.string(),
+        content: zod_1.z.string(),
+        question: zod_1.z.string(),
+        locale: zod_1.z.string(),
+    }),
+    outputSchema: zod_1.z.object({
+        result: zod_1.z.object({
+            total_score: zod_1.z.number(),
+            level: zod_1.z.string(),
+            dimension_scores: zod_1.z.array(zod_1.z.object({
+                name: zod_1.z.string(),
+                score: zod_1.z.number(),
+                maxScore: zod_1.z.number(),
+                comment: zod_1.z.string(),
+            })),
+            suggestions: zod_1.z.array(zod_1.z.string()),
+        }),
+    }),
+});

package/lib/cjs/agents/multi-agent-page-writer.js

@@ -24,7 +24,7 @@ const section_content_generator_js_1 = require("./section-content-generator.js")
 async function generatePageContentMultiAgent(input) {
     const { context = '', question = '', locale = '', outputSchema, onProgress, checkConnection, sectionsRules, modelName = 'gpt-4o-mini', } = input;
     try {
-        logger_js_1.logger.info('Starting multi-agent page content generation');
+        logger_js_1.logger.info(`[${modelName}] Starting multi-agent page content generation`);
         // 检查客户端连接状态的辅助函数
         const isClientConnected = () => {
             // 如果没有提供检查函数，默认认为客户端仍连接
@@ -45,10 +45,10 @@ async function generatePageContentMultiAgent(input) {
             });
         }
         // 2. 初始页面结构规划
-        logger_js_1.logger.info('Planning initial page structure');
+        logger_js_1.logger.info(`[${modelName}] Planning initial page structure`);
         // 检查客户端是否仍然连接
         if (!isClientConnected()) {
-            logger_js_1.logger.warn('Client disconnected during structure planning, aborting generation');
+            logger_js_1.logger.warn(`[${modelName}] Client disconnected during structure planning, aborting generation`);
             throw new Error('Client disconnected');
         }
         let structurePlanMap = {};
@@ -58,7 +58,7 @@ async function generatePageContentMultiAgent(input) {
             model,
             agents: [page_structure_planner_js_1.pageStructurePlannerAgent, page_structure_validator_js_1.pageStructureValidatorAgent],
             limits: {
-                maxAgentInvokes: 20,
+                maxAgentInvokes: 30,
             },
         });
         await planningEngine.invoke(page_structure_planner_js_1.pageStructurePlannerAgent, {
@@ -74,8 +74,7 @@ async function generatePageContentMultiAgent(input) {
         });
         const structurePlanResult = await planningEngine.subscribe(core_1.UserOutputTopic);
         const { structurePlan } = structurePlanResult.message;
-        // const { structurePlan } = structurePlanResult;
-        logger_js_1.logger.info('Initial structure plan generated', structurePlan);
+        logger_js_1.logger.info(`[${modelName}] Initial structure plan generated`, structurePlan);
         await planningEngine.shutdown();
         // 触发页面结构生成成功事件
         if (onProgress) {
@@ -104,7 +103,7 @@ async function generatePageContentMultiAgent(input) {
                 logger_js_1.logger.warn(`Client disconnected before generating section ${sectionName}, aborting generation`);
                 return null;
             }
-            logger_js_1.logger.info(`Generating content for section: ${sectionName}`);
+            logger_js_1.logger.info(`[${modelName}] Generating content for section: ${sectionName}`);
             // 通知当前正在处理的section
             if (onProgress) {
                 onProgress({
@@ -129,7 +128,7 @@ async function generatePageContentMultiAgent(input) {
             // eslint-disable-next-line no-await-in-loop
             const sectionContent = await engine.invoke(sectionContentAgent, sectionContentInput);
             sectionsData[sectionName] = sectionContent;
-            logger_js_1.logger.info(`Content generated for section: ${sectionName}`);
+            logger_js_1.logger.info(`[${modelName}] Content generated for section: ${sectionName}`);
             // 通知该section已生成完成
             if (onProgress) {
                 const sectionDataObj = {};
@@ -143,10 +142,10 @@ async function generatePageContentMultiAgent(input) {
             }
         }
         // 6. 生成页面元数据
-        logger_js_1.logger.info('Generating page metadata');
+        logger_js_1.logger.info(`[${modelName}] Generating page metadata`);
         // 检查客户端是否仍然连接
         if (!isClientConnected()) {
-            logger_js_1.logger.warn('Client disconnected before generating metadata, aborting generation');
+            logger_js_1.logger.warn(`[${modelName}] Client disconnected before generating metadata, aborting generation`);
             throw new Error('Client disconnected');
         }
         const metadata = await engine.invoke(page_metadata_generator_js_1.generatePageMetadataAgent, {
@@ -161,11 +160,11 @@ async function generatePageContentMultiAgent(input) {
             description: metadata.description,
             sectionsData,
         };
-        logger_js_1.logger.info('Multi-agent page content generation completed');
+        logger_js_1.logger.info(`[${modelName}] Multi-agent page content generation completed`);
         return result;
     }
     catch (error) {
-        logger_js_1.logger.error('Error in multi-agent page content generation:', error);
+        logger_js_1.logger.error(`[${modelName}] Error in multi-agent page content generation:`, error);
         throw error;
     }
 }

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

@@ -72,7 +72,6 @@ exports.pageStructurePlannerAgent = core_1.AIAgent.from({
         logger_js_1.logger.info('page structure planner output', JSON.stringify(output.structurePlan));
         return 'validation_request';
     },
-    // publishTopic: UserOutputTopic,
     instructions: prompt,
     inputSchema: zod_1.z.object({
         context: zod_1.z.string(),