From 8ea8ae4d37d94279a80cfd4af5f3fd0fea594915 Mon Sep 17 00:00:00 2001 From: Ali Ismail Date: Fri, 26 Sep 2025 17:02:21 -0700 Subject: [PATCH 1/3] test(openai): add openai unit test for streaming error in _generate method - Add test for streaming error with multiple prompts --- .../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 f0b027f9b7b3b..bae96ab7ca3b7 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"]) From d447108df9a6f8011399e5cac02a022416232997 Mon Sep 17 00:00:00 2001 From: Ali Ismail Date: Tue, 30 Sep 2025 15:21:28 -0700 Subject: [PATCH 2/3] test: pr-title-labler.yml not found in github ci tests. Add # to test file --- .github/pr-title-labeler.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/pr-title-labeler.yml b/.github/pr-title-labeler.yml index 24aedf6dceaaa..743731be4b278 100644 --- a/.github/pr-title-labeler.yml +++ b/.github/pr-title-labeler.yml @@ -38,4 +38,4 @@ label-mapping: refactor: ["refactor"] release: ["release"] revert: ["revert"] - tests: ["test"] + tests: ["test"] # From 3bab7cf1773f0a62ea7387b325270f483e16f79c Mon Sep 17 00:00:00 2001 From: Ali Ismail Date: Tue, 30 Sep 2025 15:27:32 -0700 Subject: [PATCH 3/3] revert hashtag for testing --- .github/pr-title-labeler.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/pr-title-labeler.yml b/.github/pr-title-labeler.yml index 743731be4b278..24aedf6dceaaa 100644 --- a/.github/pr-title-labeler.yml +++ b/.github/pr-title-labeler.yml @@ -38,4 +38,4 @@ label-mapping: refactor: ["refactor"] release: ["release"] revert: ["revert"] - tests: ["test"] # + tests: ["test"]