@autonomys/auto-mcp-servers 0.1.1-alpha.8 → 0.1.2

package/README.md

@@ -1,95 +1,101 @@
-# Autonomys MCP Servers
-
-This package provides Model Context Protocol (MCP) servers for Autonomys services. These servers expose Autonomys SDK functionality as MCP tools that can be used with Claude Desktop and other MCP clients, or integrated into agent frameworks.
-
-## Available Servers
-
-### Auto Drive Server
-
-The Auto Drive server exposes functionality from the `@autonomys/auto-drive` package as MCP tools.
-
-#### Tools
-
-The Auto Drive server provides the following tools:
-
-- `uploadObject`: Upload an object to the Autonomys network.
-
-More servers and tools will be coming soon!
-
-## Usage
-
-### With Claude Desktop
-
-1. Install Claude Desktop
-2. Edit your `claude_desktop_config.json` file:
-
-```json
-{
-  "mcpServers": {
-    "auto-drive": {
-      "command": "npx",
-      "args": ["-y", "@autonomys/auto-mcp-servers", "auto-drive"],
-      "env": { "AUTO_DRIVE_API_KEY": "your-api-key" }
-    }
-  }
-}
-```
-
-4. Restart Claude Desktop
-
-### In Agent Frameworks
-
-You can use these MCP servers as tools with agent frameworks such as LangChain.
-
-1. Install the Auto Drive server into your project:
-
-```bash
-npm install @autonomys/auto-mcp-servers
-# or
-yarn add @autonomys/auto-mcp-servers
-```
-
-2. Use the Auto Drive server as a tool in your agent project:
-
-```typescript
-import { Client } from '@modelcontextprotocol/sdk/client/index.js'
-import {
-  StdioClientTransport,
-  StdioServerParameters,
-} from '@modelcontextprotocol/sdk/client/stdio.js'
-import { loadMcpTools } from '@langchain/mcp-adapters'
-import { StructuredToolInterface } from '@langchain/core/tools'
-import { ChatOpenAI } from '@langchain/openai'
-
-// Create Auto Drive tools with a single function
-const createAutoDriveTools = async (apiKey: string): Promise<StructuredToolInterface[]> => {
-  // Set up transport for Auto Drive server
-  const transport = new StdioClientTransport({
-    command: process.execPath,
-    args: ['node_modules/.bin/auto-drive-server'],
-    env: { AUTO_DRIVE_API_KEY: apiKey },
-  })
-
-  // Initialize client and connect
-  const client = new Client({ name: 'auto-drive', version: '0.0.1' })
-  client.connect(transport)
-
-  // Load MCP tools
-  return await loadMcpTools('auto-drive', client)
-}
-
-// Create tools with your API key
-const tools = await createAutoDriveTools('your-api-key')
-
-// Initialize the ChatOpenAI model
-const model = new ChatOpenAI({ modelName: 'gpt-4o' })
-
-const result = await model
-  .bindTools(tools)
-  .invoke('Upload a profound thought to the Autonomys network')
-console.log(result)
-```
-
-## License
-
-MIT
+# Autonomys MCP Servers
+
+This package provides Model Context Protocol (MCP) servers for Autonomys services. These servers expose Autonomys SDK functionality as MCP tools that can be used with Claude Desktop and other MCP clients, or integrated into agent frameworks.
+
+## Available Servers
+
+### Auto Drive Server
+
+The Auto Drive server exposes functionality from the `@autonomys/auto-drive` package as MCP tools.
+
+#### Tools
+
+The Auto Drive server provides the following tools:
+
+- `upload-object`: Upload an object (as JSON data) to the Autonomys network.
+- `download-object`: Download a text-based object (`text/*` or `application/json`) from the Autonomys network using its CID.
+- `search-objects`: Search for objects on the Autonomys network by name or CID fragment. Returns a JSON object containing an array of results, each including the object's name, CID, type, size, and mimeType (for files).
+
+More servers and tools will be coming soon!
+
+## Usage
+
+### With Claude Desktop
+
+1. Install Claude Desktop
+2. Edit your `claude_desktop_config.json` file:
+
+```json
+{
+  "mcpServers": {
+    "auto-drive": {
+      "command": "npx",
+      "args": ["-y", "@autonomys/auto-mcp-servers", "auto-drive"],
+      "env": {
+        "AUTO_DRIVE_API_KEY": "your-api-key",
+        "ENCRYPTION_PASSWORD": "my-password (optional)",
+        "NETWORK": "mainnet or taurus (optional, defaults to mainnet)"
+      }
+    }
+  }
+}
+```
+
+4. Restart Claude Desktop
+
+### In Agent Frameworks
+
+You can use these MCP servers as tools with agent frameworks such as LangChain.
+
+1. Install the Auto Drive server into your project:
+
+```bash
+npm install @autonomys/auto-mcp-servers
+# or
+yarn add @autonomys/auto-mcp-servers
+```
+
+2. Use the Auto Drive server as a tool in your agent project:
+
+```typescript
+import { Client } from '@modelcontextprotocol/sdk/client/index.js'
+import {
+  StdioClientTransport,
+  StdioServerParameters,
+} from '@modelcontextprotocol/sdk/client/stdio.js'
+import { loadMcpTools } from '@langchain/mcp-adapters'
+import { StructuredToolInterface } from '@langchain/core/tools'
+import { ChatOpenAI } from '@langchain/openai'
+
+// Create Auto Drive tools with a single function
+const createAutoDriveTools = async (apiKey: string): Promise<StructuredToolInterface[]> => {
+  // Set up transport for Auto Drive server
+  const transport = new StdioClientTransport({
+    command: process.execPath,
+    args: ['node_modules/.bin/auto-drive-server'],
+    env: { AUTO_DRIVE_API_KEY: apiKey },
+  })
+
+  // Initialize client and connect
+  const client = new Client({ name: 'auto-drive', version: '0.0.1' })
+  client.connect(transport)
+
+  // Load MCP tools
+  return await loadMcpTools('auto-drive', client)
+}
+
+// Create tools with your API key
+const tools = await createAutoDriveTools('your-api-key')
+
+// Initialize the ChatOpenAI model
+const model = new ChatOpenAI({ modelName: 'gpt-4o' })
+
+const result = await model
+  .bindTools(tools)
+  .invoke('Upload a profound thought to the Autonomys network')
+console.log(result)
+```
+
+## License
+
+MIT

