@defai.digital/ax-cli 2.2.1 → 2.3.1

package/dist/utils/onboarding-manager.js

@@ -0,0 +1,105 @@
+/**
+ * Onboarding Manager - First-run detection and state management
+ */
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+export class OnboardingManager {
+    static FIRST_RUN_MARKER = '.first-run-complete';
+    static USER_CONFIG_DIR = '.ax-cli';
+    /**
+     * Get the user's AX CLI configuration directory
+     */
+    static getUserConfigDir() {
+        return path.join(os.homedir(), OnboardingManager.USER_CONFIG_DIR);
+    }
+    /**
+     * Get the path to the first-run marker file
+     */
+    static getMarkerPath() {
+        return path.join(OnboardingManager.getUserConfigDir(), OnboardingManager.FIRST_RUN_MARKER);
+    }
+    /**
+     * Detect if this is the user's first run of AX CLI
+     */
+    static detectFirstRun() {
+        const markerPath = OnboardingManager.getMarkerPath();
+        return !fs.existsSync(markerPath);
+    }
+    /**
+     * Mark onboarding as completed
+     */
+    static markCompleted() {
+        const configDir = OnboardingManager.getUserConfigDir();
+        const markerPath = OnboardingManager.getMarkerPath();
+        // Create config directory if it doesn't exist
+        if (!fs.existsSync(configDir)) {
+            fs.mkdirSync(configDir, { recursive: true });
+        }
+        // Create marker file with timestamp
+        const state = {
+            isFirstRun: false,
+            setupCompleted: true,
+            completedAt: new Date().toISOString(),
+            version: this.getVersion(),
+        };
+        fs.writeFileSync(markerPath, JSON.stringify(state, null, 2), 'utf-8');
+    }
+    /**
+     * Get current onboarding state
+     */
+    static getCurrentState() {
+        const markerPath = OnboardingManager.getMarkerPath();
+        if (!fs.existsSync(markerPath)) {
+            return {
+                isFirstRun: true,
+                setupCompleted: false,
+                version: this.getVersion(),
+            };
+        }
+        try {
+            const content = fs.readFileSync(markerPath, 'utf-8');
+            return JSON.parse(content);
+        }
+        catch {
+            // If marker file is corrupted, treat as first run
+            return {
+                isFirstRun: true,
+                setupCompleted: false,
+                version: this.getVersion(),
+            };
+        }
+    }
+    /**
+     * Reset onboarding state (useful for testing)
+     */
+    static resetOnboarding() {
+        const markerPath = OnboardingManager.getMarkerPath();
+        if (fs.existsSync(markerPath)) {
+            fs.unlinkSync(markerPath);
+        }
+    }
+    /**
+     * Check if welcome screen should be shown
+     */
+    static shouldShowWelcome() {
+        return OnboardingManager.detectFirstRun();
+    }
+    /**
+     * Get current CLI version
+     */
+    static getVersion() {
+        try {
+            const packageJsonPath = path.join(process.cwd(), 'package.json');
+            if (fs.existsSync(packageJsonPath)) {
+                const pkg = JSON.parse(fs.readFileSync(packageJsonPath, 'utf-8'));
+                return pkg.version || '0.0.0';
+            }
+        }
+        catch {
+            // Ignore errors
+        }
+        return '0.0.0';
+    }
+}
+//# sourceMappingURL=onboarding-manager.js.map

package/dist/utils/onboarding-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"onboarding-manager.js","sourceRoot":"","sources":["../../src/utils/onboarding-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AASzB,MAAM,OAAO,iBAAiB;IACpB,MAAM,CAAU,gBAAgB,GAAG,qBAAqB,CAAC;IACzD,MAAM,CAAU,eAAe,GAAG,SAAS,CAAC;IAEpD;;OAEG;IACK,MAAM,CAAC,gBAAgB;QAC7B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,iBAAiB,CAAC,eAAe,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,aAAa;QAC1B,OAAO,IAAI,CAAC,IAAI,CACd,iBAAiB,CAAC,gBAAgB,EAAE,EACpC,iBAAiB,CAAC,gBAAgB,CACnC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,cAAc;QACnB,MAAM,UAAU,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC;QACrD,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,aAAa;QAClB,MAAM,SAAS,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,UAAU,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC;QAErD,8CAA8C;QAC9C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,oCAAoC;QACpC,MAAM,KAAK,GAAoB;YAC7B,UAAU,EAAE,KAAK;YACjB,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACrC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;SAC3B,CAAC;QAEF,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACxE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,eAAe;QACpB,MAAM,UAAU,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC;QAErD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,UAAU,EAAE,IAAI;gBAChB,cAAc,EAAE,KAAK;gBACrB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;aAC3B,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACrD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAoB,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,kDAAkD;YAClD,OAAO;gBACL,UAAU,EAAE,IAAI;gBAChB,cAAc,EAAE,KAAK;gBACrB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;aAC3B,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,eAAe;QACpB,MAAM,UAAU,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC;QACrD,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,iBAAiB;QACtB,OAAO,iBAAiB,CAAC,cAAc,EAAE,CAAC;IAC5C,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,UAAU;QACvB,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,cAAc,CAAC,CAAC;YACjE,IAAI,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC;gBAClE,OAAO,GAAG,CAAC,OAAO,IAAI,OAAO,CAAC;YAChC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,gBAAgB;QAClB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC"}

