enh: langfuse filter

open-webui · Jun 2, 2024 · 5478b92 · 5478b92
1 parent 179c66d
commit 5478b92
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/examples/filters/langfuse_filter_pipeline.py b/examples/filters/langfuse_filter_pipeline.py
@@ -101,3 +101,18 @@ async def inlet(self, body: dict, user: Optional[dict] = None) -> dict:
         print(trace.get_trace_url())
 
         return body
+
+    async def outlet(self, body: dict, user: Optional[dict] = None) -> dict:
+        print(f"outlet:{__name__}")
+
+        trace = self.langfuse.trace(
+            name=f"filter:{__name__}",
+            input=body,
+            user_id=user["id"],
+            metadata={"name": user["name"]},
+            session_id=body["chat_id"],
+        )
+
+        print(trace.get_trace_url())
+
+        return body