package/dist/auto-drive/handlers.d.ts

@@ -0,0 +1,14 @@
+import { CallToolResult } from '@modelcontextprotocol/sdk/types.js';
+export declare const createAutoDriveHandlers: (apiKey: string, network: "taurus" | "mainnet", encryptionPassword?: string) => {
+    uploadObjectHandler: ({ filename, data, }: {
+        filename: string;
+        data: Record<string, any>;
+    }) => Promise<CallToolResult>;
+    downloadObjectHandler: ({ cid }: {
+        cid: string;
+    }) => Promise<CallToolResult>;
+    searchObjectsHandler: ({ query }: {
+        query: string;
+    }) => Promise<CallToolResult>;
+};
+//# sourceMappingURL=handlers.d.ts.map

package/dist/auto-drive/handlers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handlers.d.ts","sourceRoot":"","sources":["../../src/auto-drive/handlers.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAA;AAEnE,eAAO,MAAM,uBAAuB,GAClC,QAAQ,MAAM,EACd,SAAS,QAAQ,GAAG,SAAS,EAC7B,qBAAqB,MAAM;+CAWtB;QACD,QAAQ,EAAE,MAAM,CAAA;QAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAC1B,KAAG,OAAO,CAAC,cAAc,CAAC;qCAIY;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,KAAG,OAAO,CAAC,cAAc,CAAC;sCAsExC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,KAAG,OAAO,CAAC,cAAc,CAAC;CActF,CAAA"}

package/dist/auto-drive/handlers.js

