@aigne/example-workflow-code-execution 1.1.0 → 1.2.1

package/README.md

@@ -90,13 +90,14 @@ The following example demonstrates how to build a code-execution workflow:
 
 ```typescript
 import assert from "node:assert";
-import { AIAgent, ChatModelOpenAI, ExecutionEngine, FunctionAgent } from "@aigne/core-next";
+import { AIAgent, ExecutionEngine, FunctionAgent } 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,
 });
 
@@ -125,11 +126,11 @@ Work with the sandbox to execute your code.
 
 const engine = new ExecutionEngine({ model });
 
-const result = await engine.run("10! = ?", coder);
+const result = await engine.call(coder, "10! = ?");
 console.log(result);
 // Output:
 // {
-//   text: "The value of \\(10!\\) (10 factorial) is 3,628,800.",
+//   $message: "The value of \\(10!\\) (10 factorial) is 3,628,800.",
 // }
 ```
 

package/index.ts

@@ -1,19 +1,15 @@
 #!/usr/bin/env npx -y bun
 
 import assert from "node:assert";
-import {
-  AIAgent,
-  ChatModelOpenAI,
-  ExecutionEngine,
-  FunctionAgent,
-  runChatLoopInTerminal,
-} from "@aigne/core-next";
+import { AIAgent, 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,
 });
 
@@ -38,13 +34,14 @@ You are a proficient coder. You write code to solve problems.
 Work with the sandbox to execute your code.
 `,
   tools: [sandbox],
+  memory: true,
 });
 
 const engine = new ExecutionEngine({ model });
 
-const userAgent = await engine.run(coder);
+const user = engine.call(coder);
 
-await runChatLoopInTerminal(userAgent, {
+await runChatLoopInTerminal(user, {
   welcome:
     "Welcome to the code execution workflow! you can ask me anything can be resolved by running code.",
   defaultQuestion: "10! = ?",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-workflow-code-execution",
-  "version": "1.1.0",
+  "version": "1.2.1",
   "description": "A demonstration of using AIGNE Framework to build a code-execution workflow",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/workflow-code-execution",
@@ -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/usage.ts

@@ -1,13 +1,12 @@
-#!/usr/bin/env npx -y bun
-
 import assert from "node:assert";
-import { AIAgent, ChatModelOpenAI, ExecutionEngine, FunctionAgent } from "@aigne/core-next";
+import { AIAgent, ExecutionEngine, FunctionAgent } 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,
 });
 
@@ -36,9 +35,9 @@ Work with the sandbox to execute your code.
 
 const engine = new ExecutionEngine({ model });
 
-const result = await engine.run("10! = ?", coder);
+const result = await engine.call(coder, "10! = ?");
 console.log(result);
 // Output:
 // {
-//   text: "The value of \\(10!\\) (10 factorial) is 3,628,800.",
+//   $message: "The value of \\(10!\\) (10 factorial) is 3,628,800.",
 // }