@blocklet/pages-kit-agents 0.5.12

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

@@ -0,0 +1,3 @@
+import { ClaudeChatModel } from '@aigne/core/models/claude-chat-model.js';
+import { OpenAIChatModel } from '@aigne/core/models/openai-chat-model.js';
+export declare function getModel(modelName?: string): OpenAIChatModel | ClaudeChatModel | undefined;

package/lib/esm/utils/agent-utils.js

@@ -0,0 +1,38 @@
+import { ClaudeChatModel } from '@aigne/core/models/claude-chat-model.js';
+import { GeminiChatModel } from '@aigne/core/models/gemini-chat-model.js';
+import { OpenAIChatModel } from '@aigne/core/models/openai-chat-model.js';
+import assert from 'node:assert';
+const { OPENAI_API_KEY, GEMINI_API_KEY, CLAUDE_API_KEY } = process.env;
+export function getModel(modelName = 'openai') {
+    if (modelName?.includes('gpt')) {
+        assert(OPENAI_API_KEY, 'Please set the OPENAI_API_KEY environment variable');
+        const openai = new OpenAIChatModel({
+            apiKey: OPENAI_API_KEY,
+            model: modelName,
+            modelOptions: {
+                temperature: 0.8,
+            },
+        });
+        return openai;
+    }
+    if (modelName?.includes('gemini')) {
+        assert(GEMINI_API_KEY, 'Please set the GEMINI_API_KEY environment variable');
+        const gemini = new GeminiChatModel({
+            apiKey: GEMINI_API_KEY,
+            model: modelName,
+        });
+        return gemini;
+    }
+    if (modelName?.includes('claude')) {
+        assert(CLAUDE_API_KEY, 'Please set the CLAUDE_API_KEY environment variable');
+        const model = new ClaudeChatModel({
+            apiKey: CLAUDE_API_KEY,
+            model: modelName,
+            modelOptions: {
+                temperature: 0.7,
+            },
+        });
+        return model;
+    }
+    return undefined;
+}

package/lib/esm/utils/file-utils.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Updates the file list JSON by scanning the sample-output directory
+ * and generating a list of all HTML files in each subfolder
+ */
+export declare function updateFileList(): Promise<void>;

package/lib/esm/utils/file-utils.js

@@ -0,0 +1,61 @@
+import { existsSync } from 'fs';
+import * as fs from 'fs/promises';
+import path from 'path';
+/**
+ * Updates the file list JSON by scanning the sample-output directory
+ * and generating a list of all HTML files in each subfolder
+ */
+export async function updateFileList() {
+    const outputDir = path.join(process.cwd(), 'sample-output');
+    const indexPath = path.join(outputDir, 'index.html');
+    // 存储所有子文件夹及其HTML文件信息
+    const fileList = [];
+    try {
+        // 读取sample-output目录下的所有子文件夹
+        const folders = await fs.readdir(outputDir, { withFileTypes: true });
+        // 过滤出目录，并排除特定文件
+        const dirNames = folders
+            .filter((dirent) => dirent.isDirectory())
+            .map((dirent) => dirent.name)
+            .filter((name) => name !== '.git' && !name.startsWith('.'));
+        // 按照文件夹名称排序（时间戳格式，新的在前）
+        dirNames.sort().reverse();
+        // 处理每个子文件夹
+        for (const folderName of dirNames) {
+            const folderPath = path.join(outputDir, folderName);
+            // eslint-disable-next-line no-await-in-loop
+            const folderFiles = await fs.readdir(folderPath, { withFileTypes: true });
+            // 过滤出HTML文件
+            const htmlFiles = folderFiles
+                .filter((file) => file.isFile() && file.name.endsWith('.html'))
+                .map((file) => {
+                const fileName = file.name.replace(/\.html$/, '');
+                return {
+                    name: fileName,
+                    path: `${folderName}/${file.name}`,
+                    createdAt: new Date().toISOString(),
+                };
+            });
+            // 将文件夹信息添加到列表
+            fileList.push({
+                folder: folderName,
+                files: htmlFiles,
+            });
+        }
+        // 更新index.html，内联JSON数据
+        if (existsSync(indexPath)) {
+            let indexHtml = await fs.readFile(indexPath, 'utf8');
+            // 使用更简洁的替换方式
+            const fileListPlaceholder = /const FILE_LIST_DATA = .*?;/s;
+            if (fileListPlaceholder.test(indexHtml)) {
+                // 替换占位符为实际数据
+                const jsonString = JSON.stringify(fileList, null, 2);
+                indexHtml = indexHtml.replace(fileListPlaceholder, `const FILE_LIST_DATA = ${jsonString};`);
+                await fs.writeFile(indexPath, indexHtml, 'utf8');
+            }
+        }
+    }
+    catch (error) {
+        console.error('Error updating file list:', error);
+    }
+}

