Datarobot

Learn how to use Datarobot with Composio

Overview

SLUG: DATAROBOT

Description

DataRobot is a machine learning platform that automates model building, deployment, and monitoring, enabling organizations to derive predictive insights from large datasets

Authentication Details

generic_api_key
stringRequired

Connecting to Datarobot

Create an auth config

Use the dashboard to create an auth config for the Datarobot toolkit. This allows you to connect multiple Datarobot accounts to Composio for agents to use.

1

Select App

Navigate to Datarobot.

2

Configure Auth Config Settings

Select among the supported auth schemes of and configure them here.

3

Create and Get auth config ID

Click “Create Datarobot Auth Config”. After creation, copy the displayed ID starting with ac_. This is your auth config ID. This is not a sensitive ID — you can save it in environment variables or a database. This ID will be used to create connections to the toolkit for a given user.

Connect Your Account

Using API Key

1from composio import Composio
2
3# Replace these with your actual values
4datarobot_auth_config_id = "ac_YOUR_DATAROBOT_CONFIG_ID" # Auth config ID created above
5user_id = "0000-0000-0000" # UUID from database/app
6
7composio = Composio()
8
9def authenticate_toolkit(user_id: str, auth_config_id: str):
10 # Replace this with a method to retrieve an API key from the user.
11 # Or supply your own.
12 user_api_key = input("[!] Enter API key")
13
14 connection_request = composio.connected_accounts.initiate(
15 user_id=user_id,
16 auth_config_id=auth_config_id,
17 config={"auth_scheme": "API_KEY", "val": {"generic_api_key": user_api_key}}
18 )
19
20 # API Key authentication is immediate - no redirect needed
21 print(f"Successfully connected Datarobot for user {user_id}")
22 print(f"Connection status: {connection_request.status}")
23
24 return connection_request.id
25
26
27connection_id = authenticate_toolkit(user_id, datarobot_auth_config_id)
28
29# You can verify the connection using:
30connected_account = composio.connected_accounts.get(connection_id)
31print(f"Connected account: {connected_account}")

Tools

Executing tools

To prototype you can execute some tools to see the responses and working on the Datarobot toolkit’s playground

For code examples, see the Tool calling guide and Provider examples.

Tool List

Tool Name: Add Users to Group

Description

Tool to add one or more users to a DataRobot user group by groupId. Use when you have a valid groupId and want to add existing usernames. Limit 100 users per request.

Action Parameters

groupId
stringRequired
users
arrayRequired

Action Response

count
integerRequired
data
arrayRequired
error
successful
booleanRequired

Tool Name: Add User to Organization

Description

Tool to add a user to an existing organization. Use when you have an organizationId and wish to add or create a user within it.

Action Parameters

accessRoleIds
create
boolean
firstName
language
lastName
orgAdmin
boolean
organizationId
stringRequired
password
requireClickthrough
boolean
username
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Build Java Scoring Code Package

Description

Tool to build a Java package containing Scoring Code with agent integration. Use after confirming the deployment is ready.

Action Parameters

deploymentId
stringRequired
options

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Cancel Project Job

Description

Tool to cancel a pending job for a project. Use when you need to stop a queued or running job after confirming its ID.

Action Parameters

job_id
stringRequired
project_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Check Project Status

Description

Tool to check the status of a DataRobot project. Use after creating or loading a project to monitor its stage and Autopilot completion.

Action Parameters

projectId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Create Access Role

Description

Tool to create a custom Access Role. Use when adding tailored permissions for an organization.

Action Parameters

name
stringRequired
organizationId
permissions
objectRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Create Batch Prediction Job Definition

Description

Tool to create a Batch Prediction job definition. Use when you need to manually or scheduled scoring on large datasets.

Action Parameters

abortOnError
batchJobType
chunkSize
deploymentId
stringRequired
enabled
includePredictionStatus
includeProbabilities
includeProbabilitiesClasses
intakeSettings
objectRequired
name
outputSettings
schedule

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Create Deployment

