@assethub/cli 0.1.18 → 0.1.19

package/dist/nodeMeshBatch.d.ts

@@ -0,0 +1,30 @@
+import type { CanvasExecution, ExecutionContext } from '@assethub/api-client';
+import { executeRecorded, resumeRecorded } from './execution.js';
+type Session = Parameters<typeof executeRecorded>[0];
+type NodeMeshBatch = {
+    operationId: string;
+    nodeId: string;
+    canvasId: number;
+    baseUrl: string;
+    ownerId: string;
+    body: Record<string, unknown>;
+    children: Array<{
+        nodeId: string;
+        operationId: string;
+    }>;
+    skipped: Array<{
+        nodeId: string;
+        meshStatus?: string;
+    }>;
+    agent?: ExecutionContext['agent'];
+    parentRunId?: string;
+};
+export type NodeMeshBatchResult = Pick<NodeMeshBatch, 'operationId' | 'nodeId' | 'skipped'> & {
+    executions: CanvasExecution[];
+};
+export declare const executeNodeMeshBatch: (session: Session, input: Omit<NodeMeshBatch, "baseUrl" | "ownerId" | "canvasId" | "children"> & {
+    children: string[];
+}) => Promise<NodeMeshBatchResult>;
+export declare const resumeNodeMeshBatch: (options: Parameters<typeof resumeRecorded>[0]) => Promise<NodeMeshBatchResult | undefined>;
+export {};
+//# sourceMappingURL=nodeMeshBatch.d.ts.map

package/dist/nodeMeshBatch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"nodeMeshBatch.d.ts","sourceRoot":"","sources":["../src/nodeMeshBatch.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,eAAe,EACf,gBAAgB,EAEjB,MAAM,sBAAsB,CAAA;AAE7B,OAAO,EAIL,eAAe,EAEf,cAAc,EAEf,MAAM,gBAAgB,CAAA;AAEvB,KAAK,OAAO,GAAG,UAAU,CAAC,OAAO,eAAe,CAAC,CAAC,CAAC,CAAC,CAAA;AACpD,KAAK,aAAa,GAAG;IACnB,WAAW,EAAE,MAAM,CAAA;IACnB,MAAM,EAAE,MAAM,CAAA;IACd,QAAQ,EAAE,MAAM,CAAA;IAChB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,MAAM,CAAA;IACf,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC7B,QAAQ,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAC,CAAC,CAAA;IACtD,OAAO,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAC,CAAC,CAAA;IACrD,KAAK,CAAC,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAA;IACjC,WAAW,CAAC,EAAE,MAAM,CAAA;CACrB,CAAA;AACD,MAAM,MAAM,mBAAmB,GAAG,IAAI,CACpC,aAAa,EACb,aAAa,GAAG,QAAQ,GAAG,SAAS,CACrC,GAAG;IAAC,UAAU,EAAE,eAAe,EAAE,CAAA;CAAC,CAAA;AAwDnC,eAAO,MAAM,oBAAoB,GAC/B,SAAS,OAAO,EAChB,OAAO,IAAI,CACT,aAAa,EACb,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,UAAU,CAChD,GAAG;IAAC,QAAQ,EAAE,MAAM,EAAE,CAAA;CAAC,KACvB,OAAO,CAAC,mBAAmB,CAsC7B,CAAA;AAED,eAAO,MAAM,mBAAmB,GAC9B,SAAS,UAAU,CAAC,OAAO,cAAc,CAAC,CAAC,CAAC,CAAC,KAC5C,OAAO,CAAC,mBAAmB,GAAG,SAAS,CA0DzC,CAAA"}

package/dist/nodeMeshBatch.js