package/dist/utils/usage-tracker.d.ts

@@ -0,0 +1,59 @@
+/**
+ * Usage Tracker - Session-based API usage tracking
+ *
+ * Tracks token usage across API calls in the current session.
+ * Phase 1: Support for z.ai and session-based tracking
+ * Phase 2: Add support for historical tracking and other providers
+ */
+export interface UsageStats {
+    requests: number;
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+    reasoningTokens: number;
+}
+export interface SessionStats {
+    totalRequests: number;
+    totalPromptTokens: number;
+    totalCompletionTokens: number;
+    totalTokens: number;
+    totalReasoningTokens: number;
+    byModel: Map<string, UsageStats>;
+}
+/**
+ * UsageTracker - Singleton for tracking API usage
+ */
+export declare class UsageTracker {
+    private static instance;
+    private sessionStats;
+    private constructor();
+    /**
+     * Get singleton instance
+     */
+    static getInstance(): UsageTracker;
+    /**
+     * Track usage from API response
+     */
+    trackUsage(model: string, usage: {
+        prompt_tokens?: number;
+        completion_tokens?: number;
+        total_tokens?: number;
+        reasoning_tokens?: number;
+    }): void;
+    /**
+     * Get current session statistics
+     */
+    getSessionStats(): SessionStats;
+    /**
+     * Reset session statistics
+     */
+    resetSession(): void;
+    /**
+     * Get statistics for a specific model
+     */
+    getModelStats(model: string): UsageStats | null;
+}
+/**
+ * Get the singleton usage tracker instance
+ */
+export declare function getUsageTracker(): UsageTracker;

package/dist/utils/usage-tracker.js

@@ -0,0 +1,97 @@
+/**
+ * Usage Tracker - Session-based API usage tracking
+ *
+ * Tracks token usage across API calls in the current session.
+ * Phase 1: Support for z.ai and session-based tracking
+ * Phase 2: Add support for historical tracking and other providers
+ */
+/**
+ * UsageTracker - Singleton for tracking API usage
+ */
+export class UsageTracker {
+    static instance = null;
+    sessionStats = {
+        totalRequests: 0,
+        totalPromptTokens: 0,
+        totalCompletionTokens: 0,
+        totalTokens: 0,
+        totalReasoningTokens: 0,
+        byModel: new Map()
+    };
+    constructor() {
+        // Private constructor for singleton
+    }
+    /**
+     * Get singleton instance
+     */
+    static getInstance() {
+        if (!UsageTracker.instance) {
+            UsageTracker.instance = new UsageTracker();
+        }
+        return UsageTracker.instance;
+    }
+    /**
+     * Track usage from API response
+     */
+    trackUsage(model, usage) {
+        const promptTokens = usage.prompt_tokens || 0;
+        const completionTokens = usage.completion_tokens || 0;
+        const totalTokens = usage.total_tokens || (promptTokens + completionTokens);
+        const reasoningTokens = usage.reasoning_tokens || 0;
+        // Update total stats
+        this.sessionStats.totalRequests++;
+        this.sessionStats.totalPromptTokens += promptTokens;
+        this.sessionStats.totalCompletionTokens += completionTokens;
+        this.sessionStats.totalTokens += totalTokens;
+        this.sessionStats.totalReasoningTokens += reasoningTokens;
+        // Update per-model stats
+        const modelStats = this.sessionStats.byModel.get(model) || {
+            requests: 0,
+            promptTokens: 0,
+            completionTokens: 0,
+            totalTokens: 0,
+            reasoningTokens: 0
+        };
+        modelStats.requests++;
+        modelStats.promptTokens += promptTokens;
+        modelStats.completionTokens += completionTokens;
+        modelStats.totalTokens += totalTokens;
+        modelStats.reasoningTokens += reasoningTokens;
+        this.sessionStats.byModel.set(model, modelStats);
+    }
+    /**
+     * Get current session statistics
+     */
+    getSessionStats() {
+        return {
+            ...this.sessionStats,
+            byModel: new Map(this.sessionStats.byModel)
+        };
+    }
+    /**
+     * Reset session statistics
+     */
+    resetSession() {
+        this.sessionStats = {
+            totalRequests: 0,
+            totalPromptTokens: 0,
+            totalCompletionTokens: 0,
+            totalTokens: 0,
+            totalReasoningTokens: 0,
+            byModel: new Map()
+        };
+    }
+    /**
+     * Get statistics for a specific model
+     */
+    getModelStats(model) {
+        return this.sessionStats.byModel.get(model) || null;
+    }
+}
+/**
+ * Get the singleton usage tracker instance
+ */
+export function getUsageTracker() {
+    return UsageTracker.getInstance();
+}
+//# sourceMappingURL=usage-tracker.js.map

