langtrain 0.1.24 → 0.1.26

package/dist/index.d.mts

@@ -136,6 +136,11 @@ interface SubscriptionInfo {
     expires_at?: string;
     features: string[];
     limits: any;
+    usage?: {
+        tokensUsedThisMonth?: number;
+        tokenLimit?: number;
+        apiCalls?: number;
+    };
 }
 interface FeatureCheck {
     feature: string;
@@ -249,4 +254,38 @@ declare class GuardrailClient {
     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 };
+interface UsageSummary {
+    workspace_id: string;
+    plan: string;
+    quotas: Record<string, any>;
+    billing?: {
+        plan_id: string;
+        tokens_used: number;
+        tokens_limit: number;
+        period_end: string;
+    };
+}
+interface UsageHistoryPoint {
+    date: string;
+    tokens: number;
+    agent_runs: number;
+    cost: number;
+}
+declare class UsageClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    /**
+     * Get current usage summary for a workspace.
+     */
+    getSummary(workspaceId: string): Promise<UsageSummary>;
+    /**
+     * Get historical usage data for charts.
+     */
+    getHistory(workspaceId: string, days?: number): Promise<UsageHistoryPoint[]>;
+}
+
+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, UsageClient, type UsageHistoryPoint, type UsageSummary };

package/dist/index.d.ts

@@ -136,6 +136,11 @@ interface SubscriptionInfo {
     expires_at?: string;
     features: string[];
     limits: any;
+    usage?: {
+        tokensUsedThisMonth?: number;
+        tokenLimit?: number;
+        apiCalls?: number;
+    };
 }
 interface FeatureCheck {
     feature: string;
@@ -249,4 +254,38 @@ declare class GuardrailClient {
     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 };
+interface UsageSummary {
+    workspace_id: string;
+    plan: string;
+    quotas: Record<string, any>;
+    billing?: {
+        plan_id: string;
+        tokens_used: number;
+        tokens_limit: number;
+        period_end: string;
+    };
+}
+interface UsageHistoryPoint {
+    date: string;
+    tokens: number;
+    agent_runs: number;
+    cost: number;
+}
+declare class UsageClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    /**
+     * Get current usage summary for a workspace.
+     */
+    getSummary(workspaceId: string): Promise<UsageSummary>;
+    /**
+     * Get historical usage data for charts.
+     */
+    getHistory(workspaceId: string, days?: number): Promise<UsageHistoryPoint[]>;
+}
+
+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, UsageClient, type UsageHistoryPoint, type UsageSummary };

package/dist/index.js

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

package/dist/index.mjs

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langtrain",
-  "version": "0.1.24",
+  "version": "0.1.26",
   "description": "Unified JavaScript SDK for Langtrain Ecosystem",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

package/src/cli/auth.ts

@@ -1,12 +1,25 @@
-import { password, isCancel, cancel, intro, green, yellow, red, bgMagenta, black, spinner, gray } from './ui';
+import { password, isCancel, cancel, intro, green, yellow, red, bgMagenta, black, spinner, gray, cyan, dim, bold } from './ui';
 import { getConfig, saveConfig } from './config';
 import { SubscriptionClient, SubscriptionInfo } from '../index';
 
