weflayr 0.9.0 → 0.10.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "weflayr",
-  "version": "0.9.0",
+  "version": "0.10.3",
   "description": "Weflayr Node.js SDK — instrument any LLM client via JS Proxy",
   "main": "src/index.js",
   "types": "index.d.ts",

package/src/instrument.js

@@ -3,13 +3,15 @@
 const https = require('https');
 const http = require('http');
 const { randomUUID } = require('crypto');
+const { AsyncLocalStorage } = require('async_hooks');
 const { version: SDK_VERSION } = require('../package.json');
 const SDK_LANGUAGE = 'javascript';
 
 let _config = null;
-let _propagated = {};
+const _asyncStore = new AsyncLocalStorage();
 
 function configure(intakeUrl, clientId, clientSecret, settings) {
+  _asyncStore.enterWith({});
   if (settings.ignore_fields && settings.allow_fields) {
     console.warn('[weflayr] both ignore_fields and allow_fields are set - no events will be sent to the intake API');
     _config = { intakeUrl, clientId, clientSecret, settings, _blocked: true };
@@ -194,7 +196,7 @@ function _extractTags(args) {
 
   const tags = {
     ...(_config.settings.default_tags || {}),
-    ..._propagated,
+    ..._asyncStore.getStore(),
     ...callTags,
   };
 
@@ -202,9 +204,10 @@ function _extractTags(args) {
 }
 
 function weflayr_propagate(key, value) {
-  _propagated[key] = value;
+  _asyncStore.getStore()[key] = value;
 }
 
+
 function _isAsyncIterable(value) {
   return value != null && typeof value[Symbol.asyncIterator] === 'function';
 }