edgegate-mcp 0.4.1 → 0.8.0

package/dist/tools/check_byo_bucket.js

@@ -0,0 +1,89 @@
+import { z } from "zod";
+import { EdgeGateError } from "../client.js";
+export const checkByoBucketInputSchema = z.object({
+    workspace_id: z.string().uuid(),
+});
+export async function checkByoBucketHandler(client, input) {
+    try {
+        const grant = await client.verifyByoGrant(input.workspace_id);
+        return renderGrantStatus(grant);
+    }
+    catch (err) {
+        if (err instanceof EdgeGateError) {
+            if (err.status === 402) {
+                return {
+                    isError: true,
+                    content: [
+                        {
+                            type: "text",
+                            text: `BYO storage requires the Enterprise plan. ` +
+                                `Contact sales: https://edgegate.frozo.ai/enterprise.`,
+                        },
+                    ],
+                };
+            }
+            if (err.status === 404) {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: [
+                                `No BYO storage grant is registered for this workspace.`,
+                                ``,
+                                `Register one with \`edgegate_register_byo_bucket\` first. ` +
+                                    `Until then, this workspace uses EdgeGate-managed storage.`,
+                            ].join("\n"),
+                        },
+                    ],
+                };
+            }
+            return {
+                isError: true,
+                content: [{ type: "text", text: `EdgeGate returned ${err.status}: ${err.detail}` }],
+            };
+        }
+        throw err;
+    }
+}
+function renderGrantStatus(grant) {
+    const lines = [
+        `BYO storage probe result: **${grant.status}**`,
+        ``,
+        `- bucket: \`${grant.bucket}\` (${grant.region})`,
+        `- role: \`${grant.role_arn}\``,
+        grant.kms_key_id
+            ? `- kms key: \`${grant.kms_key_id}\``
+            : `- kms key: (none)`,
+        `- last_verified_at: ${grant.last_verified_at ?? "(never)"}`,
+    ];
+    if (grant.status === "active") {
+        lines.push(``);
+        lines.push(`Probe passed — AssumeRole + HeadObject succeeded. You can register ` +
+            `artifacts with \`edgegate_register_byo_artifact\` and trigger runs as usual.`);
+    }
+    else if (grant.status === "failed" && grant.last_verify_error) {
+        lines.push(``);
+        lines.push(`Last error: \`${grant.last_verify_error}\``);
+        lines.push(``);
+        lines.push(`Common causes:`);
+        lines.push(`- **BYO_ASSUME_ROLE_FAILED** — the External ID in your role's trust policy ` +
+            `doesn't match. Copy the \`external_id\` from \`edgegate_register_byo_bucket\` ` +
+            `or the dashboard into your trust policy's \`sts:ExternalId\` condition.`);
+        lines.push(`- **BYO_OBJECT_ACCESS_DENIED** / **BYO_BUCKET_GONE** — the IAM role lacks ` +
+            `s3:GetObject/ListBucket on the registered bucket, or the bucket name was ` +
+            `mistyped.`);
+        lines.push(`- **BYO_KMS_ACCESS_DENIED** — bucket uses SSE-KMS but the role lacks ` +
+            `kms:Decrypt on the key.`);
+        lines.push(`- **BYO_REGION_MISMATCH** — the bucket isn't in the region you registered.`);
+        lines.push(``);
+        lines.push(`Fix the IAM/bucket config in your AWS account, then re-run ` +
+            `\`edgegate_check_byo_bucket\` to re-probe.`);
+    }
+    else if (grant.status === "revoked") {
+        lines.push(``);
+        lines.push(`The grant is revoked — no further reads will be attempted. Register a fresh ` +
+            `grant via \`edgegate_register_byo_bucket\` if you want to re-enable BYO storage.`);
+    }
+    return { content: [{ type: "text", text: lines.join("\n") }] };
+}
+//# sourceMappingURL=check_byo_bucket.js.map