@@ -0,0 +1,117 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
+import { createAutoDriveApi } from '@autonomys/auto-drive';
+export const createAutoDriveHandlers = (apiKey, network, encryptionPassword) => {
+    const autoDriveApi = createAutoDriveApi({ apiKey, network });
+    const uploadOptions = encryptionPassword
+        ? { password: encryptionPassword }
+        : undefined;
+    return {
+        uploadObjectHandler: (_a) => __awaiter(void 0, [_a], void 0, function* ({ filename, data, }) {
+            const cid = yield autoDriveApi.uploadObjectAsJSON({ data }, filename, uploadOptions);
+            return { content: [{ type: 'text', text: `Object uploaded successfully with ${cid}` }] };
+        }),
+        downloadObjectHandler: (_a) => __awaiter(void 0, [_a], void 0, function* ({ cid }) {
+            var _b, e_1, _c, _d;
+            try {
+                // Get object summary to determine type and metadata
+                const summaries = yield autoDriveApi.searchByNameOrCID(cid);
+                if (!summaries || summaries.length === 0) {
+                    throw new Error(`Object not found for CID: ${cid}`);
+                }
+                const summary = summaries[0];
+                // Handle folders
+                if (summary.type === 'folder') {
+                    return {
+                        isError: true,
+                        content: [
+                            {
+                                type: 'text',
+                                text: `Error: CID ${cid} points to a folder, which cannot be downloaded directly with this tool.`,
+                            },
+                        ],
+                    };
+                }
+                const mimeType = summary.mimeType || 'application/octet-stream';
+                const filename = summary.name || `download-${cid}`;
+                // Handle only text-based types
+                if (mimeType.startsWith('text/') || mimeType === 'application/json') {
+                    const stream = yield autoDriveApi.downloadFile(cid, encryptionPassword);
+                    let fileBuffer = Buffer.alloc(0);
+                    try {
+                        for (var _e = true, stream_1 = __asyncValues(stream), stream_1_1; stream_1_1 = yield stream_1.next(), _b = stream_1_1.done, !_b; _e = true) {
+                            _d = stream_1_1.value;
+                            _e = false;
+                            const chunk = _d;
+                            fileBuffer = Buffer.concat([fileBuffer, chunk]);
+                        }
+                    }
+                    catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                    finally {
+                        try {
+                            if (!_e && !_b && (_c = stream_1.return)) yield _c.call(stream_1);
+                        }
+                        finally { if (e_1) throw e_1.error; }
+                    }
+                    try {
+                        const text = fileBuffer.toString('utf-8');
+                        return {
+                            content: [{ type: 'text', text }],
+                        };
+                    }
+                    catch (e) {
+                        console.error(`Failed to decode text content for ${filename} (${mimeType})`, e);
+                        return {
+                            isError: true,
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: `Error: Failed to decode file content for ${filename} as UTF-8 text.`,
+                                },
+                            ],
+                        };
+                    }
+                }
+                // For all other types, return an error
+                return {
+                    isError: true,
+                    content: [
+                        {
+                            type: 'text',
+                            text: `Error: File type \'${mimeType}\' is not supported for direct download in this client. Only text/* and application/json are supported.`,
+                        },
+                    ],
+                };
+            }
+            catch (error) {
+                console.error(`Failed to download object with CID ${cid}:`, error);
+                const errorMessage = error.message || 'Unknown error occurred during download.';
+                return {
+                    isError: true,
+                    content: [{ type: 'text', text: `Error downloading object: ${errorMessage}` }],
+                };
+            }
+        }),
+        searchObjectsHandler: (_a) => __awaiter(void 0, [_a], void 0, function* ({ query }) {
+            const summaries = yield autoDriveApi.searchByNameOrCID(query);
+            const results = summaries.map((s) => (Object.assign({ name: s.name, cid: s.headCid, type: s.type, size: s.size }, (s.type === 'file' && { mimeType: s.mimeType }))));
+            return {
+                content: [{ type: 'text', text: JSON.stringify({ results }, null, 2) }],
+            };
+        }),
+    };
+};

package/dist/auto-drive/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/auto-drive/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAA;AAanE,eAAO,MAAM,eAAe,WAA0D,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/auto-drive/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAA;AAgBnE,eAAO,MAAM,eAAe,WAA0D,CAAA"}

package/dist/auto-drive/index.js

@@ -7,122 +7,34 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-var __asyncValues = (this && this.__asyncValues) || function (o) {
-    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-    var m = o[Symbol.asyncIterator], i;
-    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
-    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
-    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
-};
-import { createAutoDriveApi } from '@autonomys/auto-drive';
+var _a;
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { z } from 'zod';
-const AUTO_DRIVE_API_KEY = process.env.AUTO_DRIVE_API_KEY;
+import { createAutoDriveHandlers } from './handlers.js';
+const AUTO_DRIVE_API_KEY = (_a = process.env.AUTO_DRIVE_API_KEY) !== null && _a !== void 0 ? _a : (() => {
+    throw new Error('AUTO_DRIVE_API_KEY environment variable is not set');
+})();
 const NETWORK = process.env.NETWORK === 'taurus' ? 'taurus' : 'mainnet';
 const ENCRYPTION_PASSWORD = process.env.ENCRYPTION_PASSWORD;