Description

Tool to create a DataRobot deployment from a model package. Use after obtaining a modelPackageId to deploy a trained model for predictions.

Action Parameters

defaultPredictionServerId
description
stringRequired
label
stringRequired
modelPackageId
stringRequired
predictionEnvironmentId

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Create Model Package

Description

Tool to create a model package from a DataRobot Leaderboard model. Use after a model is trained and you need an offline package.

Action Parameters

computeAllTsIntervals
description
modelId
stringRequired
name
predictionThreshold

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Create DataRobot Project

Description

Tool to create a new DataRobot project. Use after specifying a dataset URL or existing dataset ID. Returns project ID and a status URL to poll.

Action Parameters

credentialId
dataSourceId
datasetId
datasetVersionId
password
projectName
recipeId
url
useKerberos
user

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Create User Group

Description

Tool to create a new user group. Use when you need to add a group to an existing DataRobot organization after confirming orgId.

Action Parameters

accessRoleId
description
email
name
stringRequired
orgId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Delete Access Role

Description

Tool to delete a custom Access Role. Use when you need to remove a custom role by its ID after verifying the ID.

Action Parameters

roleId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Delete Deployment

Description

Tool to delete a DataRobot deployment. Use when you need to permanently remove a deployment by its unique ID.

Action Parameters

deploymentId
stringRequired
ignoreManagementAgent

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Delete Project

Description

Tool to delete a DataRobot project. Use when you need to permanently remove a project by its unique ID. Use after confirming the project is no longer needed.

Action Parameters

projectId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Delete User Group

Description

Tool to delete a user group by its ID. Use after confirming the group ID to remove that group.

Action Parameters

group_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Download Scoring Code

Description

Tool to download scoring code for a DataRobot deployment. Use after deployment is active.

Action Parameters

deploymentId
stringRequired
includeAgent
includePredictionExplanations
includePredictionIntervals
sourceCode

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Export Tenant Usage

Description

Tool to export tenant resource usage. Use when you need to retrieve usage data for a specific tenant over a date range.

Action Parameters

end
stringRequired
start
stringRequired
tenantId
stringRequired
userId
workloadCategory

Action Response

data
arrayRequired
error
successful
booleanRequired

Tool Name: Get Access Role

Description

Tool to retrieve details for a specific Access Role by ID. Use when you need confirmation of role permissions.

Action Parameters

role_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Account Info

Description

Tool to fetch the current account information including organization, user, and tenant details. Use when you need the current organization ID for organization user listing.

Action Parameters

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Accuracy Metrics Config

Description

Tool to retrieve which accuracy metrics are displayed and their order for a deployment. Use when you need the configured metrics order for a deployment.

Action Parameters

deploymentId
stringRequired

Action Response

data
arrayRequired
error
successful
booleanRequired

Tool Name: Get Accuracy Over Time

Description

Tool to retrieve baseline and accuracy metric values over time buckets for a deployment. Use when analyzing model performance trends; call after confirming the deployment is live.

Action Parameters

bucketSize
deploymentId
stringRequired
end
start

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Datetime Partitioning

Description

Tool to retrieve datetime partitioning configuration for a project. Use when you need to inspect a project's time-series settings before modeling.

Action Parameters

projectId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Deployment

Description

Tool to retrieve a deployment by ID. Use after creating or updating a deployment to fetch its full metadata and status.

Action Parameters

deploymentId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Deployment Accuracy

Description

Tool to retrieve accuracy metrics for a deployment over a time period. Use when you need to analyze model performance trends or drift for a specific deployment.

Action Parameters

baselineModelId
batchId
deploymentId
stringRequired
end
metric
modelId
segmentAttribute
segmentValue
start
targetClass

Action Response

batchIds
data
arrayRequired
error
metrics
modelId
period
segmentAttribute
segmentValue
successful
booleanRequired

Tool Name: Get Deployment Capabilities

