feat: Removed GitHub sign in requirement for non-AI assisted usage. Users will only be prompted to sign in to GitHub if the extension needs to use GitHub API's for search.

Author: VikashLoomba

src/McpAgent.ts

@@ -6,7 +6,7 @@ import {
 	SCOPES,
 } from "./utilities/const";
 import { CopilotChatProvider } from "./utilities/CopilotChat";
-import { AxAgent, AxFunction, ax, f, AxFunctionProcessor, agent } from "@ax-llm/ax";
+import { AxAgent, AxFunction, AxAI, ax, f, AxFunctionProcessor, agent } from "@ax-llm/ax";
 import { getReadme, searchMcpServers2 } from "./utilities/repoSearch";
 import { 
 	logChatSearch, 
@@ -20,6 +20,22 @@ import {
 import { TelemetryEvents } from "./telemetry/types";
 import { outputLogger } from "./utilities/outputLogger";
 
+async function configureCopilotProvider(debug: boolean) {
+	const copilot = CopilotChatProvider.getInstance();
+	const provider = await copilot.getProvider({ interactive: true });
+	provider.setOptions({
+		debug,
+		fetch: (input: RequestInfo | URL, init?: RequestInit) => {
+			init!.headers = {
+				...init?.headers,
+				...copilot.headers,
+			};
+			return fetch(input, init);
+		},
+	});
+	return { provider, copilot };
+}
+
 const getRepoReadme: AxFunction = {
 	func: getReadme,
 	name: "getReadme",
@@ -165,19 +181,13 @@ export const handler: vscode.ChatRequestHandler = async (
 		prompt: request.prompt 
 	});
 
-	const copilot = CopilotChatProvider.getInstance();
-	const provider = copilot.provider;
-	provider.setOptions(
-		{ 
-			debug: false, 
-			fetch: (input: RequestInfo | URL, init?: RequestInit) => {
-				init!.headers = {
-					...init?.headers,
-					...copilot.headers,
-				};
-				return fetch(input, init);
-			} 
-	});
+	let provider: AxAI;
+	try {
+		({ provider } = await configureCopilotProvider(false));
+	} catch (error) {
+		outputLogger.warn("Copilot provider unavailable for chat handler", error as Error);
+		throw new Error("GitHub authentication is required to use AI-assisted chat features.");
+	}
 
 	const session = await vscode.authentication.getSession(
 		GITHUB_AUTH_PROVIDER_ID,
@@ -329,25 +339,16 @@ class GitHubSearchTool
 		// Create the search coordinator agent with composition
 		this.searchCoordinatorAgent = this.createSearchCoordinatorAgent();
 	}
-	
+
 	private createSearchCoordinatorAgent(): ReturnType<typeof agent> {
 		// Create query generator agent
 		const queryGeneratorAgent = createQueryGeneratorAgent();
-		const copilot = CopilotChatProvider.getInstance();
-		const provider = copilot.provider;
-		outputLogger.info("headers: ", copilot.headers);
-		provider.setOptions({ debug: true, fetch: (input: RequestInfo | URL, init?: RequestInit) => {
-			init!.headers = {
-				...init?.headers,
-				...copilot.headers,
-			};
-			return fetch(input, init);
-		} });
 		// Create the search and process function
 		const generateAndSearchRepos: AxFunction = {
 			name: "generateAndSearchRepos",
 			description: "Generate a search query and search for MCP repositories",
 			func: async (args: { originalUserMessage: string }) => {
+				const { provider } = await configureCopilotProvider(true);
 
 				// Use the query generator agent
 				const queryResult = await queryGeneratorAgent.forward(
@@ -387,19 +388,7 @@ class GitHubSearchTool
 
 	async invoke(options: vscode.LanguageModelToolInvocationOptions<{ userQuery: string; }>): Promise<vscode.LanguageModelToolResult> {
 		this.stream.progress("Beginning search for installable MCP servers...");
-		const copilot = CopilotChatProvider.getInstance();
-		const provider = copilot.provider;
-		provider.setOptions(
-		{ 
-			debug: false, 
-			fetch: (input: RequestInfo | URL, init?: RequestInit) => {
-				init!.headers = {
-					...init?.headers,
-					...copilot.headers,
-				};
-				return fetch(input, init);
-			} 
-		});
+		const { provider } = await configureCopilotProvider(false);
 		outputLogger.info("GitHubSearchTool invoked", { options });
 
 		try {
@@ -476,25 +465,13 @@ class GitHubSearchTool
 }
 
 export async function readmeExtractionRequest(readme: string) {
-	// Try to get GitHub token from VSCode authentication API using the same scopes as extension activation
-	const session = await vscode.authentication.getSession(GITHUB_AUTH_PROVIDER_ID, SCOPES, { createIfNone: false });
-	const accessToken = session?.accessToken;
-	if (!accessToken) {
-		throw new Error("Copilot not set up.");
+	let provider: AxAI;
+	try {
+		({ provider } = await configureCopilotProvider(false));
+	} catch (error) {
+		outputLogger.warn("Copilot provider unavailable for README extraction", error as Error);
+		throw new Error("GitHub authentication is required to use AI-assisted setup.");
 	}
-	const copilot = CopilotChatProvider.getInstance();
-	const provider = copilot.provider;
-	provider.setOptions(
-		{ 
-			debug: false, 
-			fetch: (input: RequestInfo | URL, init?: RequestInit) => {
-				init!.headers = {
-					...init?.headers,
-					...copilot.headers,
-				};
-				return fetch(input, init);
-			} 
-	});
 
 	const extractor = ax(`
 		"Extracts the MCP server configuration from a README.md file and returns the necessary information to run it."

src/extension.ts

@@ -53,22 +53,38 @@ export async function activate(context: vscode.ExtensionContext) {
 	context.subscriptions.push(logger, { dispose: shutdownLogs });
 	context.subscriptions.push(telemetryReporter);
 	// console.dir(await vscode.authentication.getAccounts('github'), {depth: null});
-	outputLogger.debug("Getting GitHub authentication session");
-	const session = await vscode.authentication.getSession(
-		GITHUB_AUTH_PROVIDER_ID,
-		SCOPES,
-		{ createIfNone: true }
-	);
+    outputLogger.debug("Checking for existing GitHub authentication session");
+    let session: vscode.AuthenticationSession | undefined;
+    try {
+        session = await vscode.authentication.getSession(
+            GITHUB_AUTH_PROVIDER_ID,
+            SCOPES,
+            { createIfNone: false }
+        );
+    } catch (error) {
+        outputLogger.warn("Failed to retrieve existing GitHub session", error as Error);
+    }
 
-	outputLogger.info("Initializing Copilot MCP LM Provider");
-	const copilot = await CopilotChatProvider.initialize(context);
-	const models = await copilot.getModels();
-	outputLogger.info("Available Copilot models", models.map((m: any) => m.id));
+    outputLogger.info("Configuring Copilot MCP LM Provider");
+    const copilot = await CopilotChatProvider.configure(context);
+    const initialized = await copilot.tryEnsureInitialized();
+    if (initialized) {
+        try {
+            const models = await copilot.getModels();
+            outputLogger.info("Available Copilot models", models.map((m: any) => m.id));
+        } catch (modelError) {
+            outputLogger.warn("Failed to retrieve Copilot models", modelError as Error);
+        }
+    } else {
+        outputLogger.debug("Copilot provider not initialized yet; will initialize on demand");
+    }
 
 	// Initialize standardized telemetry context  
 	const extensionVersion = vscode.extensions.getExtension('AutomataLabs.copilot-mcp')?.packageJSON?.version || 'unknown';
-	initializeContext(session, extensionVersion);
-	setCommonLogAttributes({ ...session.account });
+    initializeContext(session, extensionVersion);
+    if (session) {
+        setCommonLogAttributes({ ...session.account });
+    }
 
 	// Initialize CloudMCP indexer
 	outputLogger.info("Initializing CloudMCP indexer");
@@ -106,10 +122,9 @@ export async function activate(context: vscode.ExtensionContext) {
 	context.subscriptions.push(disposable, showLogsCommand);
 
 	outputLogger.debug("Creating CopilotMcpViewProvider");
-	const provider = new CopilotMcpViewProvider(
-		context.extensionUri,
-		session.accessToken
-	);
+    const provider = new CopilotMcpViewProvider(
+        context.extensionUri
+    );
 	context.subscriptions.push(
 		vscode.window.registerWebviewViewProvider(
 			CopilotMcpViewProvider.viewType,

src/panels/ExtensionPanel.ts

@@ -40,7 +40,7 @@ import type {
 } from "../shared/types/rpcTypes";
 import axios from "axios";
 import { outputLogger } from "../utilities/outputLogger";
-import { resolve } from "dns";
+import { GITHUB_AUTH_PROVIDER_ID, SCOPES } from "../utilities/const";
 
 // Helper function to read servers from .vscode/mcp.json
 async function getServersFromMcpJsonFile(
@@ -231,8 +231,8 @@ export class CopilotMcpViewProvider implements vscode.WebviewViewProvider {
 	octokit: any;
 
 	constructor(
-		private readonly _extensionUri: vscode.Uri,
-		private readonly _accessToken: string	) {}
+		private readonly _extensionUri: vscode.Uri
+	) {}
 
 	resolveWebviewView(
 		webviewView: vscode.WebviewView,
@@ -529,9 +529,27 @@ export class CopilotMcpViewProvider implements vscode.WebviewViewProvider {
 	}
 
 	async getOctokit() {
+		if (this.octokit) {
+			return this.octokit;
+		}
+		let session: vscode.AuthenticationSession | undefined;
+		try {
+			session = await vscode.authentication.getSession(
+				GITHUB_AUTH_PROVIDER_ID,
+				SCOPES,
+				{ createIfNone: true }
+			);
+		} catch (error) {
+			outputLogger.warn("Failed to acquire GitHub session for Octokit", error as Error);
+			throw new Error("GitHub authentication is required to continue.");
+		}
+		const accessToken = session?.accessToken;
+		if (!accessToken) {
+			throw new Error("GitHub authentication is required to continue.");
+		}
 		const Octokit = await import("octokit");
 		this.octokit = new Octokit.Octokit({
-			auth: this._accessToken,
+			auth: accessToken,
 		});
 		return this.octokit;
 	}