@@ -0,0 +1,122 @@
+import { join } from 'node:path';
+import { readState, writeState } from './canvas.js';
+import { activeExecution, childOperationId, CliExecutionError, executeRecorded, openExecutionSession, resumeRecorded, hasRecordedOperation, } from './execution.js';
+const uuid = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i;
+const batchPath = (stateDir, id) => {
+    if (!uuid.test(id))
+        throw new Error('Operation ID must be a UUID');
+    return join(stateDir, 'node-batches', `${id}.json`);
+};
+const submitBatch = async (session, batch) => {
+    activeExecution.batchOperationId = batch.operationId;
+    const executions = [];
+    for (const child of batch.children) {
+        try {
+            const result = (await hasRecordedOperation(session.stateDir, child.operationId))
+                ? await resumeRecorded({
+                    ...session,
+                    operationId: child.operationId,
+                    expectedCanvasId: batch.canvasId,
+                    expectedCanvasNodeId: child.nodeId,
+                    expectedOperation: 'mesh.generate',
+                    expectedBody: batch.body,
+                })
+                : await executeRecorded(session, 'mesh.generate', batch.body, {
+                    operationId: child.operationId,
+                    canvasNode: { nodeId: child.nodeId },
+                    agent: batch.agent,
+                    parentRunId: batch.parentRunId,
+                });
+            executions.push(result.execution);
+        }
+        catch (error) {
+            throw new CliExecutionError(`Node ${child.nodeId} failed (${child.operationId}). Resume batch ${batch.operationId}: ${error instanceof Error ? error.message : String(error)}`, error instanceof CliExecutionError ? error.exitCode : 1, batch.operationId);
+        }
+    }
+    return {
+        operationId: batch.operationId,
+        nodeId: batch.nodeId,
+        executions,
+        skipped: batch.skipped,
+    };
+};
+export const executeNodeMeshBatch = async (session, input) => {
+    const batch = JSON.parse(JSON.stringify({
+        ...input,
+        baseUrl: session.baseUrl,
+        ownerId: session.ownerId,
+        canvasId: session.canvas.id,
+        children: input.children.map(nodeId => ({
+            nodeId,
+            operationId: childOperationId(input.operationId, nodeId),
+        })),
+    }));
+    const path = batchPath(session.stateDir, batch.operationId);
+    // Pin the entire target set before any paid dispatch. Per-child receipts retain
+    // resolved server inputs; a retry never discovers a changed split again.
+    try {
+        await writeState(path, batch, { exclusive: true });
+    }
+    catch (error) {
+        if (!error ||
+            typeof error !== 'object' ||
+            !('code' in error) ||
+            error.code !== 'EEXIST')
+            throw error;
+        const existing = await resumeNodeMeshBatch({
+            ...session,
+            operationId: input.operationId,
+            expectedCanvasId: session.canvas.id,
+            expectedCanvasNodeId: input.nodeId,
+            expectedOperation: 'mesh.generate',
+            expectedBody: input.body,
+        });
+        if (!existing)
+            throw new Error('Saved node batch not found');
+        return existing;
+    }
+    return submitBatch(session, batch);
+};
+export const resumeNodeMeshBatch = async (options) => {
+    const batch = (await readState(batchPath(options.stateDir, options.operationId)));
+    if (batch === undefined)
+        return undefined;
+    if (options.expectedOperation !== undefined &&
+        options.expectedOperation !== 'mesh.generate')
+        throw new Error('Saved operation has a different operation type');
+    if (!batch ||
+        batch.operationId !== options.operationId ||
+        typeof batch.nodeId !== 'string' ||
+        !Number.isSafeInteger(batch.canvasId) ||
+        typeof batch.baseUrl !== 'string' ||
+        typeof batch.ownerId !== 'string' ||
+        !batch.body ||
+        typeof batch.body !== 'object' ||
+        Array.isArray(batch.body) ||
+        !Array.isArray(batch.children) ||
+        !batch.children.every(child => child &&
+            typeof child.nodeId === 'string' &&
+            uuid.test(child.operationId)) ||
+        !Array.isArray(batch.skipped))
+        throw new Error('Invalid saved node batch');
+    if (batch.baseUrl !== options.client.baseUrl)
+        throw new Error('Saved operation belongs to a different API origin');
+    if (options.expectedCanvasId !== undefined &&
+        options.expectedCanvasId !== batch.canvasId)
+        throw new Error('Saved operation belongs to a different canvas');
+    if (options.expectedCanvasNodeId !== undefined &&
+        options.expectedCanvasNodeId !== batch.nodeId)
+        throw new Error('Saved operation belongs to a different canvas node');
+    if (options.expectedBody !== undefined &&
+        JSON.stringify(options.expectedBody) !== JSON.stringify(batch.body))
+        throw new Error('Operation ID already belongs to different inputs; use runs resume or a new operation ID');
+    const session = await openExecutionSession({
+        ...options,
+        canvasId: batch.canvasId,
+        operation: 'mesh.generate',
+        create: false,
+    });
+    if (session.ownerId !== batch.ownerId)
+        throw new Error('Saved operation belongs to a different organization');
+    return submitBatch(session, batch);
+};

package/dist/runsUpload/uploadRun.d.ts

