OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-12 08:30:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #387 from marcklingen/fix-output-tracking-in-langfuse-pipe

Browse Source 
fix: include assistant message as trace.output in langfuse
This commit is contained in:
Timothy Jaeryang Baek 2025-01-06 02:21:03 -08:00 committed by GitHub
commit db29eb2956
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/filters/langfuse_filter_pipeline.py
View File

@ -44,6 +44,7 @@ class Pipeline:
} }
) )
self.langfuse = None self.langfuse = None
self.chat_traces = {}
self.chat_generations = {} self.chat_generations = {}
async def on_startup(self): async def on_startup(self):
@ -107,17 +108,18 @@ class Pipeline:
metadata={"interface": "open-webui"}, metadata={"interface": "open-webui"},
) )
self.chat_traces[body["chat_id"]] = trace
self.chat_generations[body["chat_id"]] = generation self.chat_generations[body["chat_id"]] = generation
print(trace.get_trace_url())
return body return body
async def outlet(self, body: dict, user: Optional[dict] = None) -> dict: async def outlet(self, body: dict, user: Optional[dict] = None) -> dict:
print(f"outlet:{__name__}") print(f"outlet:{__name__}")
print(f"Received body: {body}") print(f"Received body: {body}")
if body["chat_id"] not in self.chat_generations: if body["chat_id"] not in self.chat_generations or body["chat_id"] not in self.chat_traces:
return body return body
trace = self.chat_traces[body["chat_id"]]
generation = self.chat_generations[body["chat_id"]] generation = self.chat_generations[body["chat_id"]]
assistant_message = get_last_assistant_message(body["messages"]) assistant_message = get_last_assistant_message(body["messages"])
@ -138,6 +140,9 @@ class Pipeline:
} }
# Update generation # Update generation
trace.update(
output=assistant_message,
)
generation.end( generation.end(
output=assistant_message, output=assistant_message,
metadata={"interface": "open-webui"}, metadata={"interface": "open-webui"},
@ -145,6 +150,7 @@ class Pipeline:
) )
# Clean up the chat_generations dictionary # Clean up the chat_generations dictionary
del self.chat_traces[body["chat_id"]]
del self.chat_generations[body["chat_id"]] del self.chat_generations[body["chat_id"]]
return body return body