Set userId in Langchain Callback handler

langfuse · Oct 12, 2023 · 8cd9be5 · 8cd9be5
1 parent 1a76edf
commit 8cd9be5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/langfuse-langchain/src/callback.ts b/langfuse-langchain/src/callback.ts
@@ -24,7 +24,9 @@ type KeyParams = {
   secretKey: string;
 } & LangfuseOptions;
 
-type ConstructorParams = RootParams | KeyParams;
+type ConstructorParams = (RootParams | KeyParams) & {
+  userId?: string;
+};
 
 export class CallbackHandler extends BaseCallbackHandler {
   name = "CallbackHandler";
@@ -33,6 +35,7 @@ export class CallbackHandler extends BaseCallbackHandler {
   observationId?: string;
   rootObservationId?: string;
   topLevelObservationId?: string;
+  userId?: string;
 
   constructor(params: ConstructorParams) {
     super();
@@ -43,6 +46,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     } else {
       this.langfuse = new Langfuse({ ...params, persistence: "memory" });
     }
+    this.userId = params.userId;
   }
 
   async flushAsync(): Promise<any> {
@@ -161,6 +165,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         id: runId,
         name: serialized.id.at(-1)?.toString(),
         metadata: this.joinTagsAndMetaData(tags, metadata),
+        userId: this.userId,
       });
       this.traceId = runId;
     }