| from api.ai import ImageGenerationRequest |
| from api.image_inputs import _payload_from_fields |
| from services.protocol.conversation import ConversationRequest |
|
|
|
|
| def test_generation_requests_default_to_url() -> None: |
| request = ImageGenerationRequest(prompt="a cat") |
|
|
| assert request.response_format == "url" |
|
|
|
|
| def test_edit_requests_default_to_url() -> None: |
| payload = _payload_from_fields({"prompt": "make it brighter"}) |
|
|
| assert payload["response_format"] == "url" |
|
|
|
|
| def test_explicit_base64_remains_supported() -> None: |
| request = ImageGenerationRequest(prompt="a cat", response_format="b64_json") |
|
|
| assert request.response_format == "b64_json" |
| assert ConversationRequest(response_format="b64_json").response_format == "b64_json" |
|
|
