@aigne/example-workflow-reflection 1.1.0 → 1.2.0

package/README.md

@@ -73,17 +73,17 @@ The following example demonstrates how to build a reflection workflow:
 import assert from "node:assert";
 import {
   AIAgent,
-  ChatModelOpenAI,
+  OpenAIChatModel,
   ExecutionEngine,
   UserInputTopic,
   UserOutputTopic,
-} from "@aigne/core-next";
+} from "@aigne/core";
 import { z } from "zod";
 
 const { OPENAI_API_KEY } = process.env;
 assert(OPENAI_API_KEY, "Please set the OPENAI_API_KEY environment variable");
 
-const model = new ChatModelOpenAI({
+const model = new OpenAIChatModel({
   apiKey: OPENAI_API_KEY,
 });
 
@@ -146,9 +146,10 @@ Please review the code. If previous feedback was provided, see if it was address
 });
 
 const engine = new ExecutionEngine({ model, agents: [coder, reviewer] });
+engine.publish(UserInputTopic, "Write a function to find the sum of all even numbers in a list.");
 
-const result = await engine.run("Write a function to find the sum of all even numbers in a list.");
-console.log(result);
+const { message } = await engine.subscribe(UserOutputTopic);
+console.log(message);
 // Output:
 // {
 //   code: "def sum_of_even_numbers(numbers):\n    \"\"\"Function to calculate the sum of all even numbers in a list.\"\"\"\n    return sum(number for number in numbers if number % 2 == 0)",

package/index.ts

@@ -3,18 +3,19 @@
 import assert from "node:assert";
 import {
   AIAgent,
-  ChatModelOpenAI,
   ExecutionEngine,
+  OpenAIChatModel,
+  UserAgent,
   UserInputTopic,
   UserOutputTopic,
   runChatLoopInTerminal,
-} from "@aigne/core-next";
+} from "@aigne/core";
 import { z } from "zod";
 
 const { OPENAI_API_KEY } = process.env;
 assert(OPENAI_API_KEY, "Please set the OPENAI_API_KEY environment variable");
 
-const model = new ChatModelOpenAI({
+const model = new OpenAIChatModel({
   apiKey: OPENAI_API_KEY,
 });
 
@@ -78,7 +79,11 @@ Please review the code. If previous feedback was provided, see if it was address
 
 const engine = new ExecutionEngine({ model, agents: [coder, reviewer] });
 
-const userAgent = await engine.run();
+const userAgent = UserAgent.from({
+  context: engine,
+  publishTopic: UserInputTopic,
+  subscribeTopic: UserOutputTopic,
+});
 
 await runChatLoopInTerminal(userAgent, {
   welcome: `Hello, I'm a coder with a reviewer. I can help you write code and get it reviewed.`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-workflow-reflection",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "A demonstration of using AIGNE Framework to build a reflection workflow",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/workflow-reflection",
@@ -16,10 +16,12 @@
     "README.md"
   ],
   "dependencies": {
+    "openai": "^4.89.0",
     "zod": "^3.24.2",
-    "@aigne/core-next": "^1.1.0"
+    "@aigne/core": "^1.3.0"
   },
   "scripts": {
-    "start": "npx -y bun run index.ts"
+    "start": "npx -y bun run index.ts",
+    "lint": "tsc --noEmit"
   }
 }

package/usages.ts

@@ -1,17 +1,17 @@
 import assert from "node:assert";
 import {
   AIAgent,
-  ChatModelOpenAI,
   ExecutionEngine,
+  OpenAIChatModel,
   UserInputTopic,
   UserOutputTopic,
-} from "@aigne/core-next";
+} from "@aigne/core";
 import { z } from "zod";
 
 const { OPENAI_API_KEY } = process.env;
 assert(OPENAI_API_KEY, "Please set the OPENAI_API_KEY environment variable");
 
-const model = new ChatModelOpenAI({
+const model = new OpenAIChatModel({
   apiKey: OPENAI_API_KEY,
 });
 
@@ -74,9 +74,10 @@ Please review the code. If previous feedback was provided, see if it was address
 });
 
 const engine = new ExecutionEngine({ model, agents: [coder, reviewer] });
+engine.publish(UserInputTopic, "Write a function to find the sum of all even numbers in a list.");
 
-const result = await engine.run("Write a function to find the sum of all even numbers in a list.");
-console.log(result);
+const { message } = await engine.subscribe(UserOutputTopic);
+console.log(message);
 // Output:
 // {
 //   code: "def sum_of_even_numbers(numbers):\n    \"\"\"Function to calculate the sum of all even numbers in a list.\"\"\"\n    return sum(number for number in numbers if number % 2 == 0)",