From cdbe2fffd998218cf9836f5303f2286dbebb52ff Mon Sep 17 00:00:00 2001
From: Bryan Morgan <bryanmorgan@google.com>
Date: Fri, 11 Jul 2025 11:25:30 -0400
Subject: Added in proper checks for customer tiers in 429/Quota error
 messaging (#3863)

Co-authored-by: Ioannis Papapanagiotou <iduckhd@hotmail.com>
---
 packages/core/src/config/config.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'packages/core/src/config')

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;
   }
-- 
cgit v1.2.3