@aigne/example-workflow-reflection 1.9.1 → 1.10.0

package/README.md

@@ -1,6 +1,6 @@
 # Workflow Reflection Demo
 
-This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) to build a reflection workflow.
+This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) to build a reflection workflow. The example now supports both one-shot and interactive chat modes, along with customizable model settings and pipeline input/output.
 
 ```mermaid
 flowchart LR
@@ -34,7 +34,14 @@ class reviewer processing
 ```bash
 export OPENAI_API_KEY=YOUR_OPENAI_API_KEY # Set your OpenAI API key
 
-npx -y @aigne/example-workflow-reflection # Run the example
+# Run in one-shot mode (default)
+npx -y @aigne/example-workflow-reflection
+
+# Run in interactive chat mode
+npx -y @aigne/example-workflow-reflection --chat
+
+# Use pipeline input
+echo "Write a function to validate email addresses" | npx -y @aigne/example-workflow-reflection
 ```
 
 ## Installation
@@ -64,7 +71,41 @@ OPENAI_API_KEY="" # Set your OpenAI API key here
 ### Run the Example
 
 ```bash
-pnpm start
+pnpm start # Run in one-shot mode (default)
+
+# Run in interactive chat mode
+pnpm start -- --chat
+
+# Use pipeline input
+echo "Write a function to validate email addresses" | pnpm start
+```
+
+### Run Options
+
+The example supports the following command-line parameters:
+
+| Parameter | Description | Default |
+|-----------|-------------|---------|
+| `--chat` | Run in interactive chat mode | Disabled (one-shot mode) |
+| `--model <provider[:model]>` | AI model to use in format 'provider[:model]' where model is optional. Examples: 'openai' or 'openai:gpt-4o-mini' | openai |
+| `--temperature <value>` | Temperature for model generation | Provider default |
+| `--top-p <value>` | Top-p sampling value | Provider default |
+| `--presence-penalty <value>` | Presence penalty value | Provider default |
+| `--frequency-penalty <value>` | Frequency penalty value | Provider default |
+| `--log-level <level>` | Set logging level (ERROR, WARN, INFO, DEBUG, TRACE) | INFO |
+| `--input`, `-i <input>` | Specify input directly | None |
+
+#### Examples
+
+```bash
+# Run in chat mode (interactive)
+pnpm start -- --chat
+
+# Set logging level
+pnpm start -- --log-level DEBUG
+
+# Use pipeline input
+echo "Write a function to validate email addresses" | pnpm start
 ```
 
 ## Example

package/index.ts

@@ -1,12 +1,9 @@
 #!/usr/bin/env bunwrapper
 
-import { runChatLoopInTerminal } from "@aigne/cli/utils/run-chat-loop.js";
-import { AIAgent, AIGNE, UserAgent, UserInputTopic, UserOutputTopic } from "@aigne/core";
-import { loadModel } from "@aigne/core/loader/index.js";
+import { runWithAIGNE } from "@aigne/cli/utils/run-with-aigne.js";
+import { AIAgent, UserAgent, UserInputTopic, UserOutputTopic } from "@aigne/core";
 import { z } from "zod";
 
-const model = await loadModel();
-
 const coder = AIAgent.from({
   name: "coder",
   subscribeTopic: [UserInputTopic, "rewrite_request"],
@@ -67,15 +64,22 @@ Please review the code. If previous feedback was provided, see if it was address
   includeInputInOutput: true,
 });
 
-const aigne = new AIGNE({ model, agents: [coder, reviewer] });
+await runWithAIGNE(
+  (aigne) => {
+    aigne.addAgent(coder, reviewer);
 
-const userAgent = UserAgent.from({
-  context: aigne.newContext(),
-  publishTopic: UserInputTopic,
-  subscribeTopic: UserOutputTopic,
-});
+    const userAgent = UserAgent.from({
+      context: aigne.newContext(),
+      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.`,
-  defaultQuestion: "Write a function to find the sum of all even numbers in a list.",
-});
+    return userAgent;
+  },
+  {
+    chatLoopOptions: {
+      welcome: `Hello, I'm a coder with a reviewer. I can help you write code and get it reviewed.`,
+      defaultQuestion: "Write a function to find the sum of all even numbers in a list.",
+    },
+  },
+);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-workflow-reflection",
-  "version": "1.9.1",
+  "version": "1.10.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,13 +16,14 @@
     "README.md"
   ],
   "dependencies": {
-    "openai": "^4.97.0",
     "zod": "^3.24.4",
-    "@aigne/cli": "^1.9.0",
-    "@aigne/core": "^1.14.0"
+    "@aigne/cli": "^1.10.0",
+    "@aigne/core": "^1.16.0",
+    "@aigne/openai": "^0.1.0"
   },
   "devDependencies": {
-    "@aigne/test-utils": "^0.2.0"
+    "@types/bun": "^1.2.9",
+    "@aigne/test-utils": "^0.3.0"
   },
   "scripts": {
     "start": "bun run index.ts",

package/usages.ts

@@ -1,6 +1,6 @@
 import assert from "node:assert";
 import { AIAgent, AIGNE, UserInputTopic, UserOutputTopic } from "@aigne/core";
-import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
+import { OpenAIChatModel } from "@aigne/openai";
 import { z } from "zod";
 
 const { OPENAI_API_KEY } = process.env;