package/dist/utils/usage-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracker.js","sourceRoot":"","sources":["../../src/utils/usage-tracker.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAmBH;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,MAAM,CAAC,QAAQ,GAAwB,IAAI,CAAC;IAE5C,YAAY,GAAiB;QACnC,aAAa,EAAE,CAAC;QAChB,iBAAiB,EAAE,CAAC;QACpB,qBAAqB,EAAE,CAAC;QACxB,WAAW,EAAE,CAAC;QACd,oBAAoB,EAAE,CAAC;QACvB,OAAO,EAAE,IAAI,GAAG,EAAE;KACnB,CAAC;IAEF;QACE,oCAAoC;IACtC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,WAAW;QAChB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;YAC3B,YAAY,CAAC,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QAC7C,CAAC;QACD,OAAO,YAAY,CAAC,QAAQ,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,KAAa,EAAE,KAKzB;QACC,MAAM,YAAY,GAAG,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;QAC9C,MAAM,gBAAgB,GAAG,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;QACtD,MAAM,WAAW,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,YAAY,GAAG,gBAAgB,CAAC,CAAC;QAC5E,MAAM,eAAe,GAAG,KAAK,CAAC,gBAAgB,IAAI,CAAC,CAAC;QAEpD,qBAAqB;QACrB,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,CAAC;QAClC,IAAI,CAAC,YAAY,CAAC,iBAAiB,IAAI,YAAY,CAAC;QACpD,IAAI,CAAC,YAAY,CAAC,qBAAqB,IAAI,gBAAgB,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,WAAW,IAAI,WAAW,CAAC;QAC7C,IAAI,CAAC,YAAY,CAAC,oBAAoB,IAAI,eAAe,CAAC;QAE1D,yBAAyB;QACzB,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;YACzD,QAAQ,EAAE,CAAC;YACX,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;YACd,eAAe,EAAE,CAAC;SACnB,CAAC;QAEF,UAAU,CAAC,QAAQ,EAAE,CAAC;QACtB,UAAU,CAAC,YAAY,IAAI,YAAY,CAAC;QACxC,UAAU,CAAC,gBAAgB,IAAI,gBAAgB,CAAC;QAChD,UAAU,CAAC,WAAW,IAAI,WAAW,CAAC;QACtC,UAAU,CAAC,eAAe,IAAI,eAAe,CAAC;QAE9C,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO;YACL,GAAG,IAAI,CAAC,YAAY;YACpB,OAAO,EAAE,IAAI,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,YAAY,GAAG;YAClB,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,CAAC;YACpB,qBAAqB,EAAE,CAAC;YACxB,WAAW,EAAE,CAAC;YACd,oBAAoB,EAAE,CAAC;YACvB,OAAO,EAAE,IAAI,GAAG,EAAE;SACnB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,KAAa;QACzB,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IACtD,CAAC;;AAGH;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,OAAO,YAAY,CAAC,WAAW,EAAE,CAAC;AACpC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@defai.digital/ax-cli",
-  "version": "2.2.1",
+  "version": "2.3.1",
   "description": "Enterprise-Class AI Command Line Interface - Primary support for GLM (General Language Model) with multi-provider AI orchestration powered by AutomatosX.",
   "type": "module",
   "workspaces": [