Description

Tool to retrieve the capabilities for a deployment. Use after creating or loading a deployment to check supported features.

Action Parameters

deploymentId
stringRequired

Action Response

data
arrayRequired
error
successful
booleanRequired

Tool Name: Get Deployment Features

Description

Tool to retrieve features in the universe dataset associated with a deployment. Use after deployment creation to explore its feature set.

Action Parameters

deploymentId
stringRequired
forSegmentedAnalysis
includeNonPredictionFeatures
limit
integer
offset
integer
orderBy
search

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: Get Deployment Settings Checklist

Description

Tool to return a checklist of deployment settings and their configuration state. Use when you need to verify which settings are set, partial, or not set for a given deployment.

Action Parameters

deploymentId
stringRequired

Action Response

data
arrayRequired
error
successful
booleanRequired

Tool Name: Get Project

Description

Tool to retrieve a DataRobot project by ID. Use when you need project metadata before further operations.

Action Parameters

project_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Project Access Control

Description

Tool to list users with their roles on a project. Use after assigning permissions or when auditing project access. Example prompt: "List access control entries for project 5f6a7b8c9d0e1f2a3b4c5d6e, first page of 20."

Action Parameters

limit
integerRequired
offset
integerRequired
projectId
stringRequired
userId
username

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired

Tool Name: Get Project Job

Description

Tool to retrieve details for an in-progress project job. Use after starting a project job to monitor its status before completion.

Action Parameters

job_id
stringRequired
project_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Get Tenant Active Users

Description

Tool to retrieve active users in a tenant over a date range. Use when an admin needs active user insights for a specific tenant between two dates.

Action Parameters

end
stringRequired
start
stringRequired
tenantId
stringRequired

Action Response

data
arrayRequired
error
successful
booleanRequired

Tool Name: Get User Group

Description

Tool to retrieve a user group by its ID. Use when you need the group's properties and permissions.

Action Parameters

groupId
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: List Access Roles

Description

Tool to list access roles. Use when you need to retrieve available access roles, with optional global filtering.

Action Parameters

globalRoles
stringRequired
limit
offset
organizationId

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Access Role Users

Description

Tool to list users assigned to an Access Role. Use when you need to fetch all users directly, via groups, or via organization by role ID.

Action Parameters

limit
namePart
offset
roleId
stringRequired

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Batch Jobs

Description

Tool to list DataRobot batch jobs. Use when you need to retrieve batch job records filtered by status or source.

Action Parameters

limit
integerRequired
offset
integerRequired
source
status

Action Response

count
integerRequired
data
arrayRequired
error
successful
booleanRequired

Tool Name: List Credentials

Description

Tool to list all available credentials. Use when you need to retrieve credentials accessible to the authenticated user.

Action Parameters

limit
offset
orderBy
types

Action Response

count
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Datasets

Description

Tool to list all datasets in the DataRobot global catalog. Use when you need to browse or filter available datasets before modeling or prediction.

Action Parameters

category
orderBy

Action Response

data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Data Sources

Description

Tool to list all available data sources. Use when retrieving the catalog of data connections.

Action Parameters

type

Action Response

data
arrayRequired
error
successful
booleanRequired

Tool Name: List Deployments

Description

Tool to list deployments a user can view. Use when retrieving paginated deployments from DataRobot.

Action Parameters

accuracyHealth
buildEnvironmentType
championModelTargetType
champion_model_execution_type
createdBy
createdByMe
defaultPredictionServerId
executionEnvironmentType
importance
lastPredictionTimestampEnd
lastPredictionTimestampStart
limit
integerDefaults to 20
modelHealth
offset
integer
orderBy
predictionEnvironmentPlatform
predictionUsageDailyAvgGreaterThan
predictionUsageDailyAvgLessThan
role
search
serviceHealth
status
tagKeys
tagValues

Action Response

count
integerRequired
data
arrayRequired
error
next
Required
previous
Required
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Feature Lists

