import { createClient } from "@toloka-platform/sdk";

const client = createClient({
  baseUrl: "https://platform.toloka.ai",
  apiKey: process.env.TOLOKA_API_KEY!,
});

const run = await client.batchInference.createRun({
  projectId: "pp.019a0000-0000-7000-8000-000000000001",
  idempotencyKey: crypto.randomUUID(),
  datasetId: "ds.019a0000-0000-7000-8000-000000000001",
  model: {
    type: "base_model",
    baseModelId: "qwen3-4b"
  },
  sampling: {
    maxOutputTokens: 256
  },
  output: {
    type: "field",
    fieldName: "answer"
  }
});
console.log(run);

curl --request POST \
  --url https://platform.toloka.ai/api/v2-beta/projects/{projectId}/batch-inference-runs \
  --header 'Authorization: <api-key>' \
  --header 'Content-Type: application/json' \
  --header 'idempotency-key: <idempotency-key>' \
  --data '
{
  "datasetId": "<string>",
  "model": {
    "type": "base_model"
  },
  "output": {
    "fieldName": "<string>",
    "type": "field"
  },
  "sampling": {
    "maxOutputTokens": 4096,
    "samplesPerRow": 1,
    "seed": 1073741823,
    "stop": [
      "<string>"
    ],
    "temperature": 0,
    "topP": 0
  }
}
'

import requests

url = "https://platform.toloka.ai/api/v2-beta/projects/{projectId}/batch-inference-runs"

payload = {
    "datasetId": "<string>",
    "model": { "type": "base_model" },
    "output": {
        "fieldName": "<string>",
        "type": "field"
    },
    "sampling": {
        "maxOutputTokens": 4096,
        "samplesPerRow": 1,
        "seed": 1073741823,
        "stop": ["<string>"],
        "temperature": 0,
        "topP": 0
    }
}
headers = {
    "idempotency-key": "<idempotency-key>",
    "Authorization": "<api-key>",
    "Content-Type": "application/json"
}

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

print(response.text)

{
  "blockedReason": "funds",
  "chargedAmount": "<string>",
  "createdAt": "2023-11-07T05:31:56Z",
  "datasetId": "<string>",
  "error": {
    "code": "requests_export_failed",
    "message": "<string>"
  },
  "id": "<string>",
  "message": "<string>",
  "mode": {
    "type": "single_turn"
  },
  "model": {
    "baseModelId": "qwen3-4b",
    "type": "base_model"
  },
  "output": {
    "datasetId": "<string>",
    "fieldId": "<string>",
    "type": "field"
  },
  "progress": {
    "phase": "queued",
    "requests": {
      "done": 4503599627370495,
      "total": 4503599627370495
    }
  },
  "projectId": "<string>",
  "quote": {
    "confirmBefore": "2023-11-07T05:31:56Z",
    "currency": "USD",
    "maximumCharge": "<string>",
    "requests": 123,
    "skippedRows": 4503599627370495
  },
  "result": {
    "errorCounts": {},
    "errorSamples": [
      {
        "code": "<string>",
        "itemId": "<string>",
        "message": "<string>"
      }
    ],
    "failed": 4503599627370495,
    "inputTokens": 4503599627370495,
    "outputTokens": 4503599627370495,
    "truncated": 4503599627370495,
    "written": 4503599627370495
  },
  "sampling": {
    "maxOutputTokens": 4096,
    "samplesPerRow": 1,
    "seed": 1073741823,
    "stop": [
      "<string>"
    ],
    "temperature": 0,
    "topP": 0
  },
  "status": "preparing",
  "updatedAt": "2023-11-07T05:31:56Z"
}