34 lines
930 B
Python
34 lines
930 B
Python
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()
|