@blocks-network/mcp-server 0.1.55 → 0.1.62

package/dist/tools.js

@@ -0,0 +1,275 @@
+/**
+ * Tool handlers for the Blocks Network MCP server.
+ *
+ * Each handler is a pure function that takes its dependencies (clients,
+ * registry helpers, fs/path utilities) so it can be unit-tested without
+ * touching the network or the filesystem.
+ */
+import { statSync } from 'node:fs';
+// ============================================================================
+// Helpers
+// ============================================================================
+const TERMINAL_STATES = new Set(['completed', 'failed', 'canceled']);
+function resolveBillingMode(task) {
+    const mode = task.billingMode;
+    return mode === 'paid' ? 'paid' : 'free';
+}
+function isTextLikeContentType(ct) {
+    return ct.startsWith('text/') || ct === 'application/json' || ct.endsWith('+json');
+}
+function isTextMimeType(mimeType) {
+    return mimeType.startsWith('text/') || mimeType === 'application/json';
+}
+async function appendArtifacts(session, refs, out) {
+    for (const ref of refs) {
+        try {
+            const downloaded = await session.downloadArtifact(ref);
+            if (isTextMimeType(downloaded.mimeType)) {
+                const text = new TextDecoder().decode(downloaded.data);
+                out.push(`[artifact: ${downloaded.fileName ?? 'unnamed'}]\n${text}`);
+            }
+            else {
+                out.push(`[artifact: ${downloaded.fileName ?? 'unnamed'}] (${downloaded.mimeType}, ${downloaded.data.length} bytes)`);
+            }
+        }
+        catch {
+            out.push(`[artifact: ${ref.fileName ?? 'unnamed'}] (download failed)`);
+        }
+    }
+}
+// ============================================================================
+// send_task
+// ============================================================================
+export async function sendTask(params, deps) {
+    const baseUrl = await deps.getBaseUrl();
+    const apiKey = deps.getApiKey();
+    const entry = await deps.getAgentByName(params.agentName, { baseUrl, apiKey });
+    const billingMode = entry?.billingMode ?? 'free';
+    const client = await deps.getTaskClient(billingMode);
+    const declaredInputs = entry?.card?.io?.inputs ?? [];
+    const textInput = declaredInputs.find((i) => isTextLikeContentType(i.contentType));
+    const fileInput = declaredInputs.find((i) => !isTextLikeContentType(i.contentType));
+    const requestParts = [];
+    const textPartId = textInput?.id ?? 'text';
+    if ((textInput || declaredInputs.length === 0) && !params.inputs?.[textPartId]) {
+        requestParts.push(deps.textPart(params.message, textPartId));
+    }
+    if (params.filePath) {
+        const safePath = deps.validateFilePath(params.filePath);
+        const size = deps.fileSize(safePath);
+        if (size > deps.maxUploadBytes) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `File too large (${size} bytes). Maximum upload size is ${deps.maxUploadBytes} bytes (25 MB).`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+        requestParts.push(await deps.filePartFromPath(safePath, {
+            partId: fileInput?.id ?? 'file',
+            contentType: fileInput?.contentType,
+        }));
+    }
+    if (params.inputs) {
+        for (const [partId, value] of Object.entries(params.inputs)) {
+            requestParts.push(deps.textPart(value, partId));
+        }
+    }
+    const session = await client.sendMessage({
+        agentName: params.agentName,
+        requestParts,
+        taskKind: params.taskKind,
+        duration: params.duration,
+    });
+    const timeout = params.timeoutMs ?? 60000;
+    const progressLines = [];
+    session.onProgress((event) => {
+        if (event.message)
+            progressLines.push(`[progress] ${event.message}`);
+    });
+    try {
+        const terminal = await session.waitForTerminal(timeout);
+        const output = [`Task ${session.taskId} ${terminal.state}`, ...progressLines];
+        await appendArtifacts(session, session.listArtifacts(), output);
+        if (terminal.state === 'failed') {
+            output.push(`Error: ${terminal.error ?? terminal.reason ?? 'unknown'}`);
+        }
+        session.close();
+        return { content: [{ type: 'text', text: output.join('\n') }] };
+    }
+    catch (err) {
+        session.close();
+        const msg = err instanceof Error ? err.message : String(err);
+        return {
+            content: [{ type: 'text', text: `Task ${session.taskId} error: ${msg}` }],
+            isError: true,
+        };
+    }
+}
+// ============================================================================
+// get_task
+// ============================================================================
+export async function getTask(params, deps) {
+    const freeClient = await deps.getTaskClient('free');
+    const task = await freeClient.getTask(params.taskId);
+    const output = [JSON.stringify(task, null, 2)];
+    if (task.state && TERMINAL_STATES.has(task.state)) {
+        try {
+            const billingMode = resolveBillingMode(task);
+            const client = await deps.getTaskClient(billingMode);
+            const session = await client.connect({ taskId: params.taskId });
+            await appendArtifacts(session, session.listArtifacts(), output);
+            session.close();
+        }
+        catch {
+            // connect failed — return task info without artifacts
+        }
+    }
+    return { content: [{ type: 'text', text: output.join('\n') }] };
+}
+// ============================================================================
+// list_tasks
+// ============================================================================
+export async function listTasks(params, deps) {
+    const client = await deps.getTaskClient();
+    const result = await client.listTasks({
+        agentName: params.agentName,
+        state: params.state,
+        limit: params.limit,
+    });
+    const lines = result.tasks.map((t) => `${t.taskId} | ${t.agentName ?? '?'} | ${t.state ?? '?'} | ${t.createdTime ?? ''}`);
+    const header = `Tasks (${result.totalCount ?? result.tasks.length} total):`;
+    return { content: [{ type: 'text', text: [header, ...lines].join('\n') }] };
+}
+// ============================================================================
+// cancel_task
+// ============================================================================
+export async function cancelTask(params, deps) {
+    const client = await deps.getTaskClient();
+    await client.cancelTask(params.taskId);
+    return { content: [{ type: 'text', text: `Task ${params.taskId} cancelled.` }] };
+}
+// ============================================================================
+// list_agents
+// ============================================================================
+export async function listAgents(params, deps) {
+    const baseUrl = await deps.getBaseUrl();
+    const apiKey = deps.getApiKey();
+    const result = await deps.listAgents({
+        baseUrl,
+        apiKey,
+        skill: params.skill,
+        listing: params.listing,
+        limit: params.limit,
+    });
+    const lines = result.agents.map((a) => {
+        const skills = a.skills?.map((s) => s.name).join(', ') ?? '';
+        return `${a.agentName} | ${a.name ?? a.agentName} | ${a.listing ?? 'public'} | ${skills}`;
+    });
+    const header = `Agents (${result.totalCount ?? result.agents.length}):`;
+    return { content: [{ type: 'text', text: [header, ...lines].join('\n') }] };
+}
+// ============================================================================
+// get_agent_card
+// ============================================================================
+export async function getAgentCard(params, deps) {
+    const baseUrl = await deps.getBaseUrl();
+    const apiKey = deps.getApiKey();
+    const entry = await deps.getAgentByName(params.agentName, { baseUrl, apiKey });
+    if (!entry) {
+        return {
+            content: [{ type: 'text', text: `Agent "${params.agentName}" not found.` }],
+            isError: true,
+        };
+    }
+    return {
+        content: [{ type: 'text', text: JSON.stringify(entry.card ?? entry, null, 2) }],
+    };
+}
+// ============================================================================
+// connect_task
+// ============================================================================
+export async function connectTask(params, deps) {
+    const freeClient = await deps.getTaskClient('free');
+    const task = await freeClient.getTask(params.taskId);
+    const billingMode = resolveBillingMode(task);
+    const client = await deps.getTaskClient(billingMode);
+    const session = await client.connect({ taskId: params.taskId });
+    const timeout = params.timeoutMs ?? 60000;
+    const progressLines = [];
+    const streamOutputs = [];
+    const streamDrains = [];
+    session.onProgress((event) => {
+        if (event.message)
+            progressLines.push(`[progress] ${event.message}`);
+    });
+    const drainStream = (streamRef) => {
+        const dir = streamRef.descriptor.localDirection;
+        if (dir !== 'inbound' && dir !== 'bidirectional')
+            return;
+        const format = streamRef.descriptor.format;
+        const label = streamRef.descriptor.declaredStream ?? streamRef.descriptor.streamId;
+        const entry = { label, chunks: [] };
+        streamOutputs.push(entry);
+        try {
+            const streamClient = streamRef.open();
+            const drain = (async () => {
+                if (format === 'events') {
+                    for await (const event of streamClient.events()) {
+                        entry.chunks.push(JSON.stringify(event) + '\n');
+                    }
+                }
+                else {
+                    const decoder = new TextDecoder();
+                    for await (const chunk of streamClient.bytes()) {
+                        entry.chunks.push(decoder.decode(chunk, { stream: true }));
+                    }
+                }
+            })();
+            streamDrains.push(drain.catch(() => { }));
+        }
+        catch {
+            // stream already ended or terminal
+        }
+    };
+    for (const ref of session.listStreams())
+        drainStream(ref);
+    session.onStream(drainStream);
+    try {
+        const terminal = await session.waitForTerminal(timeout);
+        await Promise.allSettled(streamDrains);
+        const out = [`Task ${params.taskId} ${terminal.state}`, ...progressLines];
+        for (const s of streamOutputs) {
+            if (s.chunks.length > 0) {
+                out.push(`[stream: ${s.label}]\n${s.chunks.join('')}`);
+            }
+        }
+        await appendArtifacts(session, session.listArtifacts(), out);
+        if (terminal.state === 'failed') {
+            out.push(`Error: ${terminal.error ?? terminal.reason ?? 'unknown'}`);
+        }
+        await session.asyncClose();
+        return { content: [{ type: 'text', text: out.join('\n') }] };
+    }
+    catch (err) {
+        await session.asyncClose();
+        const msg = err instanceof Error ? err.message : String(err);
+        const partial = [`Task ${params.taskId} timed out (${msg})`];
+        for (const s of streamOutputs) {
+            if (s.chunks.length > 0) {
+                partial.push(`[stream: ${s.label}]\n${s.chunks.join('')}`);
+            }
+        }
+        partial.push(...progressLines);
+        return { content: [{ type: 'text', text: partial.join('\n') }], isError: true };
+    }
+}
+// ============================================================================
+// Default file helpers (re-exported so index.ts can build a default ToolDeps)
+// ============================================================================
+export function defaultFileSize(path) {
+    return statSync(path).size;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blocks-network/mcp-server",
-  "version": "0.1.55",
+  "version": "0.1.62",
   "description": "MCP server for Blocks Network consumer operations",
   "type": "module",
   "main": "dist/index.js",
@@ -12,19 +12,31 @@
     "prebuild": "npm run build:sdk",
     "build": "tsc",
     "start": "node dist/index.js",
-    "dev": "tsx src/index.ts"
+    "dev": "tsx src/index.ts",
+    "test": "vitest run"
   },
   "dependencies": {
-    "@blocks-network/sdk": "file:../sdks/node",
-    "@modelcontextprotocol/sdk": "^1.12.1",
+    "@blocks-network/sdk": "^0.1.60",
+    "@modelcontextprotocol/sdk": "^1.29",
     "zod": "^3.25.1"
   },
   "devDependencies": {
     "@types/node": "^22.0.0",
     "tsx": "^4.19.0",
-    "typescript": "^5.7.0"
+    "typescript": "^5.7.0",
+    "vitest": "^3.2.4"
   },
   "engines": {
     "node": ">=20.0.0"
-  }
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/blocksnetwork/blocks-sdk.git",
+    "directory": "mcp"
+  },
+  "homepage": "https://github.com/blocksnetwork/blocks-sdk/tree/main/mcp#readme",
+  "bugs": {
+    "url": "https://github.com/blocksnetwork/blocks-sdk/issues"
+  },
+  "license": "SEE LICENSE IN LICENSE"
 }