@aigne/example-workflow-reflection 1.1.0 → 1.2.1

package/README.md

@@ -71,19 +71,14 @@ The following example demonstrates how to build a reflection workflow:
 
 ```typescript
 import assert from "node:assert";
-import {
-  AIAgent,
-  ChatModelOpenAI,
-  ExecutionEngine,
-  UserInputTopic,
-  UserOutputTopic,
-} from "@aigne/core-next";
+import { AIAgent, ExecutionEngine, UserInputTopic, UserOutputTopic } from "@aigne/core";
+import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
 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 +141,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

@@ -1,20 +1,15 @@
 #!/usr/bin/env npx -y bun
 
 import assert from "node:assert";
-import {
-  AIAgent,
-  ChatModelOpenAI,
-  ExecutionEngine,
-  UserInputTopic,
-  UserOutputTopic,
-  runChatLoopInTerminal,
-} from "@aigne/core-next";
+import { AIAgent, ExecutionEngine, UserAgent, UserInputTopic, UserOutputTopic } from "@aigne/core";
+import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
+import { runChatLoopInTerminal } from "@aigne/core/utils/run-chat-loop.js";
 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 +73,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.1",
   "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.1",
     "zod": "^3.24.2",
-    "@aigne/core-next": "^1.1.0"
+    "@aigne/core": "^1.5.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,12 @@
 import assert from "node:assert";
-import {
-  AIAgent,
-  ChatModelOpenAI,
-  ExecutionEngine,
-  UserInputTopic,
-  UserOutputTopic,
-} from "@aigne/core-next";
+import { AIAgent, ExecutionEngine, UserInputTopic, UserOutputTopic } from "@aigne/core";
+import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
 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 +69,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)",