Yandex Cloud AI Studio Text Classification Python API Docs | dltHub

Yandex Cloud AI Studio Text Classification API supports REST and gRPC methods for text classification, including the fewShotClassify method for binary and multi-class classification. The REST API is accessible via https://yandex.cloud/en/docs/ai-studio/text-classification/api-ref/TextClassification/. The REST API base URL is https://llm.api.cloud.yandex.net/foundationModels/v1 and Requests require a Bearer IAM token in the Authorization header..

dlt is an open-source Python library that handles authentication, pagination, and schema evolution automatically. dlthub provides AI context files that enable code assistants to generate production-ready pipelines. Install with uv pip install "dlt[workspace]" and start loading Yandex Cloud AI Studio Text Classification data in under 10 minutes.

---

What data can I load from Yandex Cloud AI Studio Text Classification?

Here are some of the endpoints you can load from Yandex Cloud AI Studio Text Classification:

---

How do I authenticate with the Yandex Cloud AI Studio Text Classification API?

Include the header Authorization: Bearer <IAM token> with each request.

1. Get your credentials

1. Open the Yandex Cloud console and navigate to the "Service Accounts" section.
2. Create a new service account or select an existing one.
3. Grant the service account the "ai.studio" role.
4. In the service account details, click "Create IAM token" (or use the CLI: yc iam create-token --service-account-id <id>).
5. Copy the generated token; it will be used as the Bearer token in API requests.

2. Add them to .dlt/secrets.toml

Copy
[sources.yandex_cloud_ai_studio_text_classification_source]
iam_token = "your_iam_token_here"

dlt reads this automatically at runtime — never hardcode tokens in your pipeline script. For production environments, see setting up credentials with dlt for environment variable and vault-based options.

---

How do I set up and run the pipeline?

Set up a virtual environment and install dlt:

Copy
uv venv && source .venv/bin/activate
uv pip install "dlt[workspace]"

1. Install the dlt AI Workbench:

Copy
dlt ai init --agent <your-agent> # <agent>: claude | cursor | codex

This installs project rules, a secrets management skill, appropriate ignore files, and configures the dlt MCP server for your agent. Learn more →

2. Install the rest-api-pipeline toolkit:

Copy
dlt ai toolkit rest-api-pipeline install

This loads the skills and context about dlt the agent uses to build the pipeline iteratively, efficiently, and safely. The agent uses MCP tools to inspect credentials — it never needs to read your secrets.toml directly. Learn more →

3. Start LLM-assisted coding:

Use /find-source to load data from the Yandex Cloud AI Studio Text Classification API into DuckDB.

The rest-api-pipeline toolkit takes over from here — it reads relevant API documentation, presents you with options for which endpoints to load, and follows a structured workflow to scaffold, debug, and validate the pipeline step by step.

4. Run the pipeline:

Copy
python yandex_cloud_ai_studio_text_classification_pipeline.py

If everything is configured correctly, you'll see output like this:

Copy
Pipeline yandex_cloud_ai_studio_text_classification_pipeline load step completed in 0.26 seconds
1 load package(s) were loaded to destination duckdb and into dataset yandex_cloud_ai_studio_text_classification_data
The duckdb destination used duckdb:/yandex_cloud_ai_studio_text_classification.duckdb location to store data
Load package 1749667187.541553 is LOADED and contains no failed jobs

Inspect your pipeline and data:

Copy
dlt pipeline yandex_cloud_ai_studio_text_classification_pipeline show

This opens the Pipeline Dashboard where you can verify pipeline state, load metrics, schema (tables, columns, types), and query the loaded data directly.

---

Python pipeline example

This example loads few_shot_classify and zero_shot_classify from the Yandex Cloud AI Studio Text Classification API into DuckDB. It mirrors the endpoint and data selector configuration from the table above:

Copy
import dlt
from dlt.sources.rest_api import RESTAPIConfig, rest_api_resources

@dlt.source
def yandex_cloud_ai_studio_text_classification_source(iam_token=dlt.secrets.value):
    config: RESTAPIConfig = {
        "client": {
            "base_url": "https://llm.api.cloud.yandex.net/foundationModels/v1",
            "auth": {
                "type": "bearer",
                "token": iam_token,
            },
        },
        "resources": [
            {"name": "few_shot_classify", "endpoint": {"path": "foundationModels/v1/fewShotTextClassification", "data_selector": "predictions"}},
            {"name": "zero_shot_classify", "endpoint": {"path": "foundationModels/v1/zeroShotTextClassification", "data_selector": "predictions"}}
        ],
    }
    yield from rest_api_resources(config)


def get_data() -> None:
    pipeline = dlt.pipeline(
        pipeline_name="yandex_cloud_ai_studio_text_classification_pipeline",
        destination="duckdb",
        dataset_name="yandex_cloud_ai_studio_text_classification_data",
    )
    load_info = pipeline.run(yandex_cloud_ai_studio_text_classification_source())
    print(load_info)

To add more endpoints, append entries from the resource table to the "resources" list using the same name, path, and data_selector pattern.

---

How do I query the loaded data?

Once the pipeline runs, dlt creates one table per resource. You can query with Python or SQL.

Python (pandas DataFrame):

Copy
import dlt

data = dlt.pipeline("yandex_cloud_ai_studio_text_classification_pipeline").dataset()
sessions_df = data.few_shot_classify.df()
print(sessions_df.head())

SQL (DuckDB example):

Copy
SELECT * FROM yandex_cloud_ai_studio_text_classification_data.few_shot_classify LIMIT 10;

In a marimo or Jupyter notebook:

Copy
import dlt

data = dlt.pipeline("yandex_cloud_ai_studio_text_classification_pipeline").dataset()
data.few_shot_classify.df().head()

See how to explore your data in marimo Notebooks and how to query your data in Python with dataset.

---

What destinations can I load Yandex Cloud AI Studio Text Classification data to?

dlt supports loading into any of these destinations — only the destination parameter changes:

Change the destination in dlt.pipeline(destination="snowflake") and add credentials in .dlt/secrets.toml. See the full destinations list.

---

Next steps

Continue your data engineering journey with the other toolkits of the dltHub AI Workbench:

• data-exploration — Build custom notebooks, charts, and dashboards for deeper analysis with marimo notebooks.
• dlthub-runtime — Deploy, schedule, and monitor your pipeline in production.

Copy
dlt ai toolkit data-exploration install
dlt ai toolkit dlthub-runtime install

• How to deploy a pipeline
• How to explore your data in marimo Notebooks
• How to query your data in Python with dataset
• Join the dlt community on Slack