@adminforth/completion-adapter-google-gemini 1.0.2 → 1.0.4

package/dist/index.js

@@ -8,30 +8,40 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 import { GoogleGenAI } from "@google/genai";
+import pRetry from 'p-retry';
+import { logger } from "adminforth";
 export default class CompletionAdapterGoogleGemini {
     constructor(options) {
         this.complete = (content_1, ...args_1) => __awaiter(this, [content_1, ...args_1], void 0, function* (content, stop = ["."], maxTokens = 50) {
-            var _a, _b;
             const ai = new GoogleGenAI({
                 apiKey: this.options.geminiApiKey,
             });
-            try {
-                const response = yield ai.models.generateContent({
-                    model: this.options.model || "gemini-3-flash-preview",
-                    contents: content,
-                    config: {
-                        temperature: (_b = (_a = this.options.expert) === null || _a === void 0 ? void 0 : _a.temperature) !== null && _b !== void 0 ? _b : 0.7,
-                        maxOutputTokens: maxTokens,
-                        stopSequences: stop,
-                    },
-                });
-                return {
-                    content: response.text,
-                };
-            }
-            catch (error) {
-                return { error: error.message };
-            }
+            const tryToGenerate = () => __awaiter(this, void 0, void 0, function* () {
+                var _a, _b;
+                logger.debug("Making Google Gemini API call");
+                try {
+                    const response = yield ai.models.generateContent({
+                        model: this.options.model || "gemini-3-flash-preview",
+                        contents: content,
+                        config: Object.assign({ temperature: (_b = (_a = this.options.expert) === null || _a === void 0 ? void 0 : _a.temperature) !== null && _b !== void 0 ? _b : 0.7, maxOutputTokens: maxTokens, stopSequences: stop }, this.options.extraRequestBodyParameters),
+                    });
+                    logger.debug(`Google Gemini SUCCESSFUL API response: ${response}`);
+                    return {
+                        content: response.text,
+                    };
+                }
+                catch (error) {
+                    logger.error(`Error during Google Gemini API call: ${error}`);
+                    throw new Error(`Error during Google Gemini API call: ${error}`);
+                }
+            });
+            const result = yield pRetry(tryToGenerate, {
+                retries: 5,
+                onFailedAttempt: ({ error, attemptNumber, retriesLeft, retriesConsumed }) => {
+                    logger.debug(`Attempt ${attemptNumber} failed. ${retriesLeft} retries left. ${retriesConsumed} retries consumed.`);
+                },
+            });
+            return result;
         });
         this.options = options;
     }

package/index.ts

@@ -1,7 +1,8 @@
 import type { AdapterOptions } from "./types.js";
 import type { CompletionAdapter } from "adminforth";
 import { GoogleGenAI } from "@google/genai";
-
+import pRetry from 'p-retry';
+import { logger } from "adminforth";
 
 export default class CompletionAdapterGoogleGemini
   implements CompletionAdapter
@@ -28,21 +29,34 @@ export default class CompletionAdapterGoogleGemini
       apiKey: this.options.geminiApiKey,
     });
 
-    try {
-      const response = await ai.models.generateContent({
-        model: this.options.model || "gemini-3-flash-preview",
-        contents: content,
-        config: {
-          temperature: this.options.expert?.temperature ?? 0.7,
-          maxOutputTokens: maxTokens,
-          stopSequences: stop,
-        },
-      });
-      return {
-        content: response.text,
-      };
-    } catch (error) {
-      return { error: (error as Error).message };
+    const tryToGenerate = async () => {
+      logger.debug("Making Google Gemini API call");
+      try {
+        const response = await ai.models.generateContent({
+          model: this.options.model || "gemini-3-flash-preview",
+          contents: content,
+          config: {
+            temperature: this.options.expert?.temperature ?? 0.7,
+            maxOutputTokens: maxTokens,
+            stopSequences: stop,
+            ...this.options.extraRequestBodyParameters,
+          },
+        });
+        logger.debug(`Google Gemini SUCCESSFUL API response: ${response}`);
+        return {
+          content: response.text,
+        };
+      } catch (error) {
+        logger.error(`Error during Google Gemini API call: ${error}`);
+        throw new Error(`Error during Google Gemini API call: ${error}`);
+      }
     }
+    const result = await pRetry(tryToGenerate, {
+      retries: 5,
+      onFailedAttempt: ({error, attemptNumber, retriesLeft, retriesConsumed}) => {
+        logger.debug(`Attempt ${attemptNumber} failed. ${retriesLeft} retries left. ${retriesConsumed} retries consumed.`);
+      },
+    })
+    return result;
   };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adminforth/completion-adapter-google-gemini",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",
@@ -15,6 +15,9 @@
   "description": "",
   "dependencies": {
     "@google/genai": "^1.35.0",
+    "p-retry": "^7.1.1"
+  },
+  "peerDependencies": {
     "adminforth": "next"
   },
   "devDependencies": {

package/types.ts

@@ -11,6 +11,12 @@ export interface AdapterOptions {
    */
   model?: string;
 
+
+  /**
+   * Additional request body parameters to include in the API request.
+   */
+  extraRequestBodyParameters?: Record<string, unknown>;
+
   /**
    * Expert settings
    */