From 0b7e60979530492328d8dc3c78098a2e75be7af2 Mon Sep 17 00:00:00 2001 From: Rich Tong <1782087+richtong@users.noreply.github.com> Date: Sat, 13 Jun 2026 14:07:58 -0700 Subject: [PATCH] fix(mlflow): call model_dump() instead of passing method reference in _add_chunk_events choice.delta.model_dump (without parens) passes the bound method object to json.dumps instead of calling it, causing _add_chunk_events to always raise an exception silently. This prevents _end_span_or_trace from being called, so zero MLflow traces are recorded for streaming responses. Fixes: streaming traces not appearing in MLflow experiment tne-costs --- litellm/integrations/mlflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/integrations/mlflow.py b/litellm/integrations/mlflow.py index 6378e55f7e1..ea929855672 100644 --- a/litellm/integrations/mlflow.py +++ b/litellm/integrations/mlflow.py @@ -157,7 +157,7 @@ class MlflowLogger(CustomLogger): SpanEvent( name="streaming_chunk", attributes={ - "delta": json.dumps(choice.delta.model_dump, default=str) + "delta": json.dumps(choice.delta.model_dump(), default=str) }, ) )