@aigne/example-mcp-blocklet 1.7.2 → 1.8.0

package/LICENSE

@@ -0,0 +1,93 @@
+Elastic License 2.0
+
+URL: https://www.elastic.co/licensing/elastic-license
+
+## Acceptance
+
+By using the software, you agree to all of the terms and conditions below.
+
+## Copyright License
+
+The licensor grants you a non-exclusive, royalty-free, worldwide,
+non-sublicensable, non-transferable license to use, copy, distribute, make
+available, and prepare derivative works of the software, in each case subject to
+the limitations and conditions below.
+
+## Limitations
+
+You may not provide the software to third parties as a hosted or managed
+service, where the service provides users with access to any substantial set of
+the features or functionality of the software.
+
+You may not move, change, disable, or circumvent the license key functionality
+in the software, and you may not remove or obscure any functionality in the
+software that is protected by the license key.
+
+You may not alter, remove, or obscure any licensing, copyright, or other notices
+of the licensor in the software. Any use of the licensor’s trademarks is subject
+to applicable law.
+
+## Patents
+
+The licensor grants you a license, under any patent claims the licensor can
+license, or becomes able to license, to make, have made, use, sell, offer for
+sale, import and have imported the software, in each case subject to the
+limitations and conditions in this license. This license does not cover any
+patent claims that you cause to be infringed by modifications or additions to
+the software. If you or your company make any written claim that the software
+infringes or contributes to infringement of any patent, your patent license for
+the software granted under these terms ends immediately. If your company makes
+such a claim, your patent license ends immediately for work on behalf of your
+company.
+
+## Notices
+
+You must ensure that anyone who gets a copy of any part of the software from you
+also gets a copy of these terms.
+
+If you modify the software, you must include in any modified copies of the
+software prominent notices stating that you have modified the software.
+
+## No Other Rights
+
+These terms do not imply any licenses other than those expressly granted in
+these terms.
+
+## Termination
+
+If you use the software in violation of these terms, such use is not licensed,
+and your licenses will automatically terminate. If the licensor provides you
+with a notice of your violation, and you cease all violation of this license no
+later than 30 days after you receive that notice, your licenses will be
+reinstated retroactively. However, if you violate these terms after such
+reinstatement, any additional violation of these terms will cause your licenses
+to terminate automatically and permanently.
+
+## No Liability
+
+*As far as the law allows, the software comes as is, without any warranty or
+condition, and the licensor will not be liable to you for any damages arising
+out of these terms or the use or nature of the software, under any kind of
+legal claim.*
+
+## Definitions
+
+The **licensor** is the entity offering these terms, and the **software** is the
+software the licensor makes available under these terms, including any portion
+of it.
+
+**you** refers to the individual or entity agreeing to these terms.
+
+**your company** is any legal entity, sole proprietorship, or other kind of
+organization that you work for, plus all organizations that have control over,
+are under the control of, or are under common control with that
+organization. **control** means ownership of substantially all the assets of an
+entity, or the power to direct its management and policies by vote, contract, or
+otherwise. Control can be direct or indirect.
+
+**your licenses** are all the licenses granted to you for the software under
+these terms.
+
+**use** means anything you do with the software requiring one of your licenses.
+
+**trademark** means trademarks, service marks, and similar rights.

package/README.md

