@defai.digital/ax-cli 4.3.6 → 4.3.9

package/dist/tools/priority.js

@@ -0,0 +1,350 @@
+/**
+ * Tool Priority System
+ *
+ * Defines priority levels and capability types for intelligent tool selection.
+ * When multiple tools can handle the same task, the system prefers higher-priority tools.
+ *
+ * Priority Order:
+ * 1. Native API capabilities (e.g., Grok's built-in search)
+ * 2. Provider-specific MCP (e.g., Z.AI MCP for GLM)
+ * 3. Domain-specific MCP (e.g., Figma MCP for design)
+ * 4. Official MCP servers
+ * 5. Community MCP servers
+ * 6. General-purpose MCP (e.g., AutomatosX)
+ * 7. Built-in tools (fallback)
+ */
+/**
+ * Priority levels for tool selection
+ * Higher values = higher priority = preferred tool
+ */
+export var ToolPriority;
+(function (ToolPriority) {
+    /** Built into provider API (highest priority) */
+    ToolPriority[ToolPriority["NATIVE_API"] = 100] = "NATIVE_API";
+    /** Provider-specific MCP (e.g., Z.AI for GLM) */
+    ToolPriority[ToolPriority["PROVIDER_MCP"] = 80] = "PROVIDER_MCP";
+    /** Domain-specific MCP (e.g., Figma for design) */
+    ToolPriority[ToolPriority["DOMAIN_SPECIFIC"] = 60] = "DOMAIN_SPECIFIC";
+    /** Official MCP servers (e.g., @modelcontextprotocol/*) */
+    ToolPriority[ToolPriority["OFFICIAL_MCP"] = 40] = "OFFICIAL_MCP";
+    /** Community MCP servers */
+    ToolPriority[ToolPriority["COMMUNITY_MCP"] = 20] = "COMMUNITY_MCP";
+    /** General-purpose MCP (e.g., AutomatosX) */
+    ToolPriority[ToolPriority["GENERAL_MCP"] = 10] = "GENERAL_MCP";
+    /** Built-in ax-cli tools (fallback) */
+    ToolPriority[ToolPriority["BUILTIN_TOOL"] = 5] = "BUILTIN_TOOL";
+})(ToolPriority || (ToolPriority = {}));
+/**
+ * Priority boost applied when a server has affinity for the current provider.
+ * This ensures provider-affinity servers beat same-level servers without affinity.
+ */
+export const PROVIDER_AFFINITY_BOOST = 10;
+/**
+ * Minimum priority difference required to consider a capability superseded.
+ * Prevents minor priority differences from hiding useful tools.
+ */
+export const SUPERSEDE_THRESHOLD = 15;
+/**
+ * Delimiters used to separate base names from variant suffixes.
+ * e.g., 'grok-beta' uses '-', 'automatosx_glm' uses '_'
+ */
+export const VARIANT_DELIMITERS = ['-', '_'];
+/**
+ * Prefix used for virtual native capability tool names.
+ * e.g., 'native_web-search' represents Grok's native search capability
+ */
+export const NATIVE_CAPABILITY_PREFIX = 'native_';
+/**
+ * Check if a name starts with a base name followed by a variant delimiter.
+ * Used to match variant names like 'grok-beta' to base name 'grok'.
+ *
+ * @param fullName - The full name to check (e.g., 'grok-beta')
+ * @param baseName - The base name to match against (e.g., 'grok')
+ * @returns True if fullName is a variant of baseName
+ */
+export function isVariantOf(fullName, baseName) {
+    return VARIANT_DELIMITERS.some(delimiter => fullName.startsWith(baseName + delimiter));
+}
+/**
+ * Native capabilities built into provider APIs
+ * These don't require MCP - they're part of the API itself
+ */
+export const PROVIDER_NATIVE_CAPABILITIES = {
+    grok: ['web-search'], // Grok has native live search via API
+    glm: [], // GLM uses Z.AI MCP for web search
+    claude: [], // Claude has no native search in API
+    openai: [], // OpenAI standard API has no search
+    gemini: ['web-search'], // Gemini has grounding/search
+};
+/**
+ * Known MCP server capability registry
+ * Add new servers here as they become available
+ */
+export const MCP_CAPABILITY_REGISTRY = [
+    // ========================================
+    // Z.AI MCP Servers (Provider-specific for GLM)
+    // ========================================
+    {
+        serverName: 'zai-web-search',
+        capabilities: ['web-search'],
+        priority: ToolPriority.PROVIDER_MCP,
+        providerAffinity: ['glm'],
+    },
+    {
+        serverName: 'zai-web-reader',
+        capabilities: ['web-fetch'],
+        priority: ToolPriority.PROVIDER_MCP,
+        providerAffinity: ['glm'],
+    },
+    {
+        serverName: 'zai-vision',
+        capabilities: ['vision'],
+        priority: ToolPriority.PROVIDER_MCP,
+        providerAffinity: ['glm'],
+    },
+    // ========================================
+    // Domain-Specific MCPs
+    // ========================================
+    {
+        serverName: 'figma',
+        capabilities: ['design-figma', 'design-general'],
+        priority: ToolPriority.DOMAIN_SPECIFIC,
+    },
+    // ========================================
+    // Official MCP Servers
+    // ========================================
+    {
+        serverName: 'github',
+        capabilities: ['git-operations'],
+        priority: ToolPriority.OFFICIAL_MCP,
+        isOfficial: true,
+    },
+    {
+        serverName: 'postgres',
+        capabilities: ['database'],
+        priority: ToolPriority.OFFICIAL_MCP,
+        isOfficial: true,
+    },
+    {
+        serverName: 'sqlite',
+        capabilities: ['database'],
+        priority: ToolPriority.OFFICIAL_MCP,
+        isOfficial: true,
+    },
+    {
+        serverName: 'puppeteer',
+        // Note: Puppeteer is primarily for browser automation and testing.
+        // It can fetch web content but via a full browser, which is heavyweight.
+        // Don't mark as 'web-fetch' to avoid superseding lightweight HTTP fetchers.
+        capabilities: ['testing'],
+        priority: ToolPriority.OFFICIAL_MCP,
+        isOfficial: true,
+    },
+    // ========================================
+    // Community MCPs
+    // ========================================
+    {
+        serverName: 'vercel',
+        capabilities: ['deployment'],
+        priority: ToolPriority.COMMUNITY_MCP,
+    },
+    {
+        serverName: 'netlify',
+        capabilities: ['deployment'],
+        priority: ToolPriority.COMMUNITY_MCP,
+    },
+    {
+        serverName: 'supabase',
+        capabilities: ['database'],
+        priority: ToolPriority.COMMUNITY_MCP,
+    },
+    {
+        serverName: 'firebase',
+        capabilities: ['database', 'deployment'],
+        priority: ToolPriority.COMMUNITY_MCP,
+    },
+    {
+        serverName: 'sentry',
+        capabilities: ['monitoring'],
+        priority: ToolPriority.COMMUNITY_MCP,
+    },
+    // ========================================
+    // General-Purpose MCPs (lowest priority)
+    // ========================================
+    {
+        serverName: 'automatosx',
+        capabilities: ['web-search', 'web-fetch', 'memory', 'agent-delegation'],
+        priority: ToolPriority.GENERAL_MCP,
+    },
+];
+/**
+ * Cached lowercase server name index for O(1) exact lookups.
+ * Maps lowercase server name to its MCPCapabilityMapping.
+ */
+const serverNameIndex = new Map(MCP_CAPABILITY_REGISTRY.map(mapping => [mapping.serverName.toLowerCase(), mapping]));
+/**
+ * Pre-computed lowercase server names sorted by length (longest first).
+ * Used for efficient variant matching.
+ */
+const sortedServerNames = MCP_CAPABILITY_REGISTRY
+    .map(mapping => mapping.serverName.toLowerCase())
+    .sort((a, b) => b.length - a.length);
+/**
+ * Pre-computed capability to servers index for O(1) capability lookups.
+ * Maps capability to array of servers that provide it.
+ */
+const capabilityIndex = new Map();
+for (const mapping of MCP_CAPABILITY_REGISTRY) {
+    for (const capability of mapping.capabilities) {
+        let servers = capabilityIndex.get(capability);
+        if (!servers) {
+            servers = [];
+            capabilityIndex.set(capability, servers);
+        }
+        servers.push(mapping);
+    }
+}
+/**
+ * Get capability mapping for a server by name.
+ * Supports exact matches and variant matches (e.g., 'automatosx-glm' matches 'automatosx').
+ * Case-insensitive matching for robustness.
+ *
+ * @param serverName - The server name to look up (may include variant suffix)
+ * @returns The capability mapping, or undefined if not found
+ */
+export function getServerCapabilityMapping(serverName) {
+    const normalizedName = serverName.toLowerCase();
+    // O(1) exact match lookup
+    const exactMatch = serverNameIndex.get(normalizedName);
+    if (exactMatch) {
+        return exactMatch;
+    }
+    // Variant match - find longest matching base name
+    for (const baseName of sortedServerNames) {
+        if (isVariantOf(normalizedName, baseName)) {
+            return serverNameIndex.get(baseName);
+        }
+    }
+    return undefined;
+}
+/**
+ * List of known provider names derived from PROVIDER_NATIVE_CAPABILITIES.
+ * Used for flexible provider name matching.
+ */
+const KNOWN_PROVIDERS = Object.keys(PROVIDER_NATIVE_CAPABILITIES);
+/**
+ * Get the base provider name from a potentially variant provider string.
+ * e.g., 'grok-beta' -> 'grok', 'glm-4' -> 'glm', 'openai' -> 'openai'
+ *
+ * @param providerName - The provider name (may include variant suffix)
+ * @returns The base provider name, or undefined if not recognized
+ */
+function getBaseProviderName(providerName) {
+    const normalizedName = providerName.toLowerCase();
+    for (const knownProvider of KNOWN_PROVIDERS) {
+        if (normalizedName === knownProvider || isVariantOf(normalizedName, knownProvider)) {
+            return knownProvider;
+        }
+    }
+    return undefined;
+}
+/**
+ * Check if a provider has native support for a capability.
+ * Supports provider variants (e.g., 'grok-beta' matches 'grok' native capabilities)
+ *
+ * @param providerName - The provider name (may include variant suffix)
+ * @param capability - The capability to check
+ * @returns True if the provider natively supports the capability
+ */
+export function hasNativeCapability(providerName, capability) {
+    const baseProvider = getBaseProviderName(providerName);
+    if (!baseProvider) {
+        return false;
+    }
+    return PROVIDER_NATIVE_CAPABILITIES[baseProvider].includes(capability);
+}
+/**
+ * Check if a provider matches a base provider name.
+ * e.g., 'grok-beta' matches 'grok', 'glm-4' matches 'glm'
+ *
+ * @param providerName - The provider name to check (may include variant suffix)
+ * @param baseProvider - The base provider to match against
+ * @returns True if the provider matches the base provider
+ */
+export function providerMatches(providerName, baseProvider) {
+    return getBaseProviderName(providerName) === baseProvider;
+}
+/**
+ * Get the priority for a server, considering provider affinity.
+ *
+ * @param serverName - The MCP server name
+ * @param providerName - Optional provider name for affinity boosting
+ * @returns The priority value (higher = more preferred)
+ */
+export function getServerPriority(serverName, providerName) {
+    const mapping = getServerCapabilityMapping(serverName);
+    if (!mapping) {
+        // Unknown server - assign community priority
+        return ToolPriority.COMMUNITY_MCP;
+    }
+    // Boost priority if this server has affinity for the current provider
+    if (providerName && mapping.providerAffinity?.some(affinity => providerMatches(providerName, affinity))) {
+        return mapping.priority + PROVIDER_AFFINITY_BOOST;
+    }
+    return mapping.priority;
+}
+/**
+ * Check if a server should be preferred for a given capability and provider.
+ *
+ * A server is preferred if:
+ * 1. The provider does NOT have native support for this capability, AND
+ * 2. Either:
+ *    a. It has provider affinity for the current provider, OR
+ *    b. It has the highest priority among all servers providing this capability
+ *
+ * Note: If the provider has native capability support, NO MCP server should be preferred.
+ *
+ * @param serverName - The MCP server name
+ * @param capability - The capability to check
+ * @param providerName - The current provider name
+ * @returns True if the server should be preferred for this capability
+ */
+export function shouldPreferServer(serverName, capability, providerName) {
+    const mapping = getServerCapabilityMapping(serverName);
+    if (!mapping)
+        return false;
+    // Check if server provides this capability
+    if (!mapping.capabilities.includes(capability))
+        return false;
+    // If the provider has native support for this capability, no MCP server should be preferred
+    if (hasNativeCapability(providerName, capability)) {
+        return false;
+    }
+    // Check if server has affinity for this provider
+    if (mapping.providerAffinity?.some(affinity => providerMatches(providerName, affinity))) {
+        return true;
+    }
+    // Even without affinity, check if this server has the highest priority for this capability
+    const serversForCapability = getServersForCapability(capability, providerName);
+    const topServer = serversForCapability[0];
+    if (topServer && topServer.serverName.toLowerCase() === serverName.toLowerCase()) {
+        return true;
+    }
+    return false;
+}
+/**
+ * Get all servers that provide a given capability, sorted by priority.
+ *
+ * @param capability - The capability to search for
+ * @param providerName - Optional provider name for affinity-based priority boosting
+ * @returns Array of server mappings sorted by priority (highest first)
+ */
+export function getServersForCapability(capability, providerName) {
+    const servers = capabilityIndex.get(capability);
+    if (!servers?.length) {
+        return [];
+    }
+    // Clone and sort to avoid mutating the cached array
+    return [...servers].sort((a, b) => getServerPriority(b.serverName, providerName) - getServerPriority(a.serverName, providerName));
+}
+//# sourceMappingURL=priority.js.map

