@blocklet/pages-kit-agents 0.5.14 → 0.5.16

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

@@ -1,8 +1,10 @@
 import { ClaudeChatModel } from '@aigne/core/models/claude-chat-model.js';
+import { DeepSeekChatModel } from '@aigne/core/models/deepseek-chat-model.js';
 import { GeminiChatModel } from '@aigne/core/models/gemini-chat-model.js';
 import { OpenAIChatModel } from '@aigne/core/models/openai-chat-model.js';
+import { XAIChatModel } from '@aigne/core/models/xai-chat-model.js';
 import assert from 'node:assert';
-const { OPENAI_API_KEY, GEMINI_API_KEY, CLAUDE_API_KEY } = process.env;
+const { OPENAI_API_KEY, GEMINI_API_KEY, CLAUDE_API_KEY, XAI_API_KEY, DEEPSEEK_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');
@@ -34,5 +36,21 @@ export function getModel(modelName = 'openai') {
         });
         return model;
     }
+    if (modelName?.includes('grok')) {
+        assert(XAI_API_KEY, 'Please set the XAI_API_KEY environment variable');
+        const grok = new XAIChatModel({
+            apiKey: XAI_API_KEY,
+            model: modelName,
+        });
+        return grok;
+    }
+    if (modelName?.includes('deepseek')) {
+        assert(DEEPSEEK_API_KEY, 'Please set the DEEPSEEK_API_KEY environment variable');
+        const deepseek = new DeepSeekChatModel({
+            apiKey: DEEPSEEK_API_KEY,
+            model: modelName,
+        });
+        return deepseek;
+    }
     return undefined;
 }

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

@@ -8,6 +8,7 @@ import path from 'path';
 export async function updateFileList() {
     const outputDir = path.join(process.cwd(), 'sample-output');
     const indexPath = path.join(outputDir, 'index.html');
+    const dataJsPath = path.join(outputDir, 'file-list-data.js');
     // 存储所有子文件夹及其HTML文件信息
     const fileList = [];
     try {
@@ -30,28 +31,71 @@ export async function updateFileList() {
                 .filter((file) => file.isFile() && file.name.endsWith('.html'))
                 .map((file) => {
                 const fileName = file.name.replace(/\.html$/, '');
+                const jsonFileName = `${fileName}.json`;
+                const jsonFilePath = path.join(folderPath, jsonFileName);
+                const htmlFilePath = `${folderName}/${file.name}`;
+                const jsonFileRelativePath = `${folderName}/${jsonFileName}`;
                 return {
                     name: fileName,
-                    path: `${folderName}/${file.name}`,
+                    path: htmlFilePath,
+                    jsonPath: existsSync(jsonFilePath) ? jsonFileRelativePath : undefined,
                     createdAt: new Date().toISOString(),
                 };
             });
+            // 读取每个HTML文件对应的JSON文件（如果存在）
+            for (const htmlFile of htmlFiles) {
+                if (htmlFile.jsonPath) {
+                    try {
+                        // eslint-disable-next-line no-await-in-loop
+                        const jsonContent = await fs.readFile(path.join(outputDir, htmlFile.jsonPath), 'utf8');
+                        const jsonData = JSON.parse(jsonContent);
+                        // 提取评分数据
+                        if (jsonData && jsonData.result) {
+                            htmlFile.scoreData = {
+                                total_score: jsonData.result.total_score,
+                                level: jsonData.result.level,
+                                suggestions: jsonData.result.suggestions,
+                                dimension_scores: jsonData.result.dimension_scores,
+                            };
+                        }
+                    }
+                    catch (jsonError) {
+                        console.error(`Error reading JSON file ${htmlFile.jsonPath}:`, jsonError);
+                    }
+                }
+            }
             // 将文件夹信息添加到列表
             fileList.push({
                 folder: folderName,
                 files: htmlFiles,
             });
         }
-        // 更新index.html，内联JSON数据
+        // 创建独立的JS文件
+        const jsonString = JSON.stringify(fileList, null, 2);
+        const jsContent = `/* eslint-disable prettier/prettier */
+/* eslint-disable @typescript-eslint/quotes */
+// 自动生成的文件列表数据 - 请勿手动修改
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+const FILE_LIST_DATA = ${jsonString};
+`;
+        await fs.writeFile(dataJsPath, jsContent, 'utf8');
+        // 确保index.html引用了这个JS文件
         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');
+            // 检查index.html是否已经包含了对file-list-data.js的引用
+            const scriptIncludeRegex = /<script src=["']file-list-data\.js["']><\/script>/;
+            if (!scriptIncludeRegex.test(indexHtml)) {
+                // 如果没有引用，则替换原来的内联数据定义为引用外部JS文件
+                const fileListPlaceholder = /const FILE_LIST_DATA = .*?;/s;
+                if (fileListPlaceholder.test(indexHtml)) {
+                    // 替换内联数据为引用外部JS文件
+                    indexHtml = indexHtml.replace(fileListPlaceholder, `// 文件列表数据已移至外部JS文件
+// 以下行将在第一次更新时被删除
+const FILE_LIST_DATA = [];`);
+                    // 在</head>前添加script标签引用外部JS文件
+                    indexHtml = indexHtml.replace('</head>', '  <script src="file-list-data.js"></script>\n  </head>');
+                    await fs.writeFile(indexPath, indexHtml, 'utf8');
+                }
             }
         }
     }

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

@@ -26,3 +26,16 @@ export declare function saveHtmlContent({ htmlContent, outputFolder, outputFileN
     outputFolder: string;
     outputFileName: string;
 }): Promise<string>;
+/**
+ * 保存JSON内容到指定目录
+ * @param jsonContent 要保存的JSON数据
+ * @param outputFolder 输出文件夹路径
+ * @param outputFileName 输出JSON文件名称，不包含扩展名
+ * @returns 生成的JSON文件的完整路径
+ * @throws 如果无法创建输出目录或写入文件则抛出异常
+ */
+export declare function saveJsonContent({ jsonContent, outputFolder, outputFileName, }: {
+    jsonContent: Record<string, any> | string;
+    outputFolder: string;
+    outputFileName: string;
+}): Promise<string>;

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

@@ -65,3 +65,32 @@ export async function saveHtmlContent({ htmlContent, outputFolder, outputFileNam
         throw new Error(`Failed to save HTML content: ${err.message}`);
     }
 }
+/**
+ * 保存JSON内容到指定目录
+ * @param jsonContent 要保存的JSON数据
+ * @param outputFolder 输出文件夹路径
+ * @param outputFileName 输出JSON文件名称，不包含扩展名
+ * @returns 生成的JSON文件的完整路径
+ * @throws 如果无法创建输出目录或写入文件则抛出异常
+ */
+export async function saveJsonContent({ jsonContent, 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}.json`);
+        // 如果传入的是对象，转换为JSON字符串
+        const content = typeof jsonContent === 'string' ? jsonContent : JSON.stringify(jsonContent, null, 2);
+        // 写入JSON内容
+        fs.writeFileSync(outputFilePath, content, 'utf-8');
+        return outputFilePath;
+    }
+    catch (error) {
+        const err = error;
+        throw new Error(`Failed to save JSON content: ${err.message}`);
+    }
+}

package/lib/types/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;
+        }[];
+    };
+}>;