langtrain 0.1.19 → 0.1.21

package/dist/index.d.mts

@@ -37,6 +37,9 @@ declare class AgentClient {
     create(agent: AgentCreate): Promise<Agent>;
     delete(agentId: string): Promise<void>;
     execute(agentId: string, input: any, messages?: any[], conversationId?: string): Promise<AgentRun>;
+    logs(agentId: string, limit?: number): Promise<{
+        logs: string[];
+    }>;
 }
 interface AgentConfig {
     system_prompt?: string;
@@ -183,4 +186,67 @@ declare namespace models {
   export { type models_Model as Model, models_ModelClient as ModelClient, type models_Permission as Permission };
 }
 
-export { type Agent, AgentClient, type AgentCreate, type AgentRun, agent as AgentTypes, type FeatureCheck, FileClient, type FileResponse, type FineTuneJobCreate, type FineTuneJobResponse, type Model, ModelClient, models as ModelTypes, SubscriptionClient, type SubscriptionInfo, TrainingClient };
+interface Secret {
+    key: string;
+    created_at: string;
+    updated_at: string;
+}
+declare class SecretClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    list(workspaceId?: string): Promise<Secret[]>;
+    set(key: string, value: string, workspaceId?: string): Promise<Secret>;
+    delete(key: string, workspaceId?: string): Promise<void>;
+}
+
+type secrets_Secret = Secret;
+type secrets_SecretClient = SecretClient;
+declare const secrets_SecretClient: typeof SecretClient;
+declare namespace secrets {
+  export { type secrets_Secret as Secret, secrets_SecretClient as SecretClient };
+}
+
+interface GuardrailConfig {
+    pii_enabled: boolean;
+    pii_entities?: string[];
+    profanity_enabled: boolean;
+    profanity_threshold?: number;
+    blocked_topics?: string[];
+    regex_patterns?: string[];
+    min_length?: number;
+    max_length?: number;
+}
+interface Guardrail {
+    id: string;
+    workspace_id: string;
+    name: string;
+    description?: string;
+    config: GuardrailConfig;
+    is_active: boolean;
+    created_at: string;
+    updated_at: string;
+}
+interface GuardrailCreate {
+    name: string;
+    description?: string;
+    config: GuardrailConfig;
+}
+declare class GuardrailClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    list(workspaceId?: string): Promise<Guardrail[]>;
+    get(guardrailId: string): Promise<Guardrail>;
+    create(data: GuardrailCreate): Promise<Guardrail>;
+    delete(guardrailId: string): Promise<void>;
+    apply(datasetId: string, guardrailId: string): Promise<any>;
+}
+
+export { type Agent, AgentClient, type AgentCreate, type AgentRun, agent as AgentTypes, type FeatureCheck, FileClient, type FileResponse, type FineTuneJobCreate, type FineTuneJobResponse, type Guardrail, GuardrailClient, type GuardrailConfig, type GuardrailCreate, type Model, ModelClient, models as ModelTypes, type Secret, SecretClient, secrets as SecretTypes, SubscriptionClient, type SubscriptionInfo, TrainingClient };

package/dist/index.d.ts

