@aispendguard/sdk 0.1.0

package/dist/wrap-gemini.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Wraps a Google Gemini GenerativeModel so every generateContent() call is
+ * automatically tracked. Returns the original response unchanged.
+ *
+ * Usage:
+ *   import { init, wrapGemini } from "@aispendguard/sdk";
+ *   import { GoogleGenerativeAI } from "@google/generative-ai";
+ *
+ *   init({ apiKey: "asg_...", defaultTags: { feature: "chat", route: "/api/chat" } });
+ *   const genAI = new GoogleGenerativeAI(process.env.GOOGLE_AI_API_KEY!);
+ *   const model = wrapGemini(genAI.getGenerativeModel({ model: "gemini-2.0-flash" }), "gemini-2.0-flash");
+ *
+ *   // Automatically tracked:
+ *   const result = await model.generateContent("Hello");
+ */
+export declare function wrapGemini<T extends object>(client: T, modelName: string): T;

package/dist/wrap-gemini.js

@@ -0,0 +1,82 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.wrapGemini = wrapGemini;
+const gemini_1 = require("./gemini");
+const index_1 = require("./index");
+/**
+ * Wraps a Google Gemini GenerativeModel so every generateContent() call is
+ * automatically tracked. Returns the original response unchanged.
+ *
+ * Usage:
+ *   import { init, wrapGemini } from "@aispendguard/sdk";
+ *   import { GoogleGenerativeAI } from "@google/generative-ai";
+ *
+ *   init({ apiKey: "asg_...", defaultTags: { feature: "chat", route: "/api/chat" } });
+ *   const genAI = new GoogleGenerativeAI(process.env.GOOGLE_AI_API_KEY!);
+ *   const model = wrapGemini(genAI.getGenerativeModel({ model: "gemini-2.0-flash" }), "gemini-2.0-flash");
+ *
+ *   // Automatically tracked:
+ *   const result = await model.generateContent("Hello");
+ */
+function wrapGemini(client, modelName) {
+    const originalGenerate = client.generateContent;
+    if (typeof originalGenerate !== "function")
+        return client;
+    client.generateContent = async function wrappedGenerate(...args) {
+        const start = Date.now();
+        const result = await originalGenerate.apply(this, args);
+        const latencyMs = Date.now() - start;
+        try {
+            const client = getClientSafe();
+            if (!client)
+                return result;
+            const res = result;
+            const response = res.response;
+            const usage = (response?.usageMetadata ?? res.usageMetadata);
+            const resolvedModel = typeof response?.modelVersion === "string"
+                ? response.modelVersion
+                : undefined;
+            // Check for asgTags in the first argument if it's an object
+            let asgTags;
+            if (args[0] && typeof args[0] === "object" && !Array.isArray(args[0])) {
+                asgTags = args[0].asgTags;
+            }
+            const tags = mergeTags(client.defaultTags, asgTags, modelName);
+            const event = (0, gemini_1.createGeminiUsageEvent)({
+                model: modelName,
+                resolvedModel,
+                usage: usage,
+                latencyMs,
+                tags,
+            });
+            client.trackUsage(event).catch(logError);
+        }
+        catch {
+            // Never break user code
+        }
+        return result;
+    };
+    return client;
+}
+function mergeTags(defaults, overrides, model) {
+    return {
+        task_type: "chat",
+        feature: "default",
+        route: "default",
+        ...defaults,
+        ...overrides,
+        source: "auto-wrap",
+        model,
+    };
+}
+function getClientSafe() {
+    try {
+        return (0, index_1.getClient)();
+    }
+    catch {
+        return null;
+    }
+}
+function logError(err) {
+    console.warn(`[aispendguard-sdk] auto-wrap tracking failed: ${err instanceof Error ? err.message : String(err)}`);
+}