package/dist/tools/check_byo_bucket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"check_byo_bucket.js","sourceRoot":"","sources":["../../src/tools/check_byo_bucket.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAI7D,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;CAChC,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAAsB,EACtB,KAA0B;IAE1B,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9D,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EACF,4CAA4C;gCAC5C,sDAAsD;yBACzD;qBACF;iBACF,CAAC;YACJ,CAAC;YACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE;gCACJ,wDAAwD;gCACxD,EAAE;gCACF,4DAA4D;oCAC1D,2DAA2D;6BAC9D,CAAC,IAAI,CAAC,IAAI,CAAC;yBACb;qBACF;iBACF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;aACpF,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAe;IACxC,MAAM,KAAK,GAAa;QACtB,+BAA+B,KAAK,CAAC,MAAM,IAAI;QAC/C,EAAE;QACF,eAAe,KAAK,CAAC,MAAM,OAAO,KAAK,CAAC,MAAM,GAAG;QACjD,aAAa,KAAK,CAAC,QAAQ,IAAI;QAC/B,KAAK,CAAC,UAAU;YACd,CAAC,CAAC,gBAAgB,KAAK,CAAC,UAAU,IAAI;YACtC,CAAC,CAAC,mBAAmB;QACvB,uBAAuB,KAAK,CAAC,gBAAgB,IAAI,SAAS,EAAE;KAC7D,CAAC;IACF,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CACR,qEAAqE;YACnE,8EAA8E,CACjF,CAAC;IACJ,CAAC;SAAM,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAChE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,iBAAiB,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;QACzD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC7B,KAAK,CAAC,IAAI,CACR,6EAA6E;YAC3E,gFAAgF;YAChF,yEAAyE,CAC5E,CAAC;QACF,KAAK,CAAC,IAAI,CACR,4EAA4E;YAC1E,2EAA2E;YAC3E,WAAW,CACd,CAAC;QACF,KAAK,CAAC,IAAI,CACR,uEAAuE;YACrE,yBAAyB,CAC5B,CAAC;QACF,KAAK,CAAC,IAAI,CACR,4EAA4E,CAC7E,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CACR,6DAA6D;YAC3D,4CAA4C,CAC/C,CAAC;IACJ,CAAC;SAAM,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CACR,8EAA8E;YAC5E,kFAAkF,CACrF,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;AACjE,CAAC"}

package/dist/tools/check_status.d.ts

@@ -5,11 +5,11 @@ export declare const checkStatusInputSchema: z.ZodObject<{
     workspace_id: z.ZodString;
     run_id: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    workspace_id: string;
     run_id: string;
-}, {
     workspace_id: string;
+}, {
     run_id: string;
+    workspace_id: string;
 }>;
 export type CheckStatusInput = z.infer<typeof checkStatusInputSchema>;
 export declare function checkStatusHandler(client: EdgeGateClient, input: CheckStatusInput): Promise<ToolResult>;

package/dist/tools/compare_runs.d.ts

@@ -23,12 +23,12 @@ export declare const compareRunsInputSchema: z.ZodObject<{
     run_id: z.ZodString;
     baseline_run_id: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
-    workspace_id: string;
     run_id: string;
+    workspace_id: string;
     baseline_run_id?: string | undefined;
 }, {
-    workspace_id: string;
     run_id: string;
+    workspace_id: string;
     baseline_run_id?: string | undefined;
 }>;
 export type CompareRunsInput = z.infer<typeof compareRunsInputSchema>;

package/dist/tools/connect_huggingface.d.ts

@@ -0,0 +1,15 @@
+import { z } from "zod";
+import { EdgeGateClient } from "../client.js";
+import type { ToolResult } from "./setup_workspace.js";
+export declare const connectHuggingfaceInputSchema: z.ZodObject<{
+    workspace_id: z.ZodString;
+    token: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    workspace_id: string;
+    token: string;
+}, {
+    workspace_id: string;
+    token: string;
+}>;
+export type ConnectHuggingfaceInput = z.infer<typeof connectHuggingfaceInputSchema>;
+export declare function connectHuggingfaceHandler(client: EdgeGateClient, input: ConnectHuggingfaceInput): Promise<ToolResult>;

package/dist/tools/connect_huggingface.js

@@ -0,0 +1,83 @@
+import { z } from "zod";
+import { EdgeGateError } from "../client.js";
+export const connectHuggingfaceInputSchema = z.object({
+    workspace_id: z.string().uuid(),
+    token: z
+        .string()
+        .min(8)
+        .describe("Personal HuggingFace access token (starts with hf_). Generate one at " +
+        "https://huggingface.co/settings/tokens — Read scope is enough for the " +
+        "import flow."),
+});
+export async function connectHuggingfaceHandler(client, input) {
+    try {
+        const result = await client.connectHuggingFaceIntegration(input.workspace_id, input.token);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: [
+                        `Connected HuggingFace as **${result.account_name}** (${result.account_type}).`,
+                        ``,
+                        `- token: \`****${result.token_last4}\` (encrypted at rest, never echoed)`,
+                        `- status: ${result.status}`,
+                        ``,
+                        `EdgeGate will now use this token for \`edgegate_import_huggingface_model\` calls in this workspace, so private / gated / Qualcomm-org repos work end-to-end.`,
+                        `Rotate by calling this tool again with a fresh token; remove with \`edgegate_disconnect_huggingface\`.`,
+                    ].join("\n"),
+                },
+            ],
+        };
+    }
+    catch (err) {
+        if (err instanceof EdgeGateError) {
+            if (err.status === 409) {
+                // Integration already exists — try rotating instead.
+                try {
+                    const rotated = await client.rotateHuggingFaceIntegration(input.workspace_id, input.token);
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: [
+                                    `Replaced existing HuggingFace token for account **${rotated.account_name}**.`,
+                                    ``,
+                                    `- token: \`****${rotated.token_last4}\``,
+                                    `- status: ${rotated.status}`,
+                                ].join("\n"),
+                            },
+                        ],
+                    };
+                }
+                catch (rotateErr) {
+                    if (rotateErr instanceof EdgeGateError) {
+                        return formatError(rotateErr);
+                    }
+                    throw rotateErr;
+                }
+            }
+            return formatError(err);
+        }
+        throw err;
+    }
+}
+function formatError(err) {
+    if (err.status === 400) {
+        return {
+            isError: true,
+            content: [
+                {
+                    type: "text",
+                    text: `HuggingFace rejected the token: ${err.detail}\n\n` +
+                        `Generate a fresh token at https://huggingface.co/settings/tokens — the ` +
+                        `default "Read" scope is sufficient.`,
+                },
+            ],
+        };
+    }
+    return {
+        isError: true,
+        content: [{ type: "text", text: `EdgeGate returned ${err.status}: ${err.detail}` }],
+    };
+}
+//# sourceMappingURL=connect_huggingface.js.map