@@ -37,6 +37,9 @@ declare class AgentClient {
     create(agent: AgentCreate): Promise<Agent>;
     delete(agentId: string): Promise<void>;
     execute(agentId: string, input: any, messages?: any[], conversationId?: string): Promise<AgentRun>;
+    logs(agentId: string, limit?: number): Promise<{
+        logs: string[];
+    }>;
 }
 interface AgentConfig {
     system_prompt?: string;
@@ -183,4 +186,67 @@ declare namespace models {
   export { type models_Model as Model, models_ModelClient as ModelClient, type models_Permission as Permission };
 }
 
-export { type Agent, AgentClient, type AgentCreate, type AgentRun, agent as AgentTypes, type FeatureCheck, FileClient, type FileResponse, type FineTuneJobCreate, type FineTuneJobResponse, type Model, ModelClient, models as ModelTypes, SubscriptionClient, type SubscriptionInfo, TrainingClient };
+interface Secret {
+    key: string;
+    created_at: string;
+    updated_at: string;
+}
+declare class SecretClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    list(workspaceId?: string): Promise<Secret[]>;
+    set(key: string, value: string, workspaceId?: string): Promise<Secret>;
+    delete(key: string, workspaceId?: string): Promise<void>;
+}
+
+type secrets_Secret = Secret;
+type secrets_SecretClient = SecretClient;
+declare const secrets_SecretClient: typeof SecretClient;
+declare namespace secrets {
+  export { type secrets_Secret as Secret, secrets_SecretClient as SecretClient };
+}
+
+interface GuardrailConfig {
+    pii_enabled: boolean;
+    pii_entities?: string[];
+    profanity_enabled: boolean;
+    profanity_threshold?: number;
+    blocked_topics?: string[];
+    regex_patterns?: string[];
+    min_length?: number;
+    max_length?: number;
+}
+interface Guardrail {
+    id: string;
+    workspace_id: string;
+    name: string;
+    description?: string;
+    config: GuardrailConfig;
+    is_active: boolean;
+    created_at: string;
+    updated_at: string;
+}
+interface GuardrailCreate {
+    name: string;
+    description?: string;
+    config: GuardrailConfig;
+}
+declare class GuardrailClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    list(workspaceId?: string): Promise<Guardrail[]>;
+    get(guardrailId: string): Promise<Guardrail>;
+    create(data: GuardrailCreate): Promise<Guardrail>;
+    delete(guardrailId: string): Promise<void>;
+    apply(datasetId: string, guardrailId: string): Promise<any>;
+}
+
+export { type Agent, AgentClient, type AgentCreate, type AgentRun, agent as AgentTypes, type FeatureCheck, FileClient, type FileResponse, type FineTuneJobCreate, type FineTuneJobResponse, type Guardrail, GuardrailClient, type GuardrailConfig, type GuardrailCreate, type Model, ModelClient, models as ModelTypes, type Secret, SecretClient, secrets as SecretTypes, SubscriptionClient, type SubscriptionInfo, TrainingClient };

package/dist/index.js

@@ -1,2 +1,2 @@
-'use strict';var chunkPAHGEWDE_js=require('./chunk-PAHGEWDE.js');Object.defineProperty(exports,"AgentClient",{enumerable:true,get:function(){return chunkPAHGEWDE_js.e}});Object.defineProperty(exports,"AgentTypes",{enumerable:true,get:function(){return chunkPAHGEWDE_js.f}});Object.defineProperty(exports,"FileClient",{enumerable:true,get:function(){return chunkPAHGEWDE_js.g}});Object.defineProperty(exports,"Langtune",{enumerable:true,get:function(){return chunkPAHGEWDE_js.c}});Object.defineProperty(exports,"Langvision",{enumerable:true,get:function(){return chunkPAHGEWDE_js.a}});Object.defineProperty(exports,"ModelClient",{enumerable:true,get:function(){return chunkPAHGEWDE_js.j}});Object.defineProperty(exports,"ModelTypes",{enumerable:true,get:function(){return chunkPAHGEWDE_js.k}});Object.defineProperty(exports,"SubscriptionClient",{enumerable:true,get:function(){return chunkPAHGEWDE_js.i}});Object.defineProperty(exports,"Text",{enumerable:true,get:function(){return chunkPAHGEWDE_js.d}});Object.defineProperty(exports,"TrainingClient",{enumerable:true,get:function(){return chunkPAHGEWDE_js.h}});Object.defineProperty(exports,"Vision",{enumerable:true,get:function(){return chunkPAHGEWDE_js.b}});//# sourceMappingURL=index.js.map
+'use strict';var chunkQZ6U7AJN_js=require('./chunk-QZ6U7AJN.js');chunkQZ6U7AJN_js.r();Object.defineProperty(exports,"AgentClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.h}});Object.defineProperty(exports,"AgentTypes",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.i}});Object.defineProperty(exports,"FileClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.j}});Object.defineProperty(exports,"GuardrailClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.q}});Object.defineProperty(exports,"Langtune",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.f}});Object.defineProperty(exports,"Langvision",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.d}});Object.defineProperty(exports,"ModelClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.m}});Object.defineProperty(exports,"ModelTypes",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.n}});Object.defineProperty(exports,"SecretClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.o}});Object.defineProperty(exports,"SecretTypes",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.p}});Object.defineProperty(exports,"SubscriptionClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.l}});Object.defineProperty(exports,"Text",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.g}});Object.defineProperty(exports,"TrainingClient",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.k}});Object.defineProperty(exports,"Vision",{enumerable:true,get:function(){return chunkQZ6U7AJN_js.e}});//# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/index.mjs

