datarobot-workload-api

datarobot-workload-api is a skill for Claude Code from datarobot-oss/datarobot-agent-skills. It costs 142 tokens per session (5,047 once invoked), scanned A, original, Apache-2.0.

A guide for using DataRobot’s Workload API, a web interface for running container images as managed services. It covers setup, deployment, scaling, troubleshooting, monitoring, and version rollouts.

In plain words
What is it for?
Deploy and configure containers, change replicas or autoscaling, choose CPU or GPU resources, add credentials, inspect logs and metrics, diagnose failures, and release new workload versions.
Why use it?
It provides the steps and REST calls needed to manage workloads when the standard DataRobot Python SDK does not include this API.

Skill for Claude Code

Written for Claude Code: shipped in a Claude Code plugin.

Part of the datarobot-agent-skills plugin — 21 skills shipped together

Good fit Deploy and configure containers, change replicas or autoscaling, choose CPU or GPU resources, add credentials, inspect logs and metrics, diagnose failures, and release new workload versions.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add datarobot-oss/datarobot-agent-skills --skill datarobot-workload-api
Clone the repo
git clone --depth 1 https://github.com/datarobot-oss/datarobot-agent-skills

Made for: Claude Code.

Or install datarobot-agent-skills, the plugin that ships this one along with the rest of its 21 skills.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for datarobot-workload-api