package/dist/tools/connect_huggingface.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connect_huggingface.js","sourceRoot":"","sources":["../../src/tools/connect_huggingface.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IACpD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACP,uEAAuE;QACrE,wEAAwE;QACxE,cAAc,CACjB;CACJ,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,MAAsB,EACtB,KAA8B;IAE9B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,6BAA6B,CACvD,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,KAAK,CACZ,CAAC;QAEF,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE;wBACJ,8BAA8B,MAAM,CAAC,YAAY,OAAO,MAAM,CAAC,YAAY,IAAI;wBAC/E,EAAE;wBACF,kBAAkB,MAAM,CAAC,WAAW,sCAAsC;wBAC1E,aAAa,MAAM,CAAC,MAAM,EAAE;wBAC5B,EAAE;wBACF,8JAA8J;wBAC9J,wGAAwG;qBACzG,CAAC,IAAI,CAAC,IAAI,CAAC;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,qDAAqD;gBACrD,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,4BAA4B,CACvD,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,KAAK,CACZ,CAAC;oBACF,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE;oCACJ,qDAAqD,OAAO,CAAC,YAAY,KAAK;oCAC9E,EAAE;oCACF,kBAAkB,OAAO,CAAC,WAAW,IAAI;oCACzC,aAAa,OAAO,CAAC,MAAM,EAAE;iCAC9B,CAAC,IAAI,CAAC,IAAI,CAAC;6BACb;yBACF;qBACF,CAAC;gBACJ,CAAC;gBAAC,OAAO,SAAS,EAAE,CAAC;oBACnB,IAAI,SAAS,YAAY,aAAa,EAAE,CAAC;wBACvC,OAAO,WAAW,CAAC,SAAS,CAAC,CAAC;oBAChC,CAAC;oBACD,MAAM,SAAS,CAAC;gBAClB,CAAC;YACH,CAAC;YACD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,GAAkB;IACrC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACvB,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EACF,mCAAmC,GAAG,CAAC,MAAM,MAAM;wBACnD,yEAAyE;wBACzE,qCAAqC;iBACxC;aACF;SACF,CAAC;IACJ,CAAC;IACD,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;KACpF,CAAC;AACJ,CAAC"}

package/dist/tools/connect_qaihub.d.ts

@@ -0,0 +1,15 @@
+import { z } from "zod";
+import { EdgeGateClient } from "../client.js";
+import type { ToolResult } from "./setup_workspace.js";
+export declare const connectQaihubInputSchema: z.ZodObject<{
+    workspace_id: z.ZodString;
+    token: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    workspace_id: string;
+    token: string;
+}, {
+    workspace_id: string;
+    token: string;
+}>;
+export type ConnectQaihubInput = z.infer<typeof connectQaihubInputSchema>;
+export declare function connectQaihubHandler(client: EdgeGateClient, input: ConnectQaihubInput): Promise<ToolResult>;

package/dist/tools/connect_qaihub.js