package/dist/tools/priority.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"priority.js","sourceRoot":"","sources":["../../src/tools/priority.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAqBH;;;GAGG;AACH,MAAM,CAAN,IAAY,YAeX;AAfD,WAAY,YAAY;IACtB,iDAAiD;IACjD,6DAAgB,CAAA;IAChB,iDAAiD;IACjD,gEAAiB,CAAA;IACjB,mDAAmD;IACnD,sEAAoB,CAAA;IACpB,2DAA2D;IAC3D,gEAAiB,CAAA;IACjB,4BAA4B;IAC5B,kEAAkB,CAAA;IAClB,6CAA6C;IAC7C,8DAAgB,CAAA;IAChB,uCAAuC;IACvC,+DAAgB,CAAA;AAClB,CAAC,EAfW,YAAY,KAAZ,YAAY,QAevB;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,EAAE,CAAC;AAE1C;;;GAGG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAEtC;;;GAGG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,GAAG,EAAE,GAAG,CAAU,CAAC;AAEtD;;;GAGG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,SAAS,CAAC;AAElD;;;;;;;GAOG;AACH,MAAM,UAAU,WAAW,CAAC,QAAgB,EAAE,QAAgB;IAC5D,OAAO,kBAAkB,CAAC,IAAI,CAC5B,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,GAAG,SAAS,CAAC,CACvD,CAAC;AACJ,CAAC;AAOD;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAA2C;IAClF,IAAI,EAAE,CAAC,YAAY,CAAC,EAAK,sCAAsC;IAC/D,GAAG,EAAE,EAAE,EAAkB,mCAAmC;IAC5D,MAAM,EAAE,EAAE,EAAe,qCAAqC;IAC9D,MAAM,EAAE,EAAE,EAAe,oCAAoC;IAC7D,MAAM,EAAE,CAAC,YAAY,CAAC,EAAG,8BAA8B;CACxD,CAAC;AAmBF;;;GAGG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAA2B;IAC7D,2CAA2C;IAC3C,+CAA+C;IAC/C,2CAA2C;IAC3C;QACE,UAAU,EAAE,gBAAgB;QAC5B,YAAY,EAAE,CAAC,YAAY,CAAC;QAC5B,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,gBAAgB,EAAE,CAAC,KAAK,CAAC;KAC1B;IACD;QACE,UAAU,EAAE,gBAAgB;QAC5B,YAAY,EAAE,CAAC,WAAW,CAAC;QAC3B,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,gBAAgB,EAAE,CAAC,KAAK,CAAC;KAC1B;IACD;QACE,UAAU,EAAE,YAAY;QACxB,YAAY,EAAE,CAAC,QAAQ,CAAC;QACxB,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,gBAAgB,EAAE,CAAC,KAAK,CAAC;KAC1B;IAED,2CAA2C;IAC3C,uBAAuB;IACvB,2CAA2C;IAC3C;QACE,UAAU,EAAE,OAAO;QACnB,YAAY,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC;QAChD,QAAQ,EAAE,YAAY,CAAC,eAAe;KACvC;IAED,2CAA2C;IAC3C,uBAAuB;IACvB,2CAA2C;IAC3C;QACE,UAAU,EAAE,QAAQ;QACpB,YAAY,EAAE,CAAC,gBAAgB,CAAC;QAChC,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,UAAU,EAAE,IAAI;KACjB;IACD;QACE,UAAU,EAAE,UAAU;QACtB,YAAY,EAAE,CAAC,UAAU,CAAC;QAC1B,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,UAAU,EAAE,IAAI;KACjB;IACD;QACE,UAAU,EAAE,QAAQ;QACpB,YAAY,EAAE,CAAC,UAAU,CAAC;QAC1B,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,UAAU,EAAE,IAAI;KACjB;IACD;QACE,UAAU,EAAE,WAAW;QACvB,mEAAmE;QACnE,yEAAyE;QACzE,4EAA4E;QAC5E,YAAY,EAAE,CAAC,SAAS,CAAC;QACzB,QAAQ,EAAE,YAAY,CAAC,YAAY;QACnC,UAAU,EAAE,IAAI;KACjB;IAED,2CAA2C;IAC3C,iBAAiB;IACjB,2CAA2C;IAC3C;QACE,UAAU,EAAE,QAAQ;QACpB,YAAY,EAAE,CAAC,YAAY,CAAC;QAC5B,QAAQ,EAAE,YAAY,CAAC,aAAa;KACrC;IACD;QACE,UAAU,EAAE,SAAS;QACrB,YAAY,EAAE,CAAC,YAAY,CAAC;QAC5B,QAAQ,EAAE,YAAY,CAAC,aAAa;KACrC;IACD;QACE,UAAU,EAAE,UAAU;QACtB,YAAY,EAAE,CAAC,UAAU,CAAC;QAC1B,QAAQ,EAAE,YAAY,CAAC,aAAa;KACrC;IACD;QACE,UAAU,EAAE,UAAU;QACtB,YAAY,EAAE,CAAC,UAAU,EAAE,YAAY,CAAC;QACxC,QAAQ,EAAE,YAAY,CAAC,aAAa;KACrC;IACD;QACE,UAAU,EAAE,QAAQ;QACpB,YAAY,EAAE,CAAC,YAAY,CAAC;QAC5B,QAAQ,EAAE,YAAY,CAAC,aAAa;KACrC;IAED,2CAA2C;IAC3C,yCAAyC;IACzC,2CAA2C;IAC3C;QACE,UAAU,EAAE,YAAY;QACxB,YAAY,EAAE,CAAC,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,kBAAkB,CAAC;QACvE,QAAQ,EAAE,YAAY,CAAC,WAAW;KACnC;CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,OAAO,CAAC,CAAC,CACpF,CAAC;AAEF;;;GAGG;AACH,MAAM,iBAAiB,GAAG,uBAAuB;KAC9C,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;KAChD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;AAEvC;;;GAGG;AACH,MAAM,eAAe,GAAG,IAAI,GAAG,EAA0C,CAAC;AAC1E,KAAK,MAAM,OAAO,IAAI,uBAAuB,EAAE,CAAC;IAC9C,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,EAAE,CAAC;YACb,eAAe,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,0BAA0B,CAAC,UAAkB;IAC3D,MAAM,cAAc,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;IAEhD,0BAA0B;IAC1B,MAAM,UAAU,GAAG,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACvD,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,kDAAkD;IAClD,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;QACzC,IAAI,WAAW,CAAC,cAAc,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC1C,OAAO,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;GAGG;AACH,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAmB,CAAC;AAEpF;;;;;;GAMG;AACH,SAAS,mBAAmB,CAAC,YAAoB;IAC/C,MAAM,cAAc,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IAElD,KAAK,MAAM,aAAa,IAAI,eAAe,EAAE,CAAC;QAC5C,IAAI,cAAc,KAAK,aAAa,IAAI,WAAW,CAAC,cAAc,EAAE,aAAa,CAAC,EAAE,CAAC;YACnF,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CAAC,YAAoB,EAAE,UAA0B;IAClF,MAAM,YAAY,GAAG,mBAAmB,CAAC,YAAY,CAAC,CAAC;IACvD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,4BAA4B,CAAC,YAAY,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;AACzE,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe,CAAC,YAAoB,EAAE,YAA0B;IAC9E,OAAO,mBAAmB,CAAC,YAAY,CAAC,KAAK,YAAY,CAAC;AAC5D,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,iBAAiB,CAC/B,UAAkB,EAClB,YAAqB;IAErB,MAAM,OAAO,GAAG,0BAA0B,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,6CAA6C;QAC7C,OAAO,YAAY,CAAC,aAAa,CAAC;IACpC,CAAC;IAED,sEAAsE;IACtE,IAAI,YAAY,IAAI,OAAO,CAAC,gBAAgB,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,eAAe,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAC;QACxG,OAAO,OAAO,CAAC,QAAQ,GAAG,uBAAuB,CAAC;IACpD,CAAC;IAED,OAAO,OAAO,CAAC,QAAQ,CAAC;AAC1B,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,kBAAkB,CAChC,UAAkB,EAClB,UAA0B,EAC1B,YAAoB;IAEpB,MAAM,OAAO,GAAG,0BAA0B,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO,KAAK,CAAC;IAE3B,2CAA2C;IAC3C,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC;QAAE,OAAO,KAAK,CAAC;IAE7D,4FAA4F;IAC5F,IAAI,mBAAmB,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,CAAC;QAClD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,iDAAiD;IACjD,IAAI,OAAO,CAAC,gBAAgB,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,eAAe,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAC;QACxF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,2FAA2F;IAC3F,MAAM,oBAAoB,GAAG,uBAAuB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IAC/E,MAAM,SAAS,GAAG,oBAAoB,CAAC,CAAC,CAAC,CAAC;IAC1C,IAAI,SAAS,IAAI,SAAS,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,UAAU,CAAC,WAAW,EAAE,EAAE,CAAC;QACjF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,uBAAuB,CACrC,UAA0B,EAC1B,YAAqB;IAErB,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;QACrB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,oDAAoD;IACpD,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAChC,iBAAiB,CAAC,CAAC,CAAC,UAAU,EAAE,YAAY,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,UAAU,EAAE,YAAY,CAAC,CAC9F,CAAC;AACJ,CAAC"}

