fix(openai): Fixing error that comes up using the Responses API with built-in tools and custom tools (#34136)

libs/partners/openai/tests/unit_tests/chat_models/test_responses_stream.py

@@ -756,3 +756,38 @@ def test_responses_stream(output_version: str, expected_content: list[dict]) ->
         dumped = _strip_none(item.model_dump())
         _ = dumped.pop("status", None)
         assert dumped == payload["input"][idx]
+
+
+def test_responses_stream_with_image_generation_multiple_calls() -> None:
+    """Test that streaming with image_generation tool works across multiple calls.
+
+    Regression test: image_generation tool should not be mutated between calls,
+    which would cause NotImplementedError on subsequent invocations.
+    """
+    tools: list[dict[str, Any]] = [
+        {"type": "image_generation"},
+        {"type": "function", "name": "my_tool", "parameters": {}},
+    ]
+    llm = ChatOpenAI(
+        model="gpt-4o",
+        use_responses_api=True,
+        streaming=True,
+    )
+    llm_with_tools = llm.bind_tools(tools)
+
+    mock_client = MagicMock()
+
+    def mock_create(*args: Any, **kwargs: Any) -> MockSyncContextManager:
+        return MockSyncContextManager(responses_stream)
+
+    mock_client.responses.create = mock_create
+
+    # First call should work
+    with patch.object(llm, "root_client", mock_client):
+        chunks = list(llm_with_tools.stream("test"))
+        assert len(chunks) > 0
+
+    # Second call should also work (would fail before fix due to tool mutation)
+    with patch.object(llm, "root_client", mock_client):
+        chunks = list(llm_with_tools.stream("test again"))
+        assert len(chunks) > 0