@@ -0,0 +1,62 @@
+import { z } from "zod";
+import { EdgeGateError } from "../client.js";
+export const connectQaihubInputSchema = z.object({
+    workspace_id: z.string().uuid(),
+    token: z
+        .string()
+        .min(8)
+        .describe("Qualcomm AI Hub API token. Generate one at " +
+        "https://app.aihub.qualcomm.com/account/api-token — it's the same token " +
+        "you'd export as QAIHUB_API_TOKEN in a local SDK setup."),
+});
+export async function connectQaihubHandler(client, input) {
+    try {
+        const result = await client.connectQaihubIntegration(input.workspace_id, input.token);
+        return successText(result, "Connected");
+    }
+    catch (err) {
+        if (err instanceof EdgeGateError) {
+            if (err.status === 409) {
+                // Already exists — rotate transparently. Users don't have to think
+                // about "is there already a token here" — the right thing happens
+                // on the second call too.
+                try {
+                    const rotated = await client.rotateQaihubIntegration(input.workspace_id, input.token);
+                    return successText(rotated, "Replaced");
+                }
+                catch (rotateErr) {
+                    if (rotateErr instanceof EdgeGateError) {
+                        return formatError(rotateErr);
+                    }
+                    throw rotateErr;
+                }
+            }
+            return formatError(err);
+        }
+        throw err;
+    }
+}
+function successText(result, verb) {
+    return {
+        content: [
+            {
+                type: "text",
+                text: [
+                    `${verb} Qualcomm AI Hub integration for this workspace.`,
+                    ``,
+                    `- token: \`****${result.token_last4}\` (encrypted at rest, never echoed)`,
+                    `- status: ${result.status}`,
+                    ``,
+                    `EdgeGate runs in this workspace will now use this token to compile + profile models on real Snapdragon devices via Qualcomm AI Hub.`,
+                ].join("\n"),
+            },
+        ],
+    };
+}
+function formatError(err) {
+    return {
+        isError: true,
+        content: [{ type: "text", text: `EdgeGate returned ${err.status}: ${err.detail}` }],
+    };
+}
+//# sourceMappingURL=connect_qaihub.js.map

package/dist/tools/connect_qaihub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connect_qaihub.js","sourceRoot":"","sources":["../../src/tools/connect_qaihub.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACP,6CAA6C;QAC3C,yEAAyE;QACzE,wDAAwD,CAC3D;CACJ,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAsB,EACtB,KAAyB;IAEzB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAClD,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,KAAK,CACZ,CAAC;QACF,OAAO,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC1C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,mEAAmE;gBACnE,kEAAkE;gBAClE,0BAA0B;gBAC1B,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAClD,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,KAAK,CACZ,CAAC;oBACF,OAAO,WAAW,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC1C,CAAC;gBAAC,OAAO,SAAS,EAAE,CAAC;oBACnB,IAAI,SAAS,YAAY,aAAa,EAAE,CAAC;wBACvC,OAAO,WAAW,CAAC,SAAS,CAAC,CAAC;oBAChC,CAAC;oBACD,MAAM,SAAS,CAAC;gBAClB,CAAC;YACH,CAAC;YACD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAClB,MAA+C,EAC/C,IAA8B;IAE9B,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE;oBACJ,GAAG,IAAI,kDAAkD;oBACzD,EAAE;oBACF,kBAAkB,MAAM,CAAC,WAAW,sCAAsC;oBAC1E,aAAa,MAAM,CAAC,MAAM,EAAE;oBAC5B,EAAE;oBACF,qIAAqI;iBACtI,CAAC,IAAI,CAAC,IAAI,CAAC;aACb;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,GAAkB;IACrC,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;KACpF,CAAC;AACJ,CAAC"}

package/dist/tools/create_api_key.d.ts

@@ -0,0 +1,18 @@
+import { z } from "zod";
+import { EdgeGateClient } from "../client.js";
+import type { ToolResult } from "./setup_workspace.js";
+export declare const createApiKeyInputSchema: z.ZodObject<{
+    workspace_id: z.ZodString;
+    name: z.ZodString;
+    expires_at: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    workspace_id: string;
+    name: string;
+    expires_at?: string | undefined;
+}, {
+    workspace_id: string;
+    name: string;
+    expires_at?: string | undefined;
+}>;
+export type CreateApiKeyInput = z.infer<typeof createApiKeyInputSchema>;
+export declare function createApiKeyHandler(client: EdgeGateClient, input: CreateApiKeyInput): Promise<ToolResult>;

package/dist/tools/create_api_key.js

