From fdea57c401966af7fa705b1785f40176671059fd Mon Sep 17 00:00:00 2001 From: Darsh Joshi Date: Thu, 15 Oct 2026 23:05:30 -0400 Subject: [PATCH] chore: return a Mapping from the inject id restore, as the type gate asks --- litellm/responses/streaming_iterator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/responses/streaming_iterator.py b/litellm/responses/streaming_iterator.py index 74d8aff2549..510df354206 100644 --- a/litellm/responses/streaming_iterator.py +++ b/litellm/responses/streaming_iterator.py @@ -2019,7 +2019,7 @@ def _restore_wrapped_ids_in_response_create(msg_obj: Mapping[str, object]) -> di return {**msg_obj, **top_fields, **restored_nested} -def _restore_wrapped_ids_in_response_inject(msg_obj: Mapping[str, object]) -> dict[str, object] | None: +def _restore_wrapped_ids_in_response_inject(msg_obj: Mapping[str, object]) -> Mapping[str, object] | None: response_id: Final = msg_obj.get("response_id") original_id: Final = ( ResponsesAPIRequestUtils.decode_previous_response_id_to_original_previous_response_id(response_id)