图片 API 对接文档

if (-not $env:AIMASKER_API_KEY) { throw "请先设置环境变量" }
$headers = @{ Authorization = "Bearer $env:AIMASKER_API_KEY" }
$body = @{
  model = "gpt-image-2"; prompt = "<图片描述>"
  size = "3840x2160"; quality = "high"; n = 1; output_format = "png"
} | ConvertTo-Json
$reply = Invoke-RestMethod -Method Post -Uri "https://aimasker.com/v1/images/generations/async" -Headers $headers -ContentType "application/json; charset=utf-8" -Body ([Text.Encoding]::UTF8.GetBytes($body)) -TimeoutSec 120
$taskId = $reply.task_id
if (-not $taskId) { throw "未收到 task_id；停止并核查，不要重复提交" }
$taskFile = Join-Path (Get-Location) ("aimasker-task-" + [guid]::NewGuid().ToString("N") + ".txt")
Set-Content -LiteralPath $taskFile -Value $taskId
"task_id=$taskId"

if (-not $env:AIMASKER_API_KEY) { throw "请先设置环境变量" }
$headers = @{ Authorization = "Bearer $env:AIMASKER_API_KEY" }
$body = @{
  model = "nano-banana-pro"; prompt = "<图片描述>"
  aspect_ratio = "16:9"; output_resolution = "4K"; n = 1
} | ConvertTo-Json
$reply = Invoke-RestMethod -Method Post -Uri "https://aimasker.com/v1/images/generations/async" -Headers $headers -ContentType "application/json; charset=utf-8" -Body ([Text.Encoding]::UTF8.GetBytes($body)) -TimeoutSec 120
$taskId = $reply.task_id
if (-not $taskId) { throw "未收到 task_id；停止并核查，不要重复提交" }
$taskFile = Join-Path (Get-Location) ("aimasker-task-" + [guid]::NewGuid().ToString("N") + ".txt")
Set-Content -LiteralPath $taskFile -Value $taskId
"task_id=$taskId"

: "${AIMASKER_API_KEY:?请先设置环境变量}"
TASK_DIR=$(mktemp -d)
curl --fail-with-body --max-time 120 -sS \
  'https://aimasker.com/v1/images/generations/async' \
  -H "Authorization: Bearer $AIMASKER_API_KEY" \
  -H 'Content-Type: application/json' \
  -d '{"model":"gpt-image-2","prompt":"<图片描述>","size":"3840x2160","quality":"high","n":1,"output_format":"png"}' \
  -D "$TASK_DIR/headers.txt" -o "$TASK_DIR/create.json" &&
  jq -er '.task_id' "$TASK_DIR/create.json" > "$TASK_DIR/task_id.txt"
# 任何失败都先停止核查；不要再次执行上面的创建命令。

: "${AIMASKER_API_KEY:?请先设置环境变量}"
TASK_DIR=$(mktemp -d)
curl --fail-with-body --max-time 120 -sS \
  'https://aimasker.com/v1/images/generations/async' \
  -H "Authorization: Bearer $AIMASKER_API_KEY" \
  -H 'Content-Type: application/json' \
  -d '{"model":"nano-banana-pro","prompt":"<图片描述>","aspect_ratio":"16:9","output_resolution":"4K","n":1}' \
  -D "$TASK_DIR/headers.txt" -o "$TASK_DIR/create.json" &&
  jq -er '.task_id' "$TASK_DIR/create.json" > "$TASK_DIR/task_id.txt"
# 任何失败都先停止核查；不要再次执行上面的创建命令。

: "${AIMASKER_API_KEY:?请先设置环境变量}"
TASK_DIR=$(mktemp -d)
curl --fail-with-body --max-time 120 -sS \
  'https://aimasker.com/v1/images/edits/async' \
  -H "Authorization: Bearer $AIMASKER_API_KEY" \
  -F 'model=gpt-image-2' -F 'prompt=<编辑要求>' \
  -F 'size=3840x2160' -F 'quality=high' -F 'n=1' -F 'output_format=png' \
  -F 'image[]=@reference-1.png' -F 'image[]=@reference-2.png' \
  -D "$TASK_DIR/headers.txt" -o "$TASK_DIR/create.json" &&
  jq -er '.task_id' "$TASK_DIR/create.json" > "$TASK_DIR/task_id.txt"
