fix(responses-bridge): return CustomStreamWrapper from the completed-response stream helper

This commit is contained in:
mateo-berri 2026-07-27 14:48:18 -07:00
parent 396554c7a9
commit 4299c6d191

View file

@ -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