wso2 · choreo-cicd · Feb 18, 2026 · Feb 23, 2026 · Feb 24, 2026 · Feb 24, 2026
@@ -61,6 +61,9 @@ module.exports = {
         if (pkg.dependencies['lodash']) {
           pkg.dependencies['lodash'] = '4.17.23';
         }
+        if (pkg.dependencies['bn.js']) {
+          pkg.dependencies['bn.js'] = '5.2.3';
+        }
       }
 
       if (pkg.devDependencies) {
@@ -109,6 +112,9 @@ module.exports = {
         if (pkg.devDependencies['lodash']) {
           pkg.devDependencies['lodash'] = '4.17.23';
         }
+        if (pkg.devDependencies['bn.js']) {
+          pkg.devDependencies['bn.js'] = '5.2.3';
+        }
       }
 
       return pkg;

@@ -6,6 +6,7 @@
         "lib": [
             "ES2020"
         ],
+        "skipLibCheck": true,
         "sourceMap": true,
         "rootDir": "src",
         "strict": true /* enable all strict type-checking options */

@@ -44,6 +44,7 @@ import {
     UIChatMessage,
     CheckpointInfo,
     AbortAIGenerationRequest,
+    UsageResponse,
 } from "./interfaces";
 
 export interface AIPanelAPI {
@@ -107,4 +108,5 @@ export interface AIPanelAPI {
     clearChat: () => Promise<void>;
     updateChatMessage: (params: UpdateChatMessageRequest) => Promise<void>;
     getActiveTempDir: () => Promise<string>;
+    getUsage: () => Promise<UsageResponse | undefined>;
 }
@@ -486,3 +486,8 @@ export interface AbortAIGenerationRequest {
     /** Thread identifier (defaults to 'default') */
     threadId?: string;
 }
+
+export interface UsageResponse {
+    remainingUsagePercentage: number;
+    resetsIn: number; // in seconds
+}
@@ -46,6 +46,7 @@ import {
     UIChatMessage,
     CheckpointInfo,
     AbortAIGenerationRequest,
+    UsageResponse,
 } from "./interfaces";
 import { RequestType, NotificationType } from "vscode-messenger-common";
 
@@ -96,3 +97,4 @@ export const restoreCheckpoint: RequestType<RestoreCheckpointRequest, void> = {
 export const clearChat: RequestType<void, void> = { method: `${_preFix}/clearChat` };
 export const updateChatMessage: RequestType<UpdateChatMessageRequest, void> = { method: `${_preFix}/updateChatMessage` };
 export const getActiveTempDir: RequestType<void, string> = { method: `${_preFix}/getActiveTempDir` };
+export const getUsage: RequestType<void, UsageResponse | undefined> = { method: `${_preFix}/getUsage` };
@@ -40,6 +40,7 @@ import {
     outputChannel,
     isWindows,
     isWSL,
+    quoteShellPath,
     isSupportedVersion,
     VERSION,
     isSupportedSLVersion,
@@ -1819,7 +1820,8 @@ export class BallerinaExtension {
             debug("[VERSION] Non-Windows platform detected, no extension needed");
         }
 
-        let ballerinaCommand = distPath + 'bal' + exeExtension + ' version';
+        // Build the executable path separately so we can quote it for shell execution
+        let balExecutablePath = distPath + 'bal' + exeExtension;
 
         // Handle WSL environment - prefer native Linux installation over Windows .bat files
         if (isWSL()) {
@@ -1829,26 +1831,28 @@ export class BallerinaExtension {
                     // Check if 'bal' command is available in PATH
                     execSync('which bal', { encoding: 'utf8', timeout: 5000 });
                     // If we get here, 'bal' is available, use it instead of .bat
-                    ballerinaCommand = 'bal version';
+                    balExecutablePath = 'bal';
                     debug("[VERSION] WSL detected native 'bal' command, using it instead of .bat file");
                 } catch (error) {
                     debug("[VERSION] No native 'bal' command found in WSL, will try .bat file");
                     // If the path contains Windows-style paths, we need to handle them properly
-                    if (ballerinaCommand.includes('\\') || ballerinaCommand.match(/^[A-Za-z]:/)) {
+                    if (balExecutablePath.includes('\\') || balExecutablePath.match(/^[A-Za-z]:/)) {
                         debug("[VERSION] WSL detected with Windows path, attempting to convert to WSL path");
                         // Try to convert Windows path to WSL path
-                        const wslPath = ballerinaCommand.replace(/^([A-Za-z]):/, '/mnt/$1').replace(/\\/g, '/').toLowerCase();
-                        debug(`[VERSION] Converted Windows path to WSL path: ${wslPath}`);
-                        ballerinaCommand = wslPath;
+                        balExecutablePath = balExecutablePath.replace(/^([A-Za-z]):/, '/mnt/$1').replace(/\\/g, '/').toLowerCase();
+                        debug(`[VERSION] Converted Windows path to WSL path: ${balExecutablePath}`);
                     }
                 }
             } else {
                 // We have a native Linux installation, use it directly
-                ballerinaCommand = 'bal version';
+                balExecutablePath = 'bal';
                 debug("[VERSION] WSL detected with native Linux installation, using 'bal version'");
             }
         }
 
+        // Quote the executable path to handle spaces in directory names
+        let ballerinaCommand = `${quoteShellPath(balExecutablePath)} version`;
+
         debug(`[VERSION] Executing command: '${ballerinaCommand}'`);
 
         let ballerinaExecutor = '';

@@ -4,6 +4,8 @@ import { StateMachine } from '../../../../stateMachine';
 import * as path from 'path';
 import { Uri } from 'vscode';
 
+export const DIAGNOSTICS_TOOL_NAME = "getCompilationErrors";
+
 /**
  * Diagnostic entry enriched with resolving hints
  */
@@ -128,6 +130,24 @@ export async function checkCompilationErrors(
             };
         }
 
+        // HACK: When the generated code includes `import ballerinax/client.config;` (without the quoted
+        // identifier), the language server returns diagnostics with the module name stripped to
+        // `ballerinax/.config as config` — omitting "client". As a workaround, we detect this and
+        // instruct the agent to use the correct quoted form `import ballerinax/'client.config;`
+        // instead of attempting to resolve the dependency automatically.
+        const hasInvalidClientModuleImport = enrichedDiagnostics.some(
+            d => d.code === "BCE2003" && d.message.includes("ballerinax/.config as config")
+        );
+        if (hasInvalidClientModuleImport) {
+            console.log(`[DiagnosticsUtils] Detected invalid client module import 'ballerinax/client.config'.`);
+            return {
+                diagnostics: enrichedDiagnostics,
+                message: `Found a module resolution error: the import 'import ballerinax/client.config;' is invalid. ` +
+                    `Fix this by replacing the import statement with 'import ballerinax/'client.config;'. ` +
+                    `After applying the fix, call the ${DIAGNOSTICS_TOOL_NAME} tool again to verify there are no remaining errors.`
+            };
+        }
+
         console.log(`[DiagnosticsUtils] Enriched Diagnostics:`, enrichedDiagnostics);
         return {
             diagnostics: enrichedDiagnostics,

@@ -19,6 +19,7 @@ import { z } from 'zod';
 import child_process from 'child_process';
 import { CopilotEventHandler } from '../../utils/events';
 import { extension } from '../../../../BalExtensionContext';
+import { quoteShellPath } from '../../../../utils/config';
 import { DIAGNOSTICS_TOOL_NAME } from './diagnostics';
 
 export const TEST_RUNNER_TOOL_NAME = "runTests";
@@ -99,7 +100,7 @@ function parseTestSummary(output: string): string {
 async function runBallerinaTests(cwd: string): Promise<TestRunResult> {
     return new Promise((resolve) => {
         const balCmd = extension.ballerinaExtInstance.getBallerinaCmd();
-        const command = `${balCmd} test`;
+        const command = `${quoteShellPath(balCmd)} test`;
 
         console.log(`[TestRunner] Running: ${command} in ${cwd}`);
 

@@ -134,7 +134,7 @@ export async function fetchWithAuth(input: string | URL | Request, options: Requ
         // Handle usage limit exceeded
         if (response.status === 429) {
             console.log("Usage limit exceeded (429)");
-            const error = new Error("Usage limit exceeded. Please try again later.");
+            const error = new Error("Usage limit exceeded.");
             error.name = "UsageLimitError";
             (error as any).statusCode = 429;
             throw error;

@@ -313,7 +313,7 @@ export function getErrorMessage(error: unknown): string {
     if (error instanceof Error) {
         // Standard Error objects have a .message property
         if (error.name === "UsageLimitError") {
-            return "Usage limit exceeded. Please try again later.";
+            return "Usage limit exceeded.";
         }
         if (error.name === "AI_RetryError") {
             return "An error occured connecting with the AI service. Please try again later.";
@@ -333,7 +333,7 @@ export function getErrorMessage(error: unknown): string {
     ) {
         // Check if it has a statusCode property indicating 429
         if ("statusCode" in error && (error as any).statusCode === 429) {
-            return "Usage limit exceeded. Please try again later.";
+            return "Usage limit exceeded.";
         }
         return (error as { message: string }).message;
     }

@@ -48,6 +48,7 @@ import {
     createVersionNumber
 } from "../../utils";
 import { getProjectWorkingDirectory } from "../../utils/file-utils";
+import { quoteShellPath } from "../../utils/config";
 import { decimal, ExecutableOptions } from 'vscode-languageclient/node';
 import { BAL_NOTEBOOK, getTempFile, NOTEBOOK_CELL_SCHEME } from '../../views/notebook';
 import fileUriToPath from 'file-uri-to-path';
@@ -603,7 +604,8 @@ class BallerinaDebugAdapterDescriptorFactory implements DebugAdapterDescriptorFa
     }
     getScriptPath(args: string[]): string {
         args.push('start-debugger-adapter');
-        return extension.ballerinaExtInstance.getBallerinaCmd();
+        // Quote the path to handle spaces in directory names (used with shell: true)
+        return quoteShellPath(extension.ballerinaExtInstance.getBallerinaCmd());
     }
     getCurrentWorkingDir(): string {
         return path.join(extension.ballerinaExtInstance.ballerinaHome, "bin");
@@ -662,7 +664,7 @@ class BIRunAdapter extends LoggingDebugSession {
                 task: 'run'
             };
 
-            let runCommand: string = `${extension.ballerinaExtInstance.getBallerinaCmd()} run`;
+            let runCommand: string = `${quoteShellPath(extension.ballerinaExtInstance.getBallerinaCmd())} run`;
 
             const programArgs = (args as any).programArgs;
             if (programArgs && programArgs.length > 0) {

@@ -18,7 +18,7 @@
 
 import { BallerinaProject } from "@wso2/ballerina-core";
 import { Terminal, window, workspace } from "vscode";
-import { isSupportedSLVersion, isWindows, createVersionNumber } from "../../../utils";
+import { isSupportedSLVersion, isWindows, createVersionNumber, quoteShellPath } from "../../../utils";
 import { extension } from "../../../BalExtensionContext";
 import { TracerMachine } from "../../../features/tracing";
 
@@ -128,7 +128,7 @@ export function runCommandWithConf(file: BallerinaProject | string, executor: st
     }
     let commandText;
     if (cmd === BALLERINA_COMMANDS.OTHER) {
-        commandText = `${executor} ${argsList}`;
+        commandText = `${quoteShellPath(executor)} ${argsList}`;
         terminal = window.createTerminal({ name: TERMINAL_NAME });
     } else {
         let env = {};
@@ -164,7 +164,7 @@ export function runCommandWithConf(file: BallerinaProject | string, executor: st
             }
         }
 
-        commandText = `${executor} ${cmd} ${argsList}`;
+        commandText = `${quoteShellPath(executor)} ${cmd} ${argsList}`;
         if (confPath !== '') {
             const configs = env['BAL_CONFIG_FILES'] ? `${env['BAL_CONFIG_FILES']}:${confPath}` : confPath;
             Object.assign(env, { BAL_CONFIG_FILES: configs });

@@ -34,6 +34,7 @@ import { BALLERINA_COMMANDS } from "../project";
 import { discoverTests, gatherTestItems } from "./discover";
 import { testController, projectRoot } from "./activator";
 import { extension } from "../../BalExtensionContext";
+import { quoteShellPath } from "../../utils/config";
 
 enum EXEC_ARG {
     TESTS = '--tests',
@@ -72,7 +73,7 @@ export function runHandler(request: TestRunRequest, cancellation: CancellationTo
             try {
                 // execute test
                 const executor = extension.ballerinaExtInstance.getBallerinaCmd();
-                const commandText = `${executor} ${BALLERINA_COMMANDS.TEST} ${EXEC_ARG.TESTS} ${testNames} ${EXEC_ARG.COVERAGE}`;
+                const commandText = `${quoteShellPath(executor)} ${BALLERINA_COMMANDS.TEST} ${EXEC_ARG.TESTS} ${testNames} ${EXEC_ARG.COVERAGE}`;
                 await runCommand(commandText, projectRoot);
 
             } catch {

@@ -91,7 +91,12 @@ export async function isModuleNotFoundDiagsExist(diagnosticsResult: Diagnostics[
 
     // Process each unique diagnostic only once
     let projectModified = false;
-    for (const [_, { uri }] of uniqueDiagnosticMap.entries()) {
+    for (const [message, { uri }] of uniqueDiagnosticMap.entries()) {
+        // Skip resolving dependencies for the invalid config import pattern
+        if (message.includes("ballerinax/.config as config")) {
+            continue;
+        }
+
         const dependenciesResponse = await langClient.resolveModuleDependencies({
             documentIdentifier: {
                 uri: uri

@@ -84,7 +84,8 @@ import {
     TaskDeclineRequest,
     updateChatMessage,
     UpdateChatMessageRequest,
-    updateRequirementSpecification
+    updateRequirementSpecification,
+    getUsage
 } from "@wso2/ballerina-core";
 import { Messenger } from "vscode-messenger";
 import { AiPanelRpcManager } from "./rpc-manager";
@@ -137,4 +138,5 @@ export function registerAiPanelRpcHandlers(messenger: Messenger) {
     messenger.onRequest(clearChat, () => rpcManger.clearChat());
     messenger.onRequest(updateChatMessage, (args: UpdateChatMessageRequest) => rpcManger.updateChatMessage(args));
     messenger.onRequest(getActiveTempDir, () => rpcManger.getActiveTempDir());
+    messenger.onRequest(getUsage, () => rpcManger.getUsage());
 }
@@ -41,7 +41,8 @@ import {
     SubmitFeedbackRequest,
     TestGenerationMentions,
     UIChatMessage,
-    UpdateChatMessageRequest
+    UpdateChatMessageRequest,
+    UsageResponse
 } from "@wso2/ballerina-core";
 import * as fs from 'fs';
 import path from "path";
@@ -65,12 +66,14 @@ import { refreshDataMapper } from "../data-mapper/utils";
 import {
     TEST_DIR_NAME
 } from "./constants";
+import { fetchWithAuth } from "../../features/ai/utils/ai-client";
+import { BACKEND_URL } from "../../features/ai/utils";
 import { addToIntegration, cleanDiagnosticMessages, searchDocumentation } from "./utils";
 
 import { onHideReviewActions } from '@wso2/ballerina-core';
 import { createExecutionContextFromStateMachine, createExecutorConfig, generateAgent } from '../../features/ai/agent/index';
 import { integrateCodeToWorkspace } from "../../features/ai/agent/utils";
-import { WI_EXTENSION_ID } from "../../features/ai/constants";
+import { LLM_API_BASE_PATH, WI_EXTENSION_ID } from "../../features/ai/constants";
 import { ContextTypesExecutor } from '../../features/ai/executors/datamapper/ContextTypesExecutor';
 import { FunctionMappingExecutor } from '../../features/ai/executors/datamapper/FunctionMappingExecutor';
 import { InlineMappingExecutor } from '../../features/ai/executors/datamapper/InlineMappingExecutor';
@@ -699,4 +702,24 @@ export class AiPanelRpcManager implements AIPanelAPI {
         console.log(">>> active temp project path", projectPath);
         return projectPath;
     }
+
+    async getUsage(): Promise<UsageResponse | undefined> {
+        const loginMethod = await getLoginMethod();
+        if (loginMethod !== LoginMethod.BI_INTEL) {
+            return undefined;
+        }
+        try {
+            const url = BACKEND_URL + LLM_API_BASE_PATH + "/usage";
+            const response = await fetchWithAuth(url, { method: "GET" });
+            if (response && response.ok) {
+                const data = await response.json();
+                return data as UsageResponse;
+            }
+            console.error("Failed to fetch usage: ", response?.status, response?.statusText);
+            return undefined;
+        } catch (error) {
+            console.error("Failed to fetch usage:", error);
+            return undefined;
+        }
+    }
 }
@@ -73,6 +73,27 @@ export function isWSL(): boolean {
     );
 }
 
+/**
+ * Wraps a file path in double quotes if it contains spaces,
+ * so it can be safely used in shell command strings.
+ * Handles already-quoted paths and escapes embedded double quotes.
+ */
+export function quoteShellPath(filePath: string): string {
+    // Strip existing surrounding quotes to normalize
+    let normalized = filePath;
+    if (normalized.length >= 2 && normalized.startsWith('"') && normalized.endsWith('"')) {
+        normalized = normalized.slice(1, -1);
+    }
+
+    if (!normalized.includes(' ')) {
+        return normalized;
+    }
+
+    // Escape any embedded double quotes
+    const escaped = normalized.replace(/"/g, '\\"');
+    return `"${escaped}"`;
+}
+
 export function isSupportedVersion(ballerinaExtInstance: BallerinaExtension, supportedRelease: VERSION,
     supportedVersion: number): boolean {
     const ballerinaVersion: string = ballerinaExtInstance.ballerinaVersion.toLocaleLowerCase();

@@ -22,6 +22,7 @@ import { extension } from "../BalExtensionContext";
 import { RPCLayer } from "../RPCLayer";
 import { VisualizerWebview } from "../views/visualizer/webview";
 import { debug } from "./logger";
+import { quoteShellPath } from "./config";
 
 const PROGRESS_COMPLETE = 100;
 
@@ -53,7 +54,7 @@ export async function pullMigrationTool(migrationToolName: string, version: stri
     }
 
     const ballerinaCmd = extension.ballerinaExtInstance.getBallerinaCmd();
-    const command = `${ballerinaCmd} tool pull ${migrationToolName}:${version}`;
+    const command = `${quoteShellPath(ballerinaCmd)} tool pull ${migrationToolName}:${version}`;
     debug(`Executing migration tool pull command: ${command}`);
 
     // 2. This function now returns a promise that wraps the exec lifecycle