langtrain 0.1.19 → 0.1.20

package/src/cli/ui.ts

@@ -1,5 +1,5 @@
 import { text, select, confirm, password, isCancel, cancel, note } from '@clack/prompts';
-import { bgCyan, black, red, green, yellow, gray, cyan, bold, dim, blue } from 'kleur/colors';
+import { bgMagenta, black, red, green, yellow, gray, cyan, bold, dim, blue, magenta, white } from 'kleur/colors';
 import gradient from 'gradient-string';
 
 // Re-export specific prompts to keep imports clean in other files
@@ -15,12 +15,15 @@ export function showBanner(version: string) {
     ███████╗██║  ██║██║ ╚████║╚██████╔╝  ██║   ██║  ██║██║  ██║██║██║ ╚████║
     ╚══════╝╚═╝  ╚═╝╚═╝  ╚═══╝ ╚═════╝   ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝╚═╝  ╚═══╝
     `;
-    console.log(gradient(['#00DC82', '#36E4DA', '#0047E1'])(banner));
-    console.log(`${bgCyan(black(` Langtrain SDK v${version} `))}\n`);
+    // Brand Gradient: Purple to Pink to Blue (Light Luxury)
+    console.log(gradient(['#A855F7', '#EC4899', '#3B82F6'])(banner));
+
+    // Elegant Badge: Black text on Magenta background
+    console.log(`${bgMagenta(black(` Langtrain SDK v${version} `))}\n`);
 }
 
 export function intro(message: string) {
-    console.log(cyan(`◆ ${message}`));
+    console.log(magenta(`◆ ${message}`));
 }
 
 export function outro(message: string) {
@@ -29,12 +32,12 @@ export function outro(message: string) {
 
 export function spinner() {
     return {
-        start: (msg: string) => process.stdout.write(`${cyan('●')} ${msg}\r`),
+        start: (msg: string) => process.stdout.write(`${magenta('●')} ${msg}\r`),
         stop: (msg?: string) => {
             if (msg) console.log(`${green('✔')} ${msg}`);
             else console.log(''); // Newline
         },
-        message: (msg: string) => process.stdout.write(`${cyan('●')} ${msg}\r`)
+        message: (msg: string) => process.stdout.write(`${magenta('●')} ${msg}\r`)
     };
 }
 
@@ -59,9 +62,9 @@ export function showDim(message: string) {
 }
 
 // Re-export for backward compatibility
-export { bgCyan, black, red, green, yellow, gray, cyan, bold, dim, blue, gradient };
+export { bgMagenta, black, red, green, yellow, gray, cyan, bold, dim, blue, gradient, magenta, white };
 
 export const colors = {
-    bgCyan, black, red, green, yellow, gray, cyan, bold, dim, blue
+    bgMagenta, black, red, green, yellow, gray, cyan, bold, dim, blue, magenta, white
 };
 

package/src/index.ts

@@ -9,11 +9,14 @@ export { FileClient, FileResponse } from './lib/files';
 export { TrainingClient, FineTuneJobCreate, FineTuneJobResponse } from './lib/training';
 export { SubscriptionClient, SubscriptionInfo, FeatureCheck } from './lib/subscription';
 export { ModelClient, Model } from './lib/models';
+export { SecretClient, Secret } from './lib/secrets';
+export { GuardrailClient, Guardrail, GuardrailConfig, GuardrailCreate } from './lib/guardrails';
 
 // Export Types with Namespaces to avoid collisions
 import * as Vision from 'langvision';
 import * as Text from 'langtune';
 import * as AgentTypes from './lib/agent';
 import * as ModelTypes from './lib/models';
+import * as SecretTypes from './lib/secrets';
 
-export { Vision, Text, AgentTypes, ModelTypes };
+export { Vision, Text, AgentTypes, ModelTypes, SecretTypes };

package/src/lib/agent.ts

@@ -65,6 +65,13 @@ export class AgentClient {
         });
         return response.data;
     }
+
+    async logs(agentId: string, limit: number = 100): Promise<{ logs: string[] }> {
+        const response = await this.client.get<{ logs: string[] }>(`/agents/${agentId}/logs`, {
+            params: { limit }
+        });
+        return response.data;
+    }
 }
 
 export interface AgentConfig {

package/src/lib/guardrails.ts

@@ -0,0 +1,72 @@
+import axios, { AxiosInstance } from 'axios';
+
+export 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;
+}
+
+export interface Guardrail {
+    id: string;
+    workspace_id: string;
+    name: string;
+    description?: string;
+    config: GuardrailConfig;
+    is_active: boolean;
+    created_at: string;
+    updated_at: string;
+}
+
+export interface GuardrailCreate {
+    name: string;
+    description?: string;
+    config: GuardrailConfig;
+}
+
+export class GuardrailClient {
+    private client: AxiosInstance;
+
+    constructor(private config: { apiKey: string, baseUrl?: string }) {
+        this.client = axios.create({
+            baseURL: config.baseUrl || 'https://api.langtrain.ai/api/v1',
+            headers: {
+                'X-API-Key': config.apiKey,
+                'Content-Type': 'application/json'
+            }
+        });
+    }
+
+    async list(workspaceId?: string): Promise<Guardrail[]> {
+        const params: any = {};
+        if (workspaceId) params.workspace_id = workspaceId;
+        const response = await this.client.get<Guardrail[]>('/guardrails/', { params });
+        return response.data;
+    }
+
+    async get(guardrailId: string): Promise<Guardrail> {
+        const response = await this.client.get<Guardrail>(`/guardrails/${guardrailId}`);
+        return response.data;
+    }
+
+    async create(data: GuardrailCreate): Promise<Guardrail> {
+        const response = await this.client.post<Guardrail>('/guardrails/', data);
+        return response.data;
+    }
+
+    async delete(guardrailId: string): Promise<void> {
+        await this.client.delete(`/guardrails/${guardrailId}`);
+    }
+
+    async apply(datasetId: string, guardrailId: string): Promise<any> {
+        const response = await this.client.post('/guardrails/apply', {
+            dataset_id: datasetId,
+            guardrail_id: guardrailId
+        });
+        return response.data;
+    }
+}

package/src/lib/secrets.ts

@@ -0,0 +1,39 @@
+import axios, { AxiosInstance } from 'axios';
+
+export interface Secret {
+    key: string;
+    created_at: string;
+    updated_at: string;
+}
+
+export class SecretClient {
+    private client: AxiosInstance;
+
+    constructor(private config: { apiKey: string, baseUrl?: string }) {
+        this.client = axios.create({
+            baseURL: config.baseUrl || 'https://api.langtrain.ai/api/v1',
+            headers: {
+                'X-API-Key': config.apiKey,
+                'Content-Type': 'application/json'
+            }
+        });
+    }
+
+    async list(workspaceId?: string): Promise<Secret[]> {
+        const params: any = {};
+        if (workspaceId) params.workspace_id = workspaceId;
+        const response = await this.client.get<{ secrets: Secret[] }>('/secrets', { params });
+        return response.data.secrets;
+    }
+
+    async set(key: string, value: string, workspaceId?: string): Promise<Secret> {
+        const response = await this.client.post<Secret>('/secrets', { key, value, workspaceId });
+        return response.data;
+    }
+
+    async delete(key: string, workspaceId?: string): Promise<void> {
+        const params: any = { key };
+        if (workspaceId) params.workspace_id = workspaceId;
+        await this.client.delete(`/secrets/${key}`, { params });
+    }
+}