@kalaa/node 1.0.9 → 1.1.1

package/costmeter.js

@@ -479,6 +479,54 @@ function patchAnthropicModule() {
 // per-instance in the Models constructor, not on Models.prototype. Prototype
 // patching is a no-op here; we wrap the instance's own methods instead, right
 // after GoogleGenAI's constructor creates them.
+function patchGeminiModule() {
+  patchGeminiLegacy();
+  patchGeminiUnified();
+}
+
+function patchGeminiLegacy() {
+  let Mod;
+  try { Mod = require('@google/generative-ai'); } catch { return; }
+  const GoogleGenerativeAI = Mod.GoogleGenerativeAI;
+  if (!GoogleGenerativeAI || GoogleGenerativeAI.__cm_patched) return;
+
+  let GenerativeModel;
+  try {
+    const probe = new GoogleGenerativeAI('cm_probe');
+    const model = probe.getGenerativeModel({ model: 'gemini-1.5-flash' });
+    GenerativeModel = Object.getPrototypeOf(model).constructor;
+  } catch (e) {
+    log('gemini (legacy) probe failed:', e.message);
+    return;
+  }
+
+  if (GenerativeModel?.prototype?.generateContent && !GenerativeModel.prototype.generateContent.__cm) {
+    const orig = GenerativeModel.prototype.generateContent;
+    const wrapped = async function (params, ...rest) {
+      const t0 = Date.now();
+      const modelName = (this.model || 'gemini').replace(/^models\//, '');
+      let _user, clean = params;
+      if (params && typeof params === 'object' && !Array.isArray(params) && '_user' in params) {
+        ({ _user, ...clean } = params);
+      }
+      try {
+        const resp = await orig.call(this, clean, ...rest);
+        const usage = resp?.response?.usageMetadata || {};
+        track({ provider: 'gemini', model: modelName, usage, latencyMs: Date.now() - t0, endUser: _user });
+        return resp;
+      } catch (err) {
+        track({ provider: 'gemini', model: modelName, usage: {}, latencyMs: Date.now() - t0, status: 'error', meta: { error: String(err.message).slice(0, 200) }, endUser: _user });
+        throw err;
+      }
+    };
+    wrapped.__cm = true;
+    GenerativeModel.prototype.generateContent = wrapped;
+  }
+
+  GoogleGenerativeAI.__cm_patched = true;
+  log('gemini (legacy @google/generative-ai) module patched (prototype-level)');
+}
+
 function patchGeminiUnified() {
   let Mod;
   try { Mod = require('@google/genai'); } catch { return; } // not installed, skip

package/package.json

@@ -1,6 +1,6 @@
 {
 "name": "@kalaa/node",
-  "version": "1.0.9",
+  "version": "1.1.1",
   "description": "Drop-in AI cost tracking for Node.js and TypeScript. Auto-instruments OpenAI, Anthropic, and Gemini at the class level, with zero code at each call site.",
   "main": "costmeter.js",
   "types": "costmeter.d.ts",