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

full
stringDefaults to https://app.datarobot.com/api/v2Required
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

Python
1from composio import Composio
2from openai import OpenAI
3import json
4
5openai = OpenAI()
6composio = Composio()
7
8# User ID must be a valid UUID format
9user_id = "0000-0000-0000" # Replace with actual user UUID from your database
10
11tools = composio.tools.get(user_id=user_id, toolkits=["DATAROBOT"])
12
13print("[!] Tools:")
14print(json.dumps(tools))
15
16def invoke_llm(task = "What can you do?"):
17 completion = openai.chat.completions.create(
18 model="gpt-4o",
19 messages=[
20 {
21 "role": "user",
22 "content": task, # Your task here!
23 },
24 ],
25 tools=tools,
26 )
27
28 # Handle Result from tool call
29 result = composio.provider.handle_tool_calls(user_id=user_id, response=completion)
30 print(f"[!] Completion: {completion}")
31 print(f"[!] Tool call result: {result}")
32
33invoke_llm()

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
string
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
array
create
boolean
firstName
string
language
string
lastName
string
orgAdmin
boolean
organizationId
stringRequired
password
string
requireClickthrough
boolean
username
stringRequired

Action Response

data
objectRequired
error
string
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
object

Action Response

data
objectRequired
error
string
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
string
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
string
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
string
permissions
objectRequired

Action Response

data
objectRequired
error
string
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
boolean
batchJobType
string
chunkSize
string
deploymentId
stringRequired
enabled
boolean
includePredictionStatus
boolean
includeProbabilities
boolean
includeProbabilitiesClasses
array
intakeSettings
objectRequired
name
string
outputSettings
object
schedule
object

Action Response

data
objectRequired
error
string
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
string
description
stringRequired
label
stringRequired
modelPackageId
stringRequired
predictionEnvironmentId
string

Action Response

data
objectRequired
error
string
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
boolean
description
string
modelId
stringRequired
name
string
predictionThreshold
number

Action Response

data
objectRequired
error
string
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
string
dataSourceId
string
datasetId
string
datasetVersionId
string
password
string
projectName
string
recipeId
string
url
string
useKerberos
boolean
user
string

Action Response

data
objectRequired
error
string
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
string
description
string
email
string
name
stringRequired
orgId
stringRequired

Action Response

data
objectRequired
error
string
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
string
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
string

Action Response

data
objectRequired
error
string
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
string
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
string
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
boolean
includePredictionExplanations
boolean
includePredictionIntervals
boolean
sourceCode
boolean

Action Response

data
objectRequired
error
string
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
string
workloadCategory
string

Action Response

data
arrayRequired
error
string
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
string
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
string
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
string
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
string
deploymentId
stringRequired
end
string
start
string

Action Response

data
objectRequired
error
string
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
string
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
string
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
string
batchId
string
deploymentId
stringRequired
end
string
metric
string
modelId
array
segmentAttribute
string
segmentValue
string
start
string
targetClass
string

Action Response

batchIds
array
data
arrayRequired
error
string
metrics
object
modelId
string
period
object
segmentAttribute
string
segmentValue
string
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
string
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
boolean
includeNonPredictionFeatures
boolean
limit
integer
offset
integer
orderBy
string
search
string

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
string
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
string
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
string
username
string

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
string
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
string
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
string
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
integer
offset
integer
organizationId
string

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
integer
namePart
string
offset
integer
roleId
stringRequired

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
array
status
array

Action Response

count
integerRequired
data
arrayRequired
error
string
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
integer
offset
integer
orderBy
string
types
array

Action Response

count
integer
data
arrayRequired
error
string
next
string
previous
string
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
string
orderBy
string

Action Response

data
arrayRequired
error
string
next
string
previous
string
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
string

Action Response

data
arrayRequired
error
string
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
array
buildEnvironmentType
array
championModelTargetType
string
champion_model_execution_type
string
createdBy
string
createdByMe
boolean
defaultPredictionServerId
array
executionEnvironmentType
array
importance
array
lastPredictionTimestampEnd
string
lastPredictionTimestampStart
string
limit
integerDefaults to 20
modelHealth
array
offset
integer
orderBy
string
predictionEnvironmentPlatform
array
predictionUsageDailyAvgGreaterThan
integer
predictionUsageDailyAvgLessThan
integer
role
string
search
string
serviceHealth
array
status
array
tagKeys
array
tagValues
array

Action Response

count
integerRequired
data
arrayRequired
error
string
next
stringRequired
previous
stringRequired
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
string
sortBy
string

Action Response

data
objectRequired
error
string
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
integer
offset
integer
search
string

Action Response

data
objectRequired
error
string
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
string
characteristics
array
families
array
featurelists
string
labels
array
limit
integerRequired
numberOfClusters
string
offset
integerRequired
projectId
stringRequired
searchTerm
string
showInSampleScores
boolean
sortByMetric
string
sortByPartition
string
trainingFilters
string
withMetric
string

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
array
limit
integerRequired
offset
integerRequired
organizationId
stringRequired

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
string
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
integer
offset
integer

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
string

Action Response

data
objectRequired
error
string
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
string
limit
integer
offset
integer
orderBy
string
projectId
string
projectName
string

Action Response

data
objectRequired
error
string
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
boolean
limit
integer
namePart
string
offset
integer
orderBy
string
orgId
string
userId
string

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
boolean
isAdmin
boolean
limit
integerRequired
namePart
string
offset
integerRequired
orderBy
string

Action Response

count
integerRequired
data
arrayRequired
error
string
next
string
previous
string
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
string
creatorUsername
string
dialect
string
limit
integer
offset
integer
orderBy
string
recipeType
string
search
string
status
string

Action Response

data
arrayRequired
error
string
next
string
previous
string
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
string
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
array
blendBestModels
boolean
considerBlendersInRecommendation
boolean
featurelistId
stringRequired
mode
stringDefaults to auto
prepareModelForDeployment
boolean
projectId
stringRequired
runLeakageRemovedFeatureList
boolean
scoringCodeOnly
boolean
useGpu
boolean

Action Response

data
objectRequired
error
string
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
object

Action Response

data
objectRequired
error
string
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
string
permissions
object
role_id
stringRequired

Action Response

data
objectRequired
error
string
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
string
successful
booleanRequired