@adminforth/completion-adapter-google-gemini 2.0.2 → 2.0.4

package/dist/index.js

@@ -12,7 +12,7 @@ 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) {
+        this.complete = (content_1, ...args_1) => __awaiter(this, [content_1, ...args_1], void 0, function* (content, stop = ["."], maxTokens = 50, outputSchema) {
             const ai = new GoogleGenAI({
                 apiKey: this.options.geminiApiKey,
             });
@@ -22,7 +22,7 @@ export default class CompletionAdapterGoogleGemini {
                     const response = yield ai.models.generateContent({
                         model: this.options.model || "gemini-3-flash-preview",
                         contents: content,
-                        config: Object.assign({ maxOutputTokens: maxTokens }, this.options.extraRequestBodyParameters),
+                        config: Object.assign({ responseJsonSchema: outputSchema ? outputSchema : undefined, maxOutputTokens: maxTokens }, this.options.extraRequestBodyParameters),
                     });
                     logger.debug(`Google Gemini SUCCESSFUL API response: ${response}`);
                     return {
@@ -49,4 +49,17 @@ export default class CompletionAdapterGoogleGemini {
             throw new Error("geminiApiKey is required");
         }
     }
+    measureTokensCount(content) {
+        return __awaiter(this, void 0, void 0, function* () {
+            // Implement token counting logic here
+            const ai = new GoogleGenAI({
+                apiKey: this.options.geminiApiKey,
+            });
+            const countTokensResponse = yield ai.models.countTokens({
+                model: "gemini-2.0-flash",
+                contents: content,
+            });
+            return countTokensResponse.totalTokens;
+        });
+    }
 }

package/index.ts

@@ -19,7 +19,21 @@ export default class CompletionAdapterGoogleGemini
     }
   }
 
-  complete = async (content: string, stop = ["."], maxTokens = 50): Promise<{
+  async measureTokensCount(content: string): Promise<number> {
+    // Implement token counting logic here
+    const ai = new GoogleGenAI({
+      apiKey: this.options.geminiApiKey,
+    });
+    const countTokensResponse = await ai.models.countTokens({
+      model: "gemini-2.0-flash",
+      contents: content,
+    });
+
+    return countTokensResponse.totalTokens;
+  }
+
+
+  complete = async (content: string, stop = ["."], maxTokens = 50, outputSchema?: any): Promise<{
     content?: string;
     finishReason?: string;
     error?: string;
@@ -36,6 +50,7 @@ export default class CompletionAdapterGoogleGemini
           model: this.options.model || "gemini-3-flash-preview",
           contents: content,
           config: {
+            responseJsonSchema: outputSchema ? outputSchema : undefined,
             maxOutputTokens: maxTokens,
             ...this.options.extraRequestBodyParameters,
           },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adminforth/completion-adapter-google-gemini",
-  "version": "2.0.2",
+  "version": "2.0.4",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",