Description

Tool to list all feature lists for a project. Use when you need to retrieve and filter feature lists associated with a DataRobot project.

Action Parameters

projectId
stringRequired
searchFor
sortBy

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: List Model Packages

Description

Tool to list model packages. Use when you need to search or page through model packages.

Action Parameters

limit
offset
search

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: List Model Records

Description

Tool to list model records for a project. Use when retrieving model metadata with pagination and filters.

Action Parameters

blueprints
characteristics
families
featurelists
labels
limit
integerRequired
numberOfClusters
offset
integerRequired
projectId
stringRequired
searchTerm
showInSampleScores
sortByMetric
sortByPartition
trainingFilters
withMetric

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Organization Users

Description

Tool to list memberships (users) in an organization. Use when you need to page through or filter users by ID.

Action Parameters

ids
limit
integerRequired
offset
integerRequired
organizationId
stringRequired

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired

Tool Name: List Prediction Environments

Description

Tool to list prediction environments. Use when selecting an environment for deploying or scoring models.

Action Parameters

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: List Prediction Servers

Description

Tool to list prediction servers available to the user. Use after authenticating to retrieve real-time and batch scoring endpoints.

Action Parameters

limit
offset

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired

Tool Name: List Project Jobs

Description

Tool to list all jobs for a given DataRobot project. Use when you need to inspect or monitor the status of jobs within a project, optionally filtering by status.

Action Parameters

project_id
stringRequired
status

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: List Projects

Description

Tool to list all available DataRobot projects. Use when retrieving a catalog of projects to select from.

Action Parameters

featureDiscovery
limit
offset
orderBy
projectId
projectName

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: List User Groups

Description

Tool to list user groups. Use when you need to retrieve DataRobot user groups with optional filtering and pagination.

Action Parameters

excludeUserMembership
limit
namePart
offset
orderBy
orgId
userId

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Users in Group

Description

Tool to list users in a specific DataRobot user group. Use when you need to retrieve membership details for a given groupId.

Action Parameters

groupId
stringRequired
isActive
isAdmin
limit
integerRequired
namePart
offset
integerRequired
orderBy

Action Response

count
integerRequired
data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: List Wrangling Recipes

Description

Tool to list all available wrangling recipes. Use when fetching paginated recipes for data wrangling.

Action Parameters

creatorUserId
creatorUsername
dialect
limit
offset
orderBy
recipeType
search
status

Action Response

data
arrayRequired
error
next
previous
successful
booleanRequired
totalCount
integerRequired

Tool Name: Pause or Unpause Autopilot

Description

Tool to pause or unpause Autopilot for a project. Use when you need to stop or resume automated modeling jobs after confirming the project ID.

Action Parameters

command
stringRequired
project_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Start Autopilot

Description

Tool to start Autopilot on a provided feature list. Use after loading a project and preparing feature lists.

Action Parameters

autopilotClusterList
blendBestModels
considerBlendersInRecommendation
featurelistId
stringRequired
mode
stringDefaults to auto
prepareModelForDeployment
projectId
stringRequired
runLeakageRemovedFeatureList
scoringCodeOnly
useGpu

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Start DataRobot Autopilot

Description

Tool to start the data modeling (Autopilot) process for a DataRobot project. Use after uploading data and configuring the project to initiate modeling.

Action Parameters

project_id
stringRequired
settings

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Update Access Role

Description

Tool to update a custom Access Role. Use when you need to change the name or permissions of an existing custom role.

Action Parameters

name
permissions
role_id
stringRequired

Action Response

data
objectRequired
error
successful
booleanRequired

Tool Name: Update Accuracy Metrics Config

Description

Tool to update which accuracy metrics are returned by the accuracy endpoint for a deployment. Use after deployment is live to customize returned metrics.

Action Parameters

data
arrayRequired
deploymentId
stringRequired

Action Response

data
arrayRequired
error
successful
booleanRequired