package/dist/ui/hooks/use-input-handler.js

@@ -9,6 +9,7 @@ import { getSettingsManager } from "../../utils/settings-manager.js";
 import { ProjectAnalyzer } from "../../utils/project-analyzer.js";
 import { InstructionGenerator } from "../../utils/instruction-generator.js";
 import { getUsageTracker } from "../../utils/usage-tracker.js";
+import { getActiveProvider } from "../../provider/config.js";
 import { getHistoryManager } from "../../utils/history-manager.js";
 import { handleRewindCommand, handleCheckpointsCommand, handleCheckpointCleanCommand } from "../../commands/rewind.js";
 import { handlePlansCommand, handlePlanCommand, handlePhasesCommand, handlePauseCommand, handleResumeCommand, handleSkipPhaseCommand, handleAbandonCommand, handleResumableCommand, } from "../../commands/plan.js";
@@ -550,7 +551,7 @@ onLargePaste, onPasteTruncated, onKeyboardHelp, onMcpDashboardToggle, onThinking
         if (trimmedInput === "/retry") {
             // Find the last user message index and re-send it
             // Use findLastIndex instead of reverse().find() + lastIndexOf() to avoid object reference issues
-            const lastUserIndex = chatHistory.findLastIndex(entry => entry.type === "user");
+            const lastUserIndex = chatHistory.findLastIndex((entry) => entry.type === "user");
             if (lastUserIndex >= 0 && chatHistory[lastUserIndex]?.content) {
                 // Store the message content and history state before clearing
                 const messageToRetry = chatHistory[lastUserIndex].content;
@@ -562,9 +563,13 @@ onLargePaste, onPasteTruncated, onKeyboardHelp, onMcpDashboardToggle, onThinking
                 // Track timeout for cleanup on unmount
                 retryTimeoutRef.current = setTimeout(() => {
                     retryTimeoutRef.current = null;
-                    // Call async function and handle promise rejection
-                    handleInputSubmit(messageToRetry).catch(() => {
-                        // Restore history if retry fails
+                    // BUG FIX: Properly handle async errors with explicit Promise chain
+                    // The catch must be attached immediately to prevent unhandled rejection
+                    void handleInputSubmit(messageToRetry).catch((error) => {
+                        // Log error for debugging, then restore history
+                        if (process.env.DEBUG || process.env.AX_DEBUG) {
+                            console.error('Retry failed:', error);
+                        }
                         setChatHistory(historyBackup);
                     });
                 }, 50);
@@ -798,9 +803,14 @@ Examples:
         if (trimmedInput === "/usage") {
             const tracker = getUsageTracker();
             const stats = tracker.getSessionStats();
-            let usageContent = "📊 **API Usage & Limits (Z.AI)**\n\n";
+            const provider = getActiveProvider();
+            const isGrok = provider.name === 'grok';
+            const currentModel = getSettingsManager().getCurrentModel() || provider.defaultModel;
+            const providerName = isGrok ? 'xAI (Grok)' : 'Z.AI (GLM)';
+            let usageContent = `📊 **API Usage & Limits (${providerName})**\n\n`;
             // Session statistics
             usageContent += "**📱 Current Session:**\n";
+            usageContent += `  • Model: ${currentModel}\n`;
             if (stats.totalRequests === 0) {
                 usageContent += "  No API requests made yet. Ask me something to start tracking!\n";
             }
@@ -819,29 +829,87 @@ Examples:
                     }
                 }
             }
-            // Z.AI account information
-            usageContent += `\n**🔑 Z.AI Account Usage & Limits:**\n`;
-            usageContent += `  ⚠️  API does not provide programmatic access to usage data\n`;
-            usageContent += `\n  **Check your account:**\n`;
-            usageContent += `  • Billing & Usage: https://z.ai/manage-apikey/billing\n`;
-            usageContent += `  • Rate Limits: https://z.ai/manage-apikey/rate-limits\n`;
-            usageContent += `  • API Keys: https://z.ai/manage-apikey/apikey-list\n`;
-            usageContent += `\n**ℹ️  Notes:**\n`;
-            usageContent += `  • Billing reflects previous day (n-1) consumption\n`;
-            usageContent += `  • Current day usage may not be immediately visible\n`;
-            usageContent += `  • Cached content: 1/5 of original price\n`;
-            usageContent += `\n**💰 GLM-4.6 Pricing:**\n`;
-            usageContent += `  • Input: $0.11 per 1M tokens\n`;
-            usageContent += `  • Output: $0.28 per 1M tokens\n`;
-            if (stats.totalRequests > 0) {
-                // Calculate estimated cost for this session
-                const inputCost = (stats.totalPromptTokens / 1000000) * 0.11;
-                const outputCost = (stats.totalCompletionTokens / 1000000) * 0.28;
-                const totalCost = inputCost + outputCost;
-                usageContent += `\n**💵 Estimated Session Cost:**\n`;
-                usageContent += `  • Input: $${inputCost.toFixed(6)} (${stats.totalPromptTokens.toLocaleString()} tokens)\n`;
-                usageContent += `  • Output: $${outputCost.toFixed(6)} (${stats.totalCompletionTokens.toLocaleString()} tokens)\n`;
-                usageContent += `  • **Total: ~$${totalCost.toFixed(6)}**\n`;
+            if (isGrok) {
+                // xAI/Grok account information
+                usageContent += `\n**🔑 xAI Account Usage & Limits:**\n`;
+                usageContent += `  ⚠️  API does not provide programmatic access to usage data\n`;
+                usageContent += `\n  **Check your account:**\n`;
+                usageContent += `  • Usage Explorer: https://console.x.ai\n`;
+                usageContent += `  • Billing & Team: https://console.x.ai/team\n`;
+                usageContent += `  • API Keys: https://console.x.ai/api-keys\n`;
+                usageContent += `\n**ℹ️  Notes:**\n`;
+                usageContent += `  • Usage is tracked in real-time on the xAI console\n`;
+                usageContent += `  • Cached input tokens: 75% discount\n`;
+                // Grok pricing based on model
+                const modelLower = currentModel.toLowerCase();
+                if (modelLower.includes('grok-4.1-fast')) {
+                    usageContent += `\n**💰 Grok 4.1 Fast Pricing:**\n`;
+                    usageContent += `  • Input: $0.20 per 1M tokens\n`;
+                    usageContent += `  • Output: $0.50 per 1M tokens\n`;
+                }
+                else if (modelLower.includes('grok-4')) {
+                    usageContent += `\n**💰 Grok 4 Pricing:**\n`;
+                    usageContent += `  • Input: $3.00 per 1M tokens\n`;
+                    usageContent += `  • Output: $15.00 per 1M tokens\n`;
+                    usageContent += `  • Cached: $0.75 per 1M tokens\n`;
+                }
+                else if (modelLower.includes('grok-3-mini')) {
+                    usageContent += `\n**💰 Grok 3 Mini Pricing:**\n`;
+                    usageContent += `  • Input: $0.30 per 1M tokens\n`;
+                    usageContent += `  • Output: $0.50 per 1M tokens\n`;
+                }
+                else {
+                    usageContent += `\n**💰 Grok 3 Pricing:**\n`;
+                    usageContent += `  • Input: $3.00 per 1M tokens\n`;
+                    usageContent += `  • Output: $15.00 per 1M tokens\n`;
+                    usageContent += `  • Cached: $0.75 per 1M tokens\n`;
+                }
+                if (stats.totalRequests > 0) {
+                    // Calculate estimated cost based on model
+                    let inputRate = 3.0, outputRate = 15.0;
+                    if (modelLower.includes('grok-4.1-fast')) {
+                        inputRate = 0.20;
+                        outputRate = 0.50;
+                    }
+                    else if (modelLower.includes('grok-3-mini')) {
+                        inputRate = 0.30;
+                        outputRate = 0.50;
+                    }
+                    const inputCost = (stats.totalPromptTokens / 1000000) * inputRate;
+                    const outputCost = (stats.totalCompletionTokens / 1000000) * outputRate;
+                    const totalCost = inputCost + outputCost;
+                    usageContent += `\n**💵 Estimated Session Cost:**\n`;
+                    usageContent += `  • Input: $${inputCost.toFixed(6)} (${stats.totalPromptTokens.toLocaleString()} tokens)\n`;
+                    usageContent += `  • Output: $${outputCost.toFixed(6)} (${stats.totalCompletionTokens.toLocaleString()} tokens)\n`;
+                    usageContent += `  • **Total: ~$${totalCost.toFixed(6)}**\n`;
+                }
+            }
+            else {
+                // Z.AI/GLM account information
+                usageContent += `\n**🔑 Z.AI Account Usage & Limits:**\n`;
+                usageContent += `  ⚠️  API does not provide programmatic access to usage data\n`;
+                usageContent += `\n  **Check your account:**\n`;
+                usageContent += `  • Billing & Usage: https://z.ai/manage-apikey/billing\n`;
+                usageContent += `  • Rate Limits: https://z.ai/manage-apikey/rate-limits\n`;
+                usageContent += `  • API Keys: https://z.ai/manage-apikey/apikey-list\n`;
+                usageContent += `\n**ℹ️  Notes:**\n`;
+                usageContent += `  • Billing reflects previous day (n-1) consumption\n`;
+                usageContent += `  • Current day usage may not be immediately visible\n`;
+                usageContent += `  • Cached content: 1/5 of original price\n`;
+                usageContent += `\n**💰 GLM-4.6 Pricing:**\n`;
+                usageContent += `  • Input: $2.00 per 1M tokens\n`;
+                usageContent += `  • Output: $10.00 per 1M tokens\n`;
+                usageContent += `  • Cached: $0.50 per 1M tokens\n`;
+                if (stats.totalRequests > 0) {
+                    // Calculate estimated cost for this session
+                    const inputCost = (stats.totalPromptTokens / 1000000) * 2.0;
+                    const outputCost = (stats.totalCompletionTokens / 1000000) * 10.0;
+                    const totalCost = inputCost + outputCost;
+                    usageContent += `\n**💵 Estimated Session Cost:**\n`;
+                    usageContent += `  • Input: $${inputCost.toFixed(6)} (${stats.totalPromptTokens.toLocaleString()} tokens)\n`;
+                    usageContent += `  • Output: $${outputCost.toFixed(6)} (${stats.totalCompletionTokens.toLocaleString()} tokens)\n`;
+                    usageContent += `  • **Total: ~$${totalCost.toFixed(6)}**\n`;
+                }
             }
             const usageEntry = {
                 type: "assistant",
@@ -862,7 +930,7 @@ Examples:
             };
             setChatHistory((prev) => [...prev, doctorEntry]);
             // Execute doctor command asynchronously (non-blocking)
-            void (async () => {
+            (async () => {
                 try {
                     const { exec } = await import("child_process");
                     const { promisify } = await import("util");
@@ -1033,7 +1101,7 @@ Examples:
                 };
                 setChatHistory((prev) => [...prev, userEntry]);
                 // Execute the prompt asynchronously
-                void (async () => {
+                (async () => {
                     try {
                         const manager = getMCPManager();
                         const v2 = manager.getV2Instance();
@@ -1871,7 +1939,8 @@ Respond with ONLY the commit message, no additional text.`;
             "mv",
             "rm",
         ];
-        const firstWord = trimmedInput.split(" ")[0];
+        // BUG FIX: Added fallback for empty input to prevent undefined access
+        const firstWord = trimmedInput.split(" ")[0] || "";
         if (directBashCommands.includes(firstWord)) {
             const userEntry = {
                 type: "user",
@@ -1942,7 +2011,7 @@ Respond with ONLY the commit message, no additional text.`;
                 const imageInfo = imageResult.images.map((img, i) => formatAttachmentForDisplay(img, i + 1)).join('\n');
                 const infoEntry = {
                     type: "assistant",
-                    content: `📷 Processing ${imageResult.images.length} image(s):\n${imageInfo}\n\n*Using glm-4.5v for vision analysis*`,
+                    content: `📷 Processing ${imageResult.images.length} image(s):\n${imageInfo}\n\n*Auto-switching to vision model for analysis*`,
                     timestamp: new Date(),
                 };
                 setChatHistory((prev) => [...prev, infoEntry]);