package/lib/esm/utils/index.d.ts

@@ -0,0 +1,12 @@
+/**
+ * 读取指定路径的文件并返回字符串内容
+ * @param filePath 文件路径，可以是相对路径或绝对路径
+ * @param encoding 文件编码，默认为 'utf-8'
+ * @returns 文件内容字符串
+ * @throws 如果文件不存在或读取失败则抛出异常
+ */
+export declare function readFile(filePath: string, encoding?: BufferEncoding): Promise<string>;
+export * from './agent-utils';
+export * from './logger';
+export * from './template-utils';
+export * from './file-utils';

package/lib/esm/utils/index.js

@@ -0,0 +1,28 @@
+import { readFile as fsReadFile } from 'node:fs/promises';
+import path from 'node:path';
+/**
+ * 读取指定路径的文件并返回字符串内容
+ * @param filePath 文件路径，可以是相对路径或绝对路径
+ * @param encoding 文件编码，默认为 'utf-8'
+ * @returns 文件内容字符串
+ * @throws 如果文件不存在或读取失败则抛出异常
+ */
+export async function readFile(filePath, encoding = 'utf-8') {
+    try {
+        // 如果是相对路径，则从当前工作目录解析
+        const resolvedPath = path.isAbsolute(filePath) ? filePath : path.resolve(process.cwd(), filePath);
+        // 读取文件内容
+        const content = await fsReadFile(resolvedPath, { encoding });
+        return content;
+    }
+    catch (error) {
+        // 捕获并重新抛出异常，添加更明确的错误信息
+        const err = error;
+        throw new Error(`Failed to read file at path ${filePath}: ${err.message}`);
+    }
+}
+// 导出所有模块功能
+export * from './agent-utils';
+export * from './logger';
+export * from './template-utils';
+export * from './file-utils';

package/lib/esm/utils/logger.d.ts

@@ -0,0 +1 @@
+export declare const logger: Console;

package/lib/esm/utils/logger.js

@@ -0,0 +1 @@
+export const logger = console;

package/lib/esm/utils/template-utils.d.ts

@@ -0,0 +1,28 @@
+/**
+ * 根据JSON数据更新HTML模板并保存到指定目录
+ * @param jsonData 需要应用到模板的JSON数据
+ * @param templatePath 模板HTML文件路径，默认为assets目录下的sample.html
+ * @param outputFolder 输出文件夹名称，将在output目录下创建
+ * @param outputFileName 输出HTML文件名称，不包含扩展名
+ * @returns 生成的HTML文件的完整路径
+ * @throws 如果模板文件不存在或无法创建输出目录则抛出异常
+ */
+export declare function generateHtmlFromTemplate({ jsonData, outputFolder, outputFileName, templatePath, }: {
+    jsonData: Record<string, any>;
+    outputFolder: string;
+    outputFileName: string;
+    templatePath?: string;
+}): Promise<string>;
+/**
+ * 保存HTML内容到指定目录
+ * @param htmlContent 要保存的HTML内容
+ * @param outputFolder 输出文件夹路径
+ * @param outputFileName 输出HTML文件名称，不包含扩展名
+ * @returns 生成的HTML文件的完整路径
+ * @throws 如果无法创建输出目录或写入文件则抛出异常
+ */
+export declare function saveHtmlContent({ htmlContent, outputFolder, outputFileName, }: {
+    htmlContent: string;
+    outputFolder: string;
+    outputFileName: string;
+}): Promise<string>;

package/lib/esm/utils/template-utils.js

