langtrain 0.1.20 → 0.1.21

package/dist/index.d.mts

@@ -210,4 +210,43 @@ declare namespace secrets {
   export { type secrets_Secret as Secret, secrets_SecretClient as SecretClient };
 }
 
-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, type Secret, SecretClient, secrets as SecretTypes, SubscriptionClient, type SubscriptionInfo, TrainingClient };
+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

@@ -210,4 +210,43 @@ declare namespace secrets {
   export { type secrets_Secret as Secret, secrets_SecretClient as SecretClient };
 }
 
-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, type Secret, SecretClient, secrets as SecretTypes, SubscriptionClient, type SubscriptionInfo, TrainingClient };
+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 chunkZ7FYIYKL_js=require('./chunk-Z7FYIYKL.js');chunkZ7FYIYKL_js.q();Object.defineProperty(exports,"AgentClient",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.h}});Object.defineProperty(exports,"AgentTypes",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.i}});Object.defineProperty(exports,"FileClient",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.j}});Object.defineProperty(exports,"Langtune",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.f}});Object.defineProperty(exports,"Langvision",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.d}});Object.defineProperty(exports,"ModelClient",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.m}});Object.defineProperty(exports,"ModelTypes",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.n}});Object.defineProperty(exports,"SecretClient",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.o}});Object.defineProperty(exports,"SecretTypes",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.p}});Object.defineProperty(exports,"SubscriptionClient",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.l}});Object.defineProperty(exports,"Text",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.g}});Object.defineProperty(exports,"TrainingClient",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.k}});Object.defineProperty(exports,"Vision",{enumerable:true,get:function(){return chunkZ7FYIYKL_js.e}});//# 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 @@
-import {q}from'./chunk-D3O435OM.mjs';export{h as AgentClient,i as AgentTypes,j as FileClient,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-D3O435OM.mjs';q();//# 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.20",
+  "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' }))
@@ -147,7 +160,7 @@ export async function handleAgentList(client: AgentClient) {
 }
 
 export async function handleAgentRun(client: AgentClient, agentId: string, agentName: string, initialMessage?: string) {
-    intro(bgCyan(black(` Chatting with ${agentName} `)));
+    intro(bgMagenta(black(` Chatting with ${agentName} `)));
     console.log(gray('Type "exit" to quit conversation.'));
 
     let conversationId: string | undefined = undefined;

package/src/cli/handlers/logs.ts

@@ -1,4 +1,4 @@
-import { intro, spinner, red, gray, yellow, bgCyan, black, green } from '../ui';
+import { intro, spinner, red, gray, yellow, bgMagenta, black, green } from '../ui';
 import { AgentClient } from '../../index';
 import { select, text, isCancel } from '../ui';
 
@@ -54,7 +54,7 @@ export async function handleLogs(client: AgentClient, agentName?: string) {
         s2.stop('Logs fetched.');
 
         console.log(gray('------------------------------------------------'));
-        console.log(`${bgCyan(black(' Recent Logs '))}`);
+        console.log(`${bgMagenta(black(' Recent Logs '))}`);
         if (logs.logs && logs.logs.length > 0) {
             logs.logs.forEach(log => console.log(log));
         } else {

package/src/cli/handlers/subscription.ts

@@ -1,4 +1,4 @@
-import { spinner, intro, red, green, gray, bgCyan } from '../ui';
+import { spinner, intro, red, green, gray, bgMagenta } from '../ui';
 import { getConfig } from '../config';
 import { SubscriptionClient } from '../../index';
 
@@ -15,7 +15,7 @@ export async function handleSubscriptionStatus() {
         const info = await client.getStatus();
         s.stop(green('Subscription Status:'));
 
-        console.log(gray('Plan: ') + (info.plan === 'pro' ? bgCyan(' PRO ') : info.plan.toUpperCase()));
+        console.log(gray('Plan: ') + (info.plan === 'pro' ? bgMagenta(' PRO ') : info.plan.toUpperCase()));
         console.log(gray('Active: ') + (info.is_active ? green('Yes') : red('No')));
         if (info.expires_at) console.log(gray('Expires: ') + new Date(info.expires_at).toLocaleDateString());
 

package/src/cli/handlers/tune.ts

@@ -1,4 +1,4 @@
-import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, bgCyan, black, gradient, gray } from '../ui';
+import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, bgMagenta, black, gradient, gray, createTable } from '../ui';
 import { getConfig } from '../config';
 import { Langtune, ModelClient, SubscriptionClient, FileClient, TrainingClient } from '../../index';
 
@@ -73,7 +73,7 @@ export async function handleTuneFinetune(tune: Langtune, modelClient: ModelClien
                 s.stop(red('Feature "cloud_finetuning" is not available on your plan.'));
                 const upgrade = await confirm({ message: 'Upgrade to Pro for cloud tracking?' });
                 if (upgrade && !isCancel(upgrade)) {
-                    console.log(bgCyan(black(' Visit https://langtrain.ai/dashboard/billing to upgrade. ')));
+                    console.log(bgMagenta(black(' Visit https://langtrain.ai/dashboard/billing to upgrade. ')));
                 }
                 return;
             }
@@ -158,11 +158,6 @@ export async function handleTuneList(trainingClient: TrainingClient) {
     const s = spinner();
     s.start('Fetching fine-tuning jobs...');
 
-    // We need workspace ID, usually from config or first agent?
-    // For now, let's just ask or list from all available if API supports it (it requires workspace_id)
-    // Let's assume user knows it or we can find it.
-    // Simplified: Just ask for Workspace ID if not in config (we don't save it yet)
-    // BETTER: Get it from an existing agent or config.
     const config = getConfig();
     let workspaceId = config.workspace_id;
 
@@ -181,6 +176,21 @@ export async function handleTuneList(trainingClient: TrainingClient) {
             return;
         }
 
+        // Display Table
+        const table = createTable(['ID', 'Status', 'Model', 'Progress', 'Created']);
+        jobs.data.forEach(j => {
+            const statusColor = j.status === 'succeeded' ? green : (j.status === 'failed' ? red : yellow);
+            table.push([
+                j.id.substring(0, 8) + '...',
+                statusColor(j.status),
+                j.base_model,
+                (j.progress || 0) + '%',
+                new Date(j.created_at).toLocaleDateString()
+            ]);
+        });
+        console.log(table.toString());
+        console.log('');
+
         const selectedJob = await select({
             message: 'Select a job to view details:',
             options: jobs.data.map(j => ({
@@ -214,7 +224,7 @@ export async function handleTuneStatus(trainingClient: TrainingClient, jobId?: s
         s.stop(`Job Status: ${job.status.toUpperCase()}`);
 
         console.log(gray('------------------------------------------------'));
-        console.log(`${bgCyan(black(' Job Details '))}`);
+        console.log(`${bgMagenta(black(' Job Details '))}`);
         console.log(`ID:        ${job.id}`);
         console.log(`Name:      ${job.name}`);
         console.log(`Status:    ${job.status === 'succeeded' ? green(job.status) : job.status}`);

package/src/cli/handlers/vision.ts

@@ -1,4 +1,4 @@
-import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, bgCyan, black, gradient } from '../ui';
+import { text, select, confirm, isCancel, cancel, spinner, intro, red, green, yellow, bgMagenta, black, gradient } from '../ui';
 import { getConfig } from '../config';
 import { Langvision, ModelClient, SubscriptionClient, FileClient, TrainingClient } from '../../index';
 
@@ -74,7 +74,7 @@ export async function handleVisionFinetune(vision: Langvision, modelClient: Mode
                 s.stop(red('Feature "cloud_finetuning" is not available on your plan.'));
                 const upgrade = await confirm({ message: 'Upgrade to Pro for cloud tracking?' });
                 if (upgrade && !isCancel(upgrade)) {
-                    console.log(bgCyan(black(' Visit https://langtrain.ai/dashboard/billing to upgrade. ')));
+                    console.log(bgMagenta(black(' Visit https://langtrain.ai/dashboard/billing to upgrade. ')));
                 }
                 return;
             }

package/src/cli/ui.ts

@@ -68,3 +68,5 @@ export const colors = {
     bgMagenta, black, red, green, yellow, gray, cyan, bold, dim, blue, magenta, white
 };
 
+export * from './components/Table';
+