@llmindset/hf-mcp 0.2.52 → 0.2.54

package/dist/space/utils/gradio-caller.js

@@ -0,0 +1,138 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
+import { CallToolResultSchema } from '@modelcontextprotocol/sdk/types.js';
+export function extractReplicaId(headerValue) {
+    if (!headerValue)
+        return null;
+    const parts = headerValue.split('-');
+    if (parts.length < 2)
+        return null;
+    const replicaId = parts[parts.length - 1];
+    if (!replicaId || replicaId.trim() === '')
+        return null;
+    return replicaId;
+}
+export function rewriteReplicaUrlsInResult(result, proxiedReplicaHeader) {
+    if (process.env.NO_REPLICA_REWRITE)
+        return result;
+    const replicaId = extractReplicaId(proxiedReplicaHeader);
+    if (!replicaId)
+        return result;
+    const urlPattern = /https:\/\/([^\s"']+)\/gradio_api(\S*)?/g;
+    const rewriteText = (text) => text.replace(urlPattern, (_match, host, rest = '') => {
+        return `https://${host}/--replicas/${replicaId}/gradio_api${rest}`;
+    });
+    let changed = false;
+    const newContent = result.content.map((item) => {
+        if (typeof item === 'string') {
+            const rewritten = rewriteText(item);
+            if (rewritten !== item) {
+                changed = true;
+                return { type: 'text', text: rewritten };
+            }
+            return { type: 'text', text: item };
+        }
+        if (item && typeof item === 'object' && 'text' in item && typeof item.text === 'string') {
+            const rewritten = rewriteText(item.text);
+            if (rewritten !== item.text) {
+                changed = true;
+                return { ...item, text: rewritten };
+            }
+        }
+        return item;
+    });
+    if (!changed)
+        return result;
+    return {
+        ...result,
+        content: newContent,
+    };
+}
+export async function callGradioToolWithHeaders(sseUrl, toolName, parameters, hfToken, extra, options = {}) {
+    const capturedHeaders = {};
+    let loggedHeader = false;
+    const handleHeaders = (headers) => {
+        const proxiedReplica = headers.get('x-proxied-replica') ?? '';
+        if (proxiedReplica) {
+            capturedHeaders['x-proxied-replica'] = proxiedReplica;
+        }
+        if (options.logProxiedReplica && !loggedHeader) {
+            loggedHeader = true;
+        }
+        options.onHeaders?.(headers);
+    };
+    const captureHeadersFetch = async (url, init) => {
+        const response = await fetch(url, init);
+        handleHeaders(response.headers);
+        return response;
+    };
+    const buildEventSourceFetch = (extraHeaders) => (url, init) => {
+        const headers = new Headers(init?.headers);
+        if (extraHeaders) {
+            Object.entries(extraHeaders).forEach(([key, value]) => headers.set(key, value));
+        }
+        const requestInit = { ...init, headers };
+        return captureHeadersFetch(url.toString(), requestInit);
+    };
+    const remoteClient = new Client({
+        name: 'hf-mcp-gradio-client',
+        version: '1.0.0',
+    }, {
+        capabilities: {},
+    });
+    const transportOptions = {
+        fetch: captureHeadersFetch,
+    };
+    if (hfToken) {
+        const headerName = 'X-HF-Authorization';
+        const customHeaders = {
+            [headerName]: `Bearer ${hfToken}`,
+        };
+        transportOptions.requestInit = {
+            headers: customHeaders,
+        };
+        transportOptions.eventSourceInit = {
+            fetch: buildEventSourceFetch(customHeaders),
+        };
+    }
+    else {
+        transportOptions.eventSourceInit = {
+            fetch: buildEventSourceFetch(),
+        };
+    }
+    const transport = new SSEClientTransport(new URL(sseUrl), transportOptions);
+    await remoteClient.connect(transport);
+    try {
+        const progressToken = extra?._meta?.progressToken;
+        const requestOptions = {};
+        if (progressToken !== undefined && extra) {
+            requestOptions.onprogress = (progress) => {
+                void extra.sendNotification({
+                    method: 'notifications/progress',
+                    params: {
+                        progressToken,
+                        progress: progress.progress,
+                        total: progress.total,
+                        message: progress.message,
+                    },
+                });
+            };
+            requestOptions.resetTimeoutOnProgress = true;
+        }
+        const result = await remoteClient.request({
+            method: 'tools/call',
+            params: {
+                name: toolName,
+                arguments: parameters,
+                _meta: progressToken !== undefined ? { progressToken } : undefined,
+            },
+        }, CallToolResultSchema, requestOptions);
+        const proxiedReplica = capturedHeaders['x-proxied-replica'];
+        const rewritten = rewriteReplicaUrlsInResult(result, proxiedReplica);
+        return { result: rewritten, capturedHeaders };
+    }
+    finally {
+        await remoteClient.close();
+    }
+}
+//# sourceMappingURL=gradio-caller.js.map

package/dist/space/utils/gradio-caller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gradio-caller.js","sourceRoot":"","sources":["../../../src/space/utils/gradio-caller.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAkC,MAAM,yCAAyC,CAAC;AAC7G,OAAO,EAAE,oBAAoB,EAA+C,MAAM,oCAAoC,CAAC;AAmBvH,MAAM,UAAU,gBAAgB,CAAC,WAA+B;IAC/D,IAAI,CAAC,WAAW;QAAE,OAAO,IAAI,CAAC;IAC9B,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAClC,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC1C,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE;QAAE,OAAO,IAAI,CAAC;IACvD,OAAO,SAAS,CAAC;AAClB,CAAC;AAOD,MAAM,UAAU,0BAA0B,CACzC,MAAyC,EACzC,oBAAwC;IAExC,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB;QAAE,OAAO,MAAM,CAAC;IAClD,MAAM,SAAS,GAAG,gBAAgB,CAAC,oBAAoB,CAAC,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,MAAM,CAAC;IAE9B,MAAM,UAAU,GAAG,yCAAyC,CAAC;IAE7D,MAAM,WAAW,GAAG,CAAC,IAAY,EAAU,EAAE,CAC5C,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,GAAG,EAAE,EAAE,EAAE;QACpD,OAAO,WAAW,IAAI,eAAe,SAAS,cAAc,IAAI,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEJ,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC9C,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;gBACxB,OAAO,GAAG,IAAI,CAAC;gBACf,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAqC,CAAC;YAC7E,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAqC,CAAC;QACxE,CAAC;QAED,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACzF,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,IAAI,SAAS,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC7B,OAAO,GAAG,IAAI,CAAC;gBACf,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YACrC,CAAC;QACF,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC;IAC5B,OAAO;QACN,GAAG,MAAM;QACT,OAAO,EAAE,UAAU;KACnB,CAAC;AACH,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC9C,MAAc,EACd,QAAgB,EAChB,UAAmC,EACnC,OAA2B,EAC3B,KAAyE,EACzE,UAA6B,EAAE;IAE/B,MAAM,eAAe,GAA2B,EAAE,CAAC;IACnD,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,MAAM,aAAa,GAAG,CAAC,OAAgB,EAAQ,EAAE;QAChD,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,IAAI,EAAE,CAAC;QAC9D,IAAI,cAAc,EAAE,CAAC;YACpB,eAAe,CAAC,mBAAmB,CAAC,GAAG,cAAc,CAAC;QACvD,CAAC;QACD,IAAI,OAAO,CAAC,iBAAiB,IAAI,CAAC,YAAY,EAAE,CAAC;YAChD,YAAY,GAAG,IAAI,CAAC;QACrB,CAAC;QACD,OAAO,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC,CAAC;IAEF,MAAM,mBAAmB,GAAuC,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACnF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACxC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAChC,OAAO,QAAQ,CAAC;IACjB,CAAC,CAAC;IAGF,MAAM,qBAAqB,GAC1B,CAAC,YAAqC,EAAoB,EAAE,CAC5D,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;QACb,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3C,IAAI,YAAY,EAAE,CAAC;YAClB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,MAAM,WAAW,GAAgB,EAAE,GAAI,IAAoB,EAAE,OAAO,EAAE,CAAC;QACvE,OAAO,mBAAmB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,WAAW,CAAC,CAAC;IACzD,CAAC,CAAC;IAGH,MAAM,YAAY,GAAG,IAAI,MAAM,CAC9B;QACC,IAAI,EAAE,sBAAsB;QAC5B,OAAO,EAAE,OAAO;KAChB,EACD;QACC,YAAY,EAAE,EAAE;KAChB,CACD,CAAC;IAGF,MAAM,gBAAgB,GAA8B;QACnD,KAAK,EAAE,mBAAmB;KAC1B,CAAC;IACF,IAAI,OAAO,EAAE,CAAC;QACb,MAAM,UAAU,GAAG,oBAAoB,CAAC;QACxC,MAAM,aAAa,GAAG;YACrB,CAAC,UAAU,CAAC,EAAE,UAAU,OAAO,EAAE;SACjC,CAAC;QAGF,gBAAgB,CAAC,WAAW,GAAG;YAC9B,OAAO,EAAE,aAAa;SACtB,CAAC;QAGF,gBAAgB,CAAC,eAAe,GAAG;YAClC,KAAK,EAAE,qBAAqB,CAAC,aAAa,CAAC;SAC3C,CAAC;IACH,CAAC;SAAM,CAAC;QACP,gBAAgB,CAAC,eAAe,GAAG;YAClC,KAAK,EAAE,qBAAqB,EAAE;SAC9B,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC5E,MAAM,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAEtC,IAAI,CAAC;QAEJ,MAAM,aAAa,GAAG,KAAK,EAAE,KAAK,EAAE,aAAa,CAAC;QAClD,MAAM,cAAc,GAAmB,EAAE,CAAC;QAE1C,IAAI,aAAa,KAAK,SAAS,IAAI,KAAK,EAAE,CAAC;YAE1C,cAAc,CAAC,UAAU,GAAG,CAAC,QAAQ,EAAE,EAAE;gBACxC,KAAK,KAAK,CAAC,gBAAgB,CAAC;oBAC3B,MAAM,EAAE,wBAAwB;oBAChC,MAAM,EAAE;wBACP,aAAa;wBACb,QAAQ,EAAE,QAAQ,CAAC,QAAQ;wBAC3B,KAAK,EAAE,QAAQ,CAAC,KAAK;wBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;qBACzB;iBACD,CAAC,CAAC;YACJ,CAAC,CAAC;YACF,cAAc,CAAC,sBAAsB,GAAG,IAAI,CAAC;QAC9C,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,CACxC;YACC,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,UAAU;gBACrB,KAAK,EAAE,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;aAClE;SACD,EACD,oBAAoB,EACpB,cAAc,CACd,CAAC;QAEF,MAAM,cAAc,GAAG,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,0BAA0B,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAErE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC;IAC/C,CAAC;YAAS,CAAC;QACV,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;AACF,CAAC"}

package/dist/space/utils/gradio-schema.d.ts

@@ -0,0 +1,13 @@
+import type { Tool } from '@modelcontextprotocol/sdk/types.js';
+export type ParsedSchemaFormat = 'array' | 'object';
+export interface ParsedGradioSchema {
+    format: ParsedSchemaFormat;
+    tools: Array<{
+        name: string;
+        description?: string;
+        inputSchema: unknown;
+    }>;
+}
+export declare function parseGradioSchemaResponse(schemaResponse: unknown): ParsedGradioSchema;
+export declare function normalizeParsedTools(parsed: ParsedGradioSchema): Tool[];
+//# sourceMappingURL=gradio-schema.d.ts.map

package/dist/space/utils/gradio-schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gradio-schema.d.ts","sourceRoot":"","sources":["../../../src/space/utils/gradio-schema.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAE/D,MAAM,MAAM,kBAAkB,GAAG,OAAO,GAAG,QAAQ,CAAC;AAEpD,MAAM,WAAW,kBAAkB;IAClC,MAAM,EAAE,kBAAkB,CAAC;IAC3B,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;CAC3E;AAUD,wBAAgB,yBAAyB,CAAC,cAAc,EAAE,OAAO,GAAG,kBAAkB,CA2CrF;AAKD,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI,EAAE,CAQvE"}

package/dist/space/utils/gradio-schema.js

@@ -0,0 +1,44 @@
+export function parseGradioSchemaResponse(schemaResponse) {
+    if (Array.isArray(schemaResponse)) {
+        const tools = schemaResponse.filter((tool) => {
+            return (typeof tool === 'object' &&
+                tool !== null &&
+                'name' in tool &&
+                typeof tool.name === 'string' &&
+                'inputSchema' in tool);
+        });
+        if (tools.length === 0) {
+            throw new Error('Invalid schema: no tools found in array schema');
+        }
+        return {
+            format: 'array',
+            tools,
+        };
+    }
+    if (typeof schemaResponse === 'object' && schemaResponse !== null) {
+        const entries = Object.entries(schemaResponse);
+        const tools = entries.map(([name, toolSchema]) => ({
+            name,
+            description: typeof toolSchema.description === 'string' ? toolSchema.description : undefined,
+            inputSchema: toolSchema,
+        }));
+        if (tools.length === 0) {
+            throw new Error('Invalid schema: no tools found in object schema');
+        }
+        return {
+            format: 'object',
+            tools,
+        };
+    }
+    throw new Error('Invalid schema format: expected array or object');
+}
+export function normalizeParsedTools(parsed) {
+    return parsed.tools
+        .filter((t) => !t.name.toLowerCase().includes('<lambda'))
+        .map((parsedTool) => ({
+        name: parsedTool.name,
+        description: parsedTool.description || `${parsedTool.name} tool`,
+        inputSchema: parsedTool.inputSchema,
+    }));
+}
+//# sourceMappingURL=gradio-schema.js.map

package/dist/space/utils/gradio-schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gradio-schema.js","sourceRoot":"","sources":["../../../src/space/utils/gradio-schema.ts"],"names":[],"mappings":"AAiBA,MAAM,UAAU,yBAAyB,CAAC,cAAuB;IAEhE,IAAI,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QACnC,MAAM,KAAK,GAAI,cAAiC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAwE,EAAE;YACtI,OAAO,CACN,OAAO,IAAI,KAAK,QAAQ;gBACxB,IAAI,KAAK,IAAI;gBACb,MAAM,IAAI,IAAI;gBACd,OAAQ,IAA2B,CAAC,IAAI,KAAK,QAAQ;gBACrD,aAAa,IAAI,IAAI,CACrB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACnE,CAAC;QAED,OAAO;YACN,MAAM,EAAE,OAAO;YACf,KAAK;SACL,CAAC;IACH,CAAC;IAGD,IAAI,OAAO,cAAc,KAAK,QAAQ,IAAI,cAAc,KAAK,IAAI,EAAE,CAAC;QACnE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,cAAyC,CAAC,CAAC;QAC1E,MAAM,KAAK,GAAgC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/E,IAAI;YACJ,WAAW,EAAE,OAAQ,UAAwC,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAE,UAAuC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;YACzJ,WAAW,EAAE,UAAU;SACvB,CAAC,CAAC,CAAC;QAEJ,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACpE,CAAC;QAED,OAAO;YACN,MAAM,EAAE,QAAQ;YAChB,KAAK;SACL,CAAC;IACH,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;AACpE,CAAC;AAKD,MAAM,UAAU,oBAAoB,CAAC,MAA0B;IAC9D,OAAO,MAAM,CAAC,KAAK;SACjB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;SACxD,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QACrB,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,GAAG,UAAU,CAAC,IAAI,OAAO;QAChE,WAAW,EAAE,UAAU,CAAC,WAAkC;KAC1D,CAAC,CAAC,CAAC;AACN,CAAC"}

package/dist/space/utils/parameter-formatter.js

@@ -1,4 +1,4 @@
-import { FILE_INPUT_HELP_MESSAGE } from '../types.js';
+import { FILE_INPUT_HELP_MESSAGE, VIEW_PARAMETERS } from '../types.js';
 export function formatParameters(schemaResult, spaceName) {
     const { toolName, toolDescription, parameters } = schemaResult;
     let output = `# Parameters for: ${toolName}\n\n`;
@@ -126,7 +126,7 @@ export function formatValidationError(errors, spaceName) {
     for (const error of errors) {
         output += `- ${error}\n`;
     }
-    output += `\nUse the view_parameters operation to see all required parameters and their types.`;
+    output += `\nUse the ${VIEW_PARAMETERS} operation to see all required parameters and their types.`;
     return output;
 }
 //# sourceMappingURL=parameter-formatter.js.map

package/dist/space/utils/parameter-formatter.js.map

@@ -1 +1 @@
-{"version":3,"file":"parameter-formatter.js","sourceRoot":"","sources":["../../../src/space/utils/parameter-formatter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAC;AAWtD,MAAM,UAAU,gBAAgB,CAAC,YAAoC,EAAE,SAAiB;IACvF,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,YAAY,CAAC;IAE/D,IAAI,MAAM,GAAG,qBAAqB,QAAQ,MAAM,CAAC;IAEjD,IAAI,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,oBAAoB,eAAe,MAAM,CAAC;IACrD,CAAC;IAGD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QAClD,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,MAAM,IAAI,oBAAoB,CAAC;IAE/B,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAGD,MAAM,IAAI,yBAAyB,CAAC;IACpC,MAAM,IAAI,kBAAkB,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;IAEpD,OAAO,MAAM,CAAC;AACf,CAAC;AAKD,SAAS,eAAe,CAAC,KAAoB;IAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC;IAC3D,IAAI,MAAM,GAAG,OAAO,KAAK,CAAC,IAAI,IAAI,KAAK,IAAI,CAAC;IAG5C,MAAM,IAAI,eAAe,KAAK,CAAC,IAAI,IAAI,CAAC;IAGxC,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACvB,MAAM,IAAI,sBAAsB,KAAK,CAAC,WAAW,IAAI,CAAC;IACvD,CAAC;IAGD,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,IAAI,kBAAkB,UAAU,IAAI,CAAC;IAC5C,CAAC;IAGD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,IAAI,yBAAyB,OAAO,IAAI,CAAC;IAChD,CAAC;IAGD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACtB,MAAM,IAAI,eAAe,uBAAuB,IAAI,CAAC;IACtD,CAAC;IAED,MAAM,IAAI,IAAI,CAAC;IACf,OAAO,MAAM,CAAC;AACf,CAAC;AAKD,SAAS,WAAW,CAAC,KAAc;IAClC,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,MAAM,CAAC;IAClC,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,WAAW,CAAC;IAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,KAAK,GAAG,CAAC;IACnD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACJ,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,UAAU,CAAC;QACnB,CAAC;IACF,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;QAC7D,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAKD,SAAS,kBAAkB,CAAC,SAAiB,EAAE,UAA2B;IAEzE,MAAM,aAAa,GAA2B,EAAE,CAAC;IAGjD,KAAK,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1D,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC;IAGD,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7D,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,KAAK,EAAE,CAAC;YACX,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QACpD,CAAC;IACF,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;SACvD,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC;SAC1B,IAAI,CAAC,IAAI,CAAC;SACV,IAAI,EAAE,CAAC;IAET,OAAO;;;mBAGW,SAAS;mBACT,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;;OAE3C,CAAC;AACR,CAAC;AAKD,SAAS,eAAe,CAAC,KAAoB;IAE5C,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAGD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IAGD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACtB,OAAO,gCAAgC,CAAC;IACzC,CAAC;IAGD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAEzD,QAAQ,QAAQ,EAAE,CAAC;QAClB,KAAK,QAAQ;YACZ,OAAO,iBAAiB,CAAC;QAC1B,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACb,OAAO,IAAI,CAAC;QACb,KAAK,SAAS;YACb,OAAO,MAAM,CAAC;QACf,KAAK,OAAO;YACX,OAAO,oBAAoB,CAAC;QAC7B,KAAK,QAAQ;YACZ,OAAO,kBAAkB,CAAC;QAC3B;YACC,OAAO,SAAS,CAAC;IACnB,CAAC;AACF,CAAC;AAKD,MAAM,UAAU,wBAAwB,CAAC,SAAiB,EAAE,MAAc;IACzE,OAAO,wCAAwC,SAAS;;EAEvD,MAAM;;;;+DAIuD,CAAC;AAChE,CAAC;AAKD,MAAM,UAAU,qBAAqB,CAAC,MAAgB,EAAE,SAAiB;IACxE,IAAI,MAAM,GAAG,wCAAwC,SAAS,QAAQ,CAAC;IAEvE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,KAAK,IAAI,CAAC;IAC1B,CAAC;IAED,MAAM,IAAI,qFAAqF,CAAC;IAEhG,OAAO,MAAM,CAAC;AACf,CAAC"}
+{"version":3,"file":"parameter-formatter.js","sourceRoot":"","sources":["../../../src/space/utils/parameter-formatter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAWvE,MAAM,UAAU,gBAAgB,CAAC,YAAoC,EAAE,SAAiB;IACvF,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,YAAY,CAAC;IAE/D,IAAI,MAAM,GAAG,qBAAqB,QAAQ,MAAM,CAAC;IAEjD,IAAI,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,oBAAoB,eAAe,MAAM,CAAC;IACrD,CAAC;IAGD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QAClD,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,MAAM,IAAI,oBAAoB,CAAC;IAE/B,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAGD,MAAM,IAAI,yBAAyB,CAAC;IACpC,MAAM,IAAI,kBAAkB,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;IAEpD,OAAO,MAAM,CAAC;AACf,CAAC;AAKD,SAAS,eAAe,CAAC,KAAoB;IAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC;IAC3D,IAAI,MAAM,GAAG,OAAO,KAAK,CAAC,IAAI,IAAI,KAAK,IAAI,CAAC;IAG5C,MAAM,IAAI,eAAe,KAAK,CAAC,IAAI,IAAI,CAAC;IAGxC,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACvB,MAAM,IAAI,sBAAsB,KAAK,CAAC,WAAW,IAAI,CAAC;IACvD,CAAC;IAGD,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,IAAI,kBAAkB,UAAU,IAAI,CAAC;IAC5C,CAAC;IAGD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,IAAI,yBAAyB,OAAO,IAAI,CAAC;IAChD,CAAC;IAGD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACtB,MAAM,IAAI,eAAe,uBAAuB,IAAI,CAAC;IACtD,CAAC;IAED,MAAM,IAAI,IAAI,CAAC;IACf,OAAO,MAAM,CAAC;AACf,CAAC;AAKD,SAAS,WAAW,CAAC,KAAc;IAClC,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,MAAM,CAAC;IAClC,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,WAAW,CAAC;IAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,KAAK,GAAG,CAAC;IACnD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACJ,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,UAAU,CAAC;QACnB,CAAC;IACF,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;QAC7D,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAKD,SAAS,kBAAkB,CAAC,SAAiB,EAAE,UAA2B;IAEzE,MAAM,aAAa,GAA2B,EAAE,CAAC;IAGjD,KAAK,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1D,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC;IAGD,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7D,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,KAAK,EAAE,CAAC;YACX,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QACpD,CAAC;IACF,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;SACvD,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC;SAC1B,IAAI,CAAC,IAAI,CAAC;SACV,IAAI,EAAE,CAAC;IAET,OAAO;;;mBAGW,SAAS;mBACT,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;;OAE3C,CAAC;AACR,CAAC;AAKD,SAAS,eAAe,CAAC,KAAoB;IAE5C,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAGD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IAGD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACtB,OAAO,gCAAgC,CAAC;IACzC,CAAC;IAGD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAEzD,QAAQ,QAAQ,EAAE,CAAC;QAClB,KAAK,QAAQ;YACZ,OAAO,iBAAiB,CAAC;QAC1B,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACb,OAAO,IAAI,CAAC;QACb,KAAK,SAAS;YACb,OAAO,MAAM,CAAC;QACf,KAAK,OAAO;YACX,OAAO,oBAAoB,CAAC;QAC7B,KAAK,QAAQ;YACZ,OAAO,kBAAkB,CAAC;QAC3B;YACC,OAAO,SAAS,CAAC;IACnB,CAAC;AACF,CAAC;AAKD,MAAM,UAAU,wBAAwB,CAAC,SAAiB,EAAE,MAAc;IACzE,OAAO,wCAAwC,SAAS;;EAEvD,MAAM;;;;+DAIuD,CAAC;AAChE,CAAC;AAKD,MAAM,UAAU,qBAAqB,CAAC,MAAgB,EAAE,SAAiB;IACxE,IAAI,MAAM,GAAG,wCAAwC,SAAS,QAAQ,CAAC;IAEvE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,KAAK,IAAI,CAAC;IAC1B,CAAC;IAED,MAAM,IAAI,aAAa,eAAe,4DAA4D,CAAC;IAEnG,OAAO,MAAM,CAAC;AACf,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llmindset/hf-mcp",
-  "version": "0.2.52",
+  "version": "0.2.54",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -20,7 +20,9 @@ export * from './docs-search/doc-fetch.js';
 export * from './readme-utils.js';
 export * from './use-space.js';
 export * from './jobs/jobs-tool.js';
-export * from './space/space-tool.js';
+export * from './space/dynamic-space-tool.js';
+export * from './space/utils/gradio-caller.js';
+export * from './space/utils/gradio-schema.js';
 
 // Export shared types
 export * from './types/tool-result.js';

package/src/space/commands/discover.ts

@@ -1,5 +1,6 @@
 import type { ToolResult } from '../../types/tool-result.js';
 import { escapeMarkdown } from '../../utilities.js';
+import { VIEW_PARAMETERS } from '../types.js';
 
 /**
  * Prompt configuration for discover operation (from DYNAMIC_SPACE_DATA)
@@ -10,7 +11,7 @@ export const DISCOVER_PROMPTS = {
 	RESULTS_HEADER: `**Available Spaces:**
 
 These spaces can be invoked using the \`dynamic_space\` tool.
-Use \`"operation": "view_parameters"\` to inspect a space's parameters before invoking.
+Use \`"operation": "${VIEW_PARAMETERS}"\` to inspect a space's parameters before invoking.
 
 `,
 

package/src/space/commands/dynamic-find.ts

@@ -1,6 +1,7 @@
 import type { ToolResult } from '../../types/tool-result.js';
 import { SpaceSearchTool, type SpaceSearchResult } from '../../space-search.js';
 import { escapeMarkdown } from '../../utilities.js';
+import { VIEW_PARAMETERS } from '../dynamic-space-tool.js';
 
 // Default number of results to return
 const DEFAULT_RESULTS_LIMIT = 10;
@@ -51,7 +52,7 @@ To find MCP-enabled Spaces for a specific task, call this operation with a searc
 		return `# MCP Space Find Results for "${query}" (${showingText})
 
 These MCP-enabled Spaces can be invoked using the \`dynamic_space\` tool.
-Use \`"operation": "view_parameters"\` to inspect a space's parameters before invoking.
+Use \`"operation": "${VIEW_PARAMETERS}"\` to inspect a space's parameters before invoking.
 
 `;
 	},

package/src/space/commands/invoke.ts

@@ -1,12 +1,11 @@
 import type { ToolResult } from '../../types/tool-result.js';
 import type { InvokeResult } from '../types.js';
 import type { Tool, ServerNotification, ServerRequest } from '@modelcontextprotocol/sdk/types.js';
-import type { RequestHandlerExtra, RequestOptions } from '@modelcontextprotocol/sdk/shared/protocol.js';
-import { CallToolResultSchema } from '@modelcontextprotocol/sdk/types.js';
-import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { SSEClientTransport, type SSEClientTransportOptions } from '@modelcontextprotocol/sdk/client/sse.js';
+import type { RequestHandlerExtra } from '@modelcontextprotocol/sdk/shared/protocol.js';
 import { analyzeSchemaComplexity, validateParameters, applyDefaults } from '../utils/schema-validator.js';
 import { formatComplexSchemaError, formatValidationError } from '../utils/parameter-formatter.js';
+import { callGradioToolWithHeaders } from '../utils/gradio-caller.js';
+import { parseGradioSchemaResponse, normalizeParsedTools } from '../utils/gradio-schema.js';
 
 /**
  * Invokes a Gradio space with provided parameters
@@ -88,44 +87,11 @@ export async function invokeSpace(
 		// Step 7: Apply default values for missing optional parameters
 		const finalParameters = applyDefaults(inputParameters, schemaResult);
 
-		// Step 8: Create SSE connection and invoke tool
-		const sseUrl = `https://${metadata.subdomain}.hf.space/gradio_api/mcp/sse`;
-		const client = await createLazyConnection(sseUrl, hfToken);
-
-		try {
-			// Check if the client is requesting progress notifications
-			const progressToken = extra?._meta?.progressToken;
-			const requestOptions: RequestOptions = {};
-
-			if (progressToken !== undefined && extra) {
-				// Set up progress relay from remote tool to our client
-				// eslint-disable-next-line @typescript-eslint/no-misused-promises
-				requestOptions.onprogress = async (progress) => {
-					// Relay the progress notification to our client
-					await extra.sendNotification({
-						method: 'notifications/progress',
-						params: {
-							progressToken,
-							progress: progress.progress,
-							total: progress.total,
-							message: progress.message,
-						},
-					});
-				};
-			}
-
-			const result = await client.request(
-				{
-					method: 'tools/call',
-					params: {
-						name: tool.name,
-						arguments: finalParameters,
-						_meta: progressToken !== undefined ? { progressToken } : undefined,
-					},
-				},
-				CallToolResultSchema,
-				requestOptions
-			);
+			// Step 8: Create SSE connection and invoke tool (shared helper)
+			const sseUrl = `https://${metadata.subdomain}.hf.space/gradio_api/mcp/sse`;
+			const { result } = await callGradioToolWithHeaders(sseUrl, tool.name, finalParameters, hfToken, extra, {
+				logProxiedReplica: true,
+			});
 
 			// Return raw MCP result with warnings if any
 			// This ensures the space tool behaves identically to proxied gr_* tools
@@ -136,10 +102,6 @@ export async function invokeSpace(
 				resultsShared: 1,
 				isError: result.isError,
 			};
-		} finally {
-			// Clean up connection
-			await client.close();
-		}
 	} catch (error) {
 		const errorMessage = error instanceof Error ? error.message : String(error);
 		return {
@@ -230,106 +192,10 @@ async function fetchGradioSchema(subdomain: string, isPrivate: boolean, hfToken?
 
 		const schemaResponse = (await response.json()) as unknown;
 
-		// Parse schema response (handle both array and object formats)
-		return parseSchemaResponse(schemaResponse);
-	} finally {
-		clearTimeout(timeoutId);
-	}
-}
-
-/**
- * Parses schema response and extracts tools
- */
-function parseSchemaResponse(schemaResponse: unknown): Tool[] {
-	const tools: Tool[] = [];
-
-	if (Array.isArray(schemaResponse)) {
-		// Array format: [{ name: "toolName", description: "...", inputSchema: {...} }, ...]
-		for (const item of schemaResponse) {
-			if (
-				typeof item === 'object' &&
-				item !== null &&
-				'name' in item &&
-				'inputSchema' in item
-			) {
-				const itemRecord = item as Record<string, unknown>;
-				if (typeof itemRecord.name === 'string') {
-					const tool = itemRecord as { name: string; description?: string; inputSchema: unknown };
-					tools.push({
-						name: tool.name,
-						description: tool.description || `${tool.name} tool`,
-						inputSchema: {
-							type: 'object',
-							...(tool.inputSchema as Record<string, unknown>),
-						},
-					});
-				}
-			}
-		}
-	} else if (typeof schemaResponse === 'object' && schemaResponse !== null) {
-		// Object format: { "toolName": { properties: {...}, required: [...] }, ... }
-		for (const [name, toolSchema] of Object.entries(schemaResponse)) {
-			if (typeof toolSchema === 'object' && toolSchema !== null) {
-				const schema = toolSchema as { description?: string };
-				tools.push({
-					name,
-					description: schema.description || `${name} tool`,
-					inputSchema: {
-						type: 'object',
-						...(toolSchema as Record<string, unknown>),
-					},
-				});
-			}
-		}
-	}
-
-	return tools.filter((tool) => !tool.name.toLowerCase().includes('<lambda'));
-}
-
-/**
- * Creates SSE connection to endpoint for tool execution
- */
-async function createLazyConnection(sseUrl: string, hfToken?: string): Promise<Client> {
-	// Create MCP client
-	const remoteClient = new Client(
-		{
-			name: 'hf-mcp-space-client',
-			version: '1.0.0',
-		},
-		{
-			capabilities: {},
+			// Parse schema response (handle both array and object formats)
+			const parsed = parseGradioSchemaResponse(schemaResponse);
+			return normalizeParsedTools(parsed);
+		} finally {
+			clearTimeout(timeoutId);
 		}
-	);
-
-	// Create SSE transport with HF token if available
-	const transportOptions: SSEClientTransportOptions = {};
-	if (hfToken) {
-		const headerName = 'X-HF-Authorization';
-		const customHeaders = {
-			[headerName]: `Bearer ${hfToken}`,
-		};
-
-		// Headers for POST requests
-		transportOptions.requestInit = {
-			headers: customHeaders,
-		};
-
-		// Headers for SSE connection
-		transportOptions.eventSourceInit = {
-			fetch: (url, init) => {
-				const headers = new Headers(init.headers);
-				Object.entries(customHeaders).forEach(([key, value]) => {
-					headers.set(key, value);
-				});
-				return fetch(url.toString(), { ...init, headers });
-			},
-		};
 	}
-
-	const transport = new SSEClientTransport(new URL(sseUrl), transportOptions);
-
-	// Connect the client to the transport
-	await remoteClient.connect(transport);
-
-	return remoteClient;
-}