import requests

url = "https://api.datagrid.com/v1/batch-predictions"

payload = {
    "model": "gemini-3.1-flash-lite",
    "prompt": "Extract the project name, sheet title, and revision from this drawing.",
    "output_schema": {
        "type": "object",
        "additionalProperties": False,
        "properties": {
            "project_name": { "type": "string" },
            "sheet_title": { "type": "string" },
            "revision": { "type": "string" }
        },
        "required": ["project_name", "sheet_title"]
    },
    "items": [
        {
            "custom_id": "drawing_001",
            "file_id": "file_abc123",
            "page": 1
        },
        {
            "custom_id": "drawing_002",
            "file_id": "file_def456"
        }
    ],
    "completion_window": "24h",
    "metadata": { "project": "alpha" }
}
headers = {
    "Authorization": "Bearer <token>",
    "Content-Type": "application/json"
}

response = requests.post(url, json=payload, headers=headers)

print(response.text)

const options = {
  method: 'POST',
  headers: {Authorization: 'Bearer <token>', 'Content-Type': 'application/json'},
  body: JSON.stringify({
    model: 'gemini-3.1-flash-lite',
    prompt: 'Extract the project name, sheet title, and revision from this drawing.',
    output_schema: {
      type: 'object',
      additionalProperties: false,
      properties: {
        project_name: {type: 'string'},
        sheet_title: {type: 'string'},
        revision: {type: 'string'}
      },
      required: ['project_name', 'sheet_title']
    },
    items: [
      {custom_id: 'drawing_001', file_id: 'file_abc123', page: 1},
      {custom_id: 'drawing_002', file_id: 'file_def456'}
    ],
    completion_window: '24h',
    metadata: {project: 'alpha'}
  })
};

fetch('https://api.datagrid.com/v1/batch-predictions', options)
  .then(res => res.json())
  .then(res => console.log(res))
  .catch(err => console.error(err));

curl --request POST \
  --url https://api.datagrid.com/v1/batch-predictions \
  --header 'Authorization: Bearer <token>' \
  --header 'Content-Type: application/json' \
  --data '
{
  "model": "gemini-3.1-flash-lite",
  "prompt": "Extract the project name, sheet title, and revision from this drawing.",
  "output_schema": {
    "type": "object",
    "additionalProperties": false,
    "properties": {
      "project_name": {
        "type": "string"
      },
      "sheet_title": {
        "type": "string"
      },
      "revision": {
        "type": "string"
      }
    },
    "required": [
      "project_name",
      "sheet_title"
    ]
  },
  "items": [
    {
      "custom_id": "drawing_001",
      "file_id": "file_abc123",
      "page": 1
    },
    {
      "custom_id": "drawing_002",
      "file_id": "file_def456"
    }
  ],
  "completion_window": "24h",
  "metadata": {
    "project": "alpha"
  }
}
'

{
  "object": "batch_prediction",
  "id": "bpred_abc123",
  "status": "completed",
  "model": "gemini-3.1-flash-lite",
  "completion_window": "24h",
  "created_at": "2026-04-10T12:00:00.000Z",
  "expires_at": "2026-04-11T12:00:00.000Z",
  "in_progress_at": "2026-04-10T12:01:00.000Z",
  "finalizing_at": "2026-04-10T12:05:00.000Z",
  "completed_at": "2026-04-10T12:06:00.000Z",
  "failed_at": null,
  "cancelling_at": null,
  "cancelled_at": null,
  "expired_at": null,
  "request_counts": {
    "total": 2,
    "processing": 0,
    "succeeded": 2,
    "errored": 0,
    "canceled": 0,
    "expired": 0
  },
  "metadata": {
    "project": "alpha"
  },
  "error": null,
  "results_url": "/v1/batch-predictions/bpred_abc123/results"
}

{
  "type": "<string>",
  "title": "<string>",
  "status": 123,
  "detail": "<string>",
  "instance": "<string>"
}

{
  "type": "<string>",
  "title": "<string>",
  "status": 123,
  "detail": "<string>",
  "instance": "<string>"
}

{
  "type": "<string>",
  "title": "<string>",
  "status": 123,
  "detail": "<string>",
  "instance": "<string>"
}

{
  "type": "<string>",
  "title": "<string>",
  "status": 123,
  "errors": [
    {
      "pointer": "<string>",
      "code": "<string>",
      "message": "<string>",
      "custom_id": "<string>"
    }
  ],
  "detail": "<string>",
  "instance": "<string>"
}

{
  "type": "<string>",
  "title": "<string>",
  "status": 123,
  "detail": "<string>",
  "instance": "<string>"
}