README.md
[![agentmods](https://agentmods.dev/badge/skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api/github.svg)](https://agentmods.dev/skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api)
Your own site
<a href="https://agentmods.dev/skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api"><img src="https://agentmods.dev/badge/skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for datarobot-workload-api

Your own site · 80×15
<a href="https://agentmods.dev/skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api"><img src="https://agentmods.dev/badge/skills/datarobot-oss/datarobot-agent-skills/datarobot-workload-api.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 142 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 5,047 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 1 finding. A grade says what 26 rules found in the file — not that it is safe. Third-party audits
  • NVIDIA SkillSpector warn 7 Sept 2026
SkillSpector: 1 finding, up to high

These are SkillSpector’s own severities. On a checked sample its high-severity flags on skills were ~96% false positives — a documented command, a public API, a “never do X” rule — so we show them as a caution to read, not a verdict. Why →

  • high Tool Misuse · line 284
    Tool parameters are crafted to achieve unintended or unsafe behavior. Parameter abuse can bypass intended safety checks (e.g. shell=True, --force, dangerous glob patterns).
    Fix: Validate all tool parameters against an allowlist. Reject dangerous parameter values (shell=True, --force, -rf /) and use safe defaults.
How audits are shown
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00142 $0.05047
Opus 5.5 $0.00057 $0.02019
Sonnet 5.5 $0.00028 $0.01009
Haiku 4.5 $0.00014 $0.00505

Measured 11d ago against content hash f5fa9591d02d, method: parsed. Prices are Anthropic first-party input rates as of 2026-10-07, from the pricing page.

Security

Grade A, and why

datarobot-workload-api scanned grade A with 1 finding against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 11d ago.

The scan reads SKILL.md. This mod also ships 5 executable files (scripts/check_limits.py, scripts/diagnose_workload.py, scripts/wait_for_build.py, …), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Makes network callslowCapability

Not a fault in itself. Listed so you know the mod talks to something, and to what.

`DATAROBOT_ENDPOINT` (must end `/api/v2`) and `DATAROBOT_API_TOKEN`: required as env vars only for raw REST (`scripts/`, `httpx`/`curl`) or CI — these bypass CLI auth. Header: `Authorization: Bearer ${DATAROBOT_API_TOKEN
skills/datarobot-workload-api/SKILL.md · 350 lines

How it starts

The opening of the file, as written. The whole thing — 350 lines — stays where its author put it; the contents beside it link to each section on GitHub.

DataRobot Workload API

Runs containers as managed, autoscalable DataRobot services. Four jobs — pick by intent:

  1. Create / configure / scale — deploy a container; change replicas, resources, autoscaling, bundle; inject credentials
  2. Diagnose — workload stuck, errored, or crash-looping
  3. Observe — logs, traces, metrics, service stats for a running workload
  4. Artifact lifecycle — iterate drafts, build images, lock for production, roll out new versions

Prerequisites

Auth: like gh. dr auth login (or existing .env/~/.config/datarobot/drconfig.yaml) persists credentials — no per-run env vars needed. Verify with dr auth check. On failure, run datarobot-setup.

Keep the CLI current (like datarobot-agent-assist): dr self update --force. Never gate on a pinned version. If a subcommand still errors unknown command after update, use the raw-REST fallback below.

DATAROBOT_ENDPOINT (must end /api/v2) and DATAROBOT_API_TOKEN: required as env vars only for raw REST (scripts/, httpx/curl) or CI — these bypass CLI auth. Header: Authorization: Bearer ${DATAROBOT_API_TOKEN}. Not in the datarobot Python SDK — call REST directly.

Transport: examples use Python httpx (pip install httpx); curl or the pulumi-datarobot provider work too.

Bundled scripts

Python in scripts/. Each uses httpx, reads DATAROBOT_ENDPOINT + DATAROBOT_API_TOKEN:

  • wait_for_running.py <workload_id> — poll until running; exit 2 on terminal failure, 3 on timeout
  • diagnose_workload.py <workload_id> — run the 5-step debug flow, print a diagnosis (--json for machine output)
  • wait_for_build.py <artifact_id> <build_id> — poll a server-side image build; dumps last 2KB of logs on FAILED
  • wait_for_replacement.py <workload_id> — poll a rolling replacement; handles the 404-when-cleared case
  • check_limits.py — print effective org scaling limits via /account/info/

Deeper docs in references/

In references/:

  • status-vocabulary.md — status enums and transitions
  • common-error-patterns.md — CrashLoopBackOff / ImagePullBackOff / OOMKilled / probe / exec-format
  • schema-reference.md — schemas, credential-type→key maps, spec quirks
  • lifecycle-flows.md — draft→lock→prod rules, replacement preconditions, redeploy matrix
  • code-to-workload.md — deploy from source: dr CLI, codeRef, Execution Environments
  • declarative-cli-deploy.md — dr workload config/up: the one-command path from a project directory, and when the granular verbs are the better choice; .datarobot.yaml, promote, --sync-env, --spec-file, delete --purge
  • cli-command-map.md — every dr workload/dr artifact verb → REST route, incl. diagnose, events, settings, logs filters
  • web-uis-behind-the-edge.md — web UI through the endpoint: prefix, auth gate, CSRF, WebSockets

Read the full file on GitHub · 350 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 11d ago Changed · +2 lines f5fa9591d02d
  2. 18d ago Changed · +3 lines 4b057781c0c5
  3. 1mo ago Changed · +1 lines · -10 tokens per session 023285dae2f5
  4. 1mo ago Changed · +2 lines d09f3c1913ad
  5. 1mo ago First seen · 342 lines · 152 tokens per session scan A a67aab36bb72

Subscribe to this mod's changes

datarobot-workload-api is a skill published in the GitHub repository datarobot-oss/datarobot-agent-skills (32 stars, last pushed 4d ago), licensed Apache-2.0. It adds 142 tokens to every session and 5,047 once invoked, about $0.0006 per session on Opus 5.5. A static security scan graded it A with 1 finding (makes network calls). No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Related

Other skills, from other repositories

offensive-container-escape

Container escape and breakout techniques targeting Docker, containerd, and Podman runtimes. Covers privileged container breakout via host filesystem mount and nsenter, Docker socket abuse through /var/run/docker.sock, Linux capability exploitation including CAPSYSADMIN, CAPSYSPTRACE, and CAPNETADMIN, cgroup v1…

SnailSploit/Claude-Red · 196 tokens

agents-v2-py

Build container-based Foundry Agents with Azure AI Projects SDK (ImageBasedHostedAgentDefinition). Use when creating hosted agents with custom container images in Azure AI Foundry.

lingxling/awesome-skills-cn · 39 tokens

kubernetes

Use when deploying to or debugging Kubernetes. Covers workload configuration, resource requests and limits, probes, rollout strategy, networking, and the failure modes that produce CrashLoopBackOff and OOMKilled. Trigger on "pod keeps restarting", "pod stuck in crashloopbackoff", "container got oomkilled". Do not use…

nimadorostkar/Claude-Skills-collection · 97 tokens

upstash-box-remote-work

Do work in an Upstash Box, a sandboxed cloud container driven through the remote Upstash MCP server (mcp.upstash.com), instead of on the local machine. Use when the user asks to run, build, test, clone, or edit something remotely, in a sandbox, in the cloud, or in a box, when the deliverable is a pull request, a…

upstash/skills · 204 tokens

apify-actor-development

Important: Before you begin, fill in the generatedBy property in the meta section of .actor/actor.json. Replace it with the tool and model you're currently using, such as "Claude Code with Claude Sonnet 4.5". This helps Apify monitor and improve AGENTS.md for specific AI tools and models.

lingxling/awesome-skills-cn · 71 tokens

apify-actorization

Actorization converts existing software into reusable serverless applications compatible with the Apify platform. Actors are programs packaged as Docker images that accept well-defined JSON input, perform an action, and optionally produce structured JSON output.

lingxling/awesome-skills-cn · 48 tokens