feat(responses): add safe follow-up input helper

Author: raashish1601

src/openai/types/responses/response.py

@@ -1,6 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import List, Union, Optional
+from typing import Any, List, Union, Optional, cast
 from typing_extensions import Literal, TypeAlias
 
 from .tool import Tool
@@ -23,6 +23,7 @@
 from .tool_choice_function import ToolChoiceFunction
 from ..shared.responses_model import ResponsesModel
 from .tool_choice_apply_patch import ToolChoiceApplyPatch
+from .response_input_item_param import ResponseInputItemParam
 
 __all__ = ["Response", "IncompleteDetails", "ToolChoice", "Conversation"]
 
@@ -319,3 +320,16 @@ def output_text(self) -> str:
                         texts.append(content.text)
 
         return "".join(texts)
+
+    @property
+    def output_as_input(self) -> List[ResponseInputItemParam]:
+        """Convenience property that converts `output` items into follow-up `input` items.
+
+        This omits unset and `None` fields so the returned items can be passed back
+        to `responses.create(..., input=...)` for a subsequent turn.
+        """
+
+        return [
+            cast(ResponseInputItemParam, output.to_dict(mode="json", exclude_none=True))
+            for output in cast(List[Any], self.output)
+        ]

tests/lib/responses/test_responses.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import json
+from typing import cast
 from typing_extensions import TypeVar
 
 import httpx
@@ -11,7 +12,7 @@
 from openai import OpenAI, AsyncOpenAI
 from openai._utils import assert_signatures_in_sync
 from openai._compat import parse_obj
-from openai.types.responses.response import Response
+from openai.types.responses import Response, ResponseReasoningItem
 
 from ...conftest import base_url
 from ..snapshots import make_snapshot_request
@@ -181,6 +182,69 @@ async def test_async_replayed_response_output_items_can_be_counted_without_null_
     assert request_body["input"] == EXPECTED_REPLAYED_OUTPUT_INPUT
 
 
+def test_output_as_input_omits_null_only_response_fields() -> None:
+    response = Response.construct(
+        id="resp_123",
+        created_at=1754925861,
+        model="o4-mini",
+        object="response",
+        output=[
+            {
+                "id": "rs_123",
+                "summary": [{"text": "Reasoning summary", "type": "summary_text"}],
+                "type": "reasoning",
+            },
+            {
+                "id": "msg_123",
+                "type": "message",
+                "status": "completed",
+                "content": [
+                    {
+                        "type": "output_text",
+                        "annotations": [],
+                        "text": "Paris.",
+                    }
+                ],
+                "role": "assistant",
+            },
+        ],
+        parallel_tool_calls=True,
+        tool_choice="auto",
+        tools=[],
+    )
+
+    reasoning_item = cast(ResponseReasoningItem, response.output[0])
+    assert reasoning_item.model_dump() == {
+        "id": "rs_123",
+        "summary": [{"text": "Reasoning summary", "type": "summary_text"}],
+        "type": "reasoning",
+        "content": None,
+        "encrypted_content": None,
+        "status": None,
+    }
+
+    assert response.output_as_input == [
+        {
+            "id": "rs_123",
+            "summary": [{"text": "Reasoning summary", "type": "summary_text"}],
+            "type": "reasoning",
+        },
+        {
+            "id": "msg_123",
+            "type": "message",
+            "status": "completed",
+            "content": [
+                {
+                    "type": "output_text",
+                    "annotations": [],
+                    "text": "Paris.",
+                }
+            ],
+            "role": "assistant",
+        },
+    ]
+
+
 @pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
 def test_stream_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
     checking_client: OpenAI | AsyncOpenAI = client if sync else async_client