feat(langfuse-core): add traces forwarding for Defer integration (#41)

langfuse · Nov 10, 2023 · c12f969 · c12f969
1 parent 0e6c1c7
commit c12f969
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git a/langfuse-core/src/globalThis.d.ts b/langfuse-core/src/globalThis.d.ts
@@ -0,0 +1,16 @@
+export {};
+
+declare interface DeferLangFuseTrace {
+  id: string;
+  name: string;
+  url: string;
+}
+
+declare global {
+  namespace globalThis {
+    // eslint-disable-next-line no-var
+    var __deferRuntime: {
+      langfuseTraces: (traces: DeferLangFuseTrace[]) => void;
+    };
+  }
+}
diff --git a/langfuse-core/src/index.ts b/langfuse-core/src/index.ts
@@ -452,7 +452,21 @@ export abstract class LangfuseCore extends LangfuseCoreStateless {
 
   trace(body?: CreateLangfuseTraceBody): LangfuseTraceClient {
     const id = this.traceStateless(body ?? {});
-    return new LangfuseTraceClient(this, id);
+    const t = new LangfuseTraceClient(this, id);
+    if (process.env.DEFER && body) {
+      try {
+        if (globalThis.__deferRuntime) {
+          __deferRuntime.langfuseTraces([
+            {
+              id: id,
+              name: body.name || "",
+              url: t.getTraceUrl(),
+            },
+          ]);
+        }
+      } catch {}
+    }
+    return t;
   }
 
   span(body: CreateLangfuseSpanBody): LangfuseSpanClient {