@adminforth/completion-adapter-google-gemini 2.0.3 → 2.0.5

package/dist/index.js

@@ -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,6 +19,20 @@ export default class CompletionAdapterGoogleGemini
     }
   }
 
+  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;

package/package.json

@@ -1,13 +1,12 @@
 {
   "name": "@adminforth/completion-adapter-google-gemini",
-  "version": "2.0.3",
+  "version": "2.0.5",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",
   "scripts": {
     "build": "tsc && npm version patch",
-    "rollout": "npm run build && npm publish --access public",
-    "prepare": "npm link adminforth"
+    "rollout": "npm run build && npm publish --access public"
   },
   "keywords": [],
   "author": "",
@@ -18,7 +17,7 @@
     "p-retry": "^7.1.1"
   },
   "peerDependencies": {
-    "adminforth": "next"
+    "adminforth": "^2.24.0"
   },
   "devDependencies": {
     "typescript": "^5.9.3"