@aigne/example-workflow-handoff 1.1.0 → 1.2.1

package/README.md

@@ -90,12 +90,13 @@ The following example demonstrates how to build a handoff workflow:
 
 ```typescript
 import assert from "node:assert";
-import { AIAgent, ChatModelOpenAI, ExecutionEngine } from "@aigne/core-next";
+import { AIAgent, ExecutionEngine } from "@aigne/core";
+import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
 
 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,
 });
 
@@ -118,7 +119,7 @@ const agentB = AIAgent.from({
 
 const engine = new ExecutionEngine({ model });
 
-const userAgent = await engine.run(agentA);
+const userAgent = engine.call(agentA);
 
 const result1 = await userAgent.call("transfer to agent b");
 console.log(result1);

package/index.ts

@@ -1,20 +1,15 @@
 #!/usr/bin/env npx -y bun
 
 import assert from "node:assert";
-import {
-  AIAgent,
-  type Agent,
-  ChatModelOpenAI,
-  ExecutionEngine,
-  FunctionAgent,
-  runChatLoopInTerminal,
-} from "@aigne/core-next";
+import { AIAgent, type Agent, ExecutionEngine, FunctionAgent } 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,
 });
 
@@ -110,7 +105,7 @@ tell them a crazy caveat and execute their order.
 `,
   tools: [transfer_back_to_triage, execute_order_tool],
   outputKey: "sales",
-  enableHistory: true,
+  memory: true,
 });
 
 const issuesAndRepairs = AIAgent.from({
@@ -127,7 +122,7 @@ Follow the following routine with the user:
 `,
   tools: [transfer_back_to_triage, execute_refund_tool, look_up_item_tool],
   outputKey: "issuesAndRepairs",
-  enableHistory: true,
+  memory: true,
 });
 
 // Assume this is a human agent
@@ -140,7 +135,7 @@ Only transfer to another agent if user explicitly asks for it.
 `,
   tools: [transfer_back_to_triage, transfer_to_sales_agent, transfer_to_issues_and_repairs],
   outputKey: "human",
-  enableHistory: true,
+  memory: true,
 });
 
 const triage = AIAgent.from({
@@ -153,12 +148,12 @@ But make your questions subtle and natural.
 `,
   tools: [transfer_to_issues_and_repairs, transfer_to_sales_agent, transfer_to_human_manager],
   outputKey: "triage",
-  enableHistory: true,
+  memory: true,
 });
 
 const engine = new ExecutionEngine({ model });
 
-const userAgent = await engine.run(triage);
+const userAgent = engine.call(triage);
 
 await runChatLoopInTerminal(userAgent, {
   welcome: `Hello, I'm a customer service bot for ACME Inc. How can I help you today?`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-workflow-handoff",
-  "version": "1.1.0",
+  "version": "1.2.1",
   "description": "A demonstration of using AIGNE Framework to build a handoff workflow",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/workflow-handoff",
@@ -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,10 +1,11 @@
 import assert from "node:assert";
-import { AIAgent, ChatModelOpenAI, ExecutionEngine } from "@aigne/core-next";
+import { AIAgent, ExecutionEngine } from "@aigne/core";
+import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
 
 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,
 });
 
@@ -27,7 +28,7 @@ const agentB = AIAgent.from({
 
 const engine = new ExecutionEngine({ model });
 
-const userAgent = await engine.run(agentA);
+const userAgent = engine.call(agentA);
 
 const result1 = await userAgent.call("transfer to agent b");
 console.log(result1);