@@ -1,2 +1,2 @@
-export{e as AgentClient,f as AgentTypes,g as FileClient,c as Langtune,a as Langvision,j as ModelClient,k as ModelTypes,i as SubscriptionClient,d as Text,h as TrainingClient,b as Vision}from'./chunk-Q46V6ODQ.mjs';//# sourceMappingURL=index.mjs.map
+import {r}from'./chunk-TDQXC2RA.mjs';export{h as AgentClient,i as AgentTypes,j as FileClient,q as GuardrailClient,f as Langtune,d as Langvision,m as ModelClient,n as ModelTypes,o as SecretClient,p as SecretTypes,l as SubscriptionClient,g as Text,k as TrainingClient,e as Vision}from'./chunk-TDQXC2RA.mjs';r();//# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langtrain",
-  "version": "0.1.19",
+  "version": "0.1.21",
   "description": "Unified JavaScript SDK for Langtrain Ecosystem",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -41,6 +41,7 @@
   "dependencies": {
     "@clack/prompts": "^1.0.1",
     "axios": "^1.13.5",
+    "cli-table3": "^0.6.5",
     "commander": "^14.0.3",
     "gradient-string": "^3.0.0",
     "kleur": "^4.1.5",

package/src/cli/auth.ts

@@ -1,4 +1,4 @@
-import { password, isCancel, cancel, intro, green, yellow, red, bgCyan, black, spinner, gray } from './ui';
+import { password, isCancel, cancel, intro, green, yellow, red, bgMagenta, black, spinner, gray } from './ui';
 import { getConfig, saveConfig } from './config';
 import { SubscriptionClient, SubscriptionInfo } from '../index';
 
@@ -58,7 +58,7 @@ export async function getSubscription(apiKey: string): Promise<SubscriptionInfo
         const info = await client.getStatus();
 
         // Enhance: Show plan details immediately on auth check
-        const planLabel = info.plan === 'pro' ? bgCyan(' PRO ') : info.plan.toUpperCase();
+        const planLabel = info.plan === 'pro' ? bgMagenta(' PRO ') : info.plan.toUpperCase();
         s.stop(green(`Authenticated as ${planLabel}`));
 
         if (info.is_active === false) {

package/src/cli/components/Table.ts

@@ -0,0 +1,20 @@
+import Table from 'cli-table3';
+import { colors } from '../ui';
+
+export function createTable(headers: string[]) {
+    return new Table({
+        head: headers.map(h => colors.magenta(colors.bold(h))),
+        chars: {
+            'top': '─', 'top-mid': '┬', 'top-left': '┌', 'top-right': '┐',
+            'bottom': '─', 'bottom-mid': '┴', 'bottom-left': '└', 'bottom-right': '┘',
+            'left': '│', 'left-mid': '├', 'mid': '─', 'mid-mid': '┼',
+            'right': '│', 'right-mid': '┤', 'middle': '│'
+        },
+        style: {
+            'padding-left': 1,
+            'padding-right': 1,
+            head: [], // handle colors manually
+            border: ['gray']
+        }
+    });
+}

package/src/cli/handlers/agent.ts

@@ -1,4 +1,4 @@
-import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, gray, bgCyan, black, gradient } from '../ui';
+import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, gray, bgMagenta, black, gradient, createTable } from '../ui';
 import { AgentClient, ModelClient } from '../../index';
 
 export async function handleAgentCreate(client: AgentClient, modelClient: ModelClient) {
@@ -136,6 +136,19 @@ export async function handleAgentList(client: AgentClient) {
         return;
     }
 
+    // Display Table
+    const table = createTable(['ID', 'Name', 'Model', 'Created']);
+    agents.forEach(a => {
+        table.push([
+            a.id.substring(0, 8) + '...',
+            a.name,
+            (a.config as any)?.model || 'default',
+            new Date(a.created_at).toLocaleDateString()
+        ]);
+    });
+    console.log(table.toString());
+    console.log(''); // spacer
+
     const agentId = await select({
         message: 'Select an agent to run:',
         options: agents.map(a => ({ value: a.id, label: a.name, hint: a.description || 'No description' }))
@@ -146,12 +159,31 @@ export async function handleAgentList(client: AgentClient) {
     await handleAgentRun(client, agentId as string, agents.find(a => a.id === agentId)?.name || 'Agent');
 }
 
-export async function handleAgentRun(client: AgentClient, agentId: string, agentName: string) {
-    intro(bgCyan(black(` Chatting with ${agentName} `)));
+export async function handleAgentRun(client: AgentClient, agentId: string, agentName: string, initialMessage?: string) {
+    intro(bgMagenta(black(` Chatting with ${agentName} `)));
     console.log(gray('Type "exit" to quit conversation.'));
 
     let conversationId: string | undefined = undefined;
 
+    // Send initial message if provided
+    if (initialMessage) {
+        const s = spinner();
+        s.start('Agent is analyzing...');
+        try {
+            const result = await client.execute(agentId, { prompt: initialMessage }, [], conversationId);
+            s.stop();
+            if ((result.output as any)?.response) {
+                console.log(gradient.pastel(`Agent: ${(result.output as any).response}`));
+            } else {
+                console.log(gradient.pastel(`Agent: ${JSON.stringify(result.output)}`));
+            }
+            conversationId = result.conversation_id;
+        } catch (e: any) {
+            s.stop(red('Error running agent.'));
+            console.error(e);
+        }
+    }
+
     while (true) {
         const input = await text({
             message: 'You:',

package/src/cli/handlers/data.ts

@@ -1,7 +1,7 @@
 import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, gray } from '../ui';
 import { getConfig } from '../config';
-import { FileClient } from '../../index';
-import path from 'path';
+import { FileClient, AgentClient, GuardrailClient } from '../../index';
+import { handleAgentRun } from './agent';
 import fs from 'fs';
 
 export async function handleDataUpload(client: FileClient) {
@@ -73,12 +73,135 @@ export async function handleDataList(client: FileClient) {
             return;
         }
 
-        // Just list them for now, maybe select to delete later?
-        files.forEach(f => {
-            console.log(`${f.id.padEnd(30)} ${f.filename.padEnd(20)} ${f.purpose} (${f.bytes}b)`);
+        const file = await select({
+            message: 'Select file to analyze (or cancel to exit):',
+            options: files.map(f => ({ value: f.id, label: `${f.filename} (${formatBytes(f.bytes)})` }))
         });
 
+        if (isCancel(file)) return;
+
+        await handleDataAnalyze(client, file as string);
+
     } catch (e: any) {
         s.stop(red(`Failed to list files: ${e.message}`));
     }
 }
+
+export async function handleDataAnalyze(client: FileClient, fileId: string) {
+    const config = getConfig();
+    const s2 = spinner();
+    s2.start('Connecting to Data Analyst...');
+
+    try {
+        const agentClient = new AgentClient({ apiKey: config.apiKey || '', baseUrl: config.baseUrl });
+        const agents = await agentClient.list();
+        // Check name (name is optional in type but we know it exists for system agent)
+        let analyst = agents.find(a => a.name && a.name === "Langtrain Data Analyst");
+
+        if (!analyst) {
+            s2.stop(yellow('Data Analyst agent (System) not found. Please contact admin to provision it.'));
+            return;
+        }
+
+        s2.stop(green('Connected to Data Analyst.'));
+
+        console.log(gray(`\nAnalyzing dataset ${fileId}...\n`));
+
+        await handleAgentRun(agentClient, analyst.id, analyst.name, `Please analyze the dataset with ID: ${fileId}`);
+
+    } catch (e: any) {
+        s2.stop(red(`Failed to connect: ${e.message}`));
+    }
+}
+
+export async function handleDataRefine(client: FileClient, fileId?: string) {
+    const config = getConfig();
+    const gClient = new GuardrailClient({ apiKey: config.apiKey || '', baseUrl: config.baseUrl });
+
+    // 0. Select File if not provided
+    if (!fileId) {
+        const s = spinner();
+        s.start('Fetching files...');
+        try {
+            // Need workspace ID logic similar to List?
+            // client.list takes workspaceId.
+            const wId = config.workspace_id || '';
+            const files = await client.list(wId);
+            s.stop(`Found ${files.length} files`);
+
+            if (files.length === 0) {
+                console.log(yellow('No files found. Upload one first.'));
+                return;
+            }
+
+            const selection = await select({
+                message: 'Select file to refine:',
+                options: files.map(f => ({ value: f.id, label: `${f.filename} (${formatBytes(f.bytes)})` }))
+            });
+
+            if (isCancel(selection)) return;
+            fileId = selection as string;
+
+        } catch (e: any) {
+            s.stop(red(`Failed to fetch files: ${e.message}`));
+            return;
+        }
+    }
+
+    // 1. Select Guardrail
+    const s = spinner();
+    s.start('Fetching guardrails...');
+    let guardId = "";
+    try {
+        const guards = await gClient.list();
+        s.stop(`Found ${guards.length} guardrails`);
+
+        if (guards.length === 0) {
+            console.log(yellow('No guardrails found. Please create one first using "lt guardrails create".'));
+            return;
+        }
+
+        const selection = await select({
+            message: 'Select a Guardrail to apply:',
+            options: guards.map((g: any) => ({
+                value: g.id,
+                label: g.name,
+                hint: g.description
+            }))
+        });
+
+        if (isCancel(selection)) return;
+        guardId = selection as string;
+
+    } catch (e: any) {
+        s.stop(red(`Failed to fetch guardrails: ${e.message}`));
+        return;
+    }
+
+    // 2. Apply
+    const s2 = spinner();
+    s2.start('Applying guardrail (filtering dataset)...');
+
+    try {
+        const result = await gClient.apply(fileId, guardId);
+        s2.stop(green('Dataset refined successfully!'));
+
+        console.log(gray('Stats:'));
+        console.log(`Original Rows: ${result.original_rows}`);
+        console.log(`Filtered Rows: ${result.filtered_rows}`);
+        console.log(red(`Removed: ${result.removed_rows} rows`));
+        console.log(green(`New Dataset ID: ${result.new_dataset_id}`));
+
+    } catch (e: any) {
+        s2.stop(red(`Failed to refine dataset: ${e.message}`));
+    }
+}
+
+function formatBytes(bytes: number, decimals = 2) {
+    if (!+bytes) return '0 Bytes';
+    const k = 1024;
+    const dm = decimals < 0 ? 0 : decimals;
+    const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB'];
+    const i = Math.floor(Math.log(bytes) / Math.log(k));
+    return `${parseFloat((bytes / Math.pow(k, i)).toFixed(dm))} ${sizes[i]}`;
+}

package/src/cli/handlers/env.ts

@@ -0,0 +1,89 @@
+import { intro, outro, spinner, green, red, yellow, showInfo, gray, showSuccess } from '../ui';
+import { SecretClient } from '../../index';
+import { text, confirm, select, isCancel, cancel } from '../ui';
+import { getConfig } from '../config';
+
+export async function handleEnvList(client: SecretClient) {
+    const s = spinner();
+    s.start('Fetching secrets...');
+    const config = getConfig();
+    try {
+        const secrets = await client.list(config.workspace_id);
+        s.stop(`Found ${secrets.length} secrets`);
+
+        if (secrets.length === 0) {
+            console.log(gray('No secrets found. Use "lt env set" to add one.'));
+            return;
+        }
+
+        console.log(gray('------------------------------------------------'));
+        secrets.forEach(sec => {
+            console.log(`${sec.key.padEnd(30)} ${gray('******')}`);
+        });
+        console.log(gray('------------------------------------------------'));
+
+    } catch (e: any) {
+        s.stop(red(`Failed to list secrets: ${e.message}`));
+    }
+}
+
+export async function handleEnvSet(client: SecretClient, keyVal?: string) {
+    let key = '';
+    let value = '';
+
+    if (keyVal && keyVal.includes('=')) {
+        const parts = keyVal.split('=');
+        key = parts[0];
+        value = parts.slice(1).join('=');
+    } else {
+        key = await text({ message: 'Secret Key:', placeholder: 'OPENAI_API_KEY' }) as string;
+        if (isCancel(key)) return;
+
+        value = await text({ message: 'Secret Value:', placeholder: 'sk-...' }) as string;
+        if (isCancel(value)) return;
+    }
+
+    const s = spinner();
+    s.start(`Setting ${key}...`);
+    const config = getConfig();
+
+    try {
+        await client.set(key, value, config.workspace_id);
+        s.stop(green(`Secret ${key} set successfully.`));
+    } catch (e: any) {
+        s.stop(red(`Failed to set secret: ${e.message}`));
+    }
+}
+
+// Interactive menu for env
+export async function handleEnvMenu(client: SecretClient) {
+    const action = await select({
+        message: 'Manage Secrets',
+        options: [
+            { value: 'list', label: 'List Secrets' },
+            { value: 'set', label: 'Set Secret' },
+            { value: 'remove', label: 'Remove Secret' },
+            { value: 'back', label: 'Back' }
+        ]
+    });
+
+    if (isCancel(action) || action === 'back') return;
+
+    if (action === 'list') await handleEnvList(client);
+    if (action === 'set') await handleEnvSet(client);
+    if (action === 'remove') {
+        const key = await text({ message: 'Key to remove:' });
+        if (!isCancel(key)) {
+            const s = spinner();
+            s.start('Removing...');
+            try {
+                // need workspace_id
+                const config = getConfig();
+                await client.delete(key as string, config.workspace_id);
+                s.stop(green('Removed.'));
+            } catch (e: any) {
+                s.stop(red(`Failed: ${e.message}`));
+            }
+        }
+    }
+}

package/src/cli/handlers/guardrails.ts

@@ -0,0 +1,100 @@
+import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, gray } from '../ui';
+import { getConfig } from '../config';
+import { GuardrailClient } from '../../index';
+
+export async function handleGuardrailList(client: any) { // using any to match index signature, but we instantiate specific client inside
+    const config = getConfig();
+    const gClient = new GuardrailClient({ apiKey: config.apiKey || '', baseUrl: config.baseUrl });
+
+    const s = spinner();
+    s.start('Fetching guardrails...');
+
+    try {
+        const guards = await gClient.list();
+        s.stop(`Found ${guards.length} guardrails`);
+
+        if (guards.length === 0) {
+            console.log(yellow('No guardrails found. Create one with "lt guardrails create".'));
+            return;
+        }
+
+        guards.forEach((g: any) => {
+            console.log(green(`• ${g.name}`) + gray(` (ID: ${g.id})`));
+            if (g.description) console.log(gray(`  ${g.description}`));
+            console.log(gray(`  Config: PII=${g.config.pii_enabled}, MinLen=${g.config.min_length}`));
+            console.log('');
+        });
+
+    } catch (e: any) {
+        s.stop(red(`Failed to list guardrails: ${e.message}`));
+    }
+}
+
+export async function handleGuardrailCreate(client: any) {
+    const config = getConfig();
+    const gClient = new GuardrailClient({ apiKey: config.apiKey || '', baseUrl: config.baseUrl });
+
+    intro('Create a new Data Guardrail');
+
+    const name = await text({
+        message: 'Guardrail Name:',
+        placeholder: 'e.g. Strict Safety Policy',
+        validate(value) {
+            if (!value) return 'Name is required';
+        }
+    });
+    if (isCancel(name)) return;
+
+    const description = await text({
+        message: 'Description (optional):',
+        placeholder: 'Filters PII and short text',
+    });
+    if (isCancel(description)) return;
+
+    // Interactive Config
+    const minLen = await text({
+        message: 'Minimum Text Length (0 for no limit):',
+        initialValue: '0',
+        validate(value) {
+            if (isNaN(Number(value))) return 'Must be a number';
+        }
+    });
+    if (isCancel(minLen)) return;
+
+    const enablePii = await confirm({
+        message: 'Enable PII Filtering (Email/Phone)?',
+        initialValue: false
+    });
+    if (isCancel(enablePii)) return;
+
+    const patterns = await text({
+        message: 'Regex Patterns to Block (comma separated, optional):',
+        placeholder: 'e.g. bad_word, another_one',
+    });
+    if (isCancel(patterns)) return;
+
+    const s = spinner();
+    s.start('Creating guardrail...');
+
+    try {
+        const regexList = (patterns as string).split(',').map(p => p.trim()).filter(p => p.length > 0);
+
+        const payload = {
+            name,
+            description,
+            config: {
+                min_length: Number(minLen),
+                pii_enabled: enablePii,
+                regex_patterns: regexList,
+                profanity_enabled: false
+            }
+        };
+
+        const result = await gClient.create(payload);
+        s.stop(green(`Guardrail "${result.name}" created successfully!`));
+        console.log(gray(`ID: ${result.id}`));
+
+    } catch (e: any) {
+        s.stop(red(`Failed to create guardrail: ${e.message}`));
+    }
+}