-const uploadOptions = ENCRYPTION_PASSWORD
-    ? { password: ENCRYPTION_PASSWORD }
-    : undefined;
-const autoDriveApi = createAutoDriveApi({ network: NETWORK, apiKey: AUTO_DRIVE_API_KEY });
-export const autoDriveServer = new McpServer({ name: 'Auto Drive', version: '0.1.1' });
-autoDriveServer.tool('upload-object', 'Upload an object permanently to Auto Drive, any objects uploaded here will be permanently available onchain. This is useful for storing data that you want to keep forever.', {
+const { uploadObjectHandler, downloadObjectHandler, searchObjectsHandler } = createAutoDriveHandlers(AUTO_DRIVE_API_KEY, NETWORK, ENCRYPTION_PASSWORD);
+export const autoDriveServer = new McpServer({ name: 'Auto Drive', version: '0.1.2' });
+autoDriveServer.tool('upload-object', 'Upload an object permanently to the Autonomys Network using Auto Drive, any objects uploaded here will be permanently available onchain. This is useful for storing data that you want to keep forever.', {
     filename: z.string().describe('The filename to save the object as.'),
-    data: z.record(z.string(), z.any()).describe(`
-      Data you want to permanently store onchain saved as a JSON object with any key-value pairs.
-      The keys are strings that describe the type of data being stored.
-      The values are the actual data being stored.
+    data: z.record(z.string(), z.any()).describe(`
+      Data you want to permanently store onchain saved as a JSON object with any key-value pairs.
+      The keys are strings that describe the type of data being stored.
+      The values are the actual data being stored.
       `),
-}, (_a, extra_1) => __awaiter(void 0, [_a, extra_1], void 0, function* ({ filename, data }, extra) {
-    if (!AUTO_DRIVE_API_KEY) {
-        throw new Error('AUTO_DRIVE_API_KEY environment variable is not set');
-    }
-    const cid = yield autoDriveApi.uploadObjectAsJSON({ data }, filename, uploadOptions);
-    return { content: [{ type: 'text', text: `Object uploaded successfully with ${cid}` }] };
+}, (_a, _extra_1) => __awaiter(void 0, [_a, _extra_1], void 0, function* ({ filename, data }, _extra) {
+    return yield uploadObjectHandler({ filename, data });
 }));
-autoDriveServer.tool('download-object', 'Download a text-based object (text/*, application/json) from Auto Drive using its Content Identifier (CID).', {
+autoDriveServer.tool('download-object', 'Download a text-based object (text/*, application/json) from the Autonomys Network using Auto Drive using its Content Identifier (CID).', {
     cid: z.string().describe('The Content Identifier (CID) of the object to download.'),
 }, (_a, _extra_1) => __awaiter(void 0, [_a, _extra_1], void 0, function* ({ cid }, _extra) {
-    var _b, e_1, _c, _d;
-    if (!AUTO_DRIVE_API_KEY) {
-        throw new Error('AUTO_DRIVE_API_KEY environment variable is not set');
-    }
-    try {
-        // Get object summary to determine type and metadata
-        const summaries = yield autoDriveApi.searchByNameOrCID(cid);
-        if (!summaries || summaries.length === 0) {
-            throw new Error(`Object not found for CID: ${cid}`);
-        }
-        const summary = summaries[0];
-        // Handle folders
-        if (summary.type === 'folder') {
-            return {
-                isError: true,
-                content: [
-                    {
-                        type: 'text',
-                        text: `Error: CID ${cid} points to a folder, which cannot be downloaded directly with this tool.`,
-                    },
-                ],
-            };
-        }
-        // It's a file, check MIME type
-        const mimeType = summary.mimeType || 'application/octet-stream';
-        const filename = summary.name || `download-${cid}`;
-        // Handle only text-based types
-        if (mimeType.startsWith('text/') || mimeType === 'application/json') {
-            const stream = yield autoDriveApi.downloadFile(cid, ENCRYPTION_PASSWORD);
-            let fileBuffer = Buffer.alloc(0);
-            try {
-                for (var _e = true, stream_1 = __asyncValues(stream), stream_1_1; stream_1_1 = yield stream_1.next(), _b = stream_1_1.done, !_b; _e = true) {
-                    _d = stream_1_1.value;
-                    _e = false;
-                    const chunk = _d;
-                    fileBuffer = Buffer.concat([fileBuffer, chunk]);
-                }
-            }
-            catch (e_1_1) { e_1 = { error: e_1_1 }; }
-            finally {
-                try {
-                    if (!_e && !_b && (_c = stream_1.return)) yield _c.call(stream_1);
-                }
-                finally { if (e_1) throw e_1.error; }
-            }
-            try {
-                const text = fileBuffer.toString('utf-8');
-                return {
-                    content: [{ type: 'text', text }],
-                };
-            }
-            catch (e) {
-                console.error(`Failed to decode text content for ${filename} (${mimeType})`, e);
-                return {
-                    isError: true,
-                    content: [
-                        {
-                            type: 'text',
-                            text: `Error: Failed to decode file content for ${filename} as UTF-8 text.`,
-                        },
-                    ],
-                };
-            }
-        }
-        // For all other types, return an error
-        return {
-            isError: true,
-            content: [
-                {
-                    type: 'text',
-                    text: `Error: File type \'${mimeType}\' is not supported for direct download in this client. Only text/* and application/json are supported.`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        console.error(`Failed to download object with CID ${cid}:`, error);
-        const errorMessage = error.message || 'Unknown error occurred during download.';
-        return {
-            isError: true,
-            content: [{ type: 'text', text: `Error downloading object: ${errorMessage}` }],
-        };
-    }
+    return yield downloadObjectHandler({ cid });
+}));
+autoDriveServer.tool('search-objects', 'Search for objects on the Autonomys Network using Auto Drive by name or CID.', {
+    query: z.string().describe('The name or CID fragment to search for.'),
+}, (_a, _extra_1) => __awaiter(void 0, [_a, _extra_1], void 0, function* ({ query }, _extra) {
+    return yield searchObjectsHandler({ query });
 }));

package/dist/bin/main.js

@@ -11,17 +11,17 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { autoDriveServer } from '../auto-drive/index.js';
 const showHelp = () => {
-    console.log(`
-Usage: auto-mcp-servers [server-name]
-
-Available servers:
-  auto-drive    Start the Auto Drive MCP server
-
-Environment variables:
-  For Auto Drive server:
-    AUTO_DRIVE_API_KEY     API key for Auto Drive (required)
-    NETWORK                'mainnet' (default) or 'taurus'
-    ENCRYPTION_PASSWORD    Password for encryption (optional)
+    console.log(`
+Usage: auto-mcp-servers [server-name]
+
+Available servers:
+  auto-drive    Start the Auto Drive MCP server
+
+Environment variables:
+  For Auto Drive server:
+    AUTO_DRIVE_API_KEY     API key for Auto Drive (required)
+    NETWORK                'mainnet' (default) or 'taurus'
+    ENCRYPTION_PASSWORD    Password for encryption (optional)
   `);
     process.exit(1);
 };

package/package.json

@@ -1,42 +1,42 @@
-{
-  "name": "@autonomys/auto-mcp-servers",
-  "packageManager": "yarn@4.7.0",
-  "version": "0.1.1-alpha.8",
-  "description": "Autonomys Network MCP servers",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/autonomys/auto-sdk"
-  },
-  "author": {
-    "name": "Autonomys",
-    "url": "https://www.autonomys.xyz"
-  },
-  "bugs": {
-    "url": "https://github.com/autonomys/auto-sdk/issues"
-  },
-  "license": "MIT",
-  "type": "module",
-  "main": "dist/index.js",
-  "bin": {
-    "auto-drive-server": "./dist/bin/auto-drive.js",
-    "auto-mcp-servers": "./dist/bin/main.js"
-  },
-  "scripts": {
-    "build": "tsc",
-    "prepublishOnly": "npm run build && chmod +x ./dist/bin/*.js"
-  },
-  "keywords": [
-    "mcp",
-    "server"
-  ],
-  "dependencies": {
-    "@autonomys/auto-drive": "^1.4.18",
-    "@modelcontextprotocol/sdk": "^1.9.0",
-    "zod": "^3.24.2"
-  },
-  "devDependencies": {
-    "@types/node": "22.14.0",
-    "ts-node": "^10.9.1",
-    "typescript": "^5.0.2"
-  }
-}
+{
+  "name": "@autonomys/auto-mcp-servers",
+  "packageManager": "yarn@4.7.0",
+  "version": "0.1.2",
+  "description": "Autonomys Network MCP servers",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/autonomys/auto-sdk"
+  },
+  "author": {
+    "name": "Autonomys",
+    "url": "https://www.autonomys.xyz"
+  },
+  "bugs": {
+    "url": "https://github.com/autonomys/auto-sdk/issues"
+  },
+  "license": "MIT",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "auto-drive-server": "./dist/bin/auto-drive.js",
+    "auto-mcp-servers": "./dist/bin/main.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "mcp",
+    "server"
+  ],
+  "dependencies": {
+    "@autonomys/auto-drive": "^1.4.18",
+    "@modelcontextprotocol/sdk": "^1.9.0",
+    "zod": "^3.24.2"
+  },
+  "devDependencies": {
+    "@types/node": "22.14.0",
+    "ts-node": "^10.9.1",
+    "typescript": "^5.0.2"
+  }
+}

package/src/auto-drive/handlers.ts

@@ -0,0 +1,110 @@
+import { createAutoDriveApi, UploadFileOptions } from '@autonomys/auto-drive'
+
+import { CallToolResult } from '@modelcontextprotocol/sdk/types.js'
+
+export const createAutoDriveHandlers = (
+  apiKey: string,
+  network: 'taurus' | 'mainnet',
+  encryptionPassword?: string,
+) => {
+  const autoDriveApi = createAutoDriveApi({ apiKey, network })
+  const uploadOptions: UploadFileOptions | undefined = encryptionPassword
+    ? { password: encryptionPassword }
+    : undefined
+
+  return {
+    uploadObjectHandler: async ({
+      filename,
+      data,
+    }: {
+      filename: string
+      data: Record<string, any>
+    }): Promise<CallToolResult> => {
+      const cid = await autoDriveApi.uploadObjectAsJSON({ data }, filename, uploadOptions)
+      return { content: [{ type: 'text', text: `Object uploaded successfully with ${cid}` }] }
+    },
+    downloadObjectHandler: async ({ cid }: { cid: string }): Promise<CallToolResult> => {
+      try {
+        // Get object summary to determine type and metadata
+        const summaries = await autoDriveApi.searchByNameOrCID(cid)
+        if (!summaries || summaries.length === 0) {
+          throw new Error(`Object not found for CID: ${cid}`)
+        }
+        const summary = summaries[0]
+
+        // Handle folders
+        if (summary.type === 'folder') {
+          return {
+            isError: true,
+            content: [
+              {
+                type: 'text',
+                text: `Error: CID ${cid} points to a folder, which cannot be downloaded directly with this tool.`,
+              },
+            ],
+          }
+        }
+
+        const mimeType = summary.mimeType || 'application/octet-stream'
+        const filename = summary.name || `download-${cid}`
+
+        // Handle only text-based types
+        if (mimeType.startsWith('text/') || mimeType === 'application/json') {
+          const stream = await autoDriveApi.downloadFile(cid, encryptionPassword)
+          let fileBuffer = Buffer.alloc(0)
+          for await (const chunk of stream) {
+            fileBuffer = Buffer.concat([fileBuffer, chunk])
+          }
+          try {
+            const text = fileBuffer.toString('utf-8')
+            return {
+              content: [{ type: 'text', text }],
+            }
+          } catch (e) {
+            console.error(`Failed to decode text content for ${filename} (${mimeType})`, e)
+            return {
+              isError: true,
+              content: [
+                {
+                  type: 'text',
+                  text: `Error: Failed to decode file content for ${filename} as UTF-8 text.`,
+                },
+              ],
+            }
+          }
+        }
+
+        // For all other types, return an error
+        return {
+          isError: true,
+          content: [
+            {
+              type: 'text',
+              text: `Error: File type \'${mimeType}\' is not supported for direct download in this client. Only text/* and application/json are supported.`,
+            },
+          ],
+        }
+      } catch (error: any) {
+        console.error(`Failed to download object with CID ${cid}:`, error)
+        const errorMessage = error.message || 'Unknown error occurred during download.'
+        return {
+          isError: true,
+          content: [{ type: 'text', text: `Error downloading object: ${errorMessage}` }],
+        }
+      }
+    },
+    searchObjectsHandler: async ({ query }: { query: string }): Promise<CallToolResult> => {
+      const summaries = await autoDriveApi.searchByNameOrCID(query)
+      const results = summaries.map((s) => ({
+        name: s.name,
+        cid: s.headCid,
+        type: s.type,
+        size: s.size,
+        ...(s.type === 'file' && { mimeType: s.mimeType }),
+      }))
+      return {
+        content: [{ type: 'text', text: JSON.stringify({ results }, null, 2) }],
+      }
+    },
+  }
+}

