Remove dependency pull resolution and UI

Author: yasithrashan

workspaces/ballerina/ballerina-core/src/interfaces/extended-lang-client.ts

@@ -2029,22 +2029,15 @@ export interface ProjectArtifactsRequest {
 export interface ProjectArtifacts {
     artifacts: Artifacts;
 }
+
 export interface CodeMapRequest {
     projectPath: string;
 }
+
 export interface CodeMapResponse {
     content?: string;
 }
 
-export interface CodeMapResolveModuleDependenciesRequest {
-    projectPath: string;
-}
-export interface CodeMapResolveModuleDependenciesResponse {
-    success: boolean;
-    errorMsg?: string;
-    errorDetails?: string;
-}
-
 export interface ProjectInfoRequest {
     projectPath: string;
 }
@@ -2158,7 +2151,6 @@ export interface ExtendedLangClientInterface extends BIInterface {
     getDidOpenParams(): DidOpenParams;
     getProjectArtifacts(params: ProjectArtifactsRequest): Promise<ProjectArtifacts>;
     getCodeMap(params: CodeMapRequest): Promise<CodeMapResponse>;
-    codeMapResolveModuleDependencies(params: CodeMapResolveModuleDependenciesRequest): Promise<CodeMapResolveModuleDependenciesResponse>;
     getProjectInfo(params: ProjectInfoRequest): Promise<ProjectInfo>;
     getSimpleTypeOfExpression(params: GetSimpleTypeOfExpressionRequest): Promise<GetSimpleTypeOfExpressionResponse>;
     openConfigToml(params: OpenConfigTomlRequest): Promise<void>;

workspaces/ballerina/ballerina-extension/src/core/extended-language-client.ts

@@ -295,8 +295,6 @@ import {
     GetSimpleTypeOfExpressionRequest,
     AIGetPackageVersionRequest,
     AIGetPackageVersionResponse,
-    CodeMapResolveModuleDependenciesRequest,
-    CodeMapResolveModuleDependenciesResponse
 } from "@wso2/ballerina-core";
 import { BallerinaExtension } from "./index";
 import { debug, handlePullModuleProgress } from "../utils";
@@ -496,7 +494,6 @@ enum EXTENDED_APIS {
     COPILOT_FILTER_LIBRARIES = 'copilotLibraryManager/getFilteredLibraries',
     COPILOT_SEARCH_LIBRARIES = 'copilotLibraryManager/getLibrariesBySearch',
     COPILOT_GET_CODE_MAP = 'designModelService/codeMap',
-    COPILOT_CODE_MAP_RESOLVE_MODULE_DEPENDENCIES = 'designModelService/codeMapResolveModuleDependencies',
     GET_MIGRATION_TOOLS = 'projectService/getMigrationTools',
     TIBCO_TO_BI = 'projectService/importTibco',
     MULE_TO_BI = 'projectService/importMule',
@@ -1512,10 +1509,6 @@ export class ExtendedLangClient extends LanguageClient implements ExtendedLangCl
         return this.sendRequest<CodeMapResponse>(EXTENDED_APIS.COPILOT_GET_CODE_MAP, params);
     }
 
-    async codeMapResolveModuleDependencies(params: CodeMapResolveModuleDependenciesRequest): Promise<CodeMapResolveModuleDependenciesResponse> {
-        return this.sendRequest<CodeMapResolveModuleDependenciesResponse>(EXTENDED_APIS.COPILOT_CODE_MAP_RESOLVE_MODULE_DEPENDENCIES, params);
-    }
-
     // <------------ BI APIS END --------------->
 
 

workspaces/ballerina/ballerina-extension/src/features/ai/agent/index.ts

@@ -19,7 +19,6 @@ import { StateMachine } from "../../../stateMachine";
 import { chatStateStorage } from '../../../views/ai-panel/chatStateStorage';
 import { AICommandConfig } from "../executors/base/AICommandExecutor";
 import { createWebviewEventHandler } from "../utils/events";
-import { sendDependencyPullProgressToAIPanel, clearDependencyPullProgressInAIPanel } from "../utils/ai-utils";
 import { AgentExecutor } from './AgentExecutor';
 import {
     sendTelemetryEvent,
@@ -130,18 +129,6 @@ export async function generateAgent(params: GenerateAgentCodeRequest): Promise<b
         const workspacePath = StateMachine.context().workspacePath || projectRootPath;
         if (langClient) {
             try {
-                try {
-                    sendDependencyPullProgressToAIPanel('Pulling module dependencies...');
-                    const pullResponse = await langClient.codeMapResolveModuleDependencies({
-                        projectPath: workspacePath
-                    });
-                    console.debug('[generateAgent] resolveModuleDependencies response:', JSON.stringify(pullResponse, null, 2));
-                } catch (pullErr) {
-                    console.warn('[generateAgent] Failed to resolve module dependencies, continuing:', pullErr);
-                } finally {
-                    clearDependencyPullProgressInAIPanel();
-                }
-
                 const codeMapResponse = await langClient.getCodeMap({
                     projectPath: workspacePath,
                 });

workspaces/ballerina/ballerina-extension/src/features/ai/utils/ai-utils.ts

@@ -20,7 +20,6 @@ import {
     ChatNotify,
     ChatStart,
     DiagnosticEntry,
-    dependencyPullProgress,
     IntermidaryState,
     onChatNotify,
     ProjectSource,
@@ -44,7 +43,6 @@ import { AiPanelWebview } from "../../../views/ai-panel/webview";
 import { MigrationPanelWebview } from "../../../views/migration-panel/webview";
 import { VisualizerWebview } from "../../../views/visualizer/webview";
 import { GenerationType } from "./libs/libraries";
-import { send } from "xstate";
 // import { REQUIREMENTS_DOCUMENT_KEY } from "./code/np_prompts";
 
 export function populateHistory(chatHistory: ChatEntry[]): ModelMessage[] {
@@ -236,18 +234,6 @@ export function sendIntermidateStateNotification(intermediaryState: Documentatio
     sendAIPanelNotification(msg);
 }
 
-export function sendDependencyPullProgressToAIPanel(message: string): void {
-    RPCLayer._messenger.sendNotification(
-        dependencyPullProgress,
-        { type: "webview", webviewType: AiPanelWebview.viewType },
-        message
-    );
-}
-
-export function clearDependencyPullProgressInAIPanel(): void {
-    sendDependencyPullProgressToAIPanel("");
-}
-
 export function sendToolCallNotification(toolName: string, toolInput?: any, toolCallId?: string): void {
     const msg: ToolCall = {
         type: "tool_call",

workspaces/ballerina/ballerina-visualizer/src/views/AIPanel/components/AIChat/index.tsx

@@ -55,7 +55,7 @@ import { ConfigurationCollectorSegment, ConfigurationCollectionData } from "../C
 import CheckpointSeparator from "../CheckpointSeparator";
 import { Attachment, AttachmentStatus, TaskApprovalRequest } from "@wso2/ballerina-core";
 
-import { AIChatView, Header, HeaderButtons, ChatMessage, TurnGroup, AuthProviderChip, UsageBadge, ApprovalOverlay, OverlayMessage, OverlayCloseButton, StatusBanner, StatusBannerSpinner } from "../../styles";
+import { AIChatView, Header, HeaderButtons, ChatMessage, TurnGroup, AuthProviderChip, UsageBadge, ApprovalOverlay, OverlayMessage, OverlayCloseButton } from "../../styles";
 import ReferenceDropdown from "../ReferenceDropdown";
 import { VSCodeButton } from "@vscode/webview-ui-toolkit/react";
 import MarkdownRenderer from "../MarkdownRenderer";
@@ -241,7 +241,6 @@ const AIChat: React.FC = () => {
     const [showContextUsage, setShowContextUsage] = useState(false);
 
     const [runningServices, setRunningServices] = useState<RunningServiceInfo[]>([]);
-    const [depPullStatus, setDepPullStatus] = useState<string>('');
 
     //TODO: Need a better way of storing data related to last generation to be in the repair state.
     const currentDiagnosticsRef = useRef<DiagnosticEntry[]>([]);
@@ -462,12 +461,6 @@ const AIChat: React.FC = () => {
         });
     }, [rpcClient]);
 
-    useEffect(() => {
-        rpcClient?.onDependencyPullProgress((message: string) => {
-            setDepPullStatus(message);
-        });
-    }, [rpcClient]);
-
     // Initial fetch covers services started before the webview opened;
     // the subscription keeps the list live for the rest of the session.
     useEffect(() => {
@@ -1803,12 +1796,6 @@ const AIChat: React.FC = () => {
                             </Button>
                         </HeaderButtons>
                     </Header>
-                    {depPullStatus && (
-                        <StatusBanner>
-                            <StatusBannerSpinner />
-                            <span>{depPullStatus}</span>
-                        </StatusBanner>
-                    )}
                     <main style={{ flex: 1, overflowY: "auto" }}>
                         {migrationSession && (
                             <MigrationContextCard

workspaces/ballerina/ballerina-visualizer/src/views/AIPanel/styles.tsx

@@ -14,7 +14,7 @@
  * KIND, either express or implied. See the License for the
  * specific language governing permissions and limitations
  * under the License.
- * 
+ *
  * THIS FILE INCLUDES AUTO GENERATED CODE
  */
 
@@ -119,32 +119,6 @@ export const OverlayMessage = styled.div`
     gap: 12px;
 `;
 
-export const StatusBanner = styled.div`
-    display: flex;
-    align-items: center;
-    gap: 8px;
-    padding: 6px 12px;
-    background: var(--vscode-editorWidget-background, var(--vscode-editor-background));
-    border-bottom: 1px solid var(--vscode-panel-border);
-    color: var(--vscode-foreground);
-    font-size: 12px;
-    font-family: var(--vscode-font-family);
-    flex-shrink: 0;
-`;
-
-export const StatusBannerSpinner = styled.span`
-    width: 12px;
-    height: 12px;
-    border: 2px solid var(--vscode-progressBar-background, var(--vscode-foreground));
-    border-top-color: transparent;
-    border-radius: 50%;
-    animation: status-banner-spin 0.9s linear infinite;
-    flex-shrink: 0;
-    @keyframes status-banner-spin {
-        to { transform: rotate(360deg); }
-    }
-`;
-
 export const OverlayCloseButton = styled.button`
     background: transparent;
     border: none;