@aigne/example-workflow-router 1.11.1 → 1.12.0

package/README.md

@@ -1,6 +1,6 @@
 # Workflow Router Demo
 
-This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) to build a router workflow.
+This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) to build a router workflow. The example now supports both one-shot and interactive chat modes, along with customizable model settings and pipeline input/output.
 
 ```mermaid
 flowchart LR
@@ -40,7 +40,14 @@ class other processing
 ```bash
 export OPENAI_API_KEY=YOUR_OPENAI_API_KEY # Set your OpenAI API key
 
-npx -y @aigne/example-workflow-router # Run the example
+# Run in one-shot mode (default)
+npx -y @aigne/example-workflow-router
+
+# Run in interactive chat mode
+npx -y @aigne/example-workflow-router --chat
+
+# Use pipeline input
+echo "How do I return a product?" | npx -y @aigne/example-workflow-router
 ```
 
 ## Installation
@@ -70,7 +77,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 "How do I return a product?" | 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 "How do I return a product?" | pnpm start
 ```
 
 ## Example

package/index.ts

@@ -1,10 +1,7 @@
 #!/usr/bin/env bunwrapper
 
-import { runChatLoopInTerminal } from "@aigne/cli/utils/run-chat-loop.js";
-import { AIAgent, AIAgentToolChoice, AIGNE } from "@aigne/core";
-import { loadModel } from "@aigne/core/loader/index.js";
-
-const model = await loadModel();
+import { runWithAIGNE } from "@aigne/cli/utils/run-with-aigne.js";
+import { AIAgent, AIAgentToolChoice } from "@aigne/core";
 
 const productSupport = AIAgent.from({
   name: "product_support",
@@ -43,12 +40,9 @@ Be concise, accurate, and ensure efficient handoff to the correct agent.`,
   toolChoice: AIAgentToolChoice.router,
 });
 
-const aigne = new AIGNE({ model });
-
-const userAgent = aigne.invoke(triage);
-
-await runChatLoopInTerminal(userAgent, {
-  welcome: `Welcome to the support chat!
+await runWithAIGNE(triage, {
+  chatLoopOptions: {
+    welcome: `Welcome to the support chat!
 
 I can help you with any questions you have, such as
 - product-related queries: "How do I use this product?"
@@ -57,5 +51,6 @@ I can help you with any questions you have, such as
 
 How can I assist you today?
 `,
-  defaultQuestion: "How do I use this product?",
+    defaultQuestion: "How do I use this product?",
+  },
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-workflow-router",
-  "version": "1.11.1",
+  "version": "1.12.0",
   "description": "A demonstration of using AIGNE Framework to build a router workflow",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/workflow-router",
@@ -16,13 +16,13 @@
     "README.md"
   ],
   "dependencies": {
-    "openai": "^4.97.0",
-    "zod": "^3.24.4",
-    "@aigne/cli": "^1.9.1",
-    "@aigne/core": "^1.15.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, AIAgentToolChoice, AIGNE } from "@aigne/core";
-import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
+import { OpenAIChatModel } from "@aigne/openai";
 
 const { OPENAI_API_KEY } = process.env;
 assert(OPENAI_API_KEY, "Please set the OPENAI_API_KEY environment variable");