@@ -0,0 +1,67 @@
+import fs from 'node:fs';
+import { readFile as fsReadFile } from 'node:fs/promises';
+import path from 'node:path';
+/**
+ * 根据JSON数据更新HTML模板并保存到指定目录
+ * @param jsonData 需要应用到模板的JSON数据
+ * @param templatePath 模板HTML文件路径，默认为assets目录下的sample.html
+ * @param outputFolder 输出文件夹名称，将在output目录下创建
+ * @param outputFileName 输出HTML文件名称，不包含扩展名
+ * @returns 生成的HTML文件的完整路径
+ * @throws 如果模板文件不存在或无法创建输出目录则抛出异常
+ */
+export async function generateHtmlFromTemplate({ jsonData, outputFolder, outputFileName, templatePath = 'assets/sample2.html', }) {
+    try {
+        // 读取模板文件
+        const resolvedTemplatePath = path.isAbsolute(templatePath)
+            ? templatePath
+            : path.resolve(process.cwd(), templatePath);
+        const templateContent = await fsReadFile(resolvedTemplatePath, { encoding: 'utf-8' });
+        // 将JSON数据转换为字符串
+        const jsonDataString = JSON.stringify(jsonData);
+        // 替换模板中的jsonData变量
+        const updatedContent = templateContent.replace(/const jsonData = \{[\s\S]*?\};/, `const jsonData = ${jsonDataString};`);
+        // 创建输出目录
+        const outputDir = path.resolve(process.cwd(), 'sample-output', outputFolder);
+        // 确保输出目录存在
+        if (!fs.existsSync(outputDir)) {
+            fs.mkdirSync(outputDir, { recursive: true });
+        }
+        // 构建输出文件的完整路径
+        const outputFilePath = path.join(outputDir, `${outputFileName}.html`);
+        // 写入更新后的HTML内容
+        fs.writeFileSync(outputFilePath, updatedContent, 'utf-8');
+        return outputFilePath;
+    }
+    catch (error) {
+        const err = error;
+        throw new Error(`Failed to generate HTML from template: ${err.message}`);
+    }
+}
+/**
+ * 保存HTML内容到指定目录
+ * @param htmlContent 要保存的HTML内容
+ * @param outputFolder 输出文件夹路径
+ * @param outputFileName 输出HTML文件名称，不包含扩展名
+ * @returns 生成的HTML文件的完整路径
+ * @throws 如果无法创建输出目录或写入文件则抛出异常
+ */
+export async function saveHtmlContent({ htmlContent, outputFolder, outputFileName, }) {
+    try {
+        // 创建输出目录
+        const outputDir = path.resolve(process.cwd(), 'sample-output', outputFolder);
+        // 确保输出目录存在
+        if (!fs.existsSync(outputDir)) {
+            fs.mkdirSync(outputDir, { recursive: true });
+        }
+        // 构建输出文件的完整路径
+        const outputFilePath = path.join(outputDir, `${outputFileName}.html`);
+        // 写入HTML内容
+        fs.writeFileSync(outputFilePath, htmlContent, 'utf-8');
+        return outputFilePath;
+    }
+    catch (error) {
+        const err = error;
+        throw new Error(`Failed to save HTML content: ${err.message}`);
+    }
+}

package/lib/types/agents/multi-agent-page-writer.d.ts

@@ -0,0 +1,48 @@
+import type { z } from 'zod';
+export interface GeneratePageContentInput {
+    context: string;
+    question: string;
+    sectionsRules?: Record<string, string>;
+    locale: string;
+    outputSchema: z.ZodObject<any>;
+    modelName?: string;
+    onProgress?: (params: {
+        sectionName?: string;
+        sectionIndex?: number;
+        totalSections: number;
+        sectionData?: Record<string, any>;
+        eventType?: 'progress' | 'section' | 'structure_start' | 'structure_complete';
+        structureData?: any;
+    }) => void;
+    checkConnection?: () => boolean;
+}
+/**
+ * 多Agent协作生成页面内容
+ *
+ * 工作流程：
+ * 1. 使用页面结构规划Agent分析内容，规划各section内容
+ * 2. 使用结构验证Agent检查规划的完整性和有效性
+ * 3. 如果验证不通过，将反馈提供给规划Agent进行修正
+ * 4. 逐个生成各section内容
+ * 5. 基于所有section内容生成页面元数据
+ * 6. 整合所有内容返回完整页面数据
+ */
+export default function generatePageContentMultiAgent(input: GeneratePageContentInput): Promise<{
+    title: string;
+    description: string;
+    sectionsData: Record<string, any>;
+} | null>;
+/**
+ * 单独生成特定section的内容
+ * 支持独立调用，为特定section重新生成内容
+ */
+export declare function generateSingleSectionContent({ context, question, locale, sectionName, sectionSchema, sectionPlan, existingSections, modelName, }: {
+    context: string;
+    question: string;
+    locale: string;
+    sectionName: string;
+    sectionSchema: any;
+    sectionPlan?: string;
+    existingSections?: Record<string, any>;
+    modelName?: string;
+}): Promise<import("@aigne/core").Message>;