package/src/auto-drive/index.ts

@@ -1,123 +1,57 @@
-import { createAutoDriveApi, UploadFileOptions } from '@autonomys/auto-drive'
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
-import * as mcp_types from '@modelcontextprotocol/sdk/types.js'
-import { z } from 'zod'
-
-const AUTO_DRIVE_API_KEY = process.env.AUTO_DRIVE_API_KEY
-const NETWORK = process.env.NETWORK === 'taurus' ? 'taurus' : 'mainnet'
-const ENCRYPTION_PASSWORD = process.env.ENCRYPTION_PASSWORD
-const uploadOptions: UploadFileOptions | undefined = ENCRYPTION_PASSWORD
-  ? { password: ENCRYPTION_PASSWORD }
-  : undefined
-
-const autoDriveApi = createAutoDriveApi({ network: NETWORK, apiKey: AUTO_DRIVE_API_KEY })
-
-export const autoDriveServer = new McpServer({ name: 'Auto Drive', version: '0.1.1' })
-
-autoDriveServer.tool(
-  'upload-object',
-  'Upload an object permanently to Auto Drive, any objects uploaded here will be permanently available onchain. This is useful for storing data that you want to keep forever.',
-  {
-    filename: z.string().describe('The filename to save the object as.'),
-    data: z.record(z.string(), z.any()).describe(
-      `
-      Data you want to permanently store onchain saved as a JSON object with any key-value pairs.
-      The keys are strings that describe the type of data being stored.
-      The values are the actual data being stored.
-      `,
-    ),
-  } as any,
-  async ({ filename, data }, extra) => {
-    if (!AUTO_DRIVE_API_KEY) {
-      throw new Error('AUTO_DRIVE_API_KEY environment variable is not set')
-    }
-    const cid = await autoDriveApi.uploadObjectAsJSON({ data }, filename, uploadOptions)
-    return { content: [{ type: 'text', text: `Object uploaded successfully with ${cid}` }] }
-  },
-)
-
-autoDriveServer.tool(
-  'download-object',
-  'Download a text-based object (text/*, application/json) from Auto Drive using its Content Identifier (CID).',
-  {
-    cid: z.string().describe('The Content Identifier (CID) of the object to download.'),
-  } as any,
-  async (
-    { cid },
-    _extra,
-    // extra: RequestHandlerExtra, // Infer extra type
-  ): Promise<{ content: mcp_types.TextContent[]; isError?: boolean }> => {
-    if (!AUTO_DRIVE_API_KEY) {
-      throw new Error('AUTO_DRIVE_API_KEY environment variable is not set')
-    }
-    try {
-      // Get object summary to determine type and metadata
-      const summaries = await autoDriveApi.searchByNameOrCID(cid)
-      if (!summaries || summaries.length === 0) {
-        throw new Error(`Object not found for CID: ${cid}`)
-      }
-      const summary = summaries[0]
-
-      // Handle folders
-      if (summary.type === 'folder') {
-        return {
-          isError: true,
-          content: [
-            {
-              type: 'text',
-              text: `Error: CID ${cid} points to a folder, which cannot be downloaded directly with this tool.`,
-            },
-          ],
-        }
-      }
-
-      // It's a file, check MIME type
-      const mimeType = summary.mimeType || 'application/octet-stream'
-      const filename = summary.name || `download-${cid}`
-
-      // Handle only text-based types
-      if (mimeType.startsWith('text/') || mimeType === 'application/json') {
-        const stream = await autoDriveApi.downloadFile(cid, ENCRYPTION_PASSWORD)
-        let fileBuffer = Buffer.alloc(0)
-        for await (const chunk of stream) {
-          fileBuffer = Buffer.concat([fileBuffer, chunk])
-        }
-        try {
-          const text = fileBuffer.toString('utf-8')
-          return {
-            content: [{ type: 'text', text }],
-          }
-        } catch (e) {
-          console.error(`Failed to decode text content for ${filename} (${mimeType})`, e)
-          return {
-            isError: true,
-            content: [
-              {
-                type: 'text',
-                text: `Error: Failed to decode file content for ${filename} as UTF-8 text.`,
-              },
-            ],
-          }
-        }
-      }
-
-      // For all other types, return an error
-      return {
-        isError: true,
-        content: [
-          {
-            type: 'text',
-            text: `Error: File type \'${mimeType}\' is not supported for direct download in this client. Only text/* and application/json are supported.`,
-          },
-        ],
-      }
-    } catch (error: any) {
-      console.error(`Failed to download object with CID ${cid}:`, error)
-      const errorMessage = error.message || 'Unknown error occurred during download.'
-      return {
-        isError: true,
-        content: [{ type: 'text', text: `Error downloading object: ${errorMessage}` }],
-      }
-    }
-  },
-)
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
+import { CallToolResult } from '@modelcontextprotocol/sdk/types.js'
+import { z } from 'zod'
+import { createAutoDriveHandlers } from './handlers.js'
+
+const AUTO_DRIVE_API_KEY =
+  process.env.AUTO_DRIVE_API_KEY ??
+  (() => {
+    throw new Error('AUTO_DRIVE_API_KEY environment variable is not set')
+  })()
+const NETWORK = process.env.NETWORK === 'taurus' ? 'taurus' : 'mainnet'
+const ENCRYPTION_PASSWORD = process.env.ENCRYPTION_PASSWORD
+
+const { uploadObjectHandler, downloadObjectHandler, searchObjectsHandler } =
+  createAutoDriveHandlers(AUTO_DRIVE_API_KEY, NETWORK, ENCRYPTION_PASSWORD)
+
+export const autoDriveServer = new McpServer({ name: 'Auto Drive', version: '0.1.2' })
+
+autoDriveServer.tool(
+  'upload-object',
+  'Upload an object permanently to the Autonomys Network using Auto Drive, any objects uploaded here will be permanently available onchain. This is useful for storing data that you want to keep forever.',
+  {
+    filename: z.string().describe('The filename to save the object as.'),
+    data: z.record(z.string(), z.any()).describe(
+      `
+      Data you want to permanently store onchain saved as a JSON object with any key-value pairs.
+      The keys are strings that describe the type of data being stored.
+      The values are the actual data being stored.
+      `,
+    ),
+  } as any,
+  async ({ filename, data }, _extra): Promise<CallToolResult> => {
+    return await uploadObjectHandler({ filename, data })
+  },
+)
+
+autoDriveServer.tool(
+  'download-object',
+  'Download a text-based object (text/*, application/json) from the Autonomys Network using Auto Drive using its Content Identifier (CID).',
+  {
+    cid: z.string().describe('The Content Identifier (CID) of the object to download.'),
+  } as any,
+  async ({ cid }, _extra): Promise<CallToolResult> => {
+    return await downloadObjectHandler({ cid })
+  },
+)
+
+autoDriveServer.tool(
+  'search-objects',
+  'Search for objects on the Autonomys Network using Auto Drive by name or CID.',
+  {
+    query: z.string().describe('The name or CID fragment to search for.'),
+  } as any,
+  async ({ query }, _extra): Promise<CallToolResult> => {
+    return await searchObjectsHandler({ query })
+  },
+)

