mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-06-13 14:23:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(image): cover dropping null OpenAI image params

Browse Source
This commit is contained in:
Xubin Ren 2026-06-06 19:30:57 +08:00
parent a4cf0f9514
commit a1b9577224
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/providers/test_image_generation.py
View File

@ -631,6 +631,28 @@ async def test_openai_payload_and_response() -> None:
assert body["size"] == "1792x1024"
@pytest.mark.asyncio
async def test_openai_extra_body_null_drops_default_params_only() -> None:
fake = FakeClient(FakeResponse({"data": [{"b64_json": RAW_B64}]}))
client = OpenAIImageGenerationClient(
api_key="sk-openai-test",
extra_body={
"response_format": None,
"seed": 0,
"safety_checker": False,
},
client=fake, # type: ignore[arg-type]
)
await client.generate(prompt="draw", model="dall-e-3")
body = fake.calls[0]["json"]
assert "response_format" not in body
assert body["n"] == 1
assert body["seed"] == 0
assert body["safety_checker"] is False
@pytest.mark.asyncio
async def test_openai_b64_json_response_uses_detected_mime() -> None:
raw_b64 = base64.b64encode(JPEG_BYTES).decode("ascii")