@blocklet/pages-kit-agents 0.5.41 → 0.5.42

package/lib/types/utils/component-tuils.d.ts

@@ -0,0 +1,12 @@
+import { z } from 'zod';
+export declare function getSchemaByComponentIds(sections: {
+    componentId: string;
+    componentName: string;
+    sectionName: string;
+}[]): z.ZodObject<Record<string, any>, "strip", z.ZodTypeAny, {
+    [x: string]: any;
+}, {
+    [x: string]: any;
+}>;
+export declare function getComponentsSimpleMetadata(): string;
+export declare function getComponentsMetadata(): string;

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

@@ -3,3 +3,26 @@
  * and generating a list of all HTML files in each subfolder
  */
 export declare function updateFileList(): Promise<void>;
+/**
+ * File content information interface
+ */
+export interface FileContent {
+    filePath: string;
+    fileName: string;
+    content: string;
+}
+/**
+ * Recursively reads files matching a pattern from a directory and its subdirectories
+ * @param dirPath - The directory path to search in
+ * @param pattern - String or RegExp pattern to match file names
+ * @param options - Optional configuration
+ * @returns Promise<FileContent[]> - Array of file content objects
+ */
+export declare function readFilesByPattern(dirPath: string, pattern: string | RegExp, options?: {
+    recursive?: boolean;
+}): Promise<FileContent[]>;
+export declare function readFileByPath(filePath: string): Promise<string>;
+/**
+ * Save content to a file, creating directories if needed
+ */
+export declare function saveFileByPath(filePath: string, content: string): Promise<void>;

package/lib/types/workflow-agents/reflection-agent.d.ts

@@ -0,0 +1,16 @@
+import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type Message } from '@aigne/core';
+import type { PromiseOrValue } from '@aigne/core/utils/type-utils.js';
+export interface ReflectionAgentOptions<I extends Message = Message, RO extends Message = Message> extends AgentOptions<I, I> {
+    editor: Agent<I, I>;
+    reviewer: Agent<I, RO>;
+    isApproved?: (result: RO) => PromiseOrValue<boolean>;
+    maxIterations?: number;
+}
+export declare class ReflectionAgent<I extends Message = Message, RO extends Message = Message> extends Agent<I, I> {
+    constructor(options: ReflectionAgentOptions<I, RO>);
+    editor: Agent<I, I>;
+    reviewer: Agent<I, RO>;
+    isApproved?: (result: RO) => PromiseOrValue<boolean>;
+    maxIterations: number;
+    process(input: I, options: AgentInvokeOptions): Promise<AgentProcessResult<I>>;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blocklet/pages-kit-agents",
-  "version": "0.5.41",
+  "version": "0.5.42",
   "description": "AIGNE based agents for pages kit",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/blocklet/pages-kit#readme",
@@ -23,6 +23,7 @@
   },
   "keywords": [],
   "dependencies": {
+    "@aigne/agent-library": "^1.12.6",
     "@aigne/anthropic": "^0.2.7",
     "@aigne/core": "^1.18.6",
     "@aigne/deepseek": "^0.2.7",
@@ -31,14 +32,16 @@
     "@aigne/xai": "^0.2.7",
     "@types/js-yaml": "^4.0.9",
     "@types/yargs": "^17.0.33",
+    "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.41"
+    "@blocklet/pages-kit-block-studio": "0.5.42"
   },
   "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:*",

package/workflow-cli.ts

@@ -0,0 +1,132 @@
+import { AIGNE, ProcessMode, TeamAgent } 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 { 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';
+
+aigneLogger.level = LogLevel.DEBUG;
+
+const DEFAULT_LOCALE = 'zh';
+const DEFAULT_MODEL_NAME = 'gpt-4o';
+const DEFAULT_PROJECT_ID = 'DEFAULT_PROJECT_ID';
+
+const ConfigSchema = z.object({
+  inputInfoPath: z.string(),
+  locale: z.string(),
+  modelName: z.string(),
+  projectId: z.string(),
+});
+
+type WorkflowConfig = z.infer<typeof ConfigSchema>;
+
+async function loadConfig(): Promise<WorkflowConfig> {
+  const configPath = path.join(__dirname, 'workflow-config.yaml');
+  try {
+    const fileContent = await fs.readFile(configPath, 'utf-8');
+    const parsedConfig = yaml.load(fileContent) as any;
+    return ConfigSchema.parse(parsedConfig);
+  } catch (error) {
+    logger.error(`Failed to load workflow-config.yaml: ${error instanceof Error ? error.message : String(error)}`);
+    throw new Error('请在同级目录下提供完整的 workflow-config.yaml 配置文件');
+  }
+}
+
+async function runWorkflow({
+  inputInfo,
+  locale,
+  modelName,
+  projectId,
+}: {
+  inputInfo: string;
+  locale: string;
+  modelName: string;
+  projectId: string;
+}) {
+  logger.info('inputInfo', inputInfo);
+  logger.info('locale', locale);
+  logger.info('modelName', modelName);
+  logger.info('projectId', projectId);
+
+  const model = getModel(modelName);
+  const engine = new AIGNE({
+    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,
+  });
+
+  logger.info('result', result);
+}
+
+// 主函数
+async function main() {
+  try {
+    logger.info('🔧 读取 workflow-config.yaml 配置...');
+    const config = await loadConfig();
+    logger.info('📚 读取原始信息...');
+    const inputInfo = await readFileByPath(config.inputInfoPath);
+
+    await runWorkflow({
+      inputInfo,
+      locale: config.locale || DEFAULT_LOCALE,
+      modelName: config.modelName || DEFAULT_MODEL_NAME,
+      projectId: config.projectId || DEFAULT_PROJECT_ID,
+    });
+  } catch (error) {
+    logger.error(`工作流执行失败: ${error instanceof Error ? error.message : String(error)}`);
+    process.exit(1);
+  }
+}
+
+// 处理程序退出
+process.on('SIGINT', () => {
+  logger.info('\n\n工作流已被用户中断');
+  process.exit(0);
+});
+
+// 执行主函数
+if (require.main === module) {
+  main().catch((error) => {
+    logger.error(`Error: ${error instanceof Error ? error.message : String(error)}`);
+    process.exit(1);
+  });
+}
+
+export { runWorkflow };