@@ -0,0 +1,73 @@
+import { z } from "zod";
+import { EdgeGateError } from "../client.js";
+export const createApiKeyInputSchema = z.object({
+    workspace_id: z.string().uuid(),
+    name: z
+        .string()
+        .min(1)
+        .max(255)
+        .describe("Human-readable label for the key, e.g. \"GitHub Actions production\" — " +
+        "you'll need this later to identify which key to rotate or revoke."),
+    expires_at: z
+        .string()
+        .datetime()
+        .optional()
+        .describe("Optional ISO-8601 expiry timestamp. Omit for a non-expiring key " +
+        "(you can always revoke it manually)."),
+});
+export async function createApiKeyHandler(client, input) {
+    try {
+        const key = await client.createApiKey(input.workspace_id, {
+            name: input.name,
+            expires_at: input.expires_at ?? null,
+        });
+        // Plaintext is shown ONCE and never again. Flag that explicitly so the
+        // user knows to copy it now — and so the LLM client can be careful with
+        // how/where it persists this in chat scrollback.
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: [
+                        `Created API key **${key.name}**.`,
+                        ``,
+                        `**Copy this plaintext now — it cannot be recovered later:**`,
+                        ``,
+                        "```",
+                        key.plaintext,
+                        "```",
+                        ``,
+                        `Details:`,
+                        `- key_id: \`${key.id}\``,
+                        `- prefix...suffix: \`${key.prefix}...${key.suffix}\``,
+                        `- expires: ${key.expires_at ?? "never"}`,
+                        ``,
+                        `Use it as the \`EDGEGATE_API_KEY\` env var for the MCP server, or as the bearer token for direct REST calls. Revoke it any time with \`edgegate_revoke_api_key({ workspace_id, key_id: "${key.id}" })\`.`,
+                    ].join("\n"),
+                },
+            ],
+        };
+    }
+    catch (err) {
+        if (err instanceof EdgeGateError) {
+            if (err.status === 402) {
+                return {
+                    isError: true,
+                    content: [
+                        {
+                            type: "text",
+                            text: `API key creation requires a Pro tier or above: ${err.detail}\n\n` +
+                                `Upgrade at https://edgegate.frozo.ai/pricing.`,
+                        },
+                    ],
+                };
+            }
+            return {
+                isError: true,
+                content: [{ type: "text", text: `EdgeGate returned ${err.status}: ${err.detail}` }],
+            };
+        }
+        throw err;
+    }
+}
+//# sourceMappingURL=create_api_key.js.map

package/dist/tools/create_api_key.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create_api_key.js","sourceRoot":"","sources":["../../src/tools/create_api_key.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,IAAI,EAAE,CAAC;SACJ,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,GAAG,CAAC;SACR,QAAQ,CACP,yEAAyE;QACvE,mEAAmE,CACtE;IACH,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,EAAE;SACV,QAAQ,CACP,kEAAkE;QAChE,sCAAsC,CACzC;CACJ,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,MAAsB,EACtB,KAAwB;IAExB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE;YACxD,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,UAAU,EAAE,KAAK,CAAC,UAAU,IAAI,IAAI;SACrC,CAAC,CAAC;QAEH,uEAAuE;QACvE,wEAAwE;QACxE,iDAAiD;QACjD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE;wBACJ,qBAAqB,GAAG,CAAC,IAAI,KAAK;wBAClC,EAAE;wBACF,6DAA6D;wBAC7D,EAAE;wBACF,KAAK;wBACL,GAAG,CAAC,SAAS;wBACb,KAAK;wBACL,EAAE;wBACF,UAAU;wBACV,eAAe,GAAG,CAAC,EAAE,IAAI;wBACzB,wBAAwB,GAAG,CAAC,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI;wBACtD,cAAc,GAAG,CAAC,UAAU,IAAI,OAAO,EAAE;wBACzC,EAAE;wBACF,2LAA2L,GAAG,CAAC,EAAE,SAAS;qBAC3M,CAAC,IAAI,CAAC,IAAI,CAAC;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EACF,kDAAkD,GAAG,CAAC,MAAM,MAAM;gCAClE,+CAA+C;yBAClD;qBACF;iBACF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;aACpF,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/tools/create_pipeline.d.ts

@@ -34,14 +34,36 @@ export declare const createPipelineInputSchema: z.ZodObject<{
         name: z.ZodString;
         artifact_id: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        name: string;
         artifact_id: string;
-    }, {
         name: string;
+    }, {
         artifact_id: string;
+        name: string;
     }>, "many">>;
     /** Measurement repeats per cell (1–5). Defaults to API default (3). */
     repeats: z.ZodOptional<z.ZodNumber>;