package/lib/types/agents/page-metadata-generator.d.ts

@@ -0,0 +1,10 @@
+import { AIAgent } from '@aigne/core';
+export declare const generatePageMetadataAgent: AIAgent<{
+    context: string;
+    question: string;
+    locale: string;
+    sectionsData: string;
+}, {
+    description: string;
+    title: string;
+}>;

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

@@ -0,0 +1,55 @@
+import { AIAgent } from '@aigne/core';
+export interface PlanPageStructureInput {
+    context: string;
+    question: string;
+    initialStructurePlan?: Array<{
+        sectionName: string;
+        sectionContentPlan: string;
+    }>;
+    locale: string;
+    outputSchema: any;
+    validationFeedback?: {
+        isValid: boolean;
+        missingSections: string[];
+        duplicateContent: Array<{
+            sections: string[];
+            description: string;
+        }>;
+        feedback: Array<{
+            sectionName: string;
+            feedback: string;
+        }>;
+        generalFeedback: string;
+    };
+}
+export interface PageStructurePlannerOutput {
+    structurePlan: Array<{
+        sectionName: string;
+        sectionContentPlan: string;
+    }>;
+}
+export declare const pageStructurePlannerAgent: AIAgent<{
+    outputSchema: string;
+    context: string;
+    question: string;
+    locale: string;
+    expectedSections: string;
+    initialStructurePlan: {
+        sectionName: string;
+        sectionContentPlan: string;
+    }[];
+    validationFeedback?: {
+        isValid: boolean;
+        missingSections: string[];
+        duplicateContent: {
+            description: string;
+            sections: string[];
+            score: number;
+        }[];
+    } | undefined;
+}, {
+    structurePlan: {
+        sectionName: string;
+        sectionContentPlan: string;
+    }[];
+}>;

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

@@ -0,0 +1,30 @@
+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<{
+    outputSchema: string;
+    context: string;
+    question: string;
+    expectedSections: string;
+    structurePlan: {
+        sectionName: string;
+        sectionContentPlan: string;
+    }[];
+}, {
+    validationFeedback: {
+        isValid: boolean;
+        missingSections: string[];
+        duplicateContent: {
+            description: string;
+            sections: string[];
+            score: number;
+        }[];
+    };
+}>;

package/lib/types/agents/sample-html-generator.d.ts

@@ -0,0 +1,6 @@
+import { AIAgent } from '@aigne/core';
+export declare const sampleHtmlGeneratorAgent: AIAgent<{
+    jsonData: string;
+}, {
+    html: string;
+}>;

package/lib/types/agents/section-content-generator.d.ts

@@ -0,0 +1,19 @@
+import { AIAgent } from '@aigne/core';
+interface GenerateSectionContentInput {
+    context: string;
+    question: string;
+    locale: string;
+    sectionName: string;
+    sectionSchema: any;
+    sectionPlan: string;
+    generatedSections: string;
+}
+export declare function createSectionContentAgent(input: GenerateSectionContentInput): Promise<AIAgent<{
+    context: string;
+    question: string;
+    locale: string;
+    sectionName: string;
+    sectionPlan: string;
+    generatedSections: string;
+}, import("@aigne/core").Message>>;
+export {};

package/lib/types/index.d.ts

@@ -0,0 +1,7 @@
+export {};
+export { default as generatePageContentMultiAgent, generateSingleSectionContent, } from './agents/multi-agent-page-writer';
+export * from './agents/page-metadata-generator';
+export * from './agents/page-structure-planner';
+export * from './agents/page-structure-validator';
+export * from './agents/section-content-generator';
+export * from './utils/index';