@@ -35,9 +35,10 @@ export type UploadRunInput = {
     plan: RunUploadPlan;
     baseUrl: string;
     apiKey: string;
+    workspaceId?: string;
     fetchImpl?: typeof fetch;
     skipRegister?: boolean;
     onProgress?: (event: RunUploadProgressEvent) => void;
 };
-export declare const uploadRun: ({ plan, baseUrl, apiKey, fetchImpl, skipRegister, onProgress, }: UploadRunInput) => Promise<RunUploadResult>;
+export declare const uploadRun: ({ plan, baseUrl, apiKey, workspaceId, fetchImpl, skipRegister, onProgress, }: UploadRunInput) => Promise<RunUploadResult>;
 //# sourceMappingURL=uploadRun.d.ts.map

package/dist/runsUpload/uploadRun.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"uploadRun.d.ts","sourceRoot":"","sources":["../../src/runsUpload/uploadRun.ts"],"names":[],"mappings":"AAwBA,OAAO,EAAc,KAAK,WAAW,EAAE,KAAK,aAAa,EAAC,MAAM,qBAAqB,CAAA;AAGrF,MAAM,MAAM,sBAAsB,GAC9B;IAAC,IAAI,EAAE,UAAU,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAC,GACnC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,UAAU,GAAG,SAAS,CAAA;CAAC,GAC/F;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAC,CAAA;AAEhD,MAAM,MAAM,eAAe,GAAG;IAC5B,OAAO,EAAE,MAAM,CAAA;IACf,QAAQ,EAAE,MAAM,CAAA;IAChB,GAAG,EAAE,MAAM,CAAA;IACX,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,OAAO,CAAA;IACnB,aAAa,EAAE,MAAM,CAAA;IACrB,YAAY,EAAE,MAAM,CAAA;IACpB,aAAa,EAAE,MAAM,CAAA;IACrB;;;;;OAKG;IACH,MAAM,EAAE,WAAW,GAAG,WAAW,GAAG,OAAO,CAAA;IAC3C,IAAI,EAAE,OAAO,CAAA;CACd,CAAA;AAED,MAAM,MAAM,cAAc,GAAG;IAC3B,IAAI,EAAE,aAAa,CAAA;IACnB,OAAO,EAAE,MAAM,CAAA;IACf,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,CAAC,EAAE,OAAO,KAAK,CAAA;IACxB,YAAY,CAAC,EAAE,OAAO,CAAA;IACtB,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;CACrD,CAAA;AAED,eAAO,MAAM,SAAS,GAAU,iEAO7B,cAAc,KAAG,OAAO,CAAC,eAAe,CA0F1C,CAAA"}
+{"version":3,"file":"uploadRun.d.ts","sourceRoot":"","sources":["../../src/runsUpload/uploadRun.ts"],"names":[],"mappings":"AAwBA,OAAO,EAAc,KAAK,WAAW,EAAE,KAAK,aAAa,EAAC,MAAM,qBAAqB,CAAA;AAGrF,MAAM,MAAM,sBAAsB,GAC9B;IAAC,IAAI,EAAE,UAAU,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAC,GACnC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,UAAU,GAAG,SAAS,CAAA;CAAC,GAC/F;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAC,CAAA;AAEhD,MAAM,MAAM,eAAe,GAAG;IAC5B,OAAO,EAAE,MAAM,CAAA;IACf,QAAQ,EAAE,MAAM,CAAA;IAChB,GAAG,EAAE,MAAM,CAAA;IACX,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,OAAO,CAAA;IACnB,aAAa,EAAE,MAAM,CAAA;IACrB,YAAY,EAAE,MAAM,CAAA;IACpB,aAAa,EAAE,MAAM,CAAA;IACrB;;;;;OAKG;IACH,MAAM,EAAE,WAAW,GAAG,WAAW,GAAG,OAAO,CAAA;IAC3C,IAAI,EAAE,OAAO,CAAA;CACd,CAAA;AAED,MAAM,MAAM,cAAc,GAAG;IAC3B,IAAI,EAAE,aAAa,CAAA;IACnB,OAAO,EAAE,MAAM,CAAA;IACf,MAAM,EAAE,MAAM,CAAA;IACd,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,OAAO,KAAK,CAAA;IACxB,YAAY,CAAC,EAAE,OAAO,CAAA;IACtB,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;CACrD,CAAA;AAED,eAAO,MAAM,SAAS,GAAU,8EAQ7B,cAAc,KAAG,OAAO,CAAC,eAAe,CA0F1C,CAAA"}

