Add profile-aware image reference editing

test_image_upload.py

@@ -0,0 +1,21 @@
+import io
+import unittest
+from types import SimpleNamespace
+from image_upload import read_image_upload
+from image_test import decode_references
+
+class UploadTests(unittest.TestCase):
+ def test_multiple_images_keep_order_and_reject_oversized_body(self):
+  boundary='synthetic-boundary'
+  parts=[]
+  for name,value,mime in [('model',b'athena-image','text/plain'),('prompt',b'change color','text/plain'),('image[]',b'\x89PNG\r\n\x1a\nfirst','image/png'),('image[]',b'\xff\xd8\xffsecond','image/jpeg')]:
+   parts.append(f'--{boundary}\r\nContent-Disposition: form-data; name="{name}"\r\nContent-Type: {mime}\r\n\r\n'.encode()+value+b'\r\n')
+  body=b''.join(parts)+f'--{boundary}--\r\n'.encode()
+  handler=SimpleNamespace(connection=SimpleNamespace(settimeout=lambda _:None),headers={'Content-Type':'multipart/form-data; boundary='+boundary,'Content-Length':str(len(body))},rfile=io.BytesIO(body))
+  fields,images=read_image_upload(handler)
+  self.assertEqual(fields,{'model':'athena-image','prompt':'change color'})
+  self.assertEqual([raw for raw,_ in decode_references(images,4)],[b'\x89PNG\r\n\x1a\nfirst',b'\xff\xd8\xffsecond'])
+  handler.headers['Content-Length']=str(50*1024*1024)
+  with self.assertRaises(ValueError):read_image_upload(handler)
+
+if __name__=='__main__':unittest.main()