+/**
+ * Quick check if API key is stored (no network call).
+ */
+export function isAuthenticated(): boolean {
+    const config = getConfig();
+    return !!config.apiKey;
+}
+
+/**
+ * Ensure auth — if not logged in, forces login flow.
+ */
 export async function ensureAuth(): Promise<string> {
     let config = getConfig();
 
     if (!config.apiKey) {
-        intro(yellow('Authentication required to verify plan & features.'));
+        console.log('');
+        console.log(yellow('  Authentication required.'));
+        console.log(gray('  Login to access all features.\n'));
         await handleLogin();
         config = getConfig();
     }
@@ -14,13 +27,20 @@ export async function ensureAuth(): Promise<string> {
     return config.apiKey as string;
 }
 
+/**
+ * Interactive login — Claude-style API key entry with immediate verification.
+ */
 export async function handleLogin() {
     while (true) {
-        console.log(gray('\nGet your API Key at: https://langtrain.xyz/settings/keys\n'));
+        console.log(dim('  ─────────────────────────────────────'));
+        console.log(gray('  Get your API Key at: ') + cyan('https://app.langtrain.xyz/home/api'));
+        console.log(dim('  ─────────────────────────────────────\n'));
+
         const apiKey = await password({
             message: 'Enter your Langtrain API Key:',
             validate(value) {
                 if (!value || value.length === 0) return 'API Key is required';
+                if (value.length < 10) return 'Invalid key format';
             },
         });
 
@@ -32,45 +52,73 @@ export async function handleLogin() {
         const s = spinner();
         s.start('Verifying API Key...');
 
-        // Verify key immediately
         try {
             const client = new SubscriptionClient({ apiKey: apiKey as string });
             const info = await client.getStatus();
 
-            s.stop(green(`Authenticated as ${info.plan === 'pro' ? 'PRO' : info.plan.toUpperCase()}`));
+            const planBadge = info.plan === 'pro'
+                ? bgMagenta(black(' PRO '))
+                : info.plan === 'enterprise'
+                    ? bgMagenta(black(' ENTERPRISE '))
+                    : ' FREE ';
+
+            s.stop(green(`Authenticated ${planBadge}`));
+
+            // Show initial token info if available
+            if (info.usage) {
+                const used = info.usage.tokensUsedThisMonth || 0;
+                const limit = info.usage.tokenLimit || 10000;
+                const pct = Math.round((used / limit) * 100);
+                console.log(dim(`  Tokens: ${used.toLocaleString()} / ${limit.toLocaleString()} (${pct}% used)`));
+            }
 
             const config = getConfig();
             saveConfig({ ...config, apiKey: apiKey as string });
-            // intro(green('API Key saved successfully!')); // success message above is enough
-            return; // Exit loop on success
+            console.log(green('  ✔ Credentials saved to ~/.langtrain/config.json\n'));
+            return;
         } catch (e: any) {
             s.stop(red('Invalid API Key. Please try again.'));
-            // Loop continues
         }
     }
 }
 
+/**
+ * Logout — clear stored credentials.
+ */
+export async function handleLogout() {
+    const config = getConfig();
+    delete config.apiKey;
+    saveConfig(config);
+    console.log(green('\n  ✔ Logged out. Credentials cleared.\n'));
+}
+
+/**
+ * Fetch subscription info for status bar display.
+ */
 export async function getSubscription(apiKey: string): Promise<SubscriptionInfo | null> {
     const client = new SubscriptionClient({ apiKey });
     const s = spinner();
-    s.start('Verifying subscription plan...');
+    s.start('Checking subscription...');
     try {
         const info = await client.getStatus();
 
-        // Enhance: Show plan details immediately on auth check
-        const planLabel = info.plan === 'pro' ? bgMagenta(' PRO ') : info.plan.toUpperCase();
-        s.stop(green(`Authenticated as ${planLabel}`));
+        const planBadge = info.plan === 'pro'
+            ? bgMagenta(black(' PRO '))
+            : info.plan === 'enterprise'
+                ? bgMagenta(black(' ENTERPRISE '))
+                : bold(' FREE ');
+
+        s.stop(green(`Plan: ${planBadge}`));
 
         if (info.is_active === false) {
-            console.log(yellow('Warning: Your subscription is not active. Some features may be limited.'));
+            console.log(yellow('  ⚠ Subscription inactive. Some features may be limited.\n'));
         }
 
         return info;
     } catch (e: any) {
         s.stop(red('Failed to verify subscription.'));
         if (e.response && e.response.status === 401) {
-            console.log(red('Invalid API Key. Please run login again.'));
-            // Optionally clear key?
+            console.log(red('  API Key expired. Please login again.'));
         }
         return null;
     }

package/src/cli/handlers/telemetry.ts

@@ -0,0 +1,127 @@
+import { green, dim, cyan, bold, yellow, gray, spinner } from '../ui';
+import { getConfig } from '../config';
+
+// Session-level telemetry tracker
+let sessionStart = Date.now();
+let apiCallCount = 0;
+let totalLatencyMs = 0;
+let errorCount = 0;
+
+export function trackApiCall(latencyMs: number, isError: boolean = false) {
+    apiCallCount++;
+    totalLatencyMs += latencyMs;
+    if (isError) errorCount++;
+}
+
+export async function handleTokens() {
+    const config = getConfig();
+    const apiKey = config.apiKey;
+
+    console.log('');
+    console.log(bold('  ╔══════════════════════════════════════╗'));
+    console.log(bold('  ║         TOKEN USAGE                  ║'));
+    console.log(bold('  ╚══════════════════════════════════════╝'));
+    console.log('');
+
+    if (!apiKey) {
+        console.log(yellow('  Login required to view token usage.\n'));
+        return;
+    }
+
+    const s = spinner();
+    s.start('Fetching token usage...');
+
+    try {
+        const axios = require('axios');
+        const baseUrl = config.baseUrl || 'https://api.langtrain.xyz';
+        const res = await axios.get(`${baseUrl}/v1/usage/tokens`, {
+            headers: { Authorization: `Bearer ${apiKey}` }
+        });
+
+        const usage = res.data;
+        s.stop(green('Token usage retrieved'));
+        console.log('');
+
+        const used = usage.tokens_used || 0;
+        const limit = usage.token_limit || 10000;
+        const pct = Math.round((used / limit) * 100);
+        const remaining = Math.max(0, limit - used);
+
+        // Visual bar
+        const barWidth = 30;
+        const filled = Math.round((pct / 100) * barWidth);
+        const bar = '█'.repeat(filled) + '░'.repeat(barWidth - filled);
+        const barColor = pct > 90 ? '\x1b[31m' : pct > 70 ? '\x1b[33m' : '\x1b[32m';
+
+        console.log(`  ${dim('Period:')}     ${usage.period || 'Current Month'}`);
+        console.log(`  ${dim('Used:')}       ${used.toLocaleString()} tokens`);
+        console.log(`  ${dim('Limit:')}      ${limit.toLocaleString()} tokens`);
+        console.log(`  ${dim('Remaining:')}  ${remaining.toLocaleString()} tokens`);
+        console.log(`  ${dim('Usage:')}      ${barColor}${bar}\x1b[0m ${pct}%`);
+        console.log('');
+
+        if (usage.breakdown) {
+            console.log(dim('  ── Breakdown ────────────────────────'));
+            console.log(`  ${dim('Training:')}   ${(usage.breakdown.training || 0).toLocaleString()}`);
+            console.log(`  ${dim('Inference:')}  ${(usage.breakdown.inference || 0).toLocaleString()}`);
+            console.log(`  ${dim('Agents:')}     ${(usage.breakdown.agents || 0).toLocaleString()}`);
+            console.log('');
+        }
+    } catch (e: any) {
+        s.stop('');
+        // Show mock data if API not available
+        console.log(dim('  Token data not available from server.'));
+        console.log(dim('  Showing session estimates:\n'));
+
+        console.log(`  ${dim('Session calls:')}  ${apiCallCount}`);
+        console.log(`  ${dim('Est. tokens:')}    ~${apiCallCount * 150}`);
+        console.log('');
+    }
+}
+
+export async function handleTelemetry() {
+    const uptimeMs = Date.now() - sessionStart;
+    const uptimeSec = Math.round(uptimeMs / 1000);
+    const uptimeMin = Math.floor(uptimeSec / 60);
+    const uptimeStr = uptimeMin > 0 ? `${uptimeMin}m ${uptimeSec % 60}s` : `${uptimeSec}s`;
+
+    const avgLatency = apiCallCount > 0 ? Math.round(totalLatencyMs / apiCallCount) : 0;
+    const errorRate = apiCallCount > 0 ? Math.round((errorCount / apiCallCount) * 100) : 0;
+
+    console.log('');
+    console.log(bold('  ╔══════════════════════════════════════╗'));
+    console.log(bold('  ║         SESSION TELEMETRY             ║'));
+    console.log(bold('  ╚══════════════════════════════════════╝'));
+    console.log('');
+
+    console.log(`  ${dim('Session:')}      ${uptimeStr}`);
+    console.log(`  ${dim('API calls:')}    ${apiCallCount}`);
+    console.log(`  ${dim('Avg latency:')}  ${avgLatency}ms`);
+    console.log(`  ${dim('Errors:')}       ${errorCount} (${errorRate}%)`);
+    console.log('');
+
+    console.log(dim('  ── Environment ──────────────────────'));
+    console.log(`  ${dim('Node:')}         ${process.version}`);
+    console.log(`  ${dim('Platform:')}     ${process.platform} ${process.arch}`);
+    console.log(`  ${dim('Memory:')}       ${Math.round(process.memoryUsage().heapUsed / 1024 / 1024)}MB heap`);
+    console.log(`  ${dim('Config:')}       ~/.langtrain/config.json`);
+    console.log('');
+
+    // API health check
+    const config = getConfig();
+    if (config.apiKey) {
+        const s = spinner();
+        s.start('Pinging API...');
+        try {
+            const axios = require('axios');
+            const baseUrl = config.baseUrl || 'https://api.langtrain.xyz';
+            const start = Date.now();
+            await axios.get(`${baseUrl}/health`, { timeout: 5000 });
+            const latency = Date.now() - start;
+            s.stop(green(`API healthy (${latency}ms)`));
+        } catch {
+            s.stop(yellow('API unreachable'));
+        }
+    }
+    console.log('');
+}