From 5acd34ae9235c133537f6bb99a4b7200fb404d52 Mon Sep 17 00:00:00 2001
From: Ali Ismail <99437500+Ali-Ismail-1@users.noreply.github.com>
Date: Tue, 21 Oct 2025 12:08:37 -0700
Subject: [PATCH] feat(openai): add unit test for streaming error in
 `_generate` (#33134)

---
 .../partners/openai/tests/unit_tests/llms/test_base.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/libs/partners/openai/tests/unit_tests/llms/test_base.py b/libs/partners/openai/tests/unit_tests/llms/test_base.py
index f0b027f9b7b..bae96ab7ca3 100644
--- a/libs/partners/openai/tests/unit_tests/llms/test_base.py
+++ b/libs/partners/openai/tests/unit_tests/llms/test_base.py
@@ -108,3 +108,13 @@ def test_stream_response_to_generation_chunk() -> None:
     assert chunk == GenerationChunk(
         text="", generation_info={"finish_reason": None, "logprobs": None}
     )
+
+
+def test_generate_streaming_multiple_prompts_error() -> None:
+    """Ensures ValueError when streaming=True and multiple prompts."""
+    llm = OpenAI(streaming=True)
+
+    with pytest.raises(
+        ValueError, match="Cannot stream results with multiple prompts\\."
+    ):
+        llm._generate(["foo", "bar"])