zerogpu-cli 1.0.0 → 2.0.0

package/dist/commands/classifyStructured.js

@@ -0,0 +1,73 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "gliner2-base-v1";
+export function registerClassifyStructuredCommand(program) {
+    program
+        .command("classify_structured <text>")
+        .description("Classify text against a structured schema of categories and labels.")
+        .requiredOption("-s, --schema <json>", 'JSON object mapping category name to allowed labels, e.g. \'{"sentiment":["positive","negative","neutral"]}\'')
+        .action(async (text, opts) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        let schema;
+        try {
+            schema = JSON.parse(opts.schema);
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Invalid --schema JSON: ${message}`);
+            process.exit(1);
+        }
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                    metadata: {
+                        schema,
+                        usecase: "classification",
+                    },
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")
+            ?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any classification content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=classifyStructured.js.map

package/dist/commands/classifyStructured.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classifyStructured.js","sourceRoot":"","sources":["../../src/commands/classifyStructured.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,iBAAiB,CAAC;AAEhC,MAAM,UAAU,iCAAiC,CAAC,OAAgB;IAChE,OAAO;SACJ,OAAO,CAAC,4BAA4B,CAAC;SACrC,WAAW,CACV,qEAAqE,CACtE;SACA,cAAc,CACb,qBAAqB,EACrB,+GAA+G,CAChH;SACA,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,IAAwB,EAAE,EAAE;QACvD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,0BAA0B,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE;wBACR,MAAM;wBACN,OAAO,EAAE,gBAAgB;qBAC1B;iBACF,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GACX,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC;YAC9D,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QACnD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;YACtE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/commands/classifyZeroShot.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerClassifyZeroShotCommand(program: Command): void;
+//# sourceMappingURL=classifyZeroShot.d.ts.map

package/dist/commands/classifyZeroShot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classifyZeroShot.d.ts","sourceRoot":"","sources":["../../src/commands/classifyZeroShot.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAapC,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CA4FtE"}

package/dist/commands/classifyZeroShot.js

@@ -0,0 +1,75 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "deberta-v3-small";
+function collectLabel(value, previous) {
+    return previous.concat([value]);
+}
+export function registerClassifyZeroShotCommand(program) {
+    program
+        .command("classify_zero_shot <text>")
+        .description("Zero-shot classify text against a set of candidate labels using DeBERTa v3.")
+        .option("-l, --label <label>", "Candidate label (repeatable).", collectLabel, [])
+        .option("--labels <labels>", "Comma-separated list of candidate labels (alternative to repeating --label).")
+        .action(async (text, opts) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        const labels = [
+            ...opts.label,
+            ...(opts.labels
+                ? opts.labels.split(",").map((l) => l.trim()).filter(Boolean)
+                : []),
+        ];
+        if (labels.length === 0) {
+            console.error("At least one label is required. Use --label <name> (repeatable) or --labels a,b,c.");
+            process.exit(1);
+        }
+        const instructions = `[${labels.join(", ")}]`;
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                    instructions,
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any classification content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=classifyZeroShot.js.map

package/dist/commands/classifyZeroShot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classifyZeroShot.js","sourceRoot":"","sources":["../../src/commands/classifyZeroShot.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,kBAAkB,CAAC;AAEjC,SAAS,YAAY,CAAC,KAAa,EAAE,QAAkB;IACrD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,+BAA+B,CAAC,OAAgB;IAC9D,OAAO;SACJ,OAAO,CAAC,2BAA2B,CAAC;SACpC,WAAW,CACV,6EAA6E,CAC9E;SACA,MAAM,CACL,qBAAqB,EACrB,+BAA+B,EAC/B,YAAY,EACZ,EAAE,CACH;SACA,MAAM,CACL,mBAAmB,EACnB,8EAA8E,CAC/E;SACA,MAAM,CACL,KAAK,EACH,IAAY,EACZ,IAA0C,EAC1C,EAAE;QACF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,MAAM,GAAG;YACb,GAAG,IAAI,CAAC,KAAK;YACb,GAAG,CAAC,IAAI,CAAC,MAAM;gBACb,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;gBAC7D,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,CAAC,KAAK,CACX,oFAAoF,CACrF,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAE9C,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;oBACX,YAAY;iBACb,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAChC,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;YACtE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CACF,CAAC;AACN,CAAC"}

package/dist/commands/extractEntities.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerExtractEntitiesCommand(program: Command): void;
+//# sourceMappingURL=extractEntities.d.ts.map

package/dist/commands/extractEntities.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractEntities.d.ts","sourceRoot":"","sources":["../../src/commands/extractEntities.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAcpC,wBAAgB,8BAA8B,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAyGrE"}

package/dist/commands/extractEntities.js

@@ -0,0 +1,85 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "gliner2-base-v1";
+const DEFAULT_THRESHOLD = 0.3;
+function collectLabel(value, previous) {
+    return previous.concat([value]);
+}
+export function registerExtractEntitiesCommand(program) {
+    program
+        .command("extract_entities <text>")
+        .description("Extract named entities from text using the GLiNER2 model with custom labels.")
+        .option("-l, --label <label>", "Entity label to extract (repeatable).", collectLabel, [])
+        .option("--labels <labels>", "Comma-separated list of entity labels (alternative to repeating --label).")
+        .option("-t, --threshold <number>", "Confidence threshold between 0 and 1.", `${DEFAULT_THRESHOLD}`)
+        .action(async (text, opts) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        const labels = [
+            ...opts.label,
+            ...(opts.labels
+                ? opts.labels.split(",").map((l) => l.trim()).filter(Boolean)
+                : []),
+        ];
+        if (labels.length === 0) {
+            console.error("At least one label is required. Use --label <name> (repeatable) or --labels a,b,c.");
+            process.exit(1);
+        }
+        const threshold = Number(opts.threshold);
+        if (!Number.isFinite(threshold) || threshold < 0 || threshold > 1) {
+            console.error("--threshold must be a number between 0 and 1.");
+            process.exit(1);
+        }
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                    metadata: {
+                        labels,
+                        usecase: "ner",
+                        threshold,
+                    },
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any entity content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=extractEntities.js.map

package/dist/commands/extractEntities.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractEntities.js","sourceRoot":"","sources":["../../src/commands/extractEntities.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,iBAAiB,CAAC;AAChC,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAE9B,SAAS,YAAY,CAAC,KAAa,EAAE,QAAkB;IACrD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,OAAgB;IAC7D,OAAO;SACJ,OAAO,CAAC,yBAAyB,CAAC;SAClC,WAAW,CACV,8EAA8E,CAC/E;SACA,MAAM,CACL,qBAAqB,EACrB,uCAAuC,EACvC,YAAY,EACZ,EAAE,CACH;SACA,MAAM,CACL,mBAAmB,EACnB,2EAA2E,CAC5E;SACA,MAAM,CACL,0BAA0B,EAC1B,uCAAuC,EACvC,GAAG,iBAAiB,EAAE,CACvB;SACA,MAAM,CACL,KAAK,EACH,IAAY,EACZ,IAA6D,EAC7D,EAAE;QACF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,MAAM,GAAG;YACb,GAAG,IAAI,CAAC,KAAK;YACb,GAAG,CAAC,IAAI,CAAC,MAAM;gBACb,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;gBAC7D,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,CAAC,KAAK,CACX,oFAAoF,CACrF,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClE,OAAO,CAAC,KAAK,CAAC,+CAA+C,CAAC,CAAC;YAC/D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE;wBACR,MAAM;wBACN,OAAO,EAAE,KAAK;wBACd,SAAS;qBACV;iBACF,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAChC,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAC9D,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CACF,CAAC;AACN,CAAC"}

package/dist/commands/extractJson.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerExtractJsonCommand(program: Command): void;
+//# sourceMappingURL=extractJson.d.ts.map

package/dist/commands/extractJson.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractJson.d.ts","sourceRoot":"","sources":["../../src/commands/extractJson.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AASpC,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CA8EjE"}

package/dist/commands/extractJson.js

@@ -0,0 +1,72 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "gliner2-base-v1";
+export function registerExtractJsonCommand(program) {
+    program
+        .command("extract_json <text>")
+        .description("Extract structured JSON from text according to a provided schema.")
+        .requiredOption("-s, --schema <json>", 'Schema as a JSON string, e.g. \'{"contact":["name::str::Full name","email::str::Email address"]}\'')
+        .action(async (text, options) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        let schema;
+        try {
+            schema = JSON.parse(options.schema);
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Invalid --schema JSON: ${message}`);
+            process.exit(1);
+        }
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                    metadata: {
+                        schema,
+                        usecase: "json",
+                    },
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any extraction content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=extractJson.js.map

package/dist/commands/extractJson.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractJson.js","sourceRoot":"","sources":["../../src/commands/extractJson.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,iBAAiB,CAAC;AAEhC,MAAM,UAAU,0BAA0B,CAAC,OAAgB;IACzD,OAAO;SACJ,OAAO,CAAC,qBAAqB,CAAC;SAC9B,WAAW,CACV,mEAAmE,CACpE;SACA,cAAc,CACb,qBAAqB,EACrB,oGAAoG,CACrG;SACA,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,OAA2B,EAAE,EAAE;QAC1D,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,0BAA0B,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE;wBACR,MAAM;wBACN,OAAO,EAAE,MAAM;qBAChB;iBACF,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAChC,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;YAClE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/commands/extractPii.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerExtractPiiCommand(program: Command): void;
+//# sourceMappingURL=extractPii.d.ts.map

package/dist/commands/extractPii.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractPii.d.ts","sourceRoot":"","sources":["../../src/commands/extractPii.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAWpC,wBAAgB,yBAAyB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CA2FhE"}

package/dist/commands/extractPii.js

@@ -0,0 +1,77 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "gliner-multi-pii-v1";
+const DEFAULT_THRESHOLD = 0.5;
+const DEFAULT_CATEGORIES = ["identity", "contact"];
+export function registerExtractPiiCommand(program) {
+    program
+        .command("extract_pii <text>")
+        .description("Extract PII entities from text (persons, emails, phone numbers, etc.).")
+        .option("-t, --threshold <number>", "Confidence threshold for entity extraction.", String(DEFAULT_THRESHOLD))
+        .option("-c, --categories <list>", "Comma-separated list of PII categories to extract.", DEFAULT_CATEGORIES.join(","))
+        .action(async (text, opts) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        const threshold = Number(opts.threshold);
+        if (!Number.isFinite(threshold)) {
+            console.error(`Invalid threshold: ${opts.threshold}`);
+            process.exit(1);
+        }
+        const categories = opts.categories
+            .split(",")
+            .map((c) => c.trim())
+            .filter((c) => c.length > 0);
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                    metadata: {
+                        usecase: "extract-pii",
+                        threshold,
+                        categories,
+                    },
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")
+            ?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any extraction content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=extractPii.js.map

package/dist/commands/extractPii.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractPii.js","sourceRoot":"","sources":["../../src/commands/extractPii.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,qBAAqB,CAAC;AACpC,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAC9B,MAAM,kBAAkB,GAAG,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;AAEnD,MAAM,UAAU,yBAAyB,CAAC,OAAgB;IACxD,OAAO;SACJ,OAAO,CAAC,oBAAoB,CAAC;SAC7B,WAAW,CACV,wEAAwE,CACzE;SACA,MAAM,CACL,0BAA0B,EAC1B,6CAA6C,EAC7C,MAAM,CAAC,iBAAiB,CAAC,CAC1B;SACA,MAAM,CACL,yBAAyB,EACzB,oDAAoD,EACpD,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,CAC7B;SACA,MAAM,CACL,KAAK,EACH,IAAY,EACZ,IAA+C,EAC/C,EAAE;QACF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,KAAK,CAAC,sBAAsB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACtD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU;aAC/B,KAAK,CAAC,GAAG,CAAC;aACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;aACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAE/B,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE;wBACR,OAAO,EAAE,aAAa;wBACtB,SAAS;wBACT,UAAU;qBACX;iBACF,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GACX,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC;YAC9D,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QACnD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;YAClE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CACF,CAAC;AACN,CAAC"}

package/dist/commands/generateFollowups.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerGenerateFollowupsCommand(program: Command): void;
+//# sourceMappingURL=generateFollowups.d.ts.map

package/dist/commands/generateFollowups.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generateFollowups.d.ts","sourceRoot":"","sources":["../../src/commands/generateFollowups.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AASpC,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CA6DvE"}

package/dist/commands/generateFollowups.js

@@ -0,0 +1,58 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "zlm-v1-followup-questions-edge";
+export function registerGenerateFollowupsCommand(program) {
+    program
+        .command("generate_followups <text>")
+        .description("Generate contextual follow-up questions with the ZeroGPU followup edge model.")
+        .action(async (text) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any follow-up content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=generateFollowups.js.map

package/dist/commands/generateFollowups.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generateFollowups.js","sourceRoot":"","sources":["../../src/commands/generateFollowups.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,gCAAgC,CAAC;AAE/C,MAAM,UAAU,gCAAgC,CAAC,OAAgB;IAC/D,OAAO;SACJ,OAAO,CAAC,2BAA2B,CAAC;SACpC,WAAW,CACV,+EAA+E,CAChF;SACA,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,EAAE;QAC7B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;iBACZ,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAChC,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/commands/redactPii.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerRedactPiiCommand(program: Command): void;
+//# sourceMappingURL=redactPii.d.ts.map

package/dist/commands/redactPii.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"redactPii.d.ts","sourceRoot":"","sources":["../../src/commands/redactPii.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AASpC,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAiE/D"}

package/dist/commands/redactPii.js

@@ -0,0 +1,62 @@
+import { getApiKey, getProjectId } from "../lib/auth.js";
+import { RESPONSES_ENDPOINT, } from "../lib/responses.js";
+const MODEL = "gliner-multi-pii-v1";
+export function registerRedactPiiCommand(program) {
+    program
+        .command("redact_pii <text>")
+        .description("Detect and redact PII entities in text (persons, phone numbers, emails, etc.).")
+        .action(async (text) => {
+        const apiKey = getApiKey();
+        const projectId = getProjectId();
+        if (!apiKey || !projectId) {
+            console.error("You're not fully signed in yet. Run 'zerogpu login' to set your API key and project ID.");
+            process.exit(1);
+        }
+        let response;
+        try {
+            response = await fetch(RESPONSES_ENDPOINT, {
+                method: "POST",
+                headers: {
+                    "content-type": "application/json",
+                    "x-api-key": apiKey.apiKey,
+                    "x-project-id": projectId.projectId,
+                },
+                body: JSON.stringify({
+                    model: MODEL,
+                    input: text,
+                    metadata: {
+                        mask: "label",
+                        usecase: "redact",
+                    },
+                }),
+            });
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.error(`Request failed: ${message}`);
+            process.exit(1);
+        }
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`Request failed with status ${response.status}.`);
+            if (body)
+                console.error(body);
+            process.exit(1);
+        }
+        const data = (await response.json());
+        const content = data.output?.[0]?.content?.find((c) => c.type === "output_text")?.text ?? data.output?.[0]?.content?.[0]?.text;
+        if (!content) {
+            console.error("Response did not contain any redaction content.");
+            console.error(JSON.stringify(data, null, 2));
+            process.exit(1);
+        }
+        try {
+            const parsed = JSON.parse(content);
+            console.log(JSON.stringify(parsed, null, 2));
+        }
+        catch {
+            console.log(content);
+        }
+    });
+}
+//# sourceMappingURL=redactPii.js.map

package/dist/commands/redactPii.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"redactPii.js","sourceRoot":"","sources":["../../src/commands/redactPii.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,KAAK,GAAG,qBAAqB,CAAC;AAEpC,MAAM,UAAU,wBAAwB,CAAC,OAAgB;IACvD,OAAO;SACJ,OAAO,CAAC,mBAAmB,CAAC;SAC5B,WAAW,CACV,gFAAgF,CACjF;SACA,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,EAAE;QAC7B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QAEjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CACX,yFAAyF,CAC1F,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,cAAc,EAAE,SAAS,CAAC,SAAS;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE;wBACR,IAAI,EAAE,OAAO;wBACb,OAAO,EAAE,QAAQ;qBAClB;iBACF,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,IAAI;gBAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAChC,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/commands/summarize.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare function registerSummarizeCommand(program: Command): void;
+//# sourceMappingURL=summarize.d.ts.map

package/dist/commands/summarize.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"summarize.d.ts","sourceRoot":"","sources":["../../src/commands/summarize.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AASpC,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CA2D/D"}