diff options
Diffstat (limited to 'packages/core/src/config/config.ts')
| -rw-r--r-- | packages/core/src/config/config.ts | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/packages/core/src/config/config.ts b/packages/core/src/config/config.ts index 12767133..dc85c61a 100644 --- a/packages/core/src/config/config.ts +++ b/packages/core/src/config/config.ts @@ -11,6 +11,7 @@ import { ContentGeneratorConfig, createContentGeneratorConfig, } from '../core/contentGenerator.js'; +import { UserTierId } from '../code_assist/types.js'; import { ToolRegistry } from '../tools/tool-registry.js'; import { LSTool } from '../tools/ls.js'; import { ReadFileTool } from '../tools/read-file.js'; @@ -323,6 +324,14 @@ export class Config { return this.quotaErrorOccurred; } + async getUserTier(): Promise<UserTierId | undefined> { + if (!this.geminiClient) { + return undefined; + } + const generator = this.geminiClient.getContentGenerator(); + return await generator.getTier?.(); + } + getEmbeddingModel(): string { return this.embeddingModel; } |