+    /**
+     * Optional. Override AI Hub input shapes per named input.
+     *
+     * For text models like MiniLM / BERT-family, try:
+     *   `{ input_ids: { shape: [1, 128], dtype: "int64" }, attention_mask: { shape: [1, 128], dtype: "int64" } }`
+     *
+     * For image models (e.g. MobileNet) the backend auto-detects static shapes
+     * from the ONNX file — omit this field.
+     *
+     * Omit entirely to let EdgeGate auto-detect from the ONNX file (works for most
+     * models including image classification, BERT-family, MiniLM).
+     */
+    input_specs: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        shape: z.ZodArray<z.ZodNumber, "many">;
+        dtype: z.ZodDefault<z.ZodEnum<["float32", "float16", "int64", "int32", "bool"]>>;
+    }, "strip", z.ZodTypeAny, {
+        shape: number[];
+        dtype: "float32" | "float16" | "int64" | "int32" | "bool";
+    }, {
+        shape: number[];
+        dtype?: "float32" | "float16" | "int64" | "int32" | "bool" | undefined;
+    }>>>;
 }, "strip", z.ZodTypeAny, {
     workspace_id: string;
     name: string;
@@ -55,10 +77,14 @@ export declare const createPipelineInputSchema: z.ZodObject<{
         description?: string | undefined;
     }[];
     models?: {
-        name: string;
         artifact_id: string;
+        name: string;
     }[] | undefined;
     repeats?: number | undefined;
+    input_specs?: Record<string, {
+        shape: number[];
+        dtype: "float32" | "float16" | "int64" | "int32" | "bool";
+    }> | undefined;
 }, {
     workspace_id: string;
     name: string;
@@ -72,10 +98,14 @@ export declare const createPipelineInputSchema: z.ZodObject<{
     }[];
     promptpack_version?: string | undefined;
     models?: {
-        name: string;
         artifact_id: string;
+        name: string;
     }[] | undefined;
     repeats?: number | undefined;
+    input_specs?: Record<string, {
+        shape: number[];
+        dtype?: "float32" | "float16" | "int64" | "int32" | "bool" | undefined;
+    }> | undefined;
 }>;
 export type CreatePipelineInput = z.infer<typeof createPipelineInputSchema>;
 export declare function createPipelineHandler(client: EdgeGateClient, input: CreatePipelineInput): Promise<ToolResult>;

package/dist/tools/create_pipeline.js

