mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
b13ae8de50
The _RewindableImage(BytesIO) wrapper auto-rewound on every read after
EOF, which made the OpenAI SDK's multipart upload writer read the same
bytes forever instead of seeing EOF. Workers OOM'd / SIGKILL'd:
[gw0] node down: Not properly terminated
replacing crashed worker gw0
...
worker 'gw1' crashed while running
'tests/image_gen_tests/test_image_edits.py::TestOpenAIImageEditGPTImage1::test_openai_image_edit_litellm_sdk[False]'
The auto-rewind was added defensively for parametrized + flaky-retried
tests, but BaseLLMImageEditTest::test_openai_image_edit_litellm_sdk
already calls get_base_image_edit_call_args() once per invocation and
that helper now constructs fresh streams via _make_test_images(), so
rewinding inside the stream is unnecessary. Replace with plain BytesIO
seeded with the cached image bytes.
Co-authored-by: Mateo Wang <mateo-berri@users.noreply.github.com>
|
||
|---|---|---|
| .. | ||
| request_payloads | ||
| base_image_generation_test.py | ||
| conftest.py | ||
| ishaan_github.png | ||
| litellm_site.png | ||
| test_bedrock_image_gen_unit_tests.py | ||
| test_fal_ai_image_generation.py | ||
| test_image.png | ||
| test_image_edit.png | ||
| test_image_edits.py | ||
| test_image_generation.py | ||
| test_image_variation.py | ||
| test_xinference.py | ||
| vertex_key.json | ||
