From 4d53b80657b38f8179efd3728fccc7c1f45886c3 Mon Sep 17 00:00:00 2001
From: Marc Klingen <git@marcklingen.com>
Date: Fri, 6 Sep 2024 21:53:02 +0200
Subject: [PATCH] feat: track user.email in langfuse filter pipeline and remove
 wrong token count

---
 examples/filters/langfuse_filter_pipeline.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/examples/filters/langfuse_filter_pipeline.py b/examples/filters/langfuse_filter_pipeline.py
index 7cc373f..8130c20 100644
--- a/examples/filters/langfuse_filter_pipeline.py
+++ b/examples/filters/langfuse_filter_pipeline.py
@@ -2,7 +2,7 @@
 title: Langfuse Filter Pipeline
 author: open-webui
 date: 2024-05-30
-version: 1.1
+version: 1.2
 license: MIT
 description: A filter pipeline that uses Langfuse.
 requirements: langfuse
@@ -90,8 +90,8 @@ class Pipeline:
         trace = self.langfuse.trace(
             name=f"filter:{__name__}",
             input=body,
-            user_id=user["id"],
-            metadata={"name": user["name"]},
+            user_id=user["email"],
+            metadata={"user_name": user["name"], "user_id": user["id"]},
             session_id=body["chat_id"],
         )
 
@@ -119,10 +119,6 @@ class Pipeline:
 
         generation.end(
             output=generated_message,
-            usage={
-                "totalCost": (len(user_message) + len(generated_message)) / 1000,
-                "unit": "CHARACTERS",
-            },
             metadata={"interface": "open-webui"},
         )