package/dist/runsUpload/uploadRun.js

@@ -17,10 +17,10 @@ import { readFile } from 'node:fs/promises';
 import { createHash } from 'node:crypto';
 import { CONTROL_ERROR_CODES, CONTROL_NOT_ENTITLED_STATUS, controlRunUploadUrls, } from './controlEndpoints.js';
 import { RunUploadError } from './runUploadError.js';
-export const uploadRun = async ({ plan, baseUrl, apiKey, fetchImpl = globalThis.fetch.bind(globalThis), skipRegister = false, onProgress = () => { }, }) => {
+export const uploadRun = async ({ plan, baseUrl, apiKey, workspaceId, fetchImpl = globalThis.fetch.bind(globalThis), skipRegister = false, onProgress = () => { }, }) => {
     const urls = controlRunUploadUrls(baseUrl);
     // Built once, never logged, never placed in a URL or a query string.
-    const authHeaders = { authorization: `Bearer ${apiKey}` };
+    const authHeaders = { authorization: `Bearer ${apiKey}`, ...(workspaceId ? { 'X-AssetHub-Workspace': workspaceId } : {}) };
     let registered = false;
     if (!skipRegister) {
         onProgress({ kind: 'register', graphId: plan.graphId });

package/dist/setup.d.ts

@@ -1,6 +1,6 @@
 import type { Tool } from '@modelcontextprotocol/sdk/types.js';
 export declare const cliVersion: () => Promise<string>;
-export declare const mcpConfig: (client: string, baseUrl: string, account?: boolean) => string;
+export declare const mcpConfig: (client: string, baseUrl: string, account?: boolean, workspaceId?: string) => string;
 type Check = {
     name: 'api' | 'mcp' | 'configuration';
     status: 'pass' | 'fail';
@@ -17,6 +17,7 @@ type Auth = {
     baseUrl: string;
     profile: string;
     source: string;
+    workspaceId?: string;
     workspaceMfaToken?: string;
 };
 export declare const diagnose: ({ resolveAuth, includeMcp, timeoutMs, includeTools, account, }: {

package/dist/setup.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../src/setup.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,IAAI,EAAC,MAAM,oCAAoC,CAAA;AAE5D,eAAO,MAAM,UAAU,QAAa,OAAO,CAAC,MAAM,CAKvC,CAAA;AAmBX,eAAO,MAAM,SAAS,GAAI,QAAQ,MAAM,EAAE,SAAS,MAAM,EAAE,iBAAe,KAAG,MAsB5E,CAAA;AAED,KAAK,KAAK,GAAG;IACX,IAAI,EAAE,KAAK,GAAG,KAAK,GAAG,eAAe,CAAA;IACrC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAA;IACvB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAA;CACf,CAAA;AAED,KAAK,IAAI,GAAG;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,iBAAiB,CAAC,EAAE,MAAM,CAAA;CAAC,CAAA;AA2D1G,eAAO,MAAM,QAAQ,GAAU,gEAM5B;IACD,WAAW,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAA;IAChC,UAAU,EAAE,OAAO,CAAA;IACnB,SAAS,EAAE,MAAM,CAAA;IACjB,YAAY,CAAC,EAAE,OAAO,CAAA;IACtB,OAAO,CAAC,EAAE,OAAO,CAAA;CAClB;;;;;;;;;;;;;;;;;;;;;;EAiHA,CAAA"}
+{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../src/setup.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,IAAI,EAAC,MAAM,oCAAoC,CAAA;AAE5D,eAAO,MAAM,UAAU,QAAa,OAAO,CAAC,MAAM,CAKvC,CAAA;AAmBX,eAAO,MAAM,SAAS,GAAI,QAAQ,MAAM,EAAE,SAAS,MAAM,EAAE,iBAAe,EAAE,cAAc,MAAM,KAAG,MAsBlG,CAAA;AAED,KAAK,KAAK,GAAG;IACX,IAAI,EAAE,KAAK,GAAG,KAAK,GAAG,eAAe,CAAA;IACrC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAA;IACvB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAA;CACf,CAAA;AAED,KAAK,IAAI,GAAG;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,iBAAiB,CAAC,EAAE,MAAM,CAAA;CAAC,CAAA;AA2DhI,eAAO,MAAM,QAAQ,GAAU,gEAM5B;IACD,WAAW,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAA;IAChC,UAAU,EAAE,OAAO,CAAA;IACnB,SAAS,EAAE,MAAM,CAAA;IACjB,YAAY,CAAC,EAAE,OAAO,CAAA;IACtB,OAAO,CAAC,EAAE,OAAO,CAAA;CAClB;;;;;;;;;;;;;;;;;;;;;;EAkHA,CAAA"}

package/dist/setup.js

@@ -11,7 +11,7 @@ const validatedBaseUrl = (baseUrl) => {
         throw new Error('Use HTTPS for remote servers; HTTP is supported only on localhost.');
     return url.href.replace(/\/+$/, '');
 };
-export const mcpConfig = (client, baseUrl, account = false) => {
+export const mcpConfig = (client, baseUrl, account = false, workspaceId) => {
     const url = `${validatedBaseUrl(baseUrl)}${account ? '/api/workspaces/mcp' : '/api/mcp'}`;
     const name = account ? 'assethub-workspaces' : 'assethub';
     const token = account ? 'ASSETHUB_ACCESS_TOKEN' : 'ASSETHUB_API_KEY';
@@ -20,12 +20,12 @@ export const mcpConfig = (client, baseUrl, account = false) => {
             mcpServers: {
                 [name]: {
                     url,
-                    headers: { Authorization: `Bearer \${env:${token}}`, ...(account ? { Cookie: '${env:ASSETHUB_WORKSPACE_MFA_COOKIE}' } : {}) },
+                    headers: { Authorization: `Bearer \${env:${token}}`, ...(!account && workspaceId ? { 'X-AssetHub-Workspace': workspaceId } : {}), ...(account ? { Cookie: '${env:ASSETHUB_WORKSPACE_MFA_COOKIE}' } : {}) },
                 },
             },
         }, null, 2) + '\n');
     if (client === 'codex')
-        return `[mcp_servers.${name}]\nurl = ${JSON.stringify(url)}\nbearer_token_env_var = "${token}"\n${account ? 'env_http_headers = { Cookie = "ASSETHUB_WORKSPACE_MFA_COOKIE" }\n' : ''}`;
+        return `[mcp_servers.${name}]\nurl = ${JSON.stringify(url)}\nbearer_token_env_var = "${token}"\n${account ? 'env_http_headers = { Cookie = "ASSETHUB_WORKSPACE_MFA_COOKIE" }\n' : workspaceId ? `http_headers = { "X-AssetHub-Workspace" = ${JSON.stringify(workspaceId)} }\n` : ''}`;
     throw new Error('Use mcp config --client cursor|codex');
 };
 const failedCheck = (name, status, timedOut, account = false) => {
@@ -119,6 +119,7 @@ export const diagnose = async ({ resolveAuth, includeMcp, timeoutMs, includeTool
             const capabilities = await createAssetHubClient({
                 apiKey: auth.apiKey,
                 baseUrl: auth.baseUrl,
+                workspaceId: auth.workspaceId,
                 fetch: boundedFetch,
             }).v2.getCapabilities();
             if (typeof capabilities.ownerId !== 'string' || !capabilities.ownerId)
@@ -141,7 +142,7 @@ export const diagnose = async ({ resolveAuth, includeMcp, timeoutMs, includeTool
         try {
             await client.connect(new StreamableHTTPClientTransport(new URL(`${auth.baseUrl}${account ? '/api/workspaces/mcp' : '/api/mcp'}`), {
                 fetch: boundedFetch,
-                requestInit: { headers: { Authorization: `Bearer ${auth.apiKey}`, ...(account && auth.workspaceMfaToken ? { Cookie: `ah_workspace_mfa=${encodeURIComponent(auth.workspaceMfaToken)}` } : {}) } },
+                requestInit: { headers: { Authorization: `Bearer ${auth.apiKey}`, ...(!account && auth.workspaceId ? { 'X-AssetHub-Workspace': auth.workspaceId } : {}), ...(account && auth.workspaceMfaToken ? { Cookie: `ah_workspace_mfa=${encodeURIComponent(auth.workspaceMfaToken)}` } : {}) } },
             }), { signal, timeout: timeoutMs });
             let toolCount = 0;
             const tools = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assethub/cli",
-  "version": "0.1.18",
+  "version": "0.1.19",
   "description": "Command line interface for AssetHub.",
   "type": "module",
   "bin": {
@@ -30,7 +30,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@assethub/api-client": "^0.1.9",
+    "@assethub/api-client": "^0.1.10",
     "@modelcontextprotocol/sdk": "^1.30.0"
   },
   "repository": {