diff --git a/packages/cli/src/gemini.tsx b/packages/cli/src/gemini.tsx index cba18047..a0cf352e 100644 --- a/packages/cli/src/gemini.tsx +++ b/packages/cli/src/gemini.tsx @@ -34,6 +34,9 @@ import { logUserPrompt, AuthType, getOauthClient, + logIdeConnection, + IdeConnectionEvent, + IdeConnectionType, } from '@google/gemini-cli-core'; import { validateAuthMethod } from './config/auth.js'; import { setMaxSizedBoxDebugging } from './ui/components/shared/MaxSizedBox.js'; @@ -188,6 +191,11 @@ export async function main() { await config.initialize(); + if (config.getIdeMode() && config.getIdeModeFeature()) { + await config.getIdeClient().connect(); + logIdeConnection(config, new IdeConnectionEvent(IdeConnectionType.START)); + } + // Load custom themes from settings themeManager.loadCustomThemes(settings.merged.customThemes); diff --git a/packages/core/index.ts b/packages/core/index.ts index 0c2da4b0..65a214ae 100644 --- a/packages/core/index.ts +++ b/packages/core/index.ts @@ -10,3 +10,8 @@ export { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_EMBEDDING_MODEL, } from './src/config/models.js'; +export { logIdeConnection } from './src/telemetry/loggers.js'; +export { + IdeConnectionEvent, + IdeConnectionType, +} from './src/telemetry/types.js'; diff --git a/packages/core/src/config/config.ts b/packages/core/src/config/config.ts index a06c4505..d02e4153 100644 --- a/packages/core/src/config/config.ts +++ b/packages/core/src/config/config.ts @@ -322,10 +322,6 @@ export class Config { this.folderTrust = params.folderTrust ?? false; this.ideMode = params.ideMode ?? false; this.ideClient = IdeClient.getInstance(); - if (this.ideMode && this.ideModeFeature) { - this.ideClient.connect(); - logIdeConnection(this, new IdeConnectionEvent(IdeConnectionType.START)); - } this.loadMemoryFromIncludeDirectories = params.loadMemoryFromIncludeDirectories ?? false; this.chatCompression = params.chatCompression;