package/dist/wrap-openai.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Wraps an OpenAI client so every chat.completions.create() call is
+ * automatically tracked. Returns the original response unchanged.
+ *
+ * Usage:
+ *   import { init, wrapOpenAI } from "@aispendguard/sdk";
+ *   import OpenAI from "openai";
+ *
+ *   init({ apiKey: "asg_...", defaultTags: { feature: "chat", route: "/api/chat" } });
+ *   const openai = wrapOpenAI(new OpenAI());
+ *
+ *   // Automatically tracked:
+ *   const res = await openai.chat.completions.create({ model: "gpt-4o", messages: [...] });
+ */
+export declare function wrapOpenAI<T extends object>(client: T): T;

package/dist/wrap-openai.js

@@ -0,0 +1,81 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.wrapOpenAI = wrapOpenAI;
+const openai_1 = require("./openai");
+const index_1 = require("./index");
+/**
+ * Wraps an OpenAI client so every chat.completions.create() call is
+ * automatically tracked. Returns the original response unchanged.
+ *
+ * Usage:
+ *   import { init, wrapOpenAI } from "@aispendguard/sdk";
+ *   import OpenAI from "openai";
+ *
+ *   init({ apiKey: "asg_...", defaultTags: { feature: "chat", route: "/api/chat" } });
+ *   const openai = wrapOpenAI(new OpenAI());
+ *
+ *   // Automatically tracked:
+ *   const res = await openai.chat.completions.create({ model: "gpt-4o", messages: [...] });
+ */
+function wrapOpenAI(client) {
+    const chat = client.chat;
+    if (!chat || typeof chat !== "object")
+        return client;
+    const completions = chat.completions;
+    if (!completions || typeof completions !== "object")
+        return client;
+    const originalCreate = completions.create;
+    if (typeof originalCreate !== "function")
+        return client;
+    completions.create = async function wrappedCreate(params, ...rest) {
+        const start = Date.now();
+        const result = await originalCreate.call(this, params, ...rest);
+        const latencyMs = Date.now() - start;
+        try {
+            const client = getClientSafe();
+            if (!client)
+                return result;
+            const res = result;
+            const model = params.model ?? "unknown";
+            const resolvedModel = typeof res.model === "string" ? res.model : undefined;
+            const usage = res.usage;
+            const asgOpts = params;
+            const tags = mergeTags(client.defaultTags, asgOpts.asgTags, model);
+            const event = (0, openai_1.createOpenAIUsageEvent)({
+                model,
+                resolvedModel,
+                usage: usage,
+                latencyMs,
+                tags,
+            });
+            client.trackUsage(event).catch(logError);
+        }
+        catch {
+            // Never break user code
+        }
+        return result;
+    };
+    return client;
+}
+function mergeTags(defaults, overrides, model) {
+    return {
+        task_type: "chat",
+        feature: "default",
+        route: "default",
+        ...defaults,
+        ...overrides,
+        source: "auto-wrap",
+        model,
+    };
+}
+function getClientSafe() {
+    try {
+        return (0, index_1.getClient)();
+    }
+    catch {
+        return null;
+    }
+}
+function logError(err) {
+    console.warn(`[aispendguard-sdk] auto-wrap tracking failed: ${err instanceof Error ? err.message : String(err)}`);
+}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@aispendguard/sdk",
+  "version": "0.1.0",
+  "description": "Tags-only SDK for tracking AI API spend with AISpendGuard. Supports OpenAI, Anthropic, Google Gemini, LangChain.js, and OpenTelemetry.",
+  "license": "MIT",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "LICENSE",
+    "README.md"
+  ],
+  "keywords": [
+    "ai",
+    "llm",
+    "cost",
+    "spending",
+    "tracking",
+    "openai",
+    "anthropic",
+    "gemini",
+    "langchain",
+    "observability"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/AISpendGuard/aispendguard-sdk.git"
+  },
+  "homepage": "https://aispendguard.com",
+  "bugs": {
+    "url": "https://github.com/AISpendGuard/aispendguard-sdk/issues"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "typecheck": "tsc -p tsconfig.json --noEmit",
+    "test": "npm run build && node --test tests/*.test.mjs",
+    "prepublishOnly": "npm run test"
+  },
+  "devDependencies": {
+    "@types/node": "^22.13.10",
+    "typescript": "^5.8.2"
+  }
+}