How to Get Result

GET https://api.wavespeed.ai/api/v3/predictions/{task-id}

curl --location --request GET 'https://api.wavespeed.ai/api/v3/predictions/pred_abc123' \
--header 'Authorization: Bearer ${WAVESPEED_API_KEY}'

{
  "code": 200,
  "message": "success",
  "data": {
    "id": "pred_abc123",
    "model": "wavespeed-ai/flux-dev",
    "status": "processing",
    "outputs": [],
    "created_at": "2024-01-01T12:00:00.000Z"
  }
}

{
  "code": 200,
  "message": "success",
  "data": {
    "id": "pred_abc123",
    "model": "wavespeed-ai/flux-dev",
    "status": "completed",
    "outputs": [
      "https://cdn.wavespeed.ai/generated/image123.png"
    ],
    "timings": {
      "inference": 2500
    },
    "created_at": "2024-01-01T12:00:00.000Z"
  }
}

{
  "code": 200,
  "message": "success",
  "data": {
    "id": "pred_abc123",
    "status": "failed",
    "error": "Error message here",
    "created_at": "2024-01-01T12:00:00.000Z"
  }
}

import os
import requests
import time
 
api_key = os.environ.get("WAVESPEED_API_KEY")
 
def get_result(task_id, max_wait=300):
    """Poll for result with timeout."""
    url = f"https://api.wavespeed.ai/api/v3/predictions/{task_id}"
    headers = {"Authorization": f"Bearer {api_key}"}
 
    start = time.time()
    while time.time() - start < max_wait:
        response = requests.get(url, headers=headers)
        data = response.json()["data"]
 
        if data["status"] == "completed":
            return data["outputs"]
        elif data["status"] == "failed":
            raise Exception(data.get("error", "Unknown error"))
 
        time.sleep(1)  # Wait before next poll
 
    raise Exception("Timeout waiting for result")
 
# Usage
outputs = get_result("pred_abc123")
print(outputs[0])