Initial project import

test/test_image_output_tokens.py

@@ -0,0 +1,33 @@
+import json
+import urllib.error
+import urllib.request
+
+from test.utils import BASE_URL, load_auth_key
+
+
+def main() -> None:
+    payload = {
+        "prompt": "一只橘猫坐在窗台上，午后阳光，写实摄影",
+        "model": "gpt-image-2",
+        "n": 1,
+        "response_format": "url",
+    }
+    request = urllib.request.Request(
+        BASE_URL + "/v1/images/generations",
+        data=json.dumps(payload).encode(),
+        headers={"Content-Type": "application/json", "Authorization": f"Bearer {load_auth_key()}"},
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(request) as response:
+            body = response.read().decode()
+    except urllib.error.HTTPError as error:
+        body = error.read().decode()
+    try:
+        print(json.dumps(json.loads(body), ensure_ascii=False, indent=2))
+    except json.JSONDecodeError:
+        print(body)
+
+
+if __name__ == "__main__":
+    main()