package/src/bin/auto-drive.ts

@@ -1,14 +1,14 @@
-#!/usr/bin/env node
-
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
-import { autoDriveServer } from '../index.js'
-
-const main = async () => {
-  const transport = new StdioServerTransport()
-  await autoDriveServer.connect(transport)
-}
-
-main().catch((error) => {
-  console.error('Failed to start Auto Drive server:', error)
-  process.exit(1)
-})
+#!/usr/bin/env node
+
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
+import { autoDriveServer } from '../index.js'
+
+const main = async () => {
+  const transport = new StdioServerTransport()
+  await autoDriveServer.connect(transport)
+}
+
+main().catch((error) => {
+  console.error('Failed to start Auto Drive server:', error)
+  process.exit(1)
+})

package/src/bin/main.ts

@@ -1,45 +1,45 @@
-#!/usr/bin/env node
-
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
-import { autoDriveServer } from '../auto-drive/index.js'
-
-const showHelp = () => {
-  console.log(`
-Usage: auto-mcp-servers [server-name]
-
-Available servers:
-  auto-drive    Start the Auto Drive MCP server
-
-Environment variables:
-  For Auto Drive server:
-    AUTO_DRIVE_API_KEY     API key for Auto Drive (required)
-    NETWORK                'mainnet' (default) or 'taurus'
-    ENCRYPTION_PASSWORD    Password for encryption (optional)
-  `)
-  process.exit(1)
-}
-
-const main = async () => {
-  const serverName = process.argv[2] || 'auto-drive'
-  const transport = new StdioServerTransport()
-
-  switch (serverName) {
-    case 'auto-drive':
-      await autoDriveServer.connect(transport)
-      break
-    case 'help':
-    case '--help':
-    case '-h':
-      showHelp()
-      break
-    default:
-      console.error(`Unknown server: ${serverName}`)
-      showHelp()
-      break
-  }
-}
-
-main().catch((error) => {
-  console.error(`Failed to start MCP server:`, error)
-  process.exit(1)
-})
+#!/usr/bin/env node
+
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
+import { autoDriveServer } from '../auto-drive/index.js'
+
+const showHelp = () => {
+  console.log(`
+Usage: auto-mcp-servers [server-name]
+
+Available servers:
+  auto-drive    Start the Auto Drive MCP server
+
+Environment variables:
+  For Auto Drive server:
+    AUTO_DRIVE_API_KEY     API key for Auto Drive (required)
+    NETWORK                'mainnet' (default) or 'taurus'
+    ENCRYPTION_PASSWORD    Password for encryption (optional)
+  `)
+  process.exit(1)
+}
+
+const main = async () => {
+  const serverName = process.argv[2] || 'auto-drive'
+  const transport = new StdioServerTransport()
+
+  switch (serverName) {
+    case 'auto-drive':
+      await autoDriveServer.connect(transport)
+      break
+    case 'help':
+    case '--help':
+    case '-h':
+      showHelp()
+      break
+    default:
+      console.error(`Unknown server: ${serverName}`)
+      showHelp()
+      break
+  }
+}
+
+main().catch((error) => {
+  console.error(`Failed to start MCP server:`, error)
+  process.exit(1)
+})

package/src/index.ts

@@ -1,2 +1,2 @@
-// Export servers
-export { autoDriveServer } from './auto-drive/index.js'
+// Export servers
+export { autoDriveServer } from './auto-drive/index.js'

package/tsconfig.json

@@ -1,10 +1,10 @@
-{
-  "extends": "../../tsconfig.json",
-  "compilerOptions": {
-    "module": "ESNext",
-    "moduleResolution": "Bundler",
-    "outDir": "./dist",
-    "rootDir": "./src"
-  },
-  "include": ["src/**/*"]
-}
+{
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "module": "ESNext",
+    "moduleResolution": "Bundler",
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["src/**/*"]
+}