diff options
Diffstat (limited to 'packages/cli/src/ui/hooks/usePrivacySettings.ts')
| -rw-r--r-- | packages/cli/src/ui/hooks/usePrivacySettings.ts | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/packages/cli/src/ui/hooks/usePrivacySettings.ts b/packages/cli/src/ui/hooks/usePrivacySettings.ts index bc98649b..47a62588 100644 --- a/packages/cli/src/ui/hooks/usePrivacySettings.ts +++ b/packages/cli/src/ui/hooks/usePrivacySettings.ts @@ -5,7 +5,12 @@ */ import { useState, useEffect, useCallback } from 'react'; -import { Config, CodeAssistServer, UserTierId } from '@google/gemini-cli-core'; +import { + Config, + CodeAssistServer, + UserTierId, + LoggingContentGenerator, +} from '@google/gemini-cli-core'; export interface PrivacyState { isLoading: boolean; @@ -80,7 +85,13 @@ export const usePrivacySettings = (config: Config) => { }; function getCodeAssistServer(config: Config): CodeAssistServer { - const server = config.getGeminiClient().getContentGenerator(); + let server = config.getGeminiClient().getContentGenerator(); + + // Unwrap LoggingContentGenerator if present + if (server instanceof LoggingContentGenerator) { + server = server.getWrapped(); + } + // Neither of these cases should ever happen. if (!(server instanceof CodeAssistServer)) { throw new Error('Oauth not being used'); |
