ak-gemini 2.0.5 → 2.0.7

package/base.js

@@ -98,6 +98,9 @@ class BaseGemini {
 		this.resourceExhaustedRetries = options.resourceExhaustedRetries ?? 5;
 		this.resourceExhaustedDelay = options.resourceExhaustedDelay ?? 1000;
 
+		// ── Health Check ──
+		this.healthCheck = options.healthCheck ?? false;
+
 		// ── Logging ──
 		this._configureLogLevel(options.logLevel);
 
@@ -192,11 +195,13 @@ class BaseGemini {
 		const chatOptions = this._getChatCreateOptions();
 		this.chatSession = this.genAIClient.chats.create(chatOptions);
 
-		try {
-			await this.genAIClient.models.list();
-			log.debug(`${this.constructor.name}: API connection successful.`);
-		} catch (e) {
-			throw new Error(`${this.constructor.name} initialization failed: ${e.message}`);
+		if (this.healthCheck) {
+			try {
+				await this._withRetry(() => this.genAIClient.models.list());
+				log.debug(`${this.constructor.name}: API connection successful.`);
+			} catch (e) {
+				throw new Error(`${this.constructor.name} initialization failed: ${e.message}`);
+			}
 		}
 
 		log.debug(`${this.constructor.name}: Chat session initialized.`);

package/index.cjs

@@ -363,6 +363,7 @@ var BaseGemini = class {
     }
     this.resourceExhaustedRetries = options.resourceExhaustedRetries ?? 5;
     this.resourceExhaustedDelay = options.resourceExhaustedDelay ?? 1e3;
+    this.healthCheck = options.healthCheck ?? false;
     this._configureLogLevel(options.logLevel);
     this.labels = options.labels || {};
     this.enableGrounding = options.enableGrounding || false;
@@ -425,11 +426,13 @@ var BaseGemini = class {
     logger_default.debug(`Initializing ${this.constructor.name} chat session with model: ${this.modelName}...`);
     const chatOptions = this._getChatCreateOptions();
     this.chatSession = this.genAIClient.chats.create(chatOptions);
-    try {
-      await this.genAIClient.models.list();
-      logger_default.debug(`${this.constructor.name}: API connection successful.`);
-    } catch (e) {
-      throw new Error(`${this.constructor.name} initialization failed: ${e.message}`);
+    if (this.healthCheck) {
+      try {
+        await this._withRetry(() => this.genAIClient.models.list());
+        logger_default.debug(`${this.constructor.name}: API connection successful.`);
+      } catch (e) {
+        throw new Error(`${this.constructor.name} initialization failed: ${e.message}`);
+      }
     }
     logger_default.debug(`${this.constructor.name}: Chat session initialized.`);
   }

package/package.json

@@ -2,7 +2,7 @@
 	"name": "ak-gemini",
 	"author": "ak@mixpanel.com",
 	"description": "AK's Generative AI Helper for doing... everything",
-	"version": "2.0.5",
+	"version": "2.0.7",
 	"main": "index.js",
 	"files": [
 		"index.js",

package/types.d.ts

@@ -174,6 +174,9 @@ export interface BaseGeminiOptions {
   resourceExhaustedRetries?: number;
   /** Initial backoff delay in ms for 429 retries, doubles each attempt (default: 1000) */
   resourceExhaustedDelay?: number;
+
+  /** Run models.list() health check during init() (default: false) */
+  healthCheck?: boolean;
 }
 
 export interface TransformerOptions extends BaseGeminiOptions {