github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-06-09 10:17:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(openai): add unit test for streaming error in _generate (#33134)

Browse Source
This commit is contained in:
Ali Ismail
2025-10-21 12:08:37 -07:00
committed by GitHub
parent 84dbebac4f
commit 5acd34ae92
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/partners/openai/tests/unit_tests/llms/test_base.py
View File

@@ -108,3 +108,13 @@ def test_stream_response_to_generation_chunk() -> None:
assert chunk == GenerationChunk( assert chunk == GenerationChunk(
text="", generation_info={"finish_reason": None, "logprobs": None} 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"])