@fallom/trace 0.2.0 → 0.2.1

package/dist/index.d.mts

@@ -155,12 +155,10 @@ declare class FallomSession {
         version?: number;
     }): Promise<string>;
     /**
-     * Wrap a Vercel AI SDK model to trace all calls.
+     * Wrap a Vercel AI SDK model to trace all calls (PostHog style).
+     * Returns the same model type with tracing injected.
      */
-    traceModel<T extends {
-        doGenerate?: (...args: any[]) => Promise<any>;
-        doStream?: (...args: any[]) => Promise<any>;
-    }>(model: T): T;
+    traceModel<T>(model: T): T;
     /** Wrap OpenAI client. Delegates to shared wrapper. */
     wrapOpenAI<T extends {
         chat: {

package/dist/index.d.ts

@@ -155,12 +155,10 @@ declare class FallomSession {
         version?: number;
     }): Promise<string>;
     /**
-     * Wrap a Vercel AI SDK model to trace all calls.
+     * Wrap a Vercel AI SDK model to trace all calls (PostHog style).
+     * Returns the same model type with tracing injected.
      */
-    traceModel<T extends {
-        doGenerate?: (...args: any[]) => Promise<any>;
-        doStream?: (...args: any[]) => Promise<any>;
-    }>(model: T): T;
+    traceModel<T>(model: T): T;
     /** Wrap OpenAI client. Delegates to shared wrapper. */
     wrapOpenAI<T extends {
         chat: {

package/dist/index.js

@@ -2021,13 +2021,15 @@ var FallomSession = class {
     return get3(configKey, this.ctx.sessionId, opts);
   }
   /**
-   * Wrap a Vercel AI SDK model to trace all calls.
+   * Wrap a Vercel AI SDK model to trace all calls (PostHog style).
+   * Returns the same model type with tracing injected.
    */
   traceModel(model) {
     const ctx = this.ctx;
     const tracedModel = Object.create(model);
-    if (model.doGenerate) {
-      const originalDoGenerate = model.doGenerate.bind(model);
+    const m = model;
+    if (m.doGenerate) {
+      const originalDoGenerate = m.doGenerate.bind(model);
       tracedModel.doGenerate = async function(...args) {
         if (!isInitialized()) return originalDoGenerate(...args);
         const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
@@ -2083,8 +2085,8 @@ var FallomSession = class {
         }
       };
     }
-    if (model.doStream) {
-      const originalDoStream = model.doStream.bind(model);
+    if (m.doStream) {
+      const originalDoStream = m.doStream.bind(model);
       tracedModel.doStream = async function(...args) {
         if (!isInitialized()) return originalDoStream(...args);
         const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();

package/dist/index.mjs

@@ -1743,13 +1743,15 @@ var FallomSession = class {
     return get2(configKey, this.ctx.sessionId, opts);
   }
   /**
-   * Wrap a Vercel AI SDK model to trace all calls.
+   * Wrap a Vercel AI SDK model to trace all calls (PostHog style).
+   * Returns the same model type with tracing injected.
    */
   traceModel(model) {
     const ctx = this.ctx;
     const tracedModel = Object.create(model);
-    if (model.doGenerate) {
-      const originalDoGenerate = model.doGenerate.bind(model);
+    const m = model;
+    if (m.doGenerate) {
+      const originalDoGenerate = m.doGenerate.bind(model);
       tracedModel.doGenerate = async function(...args) {
         if (!isInitialized()) return originalDoGenerate(...args);
         const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
@@ -1805,8 +1807,8 @@ var FallomSession = class {
         }
       };
     }
-    if (model.doStream) {
-      const originalDoStream = model.doStream.bind(model);
+    if (m.doStream) {
+      const originalDoStream = m.doStream.bind(model);
       tracedModel.doStream = async function(...args) {
         if (!isInitialized()) return originalDoStream(...args);
         const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fallom/trace",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Model A/B testing and tracing for LLM applications. Zero latency, production-ready.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",