# 可选蒙版：在提交前加入 -F 'mask=@mask.png'。

$taskId = (Get-Content -LiteralPath $taskFile -Raw).Trim()
$task = Invoke-RestMethod -Uri "https://aimasker.com/v1/images/tasks/$taskId" -Headers $headers -TimeoutSec 60
$task.status
if ($task.status -eq "completed") { $task.result.data | ForEach-Object { $_.url } }
# processing：至少等 5 秒再查询；failed：检查 error 和 http_status。

TASK_ID=$(cat "$TASK_DIR/task_id.txt")
curl --fail-with-body --max-time 60 -sS \
  "https://aimasker.com/v1/images/tasks/$TASK_ID" \
  -H "Authorization: Bearer $AIMASKER_API_KEY" \
  -D "$TASK_DIR/query-headers.txt" -o "$TASK_DIR/task.json" &&
  jq '{status, urls: [.result.data[]?.url], error, http_status}' "$TASK_DIR/task.json"

$ErrorActionPreference = "Stop"
$stage = Join-Path $env:TEMP ("aimasker-install-" + [guid]::NewGuid().ToString("N"))
New-Item -ItemType Directory -Path $stage | Out-Null
$zip = Join-Path $stage "aimasker-image-api.zip"
Invoke-WebRequest "https://aimasker.com/image-api-docs/downloads/aimasker-image-api.zip?v=2026.09.10.1" -OutFile $zip
if ((Get-FileHash $zip -Algorithm SHA256).Hash.ToLower() -ne "560bda04539ea212150f81d9489b92bf88fdfb48bb9df4461d18c9829fa7c7aa") { throw "校验失败，请重新获取最新文档与包" }
Expand-Archive -LiteralPath $zip -DestinationPath $stage
$skillsRoot = "$env:USERPROFILE\.agents\skills"
python "$stage\aimasker-image-api\scripts\install_skill.py" $zip --skills-dir $skillsRoot
if ($LASTEXITCODE -ne 0) { throw "安装失败，请保留现场" }
$skill = Join-Path $skillsRoot "aimasker-image-api"
python -m venv "$env:LOCALAPPDATA\aimasker-client-venv"
if ($LASTEXITCODE -ne 0) { throw "创建 Python 环境失败" }
$python = "$env:LOCALAPPDATA\aimasker-client-venv\Scripts\python.exe"
& $python -m pip install -r "$skill\requirements.txt"
if ($LASTEXITCODE -ne 0) { throw "依赖安装失败" }
$client = Join-Path $skill "scripts\aimasker_async_image.py"
& $python $client --help

install_aimasker() {
  STAGE=$(mktemp -d) || return
  ZIP="$STAGE/aimasker-image-api.zip"
  curl -fLsS 'https://aimasker.com/image-api-docs/downloads/aimasker-image-api.zip?v=2026.09.10.1' -o "$ZIP" || return
  python3 -c 'import hashlib,sys; assert hashlib.sha256(open(sys.argv[1],"rb").read()).hexdigest()==sys.argv[2], "ZIP checksum mismatch"' "$ZIP" '560bda04539ea212150f81d9489b92bf88fdfb48bb9df4461d18c9829fa7c7aa' || return
  python3 -m zipfile -e "$ZIP" "$STAGE" || return
  SKILLS_ROOT="$HOME/.agents/skills"
  python3 "$STAGE/aimasker-image-api/scripts/install_skill.py" "$ZIP" --skills-dir "$SKILLS_ROOT" || return
  SKILL="$SKILLS_ROOT/aimasker-image-api"
  python3 -m venv "$HOME/.local/share/aimasker-client-venv" || return
  PYTHON="$HOME/.local/share/aimasker-client-venv/bin/python"
  "$PYTHON" -m pip install -r "$SKILL/requirements.txt" || return
  CLIENT="$SKILL/scripts/aimasker_async_image.py"
  "$PYTHON" "$CLIENT" --help
}
install_aimasker