DO

datarobot-oss/datarobot-agent-skills

Developer tools
31 stars Quality 51 Trend 34

Bring DataRobot platform capabilities to your coding agents

Overview

Agentic skills for DataRobot enterprise AI and agent workflows. Agentic skills are modular, task-specific capability packages that help an AI agent move from general reasoning to reliable execution. Each skill bundles instructions, examples, and supporting resources so that the agent can load only what it needs for the current task, reducing context overload and improving tool use within a given workflow. DataRobot skills are Agent Context Protocol (ACP) definitions for enterprise AI and agent workflows, including building, deploying, and governing agents, as well as AI/ML tasks such as model training, deployment, predictions, feature engineering, and monitoring. They work with major coding agents, including OpenAI Codex, Anthropic Claude Code, Google Gemini CLI, Cursor, and VS Code Copilot. "Skills" is an Anthropic term used in Claude AI and Claude Code, but the concept applies more broadly. OpenAI Codex uses AGENTS.

README

DataRobot Agent Skills

Homepage · Documentation · Support

Agentic skills for DataRobot enterprise AI and agent workflows.

npx ai-agent-skills install datarobot-oss/datarobot-agent-skills

Overview

Agentic skills are modular, task-specific capability packages that help an AI agent move from general reasoning to reliable execution. Each skill bundles instructions, examples, and supporting resources so that the agent can load only what it needs for the current task, reducing context overload and improving tool use within a given workflow.

DataRobot skills are Agent Context Protocol (ACP) definitions for enterprise AI and agent workflows, including building, deploying, and governing agents, as well as AI/ML tasks such as model training, deployment, predictions, feature engineering, and monitoring. They work with major coding agents, including OpenAI Codex, Anthropic Claude Code, Google Gemini CLI, Cursor, and VS Code Copilot.

[!NOTE] “Skills” is an Anthropic term used in Claude AI and Claude Code, but the concept applies more broadly. OpenAI Codex uses AGENTS.md to define agent instructions, and Gemini uses gemini-extension.json for extensions. This repository is compatible with all of them, and more.

Quick start

[!NOTE] Supported agents for DataRobot skills include: Claude Code, Cursor, Codex, Amp, VS Code Copilot (GitHub Copilot), Gemini CLI, Goose, Letta, Kilo Code, OpenCode, and Windsurf/Devin.

Install all DataRobot skills, or only the ones you need, for all your AI agents with one command by using the universal skills installer.

For all skills:

npx ai-agent-skills install datarobot-oss/datarobot-agent-skills

For a specific skill:

npx ai-agent-skills install datarobot-oss/datarobot-agent-skills/skills/datarobot-predictions

For a specific agent:

npx ai-agent-skills install datarobot-oss/datarobot-agent-skills --agent cursor
npx ai-agent-skills install datarobot-oss/datarobot-agent-skills --agent claude

[!NOTE] By default, the installer copies skills to all supported agents at the same time. No configuration is required. For agent-specific installation methods, see the Installation to your coding agent section below.

How do skills work?

Skills are self-contained folders that package instructions, scripts, and resources for a specific use case. Each folder includes a SKILL.md file with YAML frontmatter (name and description), followed by the guidance your coding agent uses while the skill is active.

[!NOTE] All DataRobot skills follow the naming convention datarobot-, where `` describes the skill’s focus area. This provides clear identification of DataRobot-specific skills, consistent naming across the skill library, and easy discovery and organization.

Installation to your coding agent

DataRobot skills are compatible with Claude Code, Codex, Gemini CLI, Cursor, VS Code Copilot, Windsurf/Devin, and more. For the universal npx installer, see Quick start above.

Click on the section that corresponds to your coding agent below for agent-specific installation instructions.

Skills

This repository contains skills for common DataRobot workflows. You can also contribute your own skills.

Available skills

Skill Folder Description Documentation
skills/datarobot-model-training/ Instructions and utilities for training models, managing projects, and running AutoML experiments. SKILL.md
skills/datarobot-model-deployment/ Tools for deploying models, managing deployments, and configuring prediction environments. SKILL.md
skills/datarobot-predictions/ Guidance for making predictions, batch scoring, real-time predictions, and generating prediction datasets. SKILL.md
skills/datarobot-feature-engineering/ Instructions for feature engineering, feature discovery, and feature importance analysis. SKILL.md
skills/datarobot-model-monitoring/ Tools for monitoring model performance, tracking data drift, and managing model health. SKILL.md
skills/datarobot-model-explainability/ Tools for model explainability, prediction explanations, SHAP values, and model diagnostics. SKILL.md
skills/datarobot-data-preparation/ Utilities for data upload, dataset management, and data validation. SKILL.md
skills/datarobot-app-framework-cicd/ Set up CI/CD pipelines for DataRobot application templates with GitLab and GitHub Actions. SKILL.md
skills/datarobot-external-agent-monitoring/ Instrument any external AI agent with OpenTelemetry to send traces, logs, and metrics to DataRobot for monitoring and observability. Supports Google ADK, LangChain, LangGraph, CrewAI, LlamaIndex, PydanticAI, and generic Python agents. SKILL.md
skills/datarobot-agent-assist/ Build AI agents and deploy them to DataRobot. Supports building LangGraph, CrewAI, LlamaIndex, NAT and Base agents. Created agents can be bundled with MCP server, backend APIs & React frontend. SKILL.md
skills/datarobot-connectivity-datastore-sql/ Run SQL through DataRobot data connections: list connections, browse tables, preview rows, run queries, and run approved DML/DDL. SKILL.md
skills/datarobot-connectivity-jdbc-sql/ Run SQL through DataRobot against a database reached by a caller-supplied JDBC URL: test connectivity, list tables, run queries, and run approved DML/DDL. SKILL.md

Using skills in your coding agent

Once a skill is installed, mention it directly in your instructions to the coding agent:

  • “Use the DataRobot model training skill to create a new project and start AutoML training.”
  • “Use the DataRobot predictions skill to generate a prediction dataset template for deployment abc123.”
  • “Use the DataRobot feature engineering skill to analyze feature importance for my model.”
  • “Use the DataRobot model monitoring skill to check data drift for deployment xyz789.”
  • “Use the DataRobot external agent monitoring skill to instrument my agent for DataRobot monitoring.”

Your coding agent automatically loads the corresponding SKILL.md instructions and any helper scripts it needs while completing the task.

Helper scripts

Some skills include helper scripts that an agent can run directly:

  • datarobot-predictions: get_deployment_features.py, generate_prediction_data_template.py, validate_prediction_data.py, make_prediction.py
  • datarobot-model-training: create_project.py, start_training.py, list_models.py
  • datarobot-model-explainability: compute_shap_matrix.py
  • datarobot-data-preparation: upload_dataset.py
  • datarobot-external-agent-monitoring: create_shell_deployment.py, verify_otel_connection.py
  • datarobot-agent-assist: select_framework.py, clone_template.py, setup_template.py, list_llm_models.py, rehearsal.py, env_utils.py
  • datarobot-connectivity-datastore-sql: datastore_sql.py
  • datarobot-connectivity-jdbc-sql: jdbc_sql.py

These scripts are located in each skill’s scripts/ directory and can be executed directly or used as references when writing code.

Additional documentation

  • Agent framework integration—how to load and inject skills when building agents with LangGraph, PydanticAI, CrewAI, LlamaIndex, and similar frameworks.
  • Contributing—how to create skills, naming conventions, validation, and CI.

Additional references

View this README on GitHub

Recommended Tools

Try a different keyword or remove a filter.

Install

npx skillfish add datarobot-oss/datarobot-agent-skills