@aigne/example-mcp-puppeteer 1.11.1 → 1.12.0

package/README.md

@@ -1,6 +1,6 @@
 # Puppeteer MCP Server Demo
 
-This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) and [Puppeteer MCP Server](https://github.com/modelcontextprotocol/servers/tree/8bd41eb0b3cf48aea0d1fe5b6c7029736092dcb1/src/puppeteer) to extract content from websites using Puppeteer.
+This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) and [Puppeteer MCP Server](https://github.com/modelcontextprotocol/servers/tree/8bd41eb0b3cf48aea0d1fe5b6c7029736092dcb1/src/puppeteer) to extract content from websites using Puppeteer. The example now supports both one-shot and interactive chat modes, along with customizable model settings and pipeline input/output.
 
 ```mermaid
 flowchart LR
@@ -66,7 +66,14 @@ AI ->> User: The content is as follows: ...
 ```bash
 export OPENAI_API_KEY=YOUR_OPENAI_API_KEY # Set your OpenAI API key
 
-npx -y @aigne/example-mcp-puppeteer # Run the example
+# Run in one-shot mode (default)
+npx -y @aigne/example-mcp-puppeteer
+
+# Run in interactive chat mode
+npx -y @aigne/example-mcp-puppeteer --chat
+
+# Use pipeline input
+echo "extract content from https://www.arcblock.io" | npx -y @aigne/example-mcp-puppeteer
 ```
 
 ## Installation
@@ -99,6 +106,34 @@ OPENAI_API_KEY="" # Set your OpenAI API key here
 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 "extract content from https://www.arcblock.io" | pnpm start
+```
+
 ## Example
 
 The following example demonstrates how to extract content from a website:

package/index.ts

@@ -1,41 +1,36 @@
 #!/usr/bin/env bunwrapper
 
-import { runChatLoopInTerminal } from "@aigne/cli/utils/run-chat-loop.js";
-import { AIAgent, AIGNE, MCPAgent } from "@aigne/core";
-import { loadModel } from "@aigne/core/loader/index.js";
-import { logger } from "@aigne/core/utils/logger.js";
-
-logger.enable(`aigne:mcp,${process.env.DEBUG}`);
-
-const model = await loadModel();
-
-const puppeteer = await MCPAgent.from({
-  command: "npx",
-  args: ["-y", "@modelcontextprotocol/server-puppeteer"],
-  env: process.env as Record<string, string>,
-});
-
-const aigne = new AIGNE({
-  model,
-  skills: [puppeteer],
-});
-
-const agent = AIAgent.from({
-  name: "example_puppeteer",
-  instructions: `\
-## Steps to extract content from a website
-1. navigate to the url
-2. evaluate document.body.innerText to get the content
-`,
-  memory: true,
-});
-
-const userAgent = aigne.invoke(agent);
-
-await runChatLoopInTerminal(userAgent, {
-  welcome:
-    "Hello! I'm a chatbot that can extract content from a website. Try asking me a question!",
-  defaultQuestion: "What is the content of https://www.arcblock.io",
-});
+import { runWithAIGNE } from "@aigne/cli/utils/run-with-aigne.js";
+import { AIAgent, MCPAgent } from "@aigne/core";
+
+await runWithAIGNE(
+  async () => {
+    const puppeteer = await MCPAgent.from({
+      command: "npx",
+      args: ["-y", "@modelcontextprotocol/server-puppeteer"],
+      env: process.env as Record<string, string>,
+    });
+
+    const agent = AIAgent.from({
+      name: "example_puppeteer",
+      instructions: `\
+  ## Steps to extract content from a website
+  1. navigate to the url
+  2. evaluate document.body.innerText to get the content
+  `,
+      skills: [puppeteer],
+      memory: true,
+    });
+
+    return agent;
+  },
+  {
+    chatLoopOptions: {
+      welcome:
+        "Hello! I'm a chatbot that can extract content from a website. Try asking me a question!",
+      defaultQuestion: "What is the content of https://www.arcblock.io",
+    },
+  },
+);
 
 process.exit(0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-mcp-puppeteer",
-  "version": "1.11.1",
+  "version": "1.12.0",
   "description": "A demonstration of using AIGNE Framework and Puppeteer MCP Server to extract content from websites using Puppeteer",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/mcp-puppeteer",
@@ -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, AIGNE, MCPAgent } 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");