From fce092a885cc0b0159600d821a6aa45baf71bc0a Mon Sep 17 00:00:00 2001
From: Pradeep Ramola <pradeep.ramola@somaiya.edu>
Date: Wed, 30 Sep 2026 18:28:32 -0400
Subject: [PATCH] fix(streaming): preserve complete custom chunks

---
 .../litellm_core_utils/streaming_handler.py   |  5 +++-
 .../test_streaming_handler.py                 | 28 +++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/litellm/litellm_core_utils/streaming_handler.py b/litellm/litellm_core_utils/streaming_handler.py
index 52cf323febf..4ea7327bdfc 100644
--- a/litellm/litellm_core_utils/streaming_handler.py
+++ b/litellm/litellm_core_utils/streaming_handler.py
@@ -1132,7 +1132,10 @@ class CustomStreamWrapper:
                 chunk.choices[0].finish_reason = None
             return _ProviderChunkEarlyReturn(chunk)
 
-        is_generic_chunk: Final = isinstance(chunk, dict) and generic_chunk_has_all_required_fields(chunk=chunk)
+        is_generic_chunk: Final = isinstance(chunk, dict) and (
+            generic_chunk_has_all_required_fields(chunk=chunk)
+            or (is_registered_custom_provider and _GCHUNK_REQUIRED_FIELDS <= chunk.keys())
+        )
         if (
             isinstance(chunk, dict)
             and not is_generic_chunk
diff --git a/tests/unit/litellm_core_utils/test_streaming_handler.py b/tests/unit/litellm_core_utils/test_streaming_handler.py
index 7b5bcaa262b..cc68551fdca 100644
--- a/tests/unit/litellm_core_utils/test_streaming_handler.py
+++ b/tests/unit/litellm_core_utils/test_streaming_handler.py
@@ -2551,6 +2551,34 @@ def test_chunk_creator_records_incomplete_usage_chunk(
     assert initialized_custom_stream_wrapper.chunks[-1].usage.prompt_tokens == 1
 
 
+def test_custom_provider_complete_generic_chunk_with_extra_fields_is_preserved(
+    monkeypatch: pytest.MonkeyPatch,
+):
+    custom_llm_provider = "my-custom-llm"
+    monkeypatch.setattr(litellm, "_custom_providers", [custom_llm_provider])
+    wrapper = CustomStreamWrapper(
+        completion_stream=iter(
+            [
+                {
+                    "text": "hello",
+                    "is_finished": True,
+                    "finish_reason": "stop",
+                    "usage": None,
+                    "custom_metadata": {"trace_id": "trace-1"},
+                }
+            ]
+        ),
+        model="custom-model",
+        logging_obj=MagicMock(),
+        custom_llm_provider=custom_llm_provider,
+    )
+
+    chunks = list(wrapper)
+
+    assert "".join(chunk.choices[0].delta.content or "" for chunk in chunks) == "hello"
+    assert chunks[-1].choices[0].finish_reason == "stop"
+
+
 def _run_dispatch(wrapper: CustomStreamWrapper, chunk):
     model_response = wrapper.model_response_creator()
     completion_obj = {"content": ""}