@@ -37,6 +37,36 @@ export const createPipelineInputSchema = z.object({
         .optional(),
     /** Measurement repeats per cell (1–5). Defaults to API default (3). */
     repeats: z.number().int().min(1).max(5).optional(),
+    /**
+     * Optional. Override AI Hub input shapes per named input.
+     *
+     * For text models like MiniLM / BERT-family, try:
+     *   `{ input_ids: { shape: [1, 128], dtype: "int64" }, attention_mask: { shape: [1, 128], dtype: "int64" } }`
+     *
+     * For image models (e.g. MobileNet) the backend auto-detects static shapes
+     * from the ONNX file — omit this field.
+     *
+     * Omit entirely to let EdgeGate auto-detect from the ONNX file (works for most
+     * models including image classification, BERT-family, MiniLM).
+     */
+    input_specs: z
+        .record(z.string(), z.object({
+        shape: z
+            .array(z.number().int().positive())
+            .min(1)
+            .max(8)
+            .describe("Tensor shape (1–8 positive integers)"),
+        dtype: z
+            .enum(["float32", "float16", "int64", "int32", "bool"])
+            .default("float32")
+            .describe("Element dtype"),
+    }))
+        .optional()
+        .describe("Optional. Override AI Hub input shapes per named input. " +
+        'For text models like MiniLM, try `{input_ids: {shape: [1, 128], dtype: "int64"}, ' +
+        'attention_mask: {shape: [1, 128], dtype: "int64"}}`. ' +
+        "Omit to let EdgeGate auto-detect from the ONNX file (works for most models " +
+        "including image classification, BERT-family, MiniLM)."),
 });
 export async function createPipelineHandler(client, input) {
     // Guard: cell count limit applies when model_matrix is provided
@@ -77,6 +107,7 @@ export async function createPipelineHandler(client, input) {
             gates,
             ...(model_matrix !== undefined ? { model_matrix } : {}),
             ...(run_policy !== undefined ? { run_policy } : {}),
+            ...(input.input_specs !== undefined ? { input_specs: input.input_specs } : {}),
         });
         return {
             content: [

package/dist/tools/create_pipeline.js.map

@@ -1 +1 @@
-{"version":3,"file":"create_pipeline.js","sourceRoot":"","sources":["../../src/tools/create_pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,SAAS,GAAG,EAAE,CAAC;AAErB,4DAA4D;AAC5D,MAAM,YAAY,GAA2B;IAC3C,IAAI,EAAE,KAAK;IACX,GAAG,EAAE,IAAI;IACT,IAAI,EAAE,KAAK;IACX,GAAG,EAAE,IAAI;IACT,IAAI,EAAE,IAAI;CACX,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;IAChC,0GAA0G;IAC1G,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAChC,8CAA8C;IAC9C,kBAAkB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IACtD,oEAAoE;IACpE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACjD,KAAK,EAAE,CAAC;SACL,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,mBAAmB,EAAE,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;QACzE,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;QAC9C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAChC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KACnC,CAAC,CACH;SACA,GAAG,CAAC,CAAC,CAAC;IACT;;;OAGG;IACH,MAAM,EAAE,CAAC;SACN,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SAC5E,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;IACb,uEAAuE;IACvE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;CACnD,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAAsB,EACtB,KAA0B;IAE1B,gEAAgE;IAChE,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;IACpD,IAAI,KAAK,CAAC,MAAM,IAAI,SAAS,GAAG,SAAS,EAAE,CAAC;QAC1C,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EACF,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,aAAa,KAAK,CAAC,OAAO,CAAC,MAAM,cAAc,SAAS,UAAU;wBACxF,sCAAsC,SAAS,+BAA+B;wBAC9E,0BAA0B;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC;IAED,0CAA0C;IAC1C,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAC7E,MAAM,cAAc,GAAG;QACrB,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,OAAO,EAAE,KAAK,CAAC,kBAAkB;KAClC,CAAC;IACF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpC,MAAM,EAAE,CAAC,CAAC,MAAM;QAChB,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ;QAChD,SAAS,EAAE,CAAC,CAAC,SAAS;QACtB,GAAG,CAAC,CAAC,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CAAC,CAAC,CAAC;IACJ,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC/F,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAEpG,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE;YAC/D,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,aAAa;YACb,cAAc;YACd,KAAK;YACL,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACpD,CAAC,CAAC;QACH,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE;wBACJ,sBAAsB,QAAQ,CAAC,IAAI,UAAU,QAAQ,CAAC,EAAE,IAAI;wBAC5D,EAAE;wBACF,KAAK,QAAQ,CAAC,YAAY,gBAAgB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,eAAe,QAAQ,CAAC,UAAU,kBAAkB;wBAC/H,KAAK,QAAQ,CAAC,UAAU,UAAU;wBAClC,iBAAiB,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,kBAAkB,EAAE;wBACxE,EAAE;wBACF,6FAA6F;qBAC9F,CAAC,IAAI,CAAC,IAAI,CAAC;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;aAC9E,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}
+{"version":3,"file":"create_pipeline.js","sourceRoot":"","sources":["../../src/tools/create_pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,SAAS,GAAG,EAAE,CAAC;AAErB,4DAA4D;AAC5D,MAAM,YAAY,GAA2B;IAC3C,IAAI,EAAE,KAAK;IACX,GAAG,EAAE,IAAI;IACT,IAAI,EAAE,KAAK;IACX,GAAG,EAAE,IAAI;IACT,IAAI,EAAE,IAAI;CACX,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;IAChC,0GAA0G;IAC1G,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAChC,8CAA8C;IAC9C,kBAAkB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IACtD,oEAAoE;IACpE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACjD,KAAK,EAAE,CAAC;SACL,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,mBAAmB,EAAE,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;QACzE,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;QAC9C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAChC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KACnC,CAAC,CACH;SACA,GAAG,CAAC,CAAC,CAAC;IACT;;;OAGG;IACH,MAAM,EAAE,CAAC;SACN,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SAC5E,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;IACb,uEAAuE;IACvE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;IAClD;;;;;;;;;;;OAWG;IACH,WAAW,EAAE,CAAC;SACX,MAAM,CACL,CAAC,CAAC,MAAM,EAAE,EACV,CAAC,CAAC,MAAM,CAAC;QACP,KAAK,EAAE,CAAC;aACL,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC;aACN,GAAG,CAAC,CAAC,CAAC;aACN,QAAQ,CAAC,sCAAsC,CAAC;QACnD,KAAK,EAAE,CAAC;aACL,IAAI,CAAC,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;aACtD,OAAO,CAAC,SAAS,CAAC;aAClB,QAAQ,CAAC,eAAe,CAAC;KAC7B,CAAC,CACH;SACA,QAAQ,EAAE;SACV,QAAQ,CACP,0DAA0D;QACxD,mFAAmF;QACnF,uDAAuD;QACvD,6EAA6E;QAC7E,uDAAuD,CAC1D;CACJ,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAAsB,EACtB,KAA0B;IAE1B,gEAAgE;IAChE,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;IACpD,IAAI,KAAK,CAAC,MAAM,IAAI,SAAS,GAAG,SAAS,EAAE,CAAC;QAC1C,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EACF,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,aAAa,KAAK,CAAC,OAAO,CAAC,MAAM,cAAc,SAAS,UAAU;wBACxF,sCAAsC,SAAS,+BAA+B;wBAC9E,0BAA0B;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC;IAED,0CAA0C;IAC1C,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAC7E,MAAM,cAAc,GAAG;QACrB,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,OAAO,EAAE,KAAK,CAAC,kBAAkB;KAClC,CAAC;IACF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpC,MAAM,EAAE,CAAC,CAAC,MAAM;QAChB,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ;QAChD,SAAS,EAAE,CAAC,CAAC,SAAS;QACtB,GAAG,CAAC,CAAC,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CAAC,CAAC,CAAC;IACJ,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC/F,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAEpG,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE;YAC/D,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,aAAa;YACb,cAAc;YACd,KAAK;YACL,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/E,CAAC,CAAC;QACH,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE;wBACJ,sBAAsB,QAAQ,CAAC,IAAI,UAAU,QAAQ,CAAC,EAAE,IAAI;wBAC5D,EAAE;wBACF,KAAK,QAAQ,CAAC,YAAY,gBAAgB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,eAAe,QAAQ,CAAC,UAAU,kBAAkB;wBAC/H,KAAK,QAAQ,CAAC,UAAU,UAAU;wBAClC,iBAAiB,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,kBAAkB,EAAE;wBACxE,EAAE;wBACF,6FAA6F;qBAC9F,CAAC,IAAI,CAAC,IAAI,CAAC;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;aAC9E,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/tools/create_workspace.d.ts

@@ -0,0 +1,12 @@
+import { z } from "zod";
+import { EdgeGateClient } from "../client.js";
+import type { ToolResult } from "./setup_workspace.js";
+export declare const createWorkspaceInputSchema: z.ZodObject<{
+    name: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    name: string;
+}, {
+    name: string;
+}>;
+export type CreateWorkspaceInput = z.infer<typeof createWorkspaceInputSchema>;
+export declare function createWorkspaceHandler(client: EdgeGateClient, input: CreateWorkspaceInput): Promise<ToolResult>;

package/dist/tools/create_workspace.js

@@ -0,0 +1,51 @@
+import { z } from "zod";
+import { EdgeGateError } from "../client.js";
+export const createWorkspaceInputSchema = z.object({
+    name: z
+        .string()
+        .min(1)
+        .max(255)
+        .describe("Display name for the new workspace, e.g. \"MobileNet Production\"."),
+});
+export async function createWorkspaceHandler(client, input) {
+    try {
+        const ws = await client.createWorkspace(input.name);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: [
+                        `Created workspace **${ws.name}** — you are the owner.`,
+                        ``,
+                        `- workspace_id: \`${ws.id}\``,
+                        `- plan: ${ws.plan ?? "(unknown)"}`,
+                        ``,
+                        `Next: connect Qualcomm AI Hub via \`edgegate_connect_qaihub\`, then start defining pipelines with \`edgegate_create_pipeline\`. Pass \`workspace_id\` to every subsequent tool call so they operate on this workspace.`,
+                    ].join("\n"),
+                },
+            ],
+        };
+    }
+    catch (err) {
+        if (err instanceof EdgeGateError) {
+            if (err.status === 403) {
+                return {
+                    isError: true,
+                    content: [
+                        {
+                            type: "text",
+                            text: `Workspace creation blocked: ${err.detail}\n\n` +
+                                `If you've hit your plan's workspace limit, upgrade at https://edgegate.frozo.ai/pricing.`,
+                        },
+                    ],
+                };
+            }
+            return {
+                isError: true,
+                content: [{ type: "text", text: `EdgeGate returned ${err.status}: ${err.detail}` }],
+            };
+        }
+        throw err;
+    }
+}
+//# sourceMappingURL=create_workspace.js.map

package/dist/tools/create_workspace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create_workspace.js","sourceRoot":"","sources":["../../src/tools/create_workspace.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,IAAI,EAAE,CAAC;SACJ,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,GAAG,CAAC;SACR,QAAQ,CAAC,oEAAoE,CAAC;CAClF,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,MAAsB,EACtB,KAA2B;IAE3B,IAAI,CAAC;QACH,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACpD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE;wBACJ,uBAAuB,EAAE,CAAC,IAAI,yBAAyB;wBACvD,EAAE;wBACF,qBAAqB,EAAE,CAAC,EAAE,IAAI;wBAC9B,WAAW,EAAE,CAAC,IAAI,IAAI,WAAW,EAAE;wBACnC,EAAE;wBACF,wNAAwN;qBACzN,CAAC,IAAI,CAAC,IAAI,CAAC;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EACF,+BAA+B,GAAG,CAAC,MAAM,MAAM;gCAC/C,0FAA0F;yBAC7F;qBACF;iBACF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;aACpF,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/tools/disconnect_byo_bucket.d.ts

@@ -0,0 +1,12 @@
+import { z } from "zod";
+import { EdgeGateClient } from "../client.js";
+import type { ToolResult } from "./setup_workspace.js";
+export declare const disconnectByoBucketInputSchema: z.ZodObject<{
+    workspace_id: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    workspace_id: string;
+}, {
+    workspace_id: string;
+}>;
+export type DisconnectByoBucketInput = z.infer<typeof disconnectByoBucketInputSchema>;
+export declare function disconnectByoBucketHandler(client: EdgeGateClient, input: DisconnectByoBucketInput): Promise<ToolResult>;