@@ -49,6 +49,12 @@ BLOCKLET_APP_URL="" # Set your Blocklet app URL here
 pnpm start
 ```
 
+or
+
+```bash
+pnpm start https://your-blocklet-app-url
+```
+
 ## License
 
 This project is licensed under the MIT License.

package/index.ts

@@ -2,11 +2,11 @@
 
 import assert from "node:assert";
 import { runChatLoopInTerminal } from "@aigne/cli/utils/run-chat-loop.js";
-import { AIAgent, ExecutionEngine, MCPAgent, PromptBuilder } from "@aigne/core";
+import { AIAgent, AIGNE, MCPAgent, PromptBuilder } from "@aigne/core";
 import { loadModel } from "@aigne/core/loader/index.js";
 import { logger } from "@aigne/core/utils/logger.js";
 import { UnauthorizedError, refreshAuthorization } from "@modelcontextprotocol/sdk/client/auth.js";
-import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 // @ts-ignore
 import JWT from "jsonwebtoken";
 
@@ -14,20 +14,28 @@ import { TerminalOAuthProvider } from "./oauth.js";
 
 logger.enable(`aigne:mcp,${process.env.DEBUG}`);
 
-const { BLOCKLET_APP_URL } = process.env;
-assert(BLOCKLET_APP_URL, "Please set the BLOCKLET_APP_URL environment variable");
-console.info("Connecting to blocklet app", BLOCKLET_APP_URL);
+const rawUrl = process.argv[2] || process.env.BLOCKLET_APP_URL;
+assert(
+  rawUrl,
+  "Please provide a blocklet url as an argument or set the BLOCKLET_APP_URL environment variable",
+);
 
-const appUrl = new URL(BLOCKLET_APP_URL);
-appUrl.pathname = "/.well-known/service/mcp/sse";
+const appUrl = new URL(rawUrl);
+appUrl.pathname = "/.well-known/service/mcp";
+console.info("Connecting to blocklet", appUrl.href);
+
+let transport: StreamableHTTPClientTransport;
 
 const provider = new TerminalOAuthProvider(appUrl.host);
 const authCodePromise = new Promise((resolve, reject) => {
-  provider.once("authorized", resolve);
+  provider.once("authorized", async (code) => {
+    await transport.finishAuth(code);
+    resolve(code);
+  });
   provider.once("error", reject);
 });
 
-const transport = new SSEClientTransport(appUrl, {
+transport = new StreamableHTTPClientTransport(appUrl, {
   authProvider: provider,
 });
 
@@ -35,22 +43,25 @@ try {
   let tokens = await provider.tokens();
   if (tokens) {
     let decoded = JWT.decode(tokens.access_token);
-    console.info("Decoded access token:", decoded);
     if (decoded) {
       const now = Date.now();
       const expiresAt = decoded.exp * 1000;
       if (now < expiresAt) {
         console.info("Tokens already exist and not expired, skipping authorization");
       } else if (tokens.refresh_token) {
+        console.info("Access token expired:", { now, expiresAt, decoded });
         decoded = JWT.decode(tokens.refresh_token);
-        console.info("Decoded refresh token:", decoded);
         if (decoded) {
           const now = Date.now();
           const expiresAt = decoded.exp * 1000;
           if (now < expiresAt) {
             console.info("Refresh token already exists and not expired, refreshing authorization");
             try {
+              const oauthUrl = new URL(appUrl.href);
+              oauthUrl.pathname = "/.well-known/oauth-authorization-server";
+              const metadata = await fetch(oauthUrl.href).then((res) => res.json());
               tokens = await refreshAuthorization(appUrl.href, {
+                metadata,
                 // biome-ignore lint/style/noNonNullAssertion: <explanation>
                 clientInformation: (await provider.clientInformation())!,
                 refreshToken: tokens.refresh_token,
@@ -72,7 +83,7 @@ try {
       }
     }
   } else {
-    console.info("No tokens found, starting authorization");
+    console.info("No tokens found, starting authorization...");
     await transport.start();
   }
 } catch (error) {
@@ -87,21 +98,20 @@ try {
   }
 }
 
-console.info("Starting connecting to blocklet mcp...");
-
 const model = await loadModel();
 
 const blocklet = await MCPAgent.from({
   url: appUrl.href,
   timeout: 8000,
+  transport: "streamableHttp",
   opts: {
     authProvider: provider,
   },
 });
 
-const engine = new ExecutionEngine({
+const aigne = new AIGNE({
   model,
-  tools: [blocklet],
+  skills: [blocklet],
 });
 
 const agent = AIAgent.from({
@@ -111,7 +121,7 @@ const agent = AIAgent.from({
   memory: true,
 });
 
-const userAgent = engine.call(agent);
+const userAgent = aigne.invoke(agent);
 
 await runChatLoopInTerminal(userAgent, {
   welcome:

package/oauth.ts

@@ -147,8 +147,10 @@ export class TerminalOAuthProvider extends EventEmitter implements OAuthClientPr
 
           if (code) {
             this.emit("authorized", code);
-            console.info("Authorization successful!", Date.now());
-            resolve();
+            console.info("Authorization code received, exchanging for tokens...");
+            setTimeout(() => {
+              resolve();
+            }, 3000);
           } else {
             this.emit("error", new Error("No authorization code received"));
             reject(new Error("No authorization code received"));
@@ -158,7 +160,7 @@ export class TerminalOAuthProvider extends EventEmitter implements OAuthClientPr
 
       // Start the local server
       server.listen(this.localServerPort, async () => {
-        console.log("Please authorize the application in your browser...");
+        console.log("Please complete the authorization in your browser...");
         // Open the authorization URL in the default browser
         await open(authorizationUrl.toString());
       });

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@aigne/example-mcp-blocklet",
-  "version": "1.7.2",
+  "version": "1.8.0",
   "description": "A demonstration of using AIGNE Framework and MCP Server hosted by the Blocklet platform",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/AIGNE-io/aigne-framework/tree/main/examples/mcp-blocklet",
-  "license": "ISC",
+  "license": "MIT",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/AIGNE-io/aigne-framework"
@@ -20,8 +20,11 @@
     "jsonwebtoken": "^9.0.2",
     "open": "^10.1.0",
     "zod": "^3.24.2",
-    "@aigne/cli": "^1.6.0",
-    "@aigne/core": "^1.10.0"
+    "@aigne/core": "^1.12.0",
+    "@aigne/cli": "^1.8.0"
+  },
+  "devDependencies": {
+    "@aigne/test-utils": "^0.1.0"
   },
   "scripts": {
     "start": "bun run index.ts",