@rheonic/sdk 0.1.0-beta.1

package/dist/tokenEstimator.js

@@ -0,0 +1,64 @@
+import { encoding_for_model, get_encoding } from "@dqbd/tiktoken";
+import { sdkNodeConfig } from "./config.js";
+const encoderCache = new Map();
+export function prewarmTokenEstimator(model = null) {
+    try {
+        getEncoder(model);
+    }
+    catch {
+        return;
+    }
+}
+export function estimateInputTokensFromRequest(payload) {
+    if (!payload || typeof payload !== "object") {
+        return null;
+    }
+    const request = payload;
+    const text = extractTextForEstimation(request);
+    if (text === null) {
+        return null;
+    }
+    try {
+        const encoder = getEncoder(typeof request.model === "string" ? request.model : null);
+        const encodedLength = encoder.encode(text).length;
+        return Math.min(sdkNodeConfig.maxInputTokenEstimate, encodedLength);
+    }
+    catch {
+        return null;
+    }
+}
+function extractTextForEstimation(request) {
+    if (Array.isArray(request.messages)) {
+        try {
+            return JSON.stringify(request.messages);
+        }
+        catch {
+            return null;
+        }
+    }
+    if (typeof request.prompt === "string") {
+        return request.prompt;
+    }
+    return null;
+}
+function getEncoder(model) {
+    const cacheKey = model ?? sdkNodeConfig.defaultTokenizerEncoding;
+    const cached = encoderCache.get(cacheKey);
+    if (cached) {
+        return cached;
+    }
+    let encoder;
+    if (model) {
+        try {
+            encoder = encoding_for_model(model);
+        }
+        catch {
+            encoder = get_encoding(sdkNodeConfig.defaultTokenizerEncoding);
+        }
+    }
+    else {
+        encoder = get_encoding(sdkNodeConfig.defaultTokenizerEncoding);
+    }
+    encoderCache.set(cacheKey, encoder);
+    return encoder;
+}

package/package.json

@@ -0,0 +1,74 @@
+{
+  "name": "@rheonic/sdk",
+  "version": "0.1.0-beta.1",
+  "description": "Node.js SDK for Rheonic observability and protect preflight enforcement.",
+  "author": "Rheonic <founder@rheonic.dev>",
+  "license": "MIT",
+  "type": "module",
+  "sideEffects": false,
+  "main": "dist/index.js",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "!dist/*.test.d.ts",
+    "!dist/*.test.js",
+    "!dist/**/*.test.d.ts",
+    "!dist/**/*.test.js",
+    "README.md",
+    "CHANGELOG.md"
+  ],
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "test": "npm run build && c8 --check-coverage --lines 80 --functions 80 --branches 80 --statements 80 --reporter=text --reporter=lcov node --test dist/*.test.js"
+  },
+  "keywords": [
+    "llm",
+    "ai",
+    "observability",
+    "guardrails",
+    "openai",
+    "anthropic",
+    "google"
+  ],
+  "homepage": "https://beta.rheonic.dev",
+  "bugs": {
+    "url": "mailto:support@rheonic.dev"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@dqbd/tiktoken": "^1.0.22"
+  },
+  "peerDependencies": {
+    "@anthropic-ai/sdk": ">=0.39.0",
+    "@google/generative-ai": ">=0.21.0",
+    "openai": ">=4.0.0"
+  },
+  "peerDependenciesMeta": {
+    "@anthropic-ai/sdk": {
+      "optional": true
+    },
+    "@google/generative-ai": {
+      "optional": true
+    },
+    "openai": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@types/node": "^22.8.7",
+    "c8": "^10.1.3",
+    "typescript": "^5.6.3"
+  }
+}