Normalize OpenAI image form fields

2026-09-21 11:38:28 +02:00
parent a65affe30a
commit 1aec6cd4ac
2 changed files with 21 additions and 0 deletions
@@ -156,6 +156,11 @@ class ImageEditMultipartTests(unittest.TestCase):
                'Content-Disposition: form-data; name="prompt"\r\n\r\n'
                "Nur die Farbe ändern\r\n"
            ).encode(),
+            (
+                f"--{boundary}\r\n"
+                'Content-Disposition: form-data; name="n"\r\n\r\n'
+                "1\r\n"
+            ).encode(),
            (
                f"--{boundary}\r\n"
                'Content-Disposition: form-data; name="image[]"; filename="a.png"\r\n'
@@ -174,6 +179,8 @@ class ImageEditMultipartTests(unittest.TestCase):

        self.assertEqual(fields["model"], "Qwen-Image-2.1-int8")
        self.assertEqual(fields["prompt"], "Nur die Farbe ändern")
+        normalized = handler._normalize_image_multipart_fields(fields)
+        self.assertEqual(normalized["n"], 1)
        self.assertEqual(
            files,
            [("image[]", "a.png", b"PNG-A"),