@office-ai/aioncli-core 0.8.1 → 0.18.4

package/dist/src/config/config.js

@@ -4,6 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import * as path from 'node:path';
+import { inspect } from 'node:util';
 import process from 'node:process';
 import { AuthType, createContentGenerator, createContentGeneratorConfig, } from '../core/contentGenerator.js';
 import { PromptRegistry } from '../prompts/prompt-registry.js';
@@ -18,7 +19,6 @@ import { SmartEditTool } from '../tools/smart-edit.js';
 import { ShellTool } from '../tools/shell.js';
 import { WriteFileTool } from '../tools/write-file.js';
 import { WebFetchTool } from '../tools/web-fetch.js';
-import { ReadManyFilesTool } from '../tools/read-many-files.js';
 import { MemoryTool, setGeminiMdFilename } from '../tools/memoryTool.js';
 import { WebSearchTool } from '../tools/web-search.js';
 import { GeminiClient } from '../core/client.js';
@@ -26,31 +26,36 @@ import { BaseLlmClient } from '../core/baseLlmClient.js';
 import { FileDiscoveryService } from '../services/fileDiscoveryService.js';
 import { GitService } from '../services/gitService.js';
 import { initializeTelemetry, DEFAULT_TELEMETRY_TARGET, DEFAULT_OTLP_ENDPOINT, uiTelemetryService, } from '../telemetry/index.js';
+import { coreEvents } from '../utils/events.js';
 import { tokenLimit } from '../core/tokenLimits.js';
-import { StartSessionEvent } from '../telemetry/index.js';
-import { DEFAULT_GEMINI_EMBEDDING_MODEL, DEFAULT_GEMINI_FLASH_MODEL, } from './models.js';
+import { DEFAULT_GEMINI_EMBEDDING_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_MODEL_AUTO, DEFAULT_THINKING_MODE, } from './models.js';
 import { shouldAttemptBrowserLaunch } from '../utils/browser.js';
 import { ideContextStore } from '../ide/ideContext.js';
 import { WriteTodosTool } from '../tools/write-todos.js';
 import { StandardFileSystemService } from '../services/fileSystemService.js';
-import { logCliConfiguration, logRipgrepFallback, } from '../telemetry/loggers.js';
+import { logRipgrepFallback } from '../telemetry/loggers.js';
 import { RipgrepFallbackEvent } from '../telemetry/types.js';
 import { ModelRouterService } from '../routing/modelRouterService.js';
 import { OutputFormat } from '../output/types.js';
+import { ModelConfigService } from '../services/modelConfigService.js';
+import { DEFAULT_MODEL_CONFIGS } from './defaultModelConfigs.js';
 import { WorkspaceContext } from '../utils/workspaceContext.js';
 import { Storage } from './storage.js';
 import { FileExclusions } from '../utils/ignorePatterns.js';
 import { MessageBus } from '../confirmation-bus/message-bus.js';
 import { PolicyEngine } from '../policy/policy-engine.js';
