weflayr 0.1.0 → 0.1.2

package/README.md

@@ -19,13 +19,13 @@ npm install weflayr
 
 Requires Node.js 18+ and `openai>=4.0.0`.
 
----
+--- 
 
 ## Implementation steps
 
 ### 1. Get your credentials
 
-Sign in at [weflayr.com](https://weflayr.com), create a **Flare**, and copy your `client_id` and `client_secret`.
+Sign in at [weflayr.com](https://weflayr.com), create a **Flayr**, and copy your `client_id` and `client_secret`.
 
 ### 2. Set environment variables
 
@@ -121,16 +121,6 @@ By default, message content and prompt text are stripped before sending. Fields
 
 ---
 
-## Optional: forward traces to your own collector
-
-Set `WEFLAYR_COLLECTOR_ENDPOINT` to also send OTLP traces to a collector of your choice (e.g. Jaeger, Grafana Tempo):
-
-```bash
-WEFLAYR_COLLECTOR_ENDPOINT=http://localhost:4318/v1/traces
-```
-
----
-
 ## Full example
 
 ```js

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "weflayr",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Drop-in instrumented wrappers for AI clients with zero-overhead telemetry",
   "type": "module",
   "exports": {
@@ -30,7 +30,7 @@
   "homepage": "https://weflayr.com",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/WeFlayr/public-mirror-js-sdk"
+    "url": "git+https://github.com/WeFlayr/public-mirror-js-sdk.git"
   },
   "bugs": {
     "url": "https://github.com/WeFlayr/public-mirror-js-sdk/issues"

package/src/openai.js

@@ -104,11 +104,13 @@ const PATCHED_PATHS = new Set(ROUTES.map((r) => r.name));
 export function instrument(client) {
   const tracer = trace.getTracer("weflayr-openai");
 
+  const providerOpts = { provider: "openai" };
+
   // 1. Patch all explicitly defined routes with their precise extractors.
   for (const route of ROUTES) {
-    route.set(client, makeWrapper(route.get(client), route, tracer));
+    route.set(client, makeWrapper(route.get(client), route, tracer, providerOpts));
   }
 
   // 2. Wrap the whole client in a Proxy that handles anything not in ROUTES.
-  return deepFallbackProxy(client, tracer, PATCHED_PATHS);
+  return deepFallbackProxy(client, tracer, PATCHED_PATHS, "", providerOpts);
 }

package/src/weflayr.js

@@ -117,13 +117,14 @@ export async function setupWeflayr() {
   new BasicTracerProvider({ spanProcessors }).register();
 }
 
-export function makeWrapper(original, route, tracer) {
+export function makeWrapper(original, route, tracer, { provider } = {}) {
   return async function ({ tags: callTags, ...params }) {
     const tags = { ...GLOBAL_TAGS, ...callTags };
+    const before = { ...(provider ? { provider } : {}), ...route.before(params) };
     const span = tracer.startSpan(route.name, {
       attributes: {
         "weflayr.event_id": randomUUID(),
-        "weflayr.before":   JSON.stringify(route.before(params)),
+        "weflayr.before":   JSON.stringify(before),
         "weflayr.tags":     JSON.stringify(tags),
       },
     });
@@ -144,7 +145,7 @@ export function makeWrapper(original, route, tracer) {
   };
 }
 
-export function makeFallbackWrapper(fn, target, name, tracer) {
+export function makeFallbackWrapper(fn, target, name, tracer, { provider } = {}) {
   return function (...args) {
     let callArgs = args;
     let tags = GLOBAL_TAGS;
@@ -156,7 +157,7 @@ export function makeFallbackWrapper(fn, target, name, tracer) {
       tags = { ...GLOBAL_TAGS, ...callTags };
     }
 
-    const before = { model: callArgs[0]?.model };
+    const before = { ...(provider ? { provider } : {}), model: callArgs[0]?.model };
     const span = tracer.startSpan(name, {
       attributes: {
         "weflayr.event_id": randomUUID(),
@@ -204,7 +205,7 @@ export function makeFallbackWrapper(fn, target, name, tracer) {
 
 // Recursively wraps every function on `obj` that is not already covered by patchedPaths.
 // `path` tracks the dotted property path (e.g. "images.generate").
-export function deepFallbackProxy(obj, tracer, patchedPaths, path = "") {
+export function deepFallbackProxy(obj, tracer, patchedPaths, path = "", opts = {}) {
   return new Proxy(obj, {
     get(target, prop) {
       if (typeof prop !== "string") return Reflect.get(target, prop);
@@ -215,11 +216,11 @@ export function deepFallbackProxy(obj, tracer, patchedPaths, path = "") {
       if (typeof val === "function") {
         // Already patched by explicit routes — return the existing wrapper as-is.
         if (patchedPaths.has(fullPath)) return val;
-        return makeFallbackWrapper(val, target, fullPath, tracer);
+        return makeFallbackWrapper(val, target, fullPath, tracer, opts);
       }
 
       if (val !== null && typeof val === "object") {
-        return deepFallbackProxy(val, tracer, patchedPaths, fullPath);
+        return deepFallbackProxy(val, tracer, patchedPaths, fullPath, opts);
       }
 
       return val;