OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-11 16:10:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Get usage from usage field in message response body

Browse Source
This commit is contained in:
Jason Jensen 2025-02-25 17:04:14 +00:00
parent ff414797e9
commit 821c403327
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/filters/langfuse_filter_pipeline.py
View File

@ -239,10 +239,10 @@ class Pipeline:
# Extract usage if available
usage = None
if assistant_message_obj:
info = assistant_message_obj.get("info", {})
if isinstance(info, dict):
input_tokens = info.get("prompt_eval_count") or info.get("prompt_tokens")
output_tokens = info.get("eval_count") or info.get("completion_tokens")
message_usage = assistant_message_obj.get("usage", {})
if isinstance(message_usage, dict):
input_tokens = message_usage.get("prompt_eval_count") or message_usage.get("prompt_tokens")
output_tokens = message_usage.get("eval_count") or message_usage.get("completion_tokens")
if input_tokens is not None and output_tokens is not None:
usage = {
"input": input_tokens,