Python SDK для Mage

Главное

Python SDK для Mage — официальный клиент с открытым исходным кодом для Mage API, опубликован как mage-space. Установите его командой "pip install mage-space", задайте MAGE_API_KEY и вызовите mage.run с id модели и конфигом, чтобы получить готовые изображения, видео или аудио от любой из 28 моделей. Он работает на Python 3.10 и новее, а каждая генерация оплачивается в Gems.

• 3.10+

  Python

• httpx

  HTTP-клиент

• 28

  Типизированные модели

• MIT

  Открытый код

Настройка

Установите Python SDK и запустите модель

1. 01

   Установите пакет

   Добавьте mage-space в свой проект.

   pip install mage-space

2. 02

   Задайте API-ключ

   Создайте ключ в разделе API → API Keys на mage.space и экспортируйте его. Клиент читает переменную MAGE_API_KEY.

   export MAGE_API_KEY="mage_sk_..."

3. 03

   Сгенерируйте

   Вызовите run с id модели и её конфигурацией. Метод дождётся результата и вернёт его; файл доступен по адресу result.url.

   from mage_space import Mage
   
   mage = Mage()  # reads MAGE_API_KEY
   
   request = mage.run("mango", {
       "prompt": "Editorial portrait in soft daylight, 35mm film look",
       "aspect_ratio": "4:5",
       "model_id": "mango-v3",
   })
   
   print(request["result"]["url"])

Примеры

Python SDK на практике

Каждый пример начинается с клиента, который читает MAGE_API_KEY.

from mage_space import Mage

mage = Mage()  # reads MAGE_API_KEY

submitted = mage.generate("cherry", {
    "prompt": "Waves rolling onto a black sand beach at sunset",
    "resolution": "720p",
    "duration": "5",
})

final = mage.requests.wait(
    submitted["request_id"],
    timeout=900,
    on_update=lambda request: print(request["status"]),
)

from mage_space import Mage

mage = Mage()  # reads MAGE_API_KEY

clip = mage.uploads.upload("clip.mp4")  # a path, bytes, or a binary file

mage.run("cherry", {
    "prompt": "Restyle this clip as a watercolor painting",
    "videos": [clip["url"]],
})

from mage_space import Mage

mage = Mage()  # reads MAGE_API_KEY

mage.characters.create(
    name="Ana",
    handle="ana",
    image="https://example.com/ana.png",
)

mage.run("mango", {"prompt": "@ana walking through a night market"})

from mage_space import Mage, MageAPIError, MageGenerationError

mage = Mage()  # reads MAGE_API_KEY

try:
    mage.run("mango", {"prompt": "A lighthouse at night"})
except MageAPIError as error:
    if error.code == "insufficient_gems":
        print("Gems needed:", error.body["error"]["gems_required"])
    else:
        raise
except MageGenerationError as error:
    print("No output:", error.code)

Зачем нужен SDK

Что SDK делает за вас

Справочник

Python SDK вкратце

Одни и те же методы работают для всех моделей; модель передаётся первым аргументом.

Модели

Начните с этих моделей

FAQ

Частые вопросы

Начните создавать с Mage

Один аккаунт, один баланс Gems, все модели. Платите только за то, что генерируете.

Похожие материалы

• SDK Mage
• Mage TypeScript SDK
• Mage для ComfyUI
• mage-space на PyPI
• Все модели API
• Документация API