-import { ProxyAgent, setGlobalDispatcher } from 'undici';
-export var ApprovalMode;
-(function (ApprovalMode) {
-    ApprovalMode["DEFAULT"] = "default";
-    ApprovalMode["AUTO_EDIT"] = "autoEdit";
-    ApprovalMode["YOLO"] = "yolo";
-})(ApprovalMode || (ApprovalMode = {}));
-import { DEFAULT_MEMORY_FILE_FILTERING_OPTIONS, DEFAULT_FILE_FILTERING_OPTIONS, } from './constants.js';
-export { DEFAULT_MEMORY_FILE_FILTERING_OPTIONS, DEFAULT_FILE_FILTERING_OPTIONS, };
+import { getCodeAssistServer } from '../code_assist/codeAssist.js';
+import { AgentRegistry } from '../agents/registry.js';
+import { setGlobalProxy } from '../utils/fetch.js';
+import { SubagentToolWrapper } from '../agents/subagent-tool-wrapper.js';
+import { getExperiments } from '../code_assist/experiments/experiments.js';
+import { ExperimentFlags } from '../code_assist/experiments/flagNames.js';
+import { debugLogger } from '../utils/debugLogger.js';
+import { ApprovalMode } from '../policy/types.js';
+import { DEFAULT_FILE_FILTERING_OPTIONS, DEFAULT_MEMORY_FILE_FILTERING_OPTIONS, } from './constants.js';
+import { SimpleExtensionLoader, } from '../utils/extensionLoader.js';
+import { McpClientManager } from '../tools/mcp-client-manager.js';
+export { DEFAULT_FILE_FILTERING_OPTIONS, DEFAULT_MEMORY_FILE_FILTERING_OPTIONS, };
 export const DEFAULT_TRUNCATE_TOOL_OUTPUT_THRESHOLD = 4_000_000;
 export const DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES = 1000;
 export class MCPServerConfig {
@@ -67,7 +72,7 @@ export class MCPServerConfig {
     description;
     includeTools;
     excludeTools;
-    extensionName;
+    extension;
     oauth;
     authProviderType;
     targetAudience;
@@ -84,7 +89,7 @@ export class MCPServerConfig {
     // Common
     timeout, trust, 
     // Metadata
-    description, includeTools, excludeTools, extensionName, 
+    description, includeTools, excludeTools, extension, 
     // OAuth configuration
     oauth, authProviderType, 
     // Service Account Configuration
@@ -105,7 +110,7 @@ export class MCPServerConfig {
         this.description = description;
         this.includeTools = includeTools;
         this.excludeTools = excludeTools;
-        this.extensionName = extensionName;
+        this.extension = extension;
         this.oauth = oauth;
         this.authProviderType = authProviderType;
         this.targetAudience = targetAudience;
@@ -120,18 +125,22 @@ export var AuthProviderType;
 })(AuthProviderType || (AuthProviderType = {}));
 export class Config {
     toolRegistry;
+    mcpClientManager;
+    allowedMcpServers;
+    blockedMcpServers;
     promptRegistry;
+    agentRegistry;
     sessionId;
     fileSystemService;
     contentGeneratorConfig;
     contentGenerator;
+    modelConfigService;
     embeddingModel;
     sandbox;
     targetDir;
     workspaceContext;
     debugMode;
     question;
-    fullContext;
     coreTools;
     allowedTools;
     excludeTools;
@@ -141,6 +150,7 @@ export class Config {
     mcpServers;
     userMemory;
     geminiMdFileCount;
+    geminiMdFilePaths;
     approvalMode;
     showMemoryUsage;
     accessibility;
@@ -157,26 +167,31 @@ export class Config {
     cwd;
     bugCommand;
     model;
-    extensionContextFilePaths;
+    previewFeatures;
     noBrowser;
-    folderTrustFeature;
     folderTrust;
     ideMode;
     inFallbackMode = false;
     maxSessionTurns;
+    listSessions;
+    deleteSession;
     listExtensions;
-    _extensions;
-    _blockedMcpServers;
+    _extensionLoader;
+    _enabledExtensions;
+    enableExtensionReloading;
     fallbackModelHandler;
     quotaErrorOccurred = false;
     summarizeToolOutput;
     experimentalZedIntegration = false;
     loadMemoryFromIncludeDirectories = false;
-    chatCompression;
+    importFormat;
+    discoveryMaxDirs;
+    compressionThreshold;
     interactive;
+    ptyInfo;
     trustedFolder;
     useRipgrep;
-    shouldUseNodePtyShell;
+    enableInteractiveShell;
     skipNextSpeakerCheck;
     shellExecutionConfig;
     extensionManagement = true;
@@ -194,7 +209,23 @@ export class Config {
     policyEngine;
     outputSettings;
     useModelRouter;
+    initialUseModelRouter;
+    disableModelRouterForAuth;
     enableMessageBusIntegration;
+    codebaseInvestigatorSettings;
+    continueOnFailedApiCall;
+    retryFetchErrors;
+    enableShellOutputEfficiency;
+    fakeResponses;
+    recordResponses;
+    disableYoloMode;
+    pendingIncludeDirectories;
+    enableHooks;
+    hooks;
+    experiments;
+    experimentsPromise;
+    previewModelFallbackMode = false;
+    previewModelBypassMode = false;
     constructor(params) {
         this.sessionId = params.sessionId;
         this.embeddingModel =
@@ -202,10 +233,11 @@ export class Config {
         this.fileSystemService = new StandardFileSystemService();
         this.sandbox = params.sandbox;
         this.targetDir = path.resolve(params.targetDir);
-        this.workspaceContext = new WorkspaceContext(this.targetDir, params.includeDirectories ?? []);
+        this.folderTrust = params.folderTrust ?? false;
+        this.workspaceContext = new WorkspaceContext(this.targetDir, []);
+        this.pendingIncludeDirectories = params.includeDirectories ?? [];
         this.debugMode = params.debugMode;
         this.question = params.question;
-        this.fullContext = params.fullContext ?? false;
         this.coreTools = params.coreTools;
         this.allowedTools = params.allowedTools;
         this.excludeTools = params.excludeTools;
@@ -213,8 +245,11 @@ export class Config {
         this.toolCallCommand = params.toolCallCommand;
         this.mcpServerCommand = params.mcpServerCommand;
         this.mcpServers = params.mcpServers;
+        this.allowedMcpServers = params.allowedMcpServers ?? [];
+        this.blockedMcpServers = params.blockedMcpServers ?? [];
         this.userMemory = params.userMemory ?? '';
         this.geminiMdFileCount = params.geminiMdFileCount ?? 0;
+        this.geminiMdFilePaths = params.geminiMdFilePaths ?? [];
         this.approvalMode = params.approvalMode ?? ApprovalMode.DEFAULT;
         this.showMemoryUsage = params.showMemoryUsage ?? false;
         this.accessibility = params.accessibility ?? {};
@@ -229,8 +264,10 @@ export class Config {
         };
         this.usageStatisticsEnabled = params.usageStatisticsEnabled ?? true;
         this.fileFiltering = {
-            respectGitIgnore: params.fileFiltering?.respectGitIgnore ?? true,
-            respectGeminiIgnore: params.fileFiltering?.respectGeminiIgnore ?? true,
+            respectGitIgnore: params.fileFiltering?.respectGitIgnore ??
+                DEFAULT_FILE_FILTERING_OPTIONS.respectGitIgnore,
+            respectGeminiIgnore: params.fileFiltering?.respectGeminiIgnore ??
+                DEFAULT_FILE_FILTERING_OPTIONS.respectGeminiIgnore,
             enableRecursiveFileSearch: params.fileFiltering?.enableRecursiveFileSearch ?? true,
             disableFuzzySearch: params.fileFiltering?.disableFuzzySearch ?? false,
         };
@@ -240,25 +277,30 @@ export class Config {
         this.fileDiscoveryService = params.fileDiscoveryService ?? null;
         this.bugCommand = params.bugCommand;
         this.model = params.model;
-        this.extensionContextFilePaths = params.extensionContextFilePaths ?? [];
+        this.previewFeatures = params.previewFeatures ?? undefined;
         this.maxSessionTurns = params.maxSessionTurns ?? -1;
         this.experimentalZedIntegration =
             params.experimentalZedIntegration ?? false;
+        this.listSessions = params.listSessions ?? false;
+        this.deleteSession = params.deleteSession;
         this.listExtensions = params.listExtensions ?? false;
-        this._extensions = params.extensions ?? [];
-        this._blockedMcpServers = params.blockedMcpServers ?? [];
+        this._extensionLoader =
+            params.extensionLoader ?? new SimpleExtensionLoader([]);
+        this._enabledExtensions = params.enabledExtensions ?? [];
         this.noBrowser = params.noBrowser ?? false;
         this.summarizeToolOutput = params.summarizeToolOutput;
-        this.folderTrustFeature = params.folderTrustFeature ?? false;
         this.folderTrust = params.folderTrust ?? false;
         this.ideMode = params.ideMode ?? false;
         this.loadMemoryFromIncludeDirectories =
             params.loadMemoryFromIncludeDirectories ?? false;
-        this.chatCompression = params.chatCompression;
+        this.importFormat = params.importFormat ?? 'tree';
+        this.discoveryMaxDirs = params.discoveryMaxDirs ?? 200;
+        this.compressionThreshold = params.compressionThreshold;
         this.interactive = params.interactive ?? false;
+        this.ptyInfo = params.ptyInfo ?? 'child_process';
         this.trustedFolder = params.trustedFolder;
         this.useRipgrep = params.useRipgrep ?? true;
-        this.shouldUseNodePtyShell = params.shouldUseNodePtyShell ?? false;
+        this.enableInteractiveShell = params.enableInteractiveShell ?? false;
         this.skipNextSpeakerCheck = params.skipNextSpeakerCheck ?? true;
         this.shellExecutionConfig = {
             terminalWidth: params.shellExecutionConfig?.terminalWidth ?? 80,
@@ -273,31 +315,79 @@ export class Config {
             params.truncateToolOutputLines ?? DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES;
         this.enableToolOutputTruncation = params.enableToolOutputTruncation ?? true;
         this.useSmartEdit = params.useSmartEdit ?? true;
-        this.useWriteTodos = params.useWriteTodos ?? false;
-        this.useModelRouter = params.useModelRouter ?? false;
+        this.useWriteTodos = params.useWriteTodos ?? true;
+        this.initialUseModelRouter = params.useModelRouter ?? false;
+        this.useModelRouter = this.initialUseModelRouter;
+        this.disableModelRouterForAuth = params.disableModelRouterForAuth ?? [];
+        this.enableHooks = params.enableHooks ?? false;
+        // Enable MessageBus integration if:
+        // 1. Explicitly enabled via setting, OR
+        // 2. Hooks are enabled and hooks are configured
+        const hasHooks = params.hooks && Object.keys(params.hooks).length > 0;
+        const hooksNeedMessageBus = this.enableHooks && hasHooks;
         this.enableMessageBusIntegration =
-            params.enableMessageBusIntegration ?? false;
+            params.enableMessageBusIntegration ??
+                (hooksNeedMessageBus ? true : false);
+        this.codebaseInvestigatorSettings = {
+            enabled: params.codebaseInvestigatorSettings?.enabled ?? true,
+            maxNumTurns: params.codebaseInvestigatorSettings?.maxNumTurns ?? 10,
+            maxTimeMinutes: params.codebaseInvestigatorSettings?.maxTimeMinutes ?? 3,
+            thinkingBudget: params.codebaseInvestigatorSettings?.thinkingBudget ??
+                DEFAULT_THINKING_MODE,
+            model: params.codebaseInvestigatorSettings?.model ?? DEFAULT_GEMINI_MODEL,
+        };
+        this.continueOnFailedApiCall = params.continueOnFailedApiCall ?? true;
+        this.enableShellOutputEfficiency =
+            params.enableShellOutputEfficiency ?? true;
         this.extensionManagement = params.extensionManagement ?? true;
+        this.enableExtensionReloading = params.enableExtensionReloading ?? false;
         this.storage = new Storage(this.targetDir);
+        this.fakeResponses = params.fakeResponses;
+        this.recordResponses = params.recordResponses;
         this.enablePromptCompletion = params.enablePromptCompletion ?? false;
         this.fileExclusions = new FileExclusions(this);
         this.eventEmitter = params.eventEmitter;
         this.policyEngine = new PolicyEngine(params.policyEngineConfig);
-        this.messageBus = new MessageBus(this.policyEngine);
+        this.messageBus = new MessageBus(this.policyEngine, this.debugMode);
         this.outputSettings = {
             format: params.output?.format ?? OutputFormat.TEXT,
         };
+        this.retryFetchErrors = params.retryFetchErrors ?? false;
+        this.disableYoloMode = params.disableYoloMode ?? false;
+        this.hooks = params.hooks;
+        this.experiments = params.experiments;
         if (params.contextFileName) {
             setGeminiMdFilename(params.contextFileName);
         }
         if (this.telemetrySettings.enabled) {
             initializeTelemetry(this);
         }
-        if (this.getProxy()) {
-            setGlobalDispatcher(new ProxyAgent(this.getProxy()));
+        const proxy = this.getProxy();
+        if (proxy) {
+            try {
+                setGlobalProxy(proxy);
+            }
+            catch (error) {
+                coreEvents.emitFeedback('error', 'Invalid proxy configuration detected. Check debug drawer for more details (F12)', error);
+            }
         }
         this.geminiClient = new GeminiClient(this);
         this.modelRouterService = new ModelRouterService(this);
+        // HACK: The settings loading logic doesn't currently merge the default
+        // generation config with the user's settings. This means if a user provides
+        // any `generation` settings (e.g., just `overrides`), the default `aliases`
+        // are lost. This hack manually merges the default aliases back in if they
+        // are missing from the user's config.
+        // TODO(12593): Fix the settings loading logic to properly merge defaults and
+        // remove this hack.
+        let modelConfigServiceConfig = params.modelConfigServiceConfig;
+        if (modelConfigServiceConfig && !modelConfigServiceConfig.aliases) {
+            modelConfigServiceConfig = {
+                ...modelConfigServiceConfig,
+                aliases: DEFAULT_MODEL_CONFIGS.aliases,
+            };
+        }
+        this.modelConfigService = new ModelConfigService(modelConfigServiceConfig ?? DEFAULT_MODEL_CONFIGS);
     }
     /**
      * Must only be called once, throws if called again.
@@ -313,30 +403,64 @@ export class Config {
             await this.getGitService();
         }
         this.promptRegistry = new PromptRegistry();
+        this.agentRegistry = new AgentRegistry(this);
+        await this.agentRegistry.initialize();
         this.toolRegistry = await this.createToolRegistry();
+        this.mcpClientManager = new McpClientManager(this.toolRegistry, this, this.eventEmitter);
+        await Promise.all([
+            await this.mcpClientManager.startConfiguredMcpServers(),
+            await this.getExtensionLoader().start(this),
+        ]);
         await this.geminiClient.initialize();
     }
     getContentGenerator() {
         return this.contentGenerator;
     }
     async refreshAuth(authMethod) {
+        this.useModelRouter = this.initialUseModelRouter;
+        if (this.disableModelRouterForAuth?.includes(authMethod)) {
+            this.useModelRouter = false;
+            if (this.model === DEFAULT_GEMINI_MODEL_AUTO) {
+                this.model = DEFAULT_GEMINI_MODEL;
+            }
+        }
         // Vertex and Genai have incompatible encryption and sending history with
-        // throughtSignature from Genai to Vertex will fail, we need to strip them
+        // thoughtSignature from Genai to Vertex will fail, we need to strip them
         if (this.contentGeneratorConfig?.authType === AuthType.USE_GEMINI &&
-            authMethod === AuthType.LOGIN_WITH_GOOGLE) {
+            authMethod !== AuthType.USE_GEMINI) {
             // Restore the conversation history to the new client
             this.geminiClient.stripThoughtsFromHistory();
         }
-        const newContentGeneratorConfig = createContentGeneratorConfig(this, authMethod);
+        const newContentGeneratorConfig = await createContentGeneratorConfig(this, authMethod);
         this.contentGenerator = await createContentGenerator(newContentGeneratorConfig, this, this.getSessionId());
         // Only assign to instance properties after successful initialization
         this.contentGeneratorConfig = newContentGeneratorConfig;
         // Initialize BaseLlmClient now that the ContentGenerator is available
         this.baseLlmClient = new BaseLlmClient(this.contentGenerator, this);
+        const previewFeatures = this.getPreviewFeatures();
+        const codeAssistServer = getCodeAssistServer(this);
+        if (codeAssistServer) {
+            this.experimentsPromise = getExperiments(codeAssistServer)
+                .then((experiments) => {
+                this.setExperiments(experiments);
+                // If preview features have not been set and the user authenticated through Google, we enable preview based on remote config only if it's true
+                if (previewFeatures === undefined) {
+                    const remotePreviewFeatures = experiments.flags[ExperimentFlags.ENABLE_PREVIEW]?.boolValue;
+                    if (remotePreviewFeatures === true) {
+                        this.setPreviewFeatures(remotePreviewFeatures);
+                    }
+                }
+            })
+                .catch((e) => {
+                debugLogger.error('Failed to fetch experiments', e);
+            });
+        }
+        else {
+            this.experiments = undefined;
+            this.experimentsPromise = undefined;
+        }
         // Reset the session flag since we're explicitly changing auth and using default model
         this.inFallbackMode = false;
-        // Logging the cli configuration here as the auth related configuration params would have been loaded by this point
-        logCliConfiguration(this, new StartSessionEvent(this, this.toolRegistry));
     }
     getUserTier() {
         return this.contentGenerator?.userTier;
@@ -359,9 +483,18 @@ export class Config {
     getSessionId() {
         return this.sessionId;
     }
+    setSessionId(sessionId) {
+        this.sessionId = sessionId;
+    }
     shouldLoadMemoryFromIncludeDirectories() {
         return this.loadMemoryFromIncludeDirectories;
     }
+    getImportFormat() {
+        return this.importFormat;
+    }
+    getDiscoveryMaxDirs() {
+        return this.discoveryMaxDirs;
+    }
     getContentGeneratorConfig() {
         return this.contentGeneratorConfig;
     }
@@ -369,11 +502,11 @@ export class Config {
         return this.model;
     }
     setModel(newModel) {
-        // Do not allow Pro usage if the user is in fallback mode.
-        if (newModel.includes('pro') && this.isInFallbackMode()) {
-            return;
+        if (this.model !== newModel || this.inFallbackMode) {
+            this.model = newModel;
+            coreEvents.emitModelChanged(newModel);
         }
-        this.model = newModel;
+        this.setFallbackMode(false);
     }
     isInFallbackMode() {
         return this.inFallbackMode;
@@ -384,6 +517,21 @@ export class Config {
     setFallbackModelHandler(handler) {
         this.fallbackModelHandler = handler;
     }
+    getFallbackModelHandler() {
+        return this.fallbackModelHandler;
+    }
+    isPreviewModelFallbackMode() {
+        return this.previewModelFallbackMode;
+    }
+    setPreviewModelFallbackMode(active) {
+        this.previewModelFallbackMode = active;
+    }
+    isPreviewModelBypassMode() {
+        return this.previewModelBypassMode;
+    }
+    setPreviewModelBypassMode(active) {
+        this.previewModelBypassMode = active;
+    }
     getMaxSessionTurns() {
         return this.maxSessionTurns;
     }
@@ -416,6 +564,9 @@ export class Config {
     getWorkspaceContext() {
         return this.workspaceContext;
     }
+    getAgentRegistry() {
+        return this.agentRegistry;
+    }
     getToolRegistry() {
         return this.toolRegistry;
     }
@@ -428,8 +579,11 @@ export class Config {
     getQuestion() {
         return this.question;
     }
-    getFullContext() {
-        return this.fullContext;
+    getPreviewFeatures() {
+        return this.previewFeatures;
+    }
+    setPreviewFeatures(previewFeatures) {
+        this.previewFeatures = previewFeatures;
     }
     getCoreTools() {
         return this.coreTools;
@@ -437,8 +591,23 @@ export class Config {
     getAllowedTools() {
         return this.allowedTools;
     }
+    /**
+     * All the excluded tools from static configuration, loaded extensions, or
+     * other sources.
+     *
+     * May change over time.
+     */
     getExcludeTools() {
-        return this.excludeTools;
+        const excludeToolsSet = new Set([...(this.excludeTools ?? [])]);
+        for (const extension of this.getExtensionLoader().getExtensions()) {
+            if (!extension.isActive) {
+                continue;
+            }
+            for (const tool of extension.excludeTools || []) {
+                excludeToolsSet.add(tool);
+            }
+        }
+        return excludeToolsSet;
     }
     getToolDiscoveryCommand() {
         return this.toolDiscoveryCommand;
@@ -449,9 +618,26 @@ export class Config {
     getMcpServerCommand() {
         return this.mcpServerCommand;
     }
+    /**
+     * The user configured MCP servers (via gemini settings files).
+     *
+     * Does NOT include mcp servers configured by extensions.
+     */
     getMcpServers() {
         return this.mcpServers;
     }
+    getMcpClientManager() {
+        return this.mcpClientManager;
+    }
+    getAllowedMcpServers() {
+        return this.allowedMcpServers;
+    }
+    getBlockedMcpServers() {
+        return this.blockedMcpServers;
+    }
+    setMcpServers(mcpServers) {
+        this.mcpServers = mcpServers;
+    }
     getUserMemory() {
         return this.userMemory;
     }
@@ -464,6 +650,12 @@ export class Config {
     setGeminiMdFileCount(count) {
         this.geminiMdFileCount = count;
     }
+    getGeminiMdFilePaths() {
+        return this.geminiMdFilePaths;
+    }
+    setGeminiMdFilePaths(paths) {
+        this.geminiMdFilePaths = paths;
+    }
     getApprovalMode() {
         return this.approvalMode;
     }
@@ -473,6 +665,15 @@ export class Config {
         }
         this.approvalMode = mode;
     }
+    isYoloModeDisabled() {
+        return this.disableYoloMode || !this.isTrustedFolder();
+    }
+    getPendingIncludeDirectories() {
+        return this.pendingIncludeDirectories;
+    }
+    clearPendingIncludeDirectories() {
+        this.pendingIncludeDirectories = [];
+    }
     getShowMemoryUsage() {
         return this.showMemoryUsage;
     }
@@ -503,6 +704,16 @@ export class Config {
     getGeminiClient() {
         return this.geminiClient;
     }
+    /**
+     * Updates the system instruction with the latest user memory.
+     * Whenever the user memory (GEMINI.md files) is updated.
+     */
+    async updateSystemInstructionIfInitialized() {
+        const geminiClient = this.getGeminiClient();
+        if (geminiClient?.isInitialized()) {
+            await geminiClient.updateSystemInstruction();
+        }
+    }
     getModelRouterService() {
         return this.modelRouterService;
     }
@@ -559,23 +770,34 @@ export class Config {
     getUsageStatisticsEnabled() {
         return this.usageStatisticsEnabled;
     }
-    getExtensionContextFilePaths() {
-        return this.extensionContextFilePaths;
-    }
     getExperimentalZedIntegration() {
         return this.experimentalZedIntegration;
     }
     getListExtensions() {
         return this.listExtensions;
     }
+    getListSessions() {
+        return this.listSessions;
+    }
+    getDeleteSession() {
+        return this.deleteSession;
+    }
     getExtensionManagement() {
         return this.extensionManagement;
     }
     getExtensions() {
-        return this._extensions;
+        return this._extensionLoader.getExtensions();
     }
-    getBlockedMcpServers() {
-        return this._blockedMcpServers;
+    getExtensionLoader() {
+        return this._extensionLoader;
+    }
+    // The list of explicitly enabled extensions, if any were given, may contain
+    // the string "none".
+    getEnabledExtensions() {
+        return this._enabledExtensions;
+    }
+    getEnableExtensionReloading() {
+        return this.enableExtensionReloading;
     }
     getNoBrowser() {
         return this.noBrowser;
@@ -589,16 +811,16 @@ export class Config {
     getIdeMode() {
         return this.ideMode;
     }
-    getFolderTrustFeature() {
-        return this.folderTrustFeature;
-    }
     /**
-     * Returns 'true' if the workspace is considered "trusted".
-     * 'false' for untrusted.
+     * Returns 'true' if the folder trust feature is enabled.
      */
     getFolderTrust() {
         return this.folderTrust;
     }
+    /**
+     * Returns 'true' if the workspace is considered "trusted".
+     * 'false' for untrusted.
+     */
     isTrustedFolder() {
         // isWorkspaceTrusted in cli/src/config/trustedFolder.js returns undefined
         // when the file based trust value is unavailable, since it is mainly used
@@ -631,8 +853,47 @@ export class Config {
     setFileSystemService(fileSystemService) {
         this.fileSystemService = fileSystemService;
     }
-    getChatCompression() {
-        return this.chatCompression;
+    async getCompressionThreshold() {
+        if (this.compressionThreshold) {
+            return this.compressionThreshold;
+        }
+        await this.ensureExperimentsLoaded();
+        const remoteThreshold = this.experiments?.flags[ExperimentFlags.CONTEXT_COMPRESSION_THRESHOLD]
+            ?.floatValue;
+        if (remoteThreshold === 0) {
+            return undefined;
+        }
+        return remoteThreshold;
+    }
+    async getUserCaching() {
+        await this.ensureExperimentsLoaded();
+        return this.experiments?.flags[ExperimentFlags.USER_CACHING]?.boolValue;
+    }
+    async getBannerTextNoCapacityIssues() {
+        await this.ensureExperimentsLoaded();
+        return (this.experiments?.flags[ExperimentFlags.BANNER_TEXT_NO_CAPACITY_ISSUES]
+            ?.stringValue ?? '');
+    }
+    async getBannerTextCapacityIssues() {
+        await this.ensureExperimentsLoaded();
+        return (this.experiments?.flags[ExperimentFlags.BANNER_TEXT_CAPACITY_ISSUES]
+            ?.stringValue ?? '');
+    }
+    async ensureExperimentsLoaded() {
+        if (!this.experimentsPromise) {
+            return;
+        }
+        try {
+            await this.experimentsPromise;
+        }
+        catch (e) {
+            debugLogger.debug('Failed to fetch experiments', e);
+        }
+    }
+    isInteractiveShellEnabled() {
+        return (this.interactive &&
+            this.ptyInfo !== 'child_process' &&
+            this.enableInteractiveShell);
     }
     isInteractive() {
         return this.interactive;
@@ -640,12 +901,21 @@ export class Config {
     getUseRipgrep() {
         return this.useRipgrep;
     }
-    getShouldUseNodePtyShell() {
-        return this.shouldUseNodePtyShell;
+    getEnableInteractiveShell() {
+        return this.enableInteractiveShell;
     }
     getSkipNextSpeakerCheck() {
         return this.skipNextSpeakerCheck;
     }
+    getContinueOnFailedApiCall() {
+        return this.continueOnFailedApiCall;
+    }
+    getRetryFetchErrors() {
+        return this.retryFetchErrors;
+    }
+    getEnableShellOutputEfficiency() {
+        return this.enableShellOutputEfficiency;
+    }
     getShellExecutionConfig() {
         return this.shellExecutionConfig;
     }
@@ -708,15 +978,24 @@ export class Config {
     getEnableMessageBusIntegration() {
         return this.enableMessageBusIntegration;
     }
+    getEnableHooks() {
+        return this.enableHooks;
+    }
+    getCodebaseInvestigatorSettings() {
+        return this.codebaseInvestigatorSettings;
+    }
     async createToolRegistry() {
-        const registry = new ToolRegistry(this, this.eventEmitter);
+        const registry = new ToolRegistry(this);
+        // Set message bus on tool registry before discovery so MCP tools can access it
+        if (this.getEnableMessageBusIntegration()) {
+            registry.setMessageBus(this.messageBus);
+        }
         // helper to create & register core tools that are enabled
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         const registerCoreTool = (ToolClass, ...args) => {
             const className = ToolClass.name;
             const toolName = ToolClass.Name || className;
             const coreTools = this.getCoreTools();
-            const excludeTools = this.getExcludeTools() || [];
             // On some platforms, the className can be minified to _ClassName.
             const normalizedClassName = className.replace(/^_+/, '');
             let isEnabled = true; // Enabled by default if coreTools is not set.
@@ -726,24 +1005,14 @@ export class Config {
                     tool.startsWith(`${toolName}(`) ||
                     tool.startsWith(`${normalizedClassName}(`));
             }
-            const isExcluded = excludeTools.some((tool) => tool === toolName || tool === normalizedClassName);
-            if (isExcluded) {
-                isEnabled = false;
-            }
             if (isEnabled) {
                 // Pass message bus to tools when feature flag is enabled
                 // This first implementation is only focused on the general case of
                 // the tool registry.
                 const messageBusEnabled = this.getEnableMessageBusIntegration();
-                if (this.debugMode && messageBusEnabled) {
-                    console.log(`[DEBUG] enableMessageBusIntegration setting: ${messageBusEnabled}`);
-                }
                 const toolArgs = messageBusEnabled
                     ? [...args, this.getMessageBus()]
                     : args;
-                if (this.debugMode && messageBusEnabled) {
-                    console.log(`[DEBUG] Registering ${className} with messageBus: ${messageBusEnabled ? 'YES' : 'NO'}`);
-                }
                 registry.registerTool(new ToolClass(...toolArgs));
             }
         };
@@ -778,16 +1047,86 @@ export class Config {
         }
         registerCoreTool(WriteFileTool, this);
         registerCoreTool(WebFetchTool, this);
-        registerCoreTool(ReadManyFilesTool, this);
         registerCoreTool(ShellTool, this);
         registerCoreTool(MemoryTool);
         registerCoreTool(WebSearchTool, this);
         if (this.getUseWriteTodos()) {
             registerCoreTool(WriteTodosTool, this);
         }
+        // Register Subagents as Tools
+        if (this.getCodebaseInvestigatorSettings().enabled) {
+            const definition = this.agentRegistry.getDefinition('codebase_investigator');
+            if (definition) {
+                // We must respect the main allowed/exclude lists for agents too.
+                const allowedTools = this.getAllowedTools();
+                const isAllowed = !allowedTools || allowedTools.includes(definition.name);
+                if (isAllowed) {
+                    const messageBusEnabled = this.getEnableMessageBusIntegration();
+                    const wrapper = new SubagentToolWrapper(definition, this, messageBusEnabled ? this.getMessageBus() : undefined);
+                    registry.registerTool(wrapper);
+                }
+            }
+        }
         await registry.discoverAllTools();
+        registry.sortTools();
         return registry;
     }
+    /**
+     * Get hooks configuration
+     */
+    getHooks() {
+        return this.hooks;
+    }
+    /**
+     * Get experiments configuration
+     */
+    getExperiments() {
+        return this.experiments;
+    }
+    /**
+     * Set experiments configuration
+     */
+    setExperiments(experiments) {
+        this.experiments = experiments;
+        const flagSummaries = Object.entries(experiments.flags ?? {})
+            .sort(([a], [b]) => a.localeCompare(b))
+            .map(([flagId, flag]) => {
+            const summary = { flagId };
+            if (flag.boolValue !== undefined) {
+                summary['boolValue'] = flag.boolValue;
+            }
+            if (flag.floatValue !== undefined) {
+                summary['floatValue'] = flag.floatValue;
+            }
+            if (flag.intValue !== undefined) {
+                summary['intValue'] = flag.intValue;
+            }
+            if (flag.stringValue !== undefined) {
+                summary['stringValue'] = flag.stringValue;
+            }
+            const int32Length = flag.int32ListValue?.values?.length ?? 0;
+            if (int32Length > 0) {
+                summary['int32ListLength'] = int32Length;
+            }
+            const stringListLength = flag.stringListValue?.values?.length ?? 0;
+            if (stringListLength > 0) {
+                summary['stringListLength'] = stringListLength;
+            }
+            return summary;
+        });
+        const summary = {
+            experimentIds: experiments.experimentIds ?? [],
+            flags: flagSummaries,
+        };
+        const summaryString = inspect(summary, {
+            depth: null,
+            maxArrayLength: null,
+            maxStringLength: null,
+            breakLength: 80,
+            compact: false,
+        });
+        debugLogger.debug('Experiments loaded', summaryString);
+    }
 }
 // Export model constants for use in CLI
 export { DEFAULT_GEMINI_FLASH_MODEL };