@aigne/example-mcp-sqlite 1.11.1 → 1.12.1

package/README.md

@@ -1,6 +1,6 @@
 # Sqlite MCP Server Demo
 
-This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) and [MCP Server SQlite](https://github.com/modelcontextprotocol/servers/tree/main/src/sqlite) to interact with SQLite databases.
+This is a demonstration of using [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) and [MCP Server SQlite](https://github.com/modelcontextprotocol/servers/tree/main/src/sqlite) to interact with SQLite databases. The example now supports both one-shot and interactive chat modes, along with customizable model settings and pipeline input/output.
 
 ```mermaid
 flowchart LR
@@ -71,7 +71,14 @@ AI ->> User: There are 10 products in the database.
 ```bash
 export OPENAI_API_KEY=YOUR_OPENAI_API_KEY # Set your OpenAI API key
 
-npx -y @aigne/example-mcp-sqlite # Run the example
+# Run in one-shot mode (default)
+npx -y @aigne/example-mcp-sqlite
+
+# Run in interactive chat mode
+npx -y @aigne/example-mcp-sqlite --chat
+
+# Use pipeline input
+echo "create a product table with columns name description and createdAt" | npx -y @aigne/example-mcp-sqlite
 ```
 
 ## Installation
@@ -101,7 +108,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 "create a product table with columns name description and createdAt" | 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 many products?" | pnpm start
 ```
 
 ## Example

package/index.ts

@@ -1,42 +1,37 @@
 #!/usr/bin/env bunwrapper
 
 import { join } from "node:path";
-import { runChatLoopInTerminal } from "@aigne/cli/utils/run-chat-loop.js";
-import { AIAgent, AIGNE, MCPAgent, PromptBuilder } 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 sqlite = await MCPAgent.from({
-  command: "uvx",
-  args: [
-    "-q",
-    "mcp-server-sqlite",
-    "--db-path",
-    join(process.cwd(), "aigne-example-sqlite-mcp-server.db"),
-  ],
-});
-
-const prompt = await sqlite.prompts["mcp-demo"]?.invoke({ topic: "product service" });
-if (!prompt) throw new Error("Prompt mcp-demo not found");
-
-const aigne = new AIGNE({
-  model,
-  skills: [sqlite],
-});
-
-const agent = AIAgent.from({
-  instructions: PromptBuilder.from(prompt),
-  memory: true,
-});
-
-const userAgent = aigne.invoke(agent);
-
-await runChatLoopInTerminal(userAgent, {
-  initialCall: {},
-});
+import { runWithAIGNE } from "@aigne/cli/utils/run-with-aigne.js";
+import { AIAgent, MCPAgent, PromptBuilder } from "@aigne/core";
+
+await runWithAIGNE(
+  async () => {
+    const sqlite = await MCPAgent.from({
+      command: "uvx",
+      args: [
+        "-q",
+        "mcp-server-sqlite",
+        "--db-path",
+        join(process.cwd(), "aigne-example-sqlite-mcp-server.db"),
+      ],
+    });
+
+    const prompt = await sqlite.prompts["mcp-demo"]?.invoke({ topic: "product service" });
+    if (!prompt) throw new Error("Prompt mcp-demo not found");
+
+    const agent = AIAgent.from({
+      instructions: PromptBuilder.from(prompt),
+      skills: [sqlite],
+      memory: true,
+    });
+
+    return agent;
+  },
+  {
+    chatLoopOptions: {
+      initialCall: {},
+    },
+  },
+);
 
 process.exit(0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/example-mcp-sqlite",
-  "version": "1.11.1",
+  "version": "1.12.1",
   "description": "A demonstration of using AIGNE Framework and Sqlite MCP Server to interact with a SQLite database",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/mcp-sqlite",
@@ -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.1",
+    "@aigne/openai": "^0.2.0",
+    "@aigne/core": "^1.17.0"
   },
   "devDependencies": {
-    "@aigne/test-utils": "^0.2.0"
+    "@types/bun": "^1.2.9",
+    "@aigne/test-utils": "^0.3.1"
   },
   "scripts": {
     "start": "bun run index.ts",

package/usages.ts

@@ -1,7 +1,7 @@
 import assert from "node:assert";
 import { join } from "node:path";
 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");