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

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

const estimate = await client.trainings.estimateCost({
  trainingId: "trg.019a0000-0000-7000-8000-000000000001",
  inputSetId: "tei.019a0000-0000-7000-8000-000000000001"
});
console.log(estimate);

curl --request POST \
  --url https://platform.toloka.ai/api/v2-beta/trainings/{trainingId}/cost-estimate \
  --header 'Authorization: <api-key>' \
  --header 'Content-Type: application/json' \
  --data '
{
  "inputSetId": "<string>"
}
'

import requests

url = "https://platform.toloka.ai/api/v2-beta/trainings/{trainingId}/cost-estimate"

payload = { "inputSetId": "<string>" }
headers = {
    "Authorization": "<api-key>",
    "Content-Type": "application/json"
}

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

print(response.text)

{
  "calculatedAt": "2023-11-07T05:31:56Z",
  "capabilityVersion": "<string>",
  "cutoff": "2023-11-07T05:31:56Z",
  "inputSetId": "<string>",
  "inputs": {
    "training": {
      "datasetExportId": "<string>",
      "datasetId": "<string>",
      "failure": {
        "code": "empty_selection",
        "message": "<string>",
        "timestamp": "2023-11-07T05:31:56Z"
      },
      "report": {
        "errors": [
          {
            "itemIndex": 4503599627370495,
            "reason": "<string>"
          }
        ],
        "invalidCount": 4503599627370495,
        "tokens": {
          "max": 4503599627370495,
          "p50": 4503599627370495,
          "p95": 4503599627370495,
          "estimatorVersion": "utf16-code-units-div-4-v1",
          "total": 4503599627370495
        },
        "validCount": 4503599627370495
      },
      "heldOutRows": 4503599627370495
    },
    "validation": {
      "datasetExportId": "<string>",
      "datasetId": "<string>",
      "failure": {
        "code": "empty_selection",
        "message": "<string>",
        "timestamp": "2023-11-07T05:31:56Z"
      },
      "report": {
        "errors": [
          {
            "itemIndex": 4503599627370495,
            "reason": "<string>"
          }
        ],
        "invalidCount": 4503599627370495,
        "tokens": {
          "max": 4503599627370495,
          "p50": 4503599627370495,
          "p95": 4503599627370495,
          "estimatorVersion": "utf16-code-units-div-4-v1",
          "total": 4503599627370495
        },
        "validCount": 4503599627370495
      },
      "heldOutRows": 4503599627370495
    }
  },
  "issues": [
    {
      "code": "unsupported_source",
      "message": "<string>",
      "purpose": "training"
    }
  ],
  "pricingVersion": "<string>",
  "status": "blocked",
  "trainingId": "<string>"
}