使用示例

curl https://models.rivus.cn/v1/images/generations \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer YOUR_API_KEY" \
  -d '{
    "model": "jimeng-4.0",
    "prompt": "一只可爱的熊猫在图书馆看书",
    "size": "1024x1024"
  }'

curl https://models.rivus.cn/v1/images/generations \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer YOUR_API_KEY" \
  -d '{
    "model": "jimeng-3.0",
    "prompt": "未来城市的夜景，赛博朋克风格",
    "n": 4,
    "size": "1792x1024"
  }'

curl https://models.rivus.cn/v1/images/edits \
  -H "Authorization: Bearer YOUR_API_KEY" \
  -F "image=@original.png" \
  -F "model=jimeng-4.0" \
  -F "prompt=添加科幻元素和霓虹灯效果" \
  -F "n=1"

from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://models.rivus.cn/v1"
)

response = client.images.generate(
    model="jimeng-4.0",
    prompt="一只可爱的熊猫在图书馆看书",
    size="1024x1024"
)

print(response.data[0].url)

with open("input.png", "rb") as image_file:
    response = client.images.edit(
        image=image_file,
        model="jimeng-4.0",
        prompt="添加科幻元素"
    )

print(response.data[0].url)

prompts = [
    "熊猫在图书馆看书",
    "熊猫在公园散步",
    "熊猫在竹林里休息"
]

for prompt in prompts:
    response = client.images.generate(
        model="jimeng-3.0",
        prompt=prompt,
        n=1
    )
    print(f"{prompt}: {response.data[0].url}")

from openai import OpenAI, APIError, RateLimitError

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://models.rivus.cn/v1"
)

try:
    response = client.images.generate(
        model="jimeng-4.0",
        prompt="测试图片",
        n=1
    )
    print(response.data[0].url)
    
except RateLimitError:
    print("请求频率过高，请稍后重试")
    
except APIError as e:
    print(f"API 错误: {e}")

import requests

response = client.images.generate(
    model="jimeng-4.0",
    prompt="一只可爱的熊猫",
    n=1
)

image_url = response.data[0].url

# 下载图片
img_data = requests.get(image_url).content
with open('output.png', 'wb') as f:
    f.write(img_data)