@aigne/example-workflow-concurrency 1.7.0 → 1.8.0

package/README.md

@@ -76,7 +76,7 @@ The following example demonstrates how to build a concurrency workflow:
 
 ```typescript
 import assert from "node:assert";
-import { AIAgent, ExecutionEngine, parallel } from "@aigne/core";
+import { AIAgent, AIGNE, TeamAgent, ProcessMode } from "@aigne/core";
 import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
 
 const { OPENAI_API_KEY } = process.env;
@@ -104,9 +104,15 @@ Product description:
   outputKey: "audience",
 });
 
-const engine = new ExecutionEngine({ model });
+const aigne = new AIGNE({ model });
 
-const result = await engine.call(parallel(featureExtractor, audienceAnalyzer), {
+// 创建一个 TeamAgent 来处理并行工作流
+const teamAgent = TeamAgent.from({
+  skills: [featureExtractor, audienceAnalyzer],
+  mode: ProcessMode.parallel
+});
+
+const result = await aigne.invoke(teamAgent, {
   product: "AIGNE is a No-code Generative AI Apps Engine",
 });
 

package/index.test.ts

@@ -1,5 +1,11 @@
-import { test } from "bun:test";
+import { expect, test } from "bun:test";
+import { runExampleTest } from "@aigne/test-utils/run-example-test.js";
 
-test("should successfully execute the workflow-concurrency", () => import("./index.js"), {
-  timeout: 60000,
-});
+test(
+  "should successfully run the workflow-concurrency",
+  async () => {
+    const { code } = await runExampleTest();
+    expect(code).toBe(0);
+  },
+  { timeout: 600000 },
+);

package/index.ts

@@ -1,7 +1,7 @@
 #!/usr/bin/env bunwrapper
 
 import { runChatLoopInTerminal } from "@aigne/cli/utils/run-chat-loop.js";
-import { AIAgent, ExecutionEngine, parallel } from "@aigne/core";
+import { AIAgent, AIGNE, ProcessMode, TeamAgent } from "@aigne/core";
 import { loadModel } from "@aigne/core/loader/index.js";
 
 const model = await loadModel();
@@ -24,9 +24,14 @@ Product description:
   outputKey: "audience",
 });
 
-const engine = new ExecutionEngine({ model });
+const aigne = new AIGNE({ model });
 
-const userAgent = engine.call(parallel(featureExtractor, audienceAnalyzer));
+const userAgent = aigne.invoke(
+  TeamAgent.from({
+    skills: [featureExtractor, audienceAnalyzer],
+    mode: ProcessMode.parallel,
+  }),
+);
 
 await runChatLoopInTerminal(userAgent, {
   welcome: `Hello, I'm a product analyst and market researcher. I can help you with extracting features and identifying target audience.`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-workflow-concurrency",
-  "version": "1.7.0",
+  "version": "1.8.0",
   "description": "A demonstration of using AIGNE Framework to build a concurrency workflow",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/workflow-concurrency",
@@ -18,8 +18,11 @@
   "dependencies": {
     "openai": "^4.94.0",
     "zod": "^3.24.2",
-    "@aigne/cli": "^1.7.0",
-    "@aigne/core": "^1.11.0"
+    "@aigne/core": "^1.12.0",
+    "@aigne/cli": "^1.8.0"
+  },
+  "devDependencies": {
+    "@aigne/test-utils": "^0.1.0"
   },
   "scripts": {
     "start": "bun run index.ts",

package/usages.ts

@@ -1,5 +1,5 @@
 import assert from "node:assert";
-import { AIAgent, ExecutionEngine, parallel } from "@aigne/core";
+import { AIAgent, AIGNE, ProcessMode, TeamAgent } from "@aigne/core";
 import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
 
 const { OPENAI_API_KEY } = process.env;
@@ -27,11 +27,17 @@ Product description:
   outputKey: "audience",
 });
 
-const engine = new ExecutionEngine({ model });
-
-const result = await engine.call(parallel(featureExtractor, audienceAnalyzer), {
-  product: "AIGNE is a No-code Generative AI Apps Engine",
-});
+const aigne = new AIGNE({ model });
+
+const result = await aigne.invoke(
+  TeamAgent.from({
+    skills: [featureExtractor, audienceAnalyzer],
+    mode: ProcessMode.parallel,
+  }),
+  {
+    product: "AIGNE is a No-code Generative AI Apps Engine",
+  },
+);
 
 console.log(result);