@imbrace/cli 0.4.0 → 0.5.0

package/dist/commands/ai-agent/create.d.ts

@@ -9,6 +9,7 @@ export default class AiAgentCreate extends BaseCommand {
         model: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         "provider-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         mode: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "agent-type": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         temperature: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         personality: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         "core-task": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;

package/dist/commands/ai-agent/create.js

@@ -18,6 +18,10 @@ export default class AiAgentCreate extends BaseCommand {
         model: Flags.string({ description: "LLM model (e.g. gpt-4o, qwen3.5-27b). Default: gpt-4o" }),
         "provider-id": Flags.string({ description: "LLM provider ID. Default 'system'. Use 'imbrace ai-agent list-providers' to discover." }),
         mode: Flags.string({ description: "Agent mode (default: standard)", options: ["standard", "advanced"] }),
+        "agent-type": Flags.string({
+            description: "Agent type (default: agent). For Document AI, use `imbrace document-ai create` instead.",
+            options: ["agent", "assistant", "conversational", "workflow"],
+        }),
         temperature: Flags.string({ description: "Model temperature 0.0-2.0 (default: 0.1). Lower = deterministic, higher = creative." }),
         // Behavior Settings (UI tab)
         personality: Flags.string({ description: "Agent personality / role (e.g. 'You are a friendly support rep')" }),
@@ -57,6 +61,7 @@ export default class AiAgentCreate extends BaseCommand {
             ...(flags.model && { model: flags.model }),
             ...(flags["provider-id"] && { provider_id: flags["provider-id"] }),
             ...(flags.mode && { mode: flags.mode }),
+            ...(flags["agent-type"] && { agent_type: flags["agent-type"] }),
             ...(flags.temperature && { temperature: parseFloat(flags.temperature) }),
             ...(flags.personality && { personality_role: flags.personality }),
             ...(flags["core-task"] && { core_task: flags["core-task"] }),

package/dist/commands/document-ai/create.d.ts

@@ -0,0 +1,18 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiCreate extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        name: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        instructions: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        model: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "provider-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        schema: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "schema-file": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        description: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "workflow-name": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        "id-only": import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/create.js

@@ -0,0 +1,76 @@
+import { Flags } from "@oclif/core";
+import { readFileSync } from "node:fs";
+import { BaseCommand } from "../../base-command.js";
+import { input } from "@inquirer/prompts";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiCreate extends BaseCommand {
+    static description = "Create a Document AI agent (extracts structured JSON from PDFs/images)";
+    static examples = [
+        'imbrace document-ai create --name "Invoice Extractor" --instructions "Extract invoice fields. Dates as YYYY-MM-DD." --model gpt-4o --schema \'{"invoice_number":{"type":"string"},"total":{"type":"number"}}\' --json',
+        'imbrace document-ai create --name "Receipt" --instructions "Extract receipt." --model gpt-4o --schema-file ./schema.json --id-only',
+    ];
+    static flags = {
+        name: Flags.string({ char: "n", description: "Agent display name" }),
+        instructions: Flags.string({ char: "i", description: "System prompt governing extraction logic" }),
+        model: Flags.string({ description: "LLM model ID (e.g. gpt-4o, qwen3.5-27b)" }),
+        "provider-id": Flags.string({ description: "Provider UUID (use 'system' for org default)" }),
+        schema: Flags.string({ description: "JSON schema for fields (inline). Either this or --schema-file." }),
+        "schema-file": Flags.string({ description: "Path to JSON schema file" }),
+        description: Flags.string({ char: "d", description: "Agent description" }),
+        "workflow-name": Flags.string({ description: "Internal workflow name (default: 'document_extraction')" }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+        "id-only": Flags.boolean({ description: "Print only the new agent ID (pipe-friendly)" }),
+    };
+    async run() {
+        const { flags } = await this.parse(DocumentAiCreate);
+        const nonInteractive = flags.json || flags["id-only"];
+        const name = flags.name ?? (nonInteractive ? this.error("--name is required with --json or --id-only") : await input({ message: "Agent name:" }));
+        const instructions = flags.instructions ?? (nonInteractive ? this.error("--instructions is required with --json or --id-only") : await input({ message: "Instructions:" }));
+        const model = flags.model ?? (nonInteractive ? this.error("--model is required with --json or --id-only") : await input({ message: "Model (e.g. gpt-4o):" }));
+        let schema;
+        if (flags.schema) {
+            try {
+                schema = JSON.parse(flags.schema);
+            }
+            catch (e) {
+                this.error(`--schema is not valid JSON: ${e.message}`);
+            }
+        }
+        else if (flags["schema-file"]) {
+            try {
+                schema = JSON.parse(readFileSync(flags["schema-file"], "utf8"));
+            }
+            catch (e) {
+                this.error(`--schema-file is not valid JSON: ${e.message}`);
+            }
+        }
+        try {
+            const client = getClient();
+            const data = await client.documentAi.createAgent({
+                name,
+                instructions,
+                model_id: model,
+                ...(flags["provider-id"] && { provider_id: flags["provider-id"] }),
+                ...(schema && { schema }),
+                ...(flags.description && { description: flags.description }),
+                ...(flags["workflow-name"] && { workflow_name: flags["workflow-name"] }),
+            });
+            if (flags["id-only"]) {
+                this.log(data?._id ?? data?.id ?? "");
+                return;
+            }
+            const message = `Document AI Agent "${name}" created`;
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, message, data }, null, 2));
+                return;
+            }
+            this.log(`\n✅ ${message}`);
+            if (data?._id || data?.id)
+                this.log(`   ID: ${data._id || data.id}`);
+            this.log("");
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/document-ai/delete.d.ts

@@ -0,0 +1,13 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiDelete extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static args: {
+        id: import("@oclif/core/interfaces").Arg<string | undefined, Record<string, unknown>>;
+    };
+    static flags: {
+        yes: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/delete.js

@@ -0,0 +1,41 @@
+import { Args, Flags } from "@oclif/core";
+import { BaseCommand } from "../../base-command.js";
+import { confirm, input } from "@inquirer/prompts";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiDelete extends BaseCommand {
+    static description = "Delete a Document AI agent";
+    static examples = [
+        "imbrace document-ai delete <agentId> --yes --json",
+    ];
+    static args = {
+        id: Args.string({ description: "Document AI agent ID" }),
+    };
+    static flags = {
+        yes: Flags.boolean({ char: "y", description: "Skip confirmation" }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+    };
+    async run() {
+        const { args, flags } = await this.parse(DocumentAiDelete);
+        const id = args.id ?? (flags.json ? this.error("ID is required") : await input({ message: "Agent ID:" }));
+        if (!flags.yes && !flags.json) {
+            const ok = await confirm({ message: `Delete Document AI agent ${id}?`, default: false });
+            if (!ok) {
+                this.log("Cancelled.");
+                return;
+            }
+        }
+        try {
+            const client = getClient();
+            await client.documentAi.deleteAgent(id);
+            const message = "Document AI Agent deleted";
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, message }, null, 2));
+                return;
+            }
+            this.log(`\n✅ ${message}\n`);
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/document-ai/get.d.ts

@@ -0,0 +1,12 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiGet extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static args: {
+        id: import("@oclif/core/interfaces").Arg<string | undefined, Record<string, unknown>>;
+    };
+    static flags: {
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/get.js

@@ -0,0 +1,45 @@
+import { Args, Flags } from "@oclif/core";
+import { BaseCommand } from "../../base-command.js";
+import { input } from "@inquirer/prompts";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiGet extends BaseCommand {
+    static description = "Get details of a Document AI agent (including extraction schema)";
+    static examples = [
+        "imbrace document-ai get <agentId>",
+        "imbrace document-ai get <agentId> --json",
+    ];
+    static args = {
+        id: Args.string({ description: "Document AI agent ID" }),
+    };
+    static flags = {
+        json: Flags.boolean({ description: "Output as JSON" }),
+    };
+    async run() {
+        const { args, flags } = await this.parse(DocumentAiGet);
+        const id = args.id ?? (flags.json ? this.error("ID is required") : await input({ message: "Agent ID:" }));
+        try {
+            const client = getClient();
+            const data = await client.documentAi.getAgent(id);
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, data }, null, 2));
+                return;
+            }
+            const a = data || {};
+            this.log(`\n  ID:           ${a._id || a.id || ""}`);
+            this.log(`  Name:         ${a.name || ""}`);
+            this.log(`  Model:        ${a.model_id || ""}`);
+            this.log(`  Provider ID:  ${a.provider_id || ""}`);
+            this.log(`  Workflow:     ${a.workflow_name || ""}`);
+            if (a.data_schema) {
+                this.log(`  Schema fields: ${Object.keys(a.data_schema).join(", ")}`);
+            }
+            if (a.instructions) {
+                this.log(`  Instructions: ${String(a.instructions).slice(0, 80)}${a.instructions.length > 80 ? "..." : ""}`);
+            }
+            this.log("");
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/document-ai/list.d.ts

@@ -0,0 +1,11 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiList extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        search: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        all: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/list.js

@@ -0,0 +1,42 @@
+import { Flags } from "@oclif/core";
+import { BaseCommand } from "../../base-command.js";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiList extends BaseCommand {
+    static description = "List Document AI agents (filter by name)";
+    static examples = [
+        "imbrace document-ai list",
+        "imbrace document-ai list --search invoice --json",
+    ];
+    static flags = {
+        search: Flags.string({ char: "s", description: "Filter by name (case-insensitive substring)" }),
+        all: Flags.boolean({ description: "Include non-Document-AI agents (default: only document_ai type)" }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+    };
+    async run() {
+        const { flags } = await this.parse(DocumentAiList);
+        try {
+            const client = getClient();
+            const data = await client.documentAi.listAgents({
+                documentAiOnly: !flags.all,
+                ...(flags.search && { nameContains: flags.search }),
+            });
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, count: data.length, data }, null, 2));
+                return;
+            }
+            this.log(`\n  Found ${data.length} Document AI agent(s):\n`);
+            this.log("  ID                                    NAME                          MODEL");
+            this.log("  ────────────────────────────────────────────────────────────────────────────");
+            for (const a of data) {
+                const id = (a._id || a.id || "").padEnd(38);
+                const name = (a.name || "").padEnd(28);
+                const model = a.model_id || "";
+                this.log(`  ${id}  ${name}  ${model}`);
+            }
+            this.log("");
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/document-ai/process.d.ts

@@ -0,0 +1,21 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiProcess extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        "agent-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        url: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        "org-id": import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        model: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        instructions: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "board-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        language: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "additional-instructions": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "chunk-size": import("@oclif/core/interfaces").OptionFlag<number | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "max-concurrent": import("@oclif/core/interfaces").OptionFlag<number | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "max-retries": import("@oclif/core/interfaces").OptionFlag<number | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "enhanced-processing": import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/process.js

@@ -0,0 +1,58 @@
+import { Flags } from "@oclif/core";
+import { BaseCommand } from "../../base-command.js";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiProcess extends BaseCommand {
+    static description = "Run a Document AI agent on a PDF/image URL — returns the extracted JSON";
+    static examples = [
+        "imbrace document-ai process --agent-id <id> --url https://example.com/invoice.pdf --org-id org_xxx --json",
+        "imbrace document-ai process --url https://example.com/doc.pdf --org-id org_xxx --model gpt-4o --instructions 'Extract X' --json",
+    ];
+    static flags = {
+        "agent-id": Flags.string({ description: "Agent ID. If provided, uses agent's model + instructions (overridden by --model/--instructions)." }),
+        url: Flags.string({ description: "URL of the document to extract", required: true }),
+        "org-id": Flags.string({ description: "Organization ID (sent in body and header)", required: true }),
+        model: Flags.string({ description: "Override agent's model. Required if --agent-id not provided." }),
+        instructions: Flags.string({ char: "i", description: "Override agent's instructions" }),
+        "board-id": Flags.string({ description: "Board ID to write extracted records into" }),
+        language: Flags.string({ description: "Document language hint" }),
+        "additional-instructions": Flags.string({ description: "Extra per-call instructions appended to agent's prompt" }),
+        "chunk-size": Flags.integer({ description: "Chunk size for large documents" }),
+        "max-concurrent": Flags.integer({ description: "Max concurrent chunks" }),
+        "max-retries": Flags.integer({ description: "Max retries on failure" }),
+        "enhanced-processing": Flags.boolean({ description: "Use enhanced processing pipeline", allowNo: true }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+    };
+    async run() {
+        const { flags } = await this.parse(DocumentAiProcess);
+        if (!flags["agent-id"] && !flags.model) {
+            this.error("Either --agent-id or --model is required");
+        }
+        try {
+            const client = getClient();
+            const data = await client.documentAi.process({
+                url: flags.url,
+                organizationId: flags["org-id"],
+                ...(flags["agent-id"] && { agentId: flags["agent-id"] }),
+                ...(flags.model && { modelName: flags.model }),
+                ...(flags.instructions && { instructions: flags.instructions }),
+                ...(flags["board-id"] && { boardId: flags["board-id"] }),
+                ...(flags.language && { language: flags.language }),
+                ...(flags["additional-instructions"] && { additionalDocumentInstructions: flags["additional-instructions"] }),
+                ...(flags["chunk-size"] !== undefined && { chunkSize: flags["chunk-size"] }),
+                ...(flags["max-concurrent"] !== undefined && { maxConcurrent: flags["max-concurrent"] }),
+                ...(flags["max-retries"] !== undefined && { maxRetries: flags["max-retries"] }),
+                ...(flags["enhanced-processing"] !== undefined && { useEnhancedProcessing: flags["enhanced-processing"] }),
+            });
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, data }, null, 2));
+                return;
+            }
+            this.log(`\n✅ Document processed`);
+            this.log(JSON.stringify(data, null, 2));
+            this.log("");
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/document-ai/suggest-schema.d.ts

@@ -0,0 +1,12 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiSuggestSchema extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        url: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        "org-id": import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        model: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/suggest-schema.js

@@ -0,0 +1,37 @@
+import { Flags } from "@oclif/core";
+import { BaseCommand } from "../../base-command.js";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiSuggestSchema extends BaseCommand {
+    static description = "Ask the LLM to inspect a sample document and suggest an extraction schema";
+    static examples = [
+        "imbrace document-ai suggest-schema --url https://example.com/sample.pdf --org-id org_xxx --json",
+        "imbrace document-ai suggest-schema --url https://example.com/sample.pdf --org-id org_xxx --model gpt-4o --json",
+    ];
+    static flags = {
+        url: Flags.string({ description: "URL of the sample document", required: true }),
+        "org-id": Flags.string({ description: "Organization ID", required: true }),
+        model: Flags.string({ description: "LLM model to use for inspection" }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+    };
+    async run() {
+        const { flags } = await this.parse(DocumentAiSuggestSchema);
+        try {
+            const client = getClient();
+            const data = await client.documentAi.suggestSchema({
+                url: flags.url,
+                organizationId: flags["org-id"],
+                ...(flags.model && { modelName: flags.model }),
+            });
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, data }, null, 2));
+                return;
+            }
+            this.log(`\n✅ Schema suggested`);
+            this.log(JSON.stringify(data, null, 2));
+            this.log("");
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/document-ai/update.d.ts

@@ -0,0 +1,20 @@
+import { BaseCommand } from "../../base-command.js";
+export default class DocumentAiUpdate extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static args: {
+        id: import("@oclif/core/interfaces").Arg<string, Record<string, unknown>>;
+    };
+    static flags: {
+        name: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        instructions: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        model: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "provider-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        schema: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "schema-file": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        description: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "workflow-name": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/document-ai/update.js

@@ -0,0 +1,76 @@
+import { Args, Flags } from "@oclif/core";
+import { readFileSync } from "node:fs";
+import { BaseCommand } from "../../base-command.js";
+import { getClient } from "../../lib/client.js";
+export default class DocumentAiUpdate extends BaseCommand {
+    static description = "Update a Document AI agent (partial — pass any subset of flags)";
+    static examples = [
+        'imbrace document-ai update <agentId> --instructions "Updated rules" --json',
+        'imbrace document-ai update <agentId> --schema-file ./new-schema.json',
+    ];
+    static args = {
+        id: Args.string({ description: "Document AI agent ID", required: true }),
+    };
+    static flags = {
+        name: Flags.string({ char: "n", description: "New display name" }),
+        instructions: Flags.string({ char: "i", description: "New system prompt" }),
+        model: Flags.string({ description: "New LLM model ID" }),
+        "provider-id": Flags.string({ description: "New provider UUID" }),
+        schema: Flags.string({ description: "New JSON schema (inline)" }),
+        "schema-file": Flags.string({ description: "Path to new JSON schema file" }),
+        description: Flags.string({ char: "d", description: "New description" }),
+        "workflow-name": Flags.string({ description: "New workflow name" }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+    };
+    async run() {
+        const { args, flags } = await this.parse(DocumentAiUpdate);
+        let schema;
+        if (flags.schema) {
+            try {
+                schema = JSON.parse(flags.schema);
+            }
+            catch (e) {
+                this.error(`--schema is not valid JSON: ${e.message}`);
+            }
+        }
+        else if (flags["schema-file"]) {
+            try {
+                schema = JSON.parse(readFileSync(flags["schema-file"], "utf8"));
+            }
+            catch (e) {
+                this.error(`--schema-file is not valid JSON: ${e.message}`);
+            }
+        }
+        const body = {};
+        if (flags.name)
+            body.name = flags.name;
+        if (flags.instructions)
+            body.instructions = flags.instructions;
+        if (flags.model)
+            body.model_id = flags.model;
+        if (flags["provider-id"])
+            body.provider_id = flags["provider-id"];
+        if (flags.description)
+            body.description = flags.description;
+        if (flags["workflow-name"])
+            body.workflow_name = flags["workflow-name"];
+        if (schema)
+            body.schema = schema;
+        if (Object.keys(body).length === 0) {
+            this.error("Provide at least one field to update. See: imbrace document-ai update -h");
+        }
+        try {
+            const client = getClient();
+            const data = await client.documentAi.updateAgent(args.id, body);
+            const message = "Document AI Agent updated";
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, message, data }, null, 2));
+                return;
+            }
+            this.log(`\n✅ ${message}\n`);
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/guardrail/create.d.ts

@@ -0,0 +1,19 @@
+import { BaseCommand } from "../../base-command.js";
+export default class GuardrailCreate extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        name: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        model: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        instructions: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "guardrail-provider-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "org-id": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        description: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "unsafe-categories": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "custom-unsafe-patterns": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        "competitor-keywords": import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        "id-only": import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}

package/dist/commands/guardrail/create.js

@@ -0,0 +1,69 @@
+import { Flags } from "@oclif/core";
+import { BaseCommand } from "../../base-command.js";
+import { input } from "@inquirer/prompts";
+import { getClient } from "../../lib/client.js";
+export default class GuardrailCreate extends BaseCommand {
+    static description = "Create a Guardrail (content safety + compliance rules attached to AI agents)";
+    static examples = [
+        'imbrace guardrail create --name "PII Filter" --instructions "Block any PII (SSN, credit card, email)" --json',
+        'imbrace guardrail create --name "Brand Safety" --model model-armor --instructions "Refuse competitor mentions" --competitor-keywords "Competitor1,Competitor2" --json',
+    ];
+    static flags = {
+        name: Flags.string({ char: "n", description: "Guardrail name" }),
+        model: Flags.string({
+            description: "Guardrail model. Default: nim-nemo (NVIDIA NIM). Other supported: 'model-armor' (Google) or a custom guardrail-provider model.",
+        }),
+        instructions: Flags.string({ char: "i", description: "Safety / compliance rules to enforce" }),
+        "guardrail-provider-id": Flags.string({ description: "Guardrail provider UUID (only when using a custom guardrail provider)" }),
+        "org-id": Flags.string({ description: "Organization ID. Auto-fetched from account if omitted." }),
+        description: Flags.string({ char: "d", description: "Human-readable description" }),
+        "unsafe-categories": Flags.string({ description: "Comma-separated unsafe categories (e.g. 'violence,hate,sexual')" }),
+        "custom-unsafe-patterns": Flags.string({ description: "Comma-separated custom regex patterns to block (nim-nemo only)" }),
+        "competitor-keywords": Flags.string({ description: "Comma-separated competitor names to flag (nim-nemo only)" }),
+        json: Flags.boolean({ description: "Output as JSON" }),
+        "id-only": Flags.boolean({ description: "Print only the new guardrail ID" }),
+    };
+    async run() {
+        const { flags } = await this.parse(GuardrailCreate);
+        const nonInteractive = flags.json || flags["id-only"];
+        const name = flags.name ?? (nonInteractive ? this.error("--name is required with --json or --id-only") : await input({ message: "Guardrail name:" }));
+        const instructions = flags.instructions ?? (nonInteractive ? this.error("--instructions is required with --json or --id-only") : await input({ message: "Instructions:" }));
+        const model = flags.model ?? "nim-nemo";
+        const split = (s) => s.split(",").map((x) => x.trim()).filter(Boolean);
+        try {
+            const client = getClient();
+            const orgId = flags["org-id"] || (await client.account.getAccount()).organization_id;
+            const isModelArmor = model === "model-armor";
+            const data = await client.ai.createGuardrail({
+                name,
+                model,
+                instructions: isModelArmor ? "" : instructions,
+                org_id: orgId,
+                ...(flags.description && { description: flags.description }),
+                ...(flags["guardrail-provider-id"] && { guardrail_provider_id: flags["guardrail-provider-id"] }),
+                ...(flags["unsafe-categories"] && { unsafe_categories: split(flags["unsafe-categories"]) }),
+                // Patterns + competitor lists only valid for nim-nemo.
+                ...(!isModelArmor && flags["custom-unsafe-patterns"] && { custom_unsafe_patterns: split(flags["custom-unsafe-patterns"]) }),
+                ...(!isModelArmor && flags["competitor-keywords"] && { competitor_keywords: split(flags["competitor-keywords"]) }),
+            });
+            // Backend returns `guardrails_config_id`, not `_id`.
+            const id = data?.guardrails_config_id ?? data?._id ?? "";
+            if (flags["id-only"]) {
+                this.log(id);
+                return;
+            }
+            const message = `Guardrail "${name}" created`;
+            if (flags.json) {
+                this.log(JSON.stringify({ ok: true, message, data: { ...data, _id: id } }, null, 2));
+                return;
+            }
+            this.log(`\n✅ ${message}`);
+            if (id)
+                this.log(`   ID: ${id}`);
+            this.log(`\n  Attach via 'imbrace ai-agent create --guardrail-id ${id || "<id>"}'.\n`);
+        }
+        catch (error) {
+            this.error(`Failed: ${error.message}`);
+        }
+    }
+}

package/dist/commands/guardrail/delete.d.ts

@@ -0,0 +1,13 @@
+import { BaseCommand } from "../../base-command.js";
+export default class GuardrailDelete extends BaseCommand {
+    static description: string;
+    static examples: string[];
+    static args: {
+        id: import("@oclif/core/interfaces").Arg<string | undefined, Record<string, unknown>>;
+    };
+    static flags: {
+        yes: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        json: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+}