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

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

const training = await client.trainings.replaceConfiguration({
  trainingId: "trg.019a0000-0000-7000-8000-000000000001",
  configuration: {
    method: "sft",
    source: {
      type: "base_model",
      baseModelId: "qwen3-4b"
    },
    sliceId: "dsl.019a0000-0000-7000-8000-000000000001"
  }
});
console.log(training);

curl --request PUT \
  --url https://platform.toloka.ai/api/v2-beta/trainings/{trainingId}/configuration \
  --header 'Authorization: <api-key>' \
  --header 'Content-Type: application/json' \
  --data '
{
  "configuration": {
    "method": "sft",
    "sliceId": "<string>",
    "source": {
      "type": "base_model"
    },
    "batchSize": 8,
    "epochs": 3,
    "learningRate": 0.00015,
    "loraRank": 16,
    "maxSequenceLength": 12288,
    "splitFieldId": "<string>",
    "trainParts": [
      "<string>"
    ]
  }
}
'

import requests

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

payload = { "configuration": {
        "method": "sft",
        "sliceId": "<string>",
        "source": { "type": "base_model" },
        "batchSize": 8,
        "epochs": 3,
        "learningRate": 0.00015,
        "loraRank": 16,
        "maxSequenceLength": 12288,
        "splitFieldId": "<string>",
        "trainParts": ["<string>"]
    } }
headers = {
    "Authorization": "<api-key>",
    "Content-Type": "application/json"
}

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

print(response.text)

{
  "archivedAt": "2023-11-07T05:31:56Z",
  "configuration": {
    "batchSize": 8,
    "epochs": 3,
    "learningRate": 0.000505,
    "loraRank": 8,
    "maxSequenceLength": 8224,
    "method": "sft",
    "source": {
      "baseModelId": "qwen3-4b",
      "type": "base_model"
    },
    "training": {
      "datasetId": "<string>",
      "itemFilter": {
        "fieldSources": [
          {
            "fieldId": "<string>",
            "sources": [
              {
                "type": "user",
                "userId": "<string>"
              }
            ]
          }
        ],
        "fieldValues": [
          {
            "fieldId": "<string>",
            "jsonPath": "<string>",
            "operator": "text_eq",
            "value": "<string>",
            "latestOnly": true,
            "sources": []
          }
        ],
        "missingRevisionFields": [
          "<string>"
        ],
        "excludeParts": [
          {
            "fieldId": "<string>",
            "names": [
              "<string>"
            ]
          }
        ],
        "parts": {
          "fieldId": "<string>",
          "names": [
            "<string>"
          ]
        },
        "slice": {
          "sliceId": "<string>"
        }
      }
    },
    "sliceId": "<string>",
    "splitFieldId": "<string>",
    "trainParts": [
      "<string>"
    ],
    "validation": {
      "datasetId": "<string>",
      "itemFilter": {
        "fieldSources": [
          {
            "fieldId": "<string>",
            "sources": [
              {
                "type": "user",
                "userId": "<string>"
              }
            ]
          }
        ],
        "fieldValues": [
          {
            "fieldId": "<string>",
            "jsonPath": "<string>",
            "operator": "text_eq",
            "value": "<string>",
            "latestOnly": true,
            "sources": []
          }
        ],
        "missingRevisionFields": [
          "<string>"
        ],
        "excludeParts": [
          {
            "fieldId": "<string>",
            "names": [
              "<string>"
            ]
          }
        ],
        "parts": {
          "fieldId": "<string>",
          "names": [
            "<string>"
          ]
        },
        "slice": {
          "sliceId": "<string>"
        }
      }
    }
  },
  "createdAt": "2023-11-07T05:31:56Z",
  "id": "<string>",
  "modelId": "<string>",
  "name": "<string>",
  "projectId": "<string>",
  "updatedAt": "2023-11-07T05:31:56Z"
}