From 4299c6d191057ac06af49700baba16028481c8fc Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Mon, 27 Jul 2026 14:48:18 -0700
Subject: [PATCH] fix(responses-bridge): return CustomStreamWrapper from the
 completed-response stream helper

---
 .../litellm_responses_transformation/handler.py                 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/completion_extras/litellm_responses_transformation/handler.py b/litellm/completion_extras/litellm_responses_transformation/handler.py
index cf517440cd5..15f5b28e30e 100644
--- a/litellm/completion_extras/litellm_responses_transformation/handler.py
+++ b/litellm/completion_extras/litellm_responses_transformation/handler.py
@@ -354,7 +354,7 @@ class ResponsesToCompletionBridgeHandler:
         custom_llm_provider: str,
         logging_obj: "LiteLLMLoggingObj",
         json_mode: bool | None,
-    ) -> Any:
+    ) -> "CustomStreamWrapper":
         from litellm.litellm_core_utils.streaming_handler import CustomStreamWrapper
         from litellm.llms.base_llm.base_model_iterator import MockResponseIterator