How to integrate Datarobot MCP with LlamaIndex

Trusted by
AWS
Glean
Zoom
Airtable

30 min · no commitment · see it on your stack

Datarobot logo
LlamaIndex logo
divider

Introduction

This guide walks you through connecting Datarobot to LlamaIndex using the Composio tool router. By the end, you'll have a working Datarobot agent that can train a new model on uploaded sales data, deploy the latest churn prediction model, get performance metrics for active projects through natural language commands.

This guide will help you understand how to give your LlamaIndex agent real control over a Datarobot account through Composio's Datarobot MCP server.

Before we dive in, let's take a quick look at the key ideas and tools involved.

Also integrate Datarobot with

TL;DR

Here's what you'll learn:
  • Set your OpenAI and Composio API keys
  • Install LlamaIndex and Composio packages
  • Create a Composio Tool Router session for Datarobot
  • Connect LlamaIndex to the Datarobot MCP server
  • Build a Datarobot-powered agent using LlamaIndex
  • Interact with Datarobot through natural language

What is LlamaIndex?

LlamaIndex is a data framework for building LLM applications. It provides tools for connecting LLMs to external data sources and services through agents and tools.

Key features include:

  • ReAct Agent: Reasoning and acting pattern for tool-using agents
  • MCP Tools: Native support for Model Context Protocol
  • Context Management: Maintain conversation context across interactions
  • Async Support: Built for async/await patterns

What is the Datarobot MCP server, and what's possible with it?

The Datarobot MCP server is an implementation of the Model Context Protocol that connects your AI agent and assistants like Claude, Cursor, etc directly to your Datarobot account. It provides structured and secure access to your machine learning projects, so your agent can automate model training, manage deployments, monitor predictions, and handle datasets on your behalf.

  • Automated model training and evaluation: Let your agent kick off new model builds, select algorithms, and evaluate model performance across datasets with ease.
  • Deployment management: Direct your agent to deploy, update, or remove models to production environments, ensuring seamless delivery of predictive insights.
  • Prediction and scoring automation: Have your agent generate batch or real-time predictions using trained models, making it easy to integrate AI-powered results into your workflows.
  • Monitoring and health checks: Ask your agent to track model accuracy, monitor drift, and receive alerts if deployed models aren't performing as expected.
  • Project and dataset orchestration: Enable your agent to create new projects, upload datasets, and manage data sources, keeping your machine learning pipeline organized and up to date.

Supported Tools & Triggers

Tools
Add Users to GroupTool to add one or more users to a DataRobot user group by groupId.
Add User to OrganizationTool to add a user to an existing organization.
Analyze Dataset DefinitionTool to analyze a dataset definition by ID.
Archive Model PackageTool to archive a DataRobot model package.
Build Java Scoring Code PackageInitiates an asynchronous build of a Java JAR package containing DataRobot Scoring Code for a deployment.
Cancel Notebook JobTool to cancel a running or pending notebook job execution.
Cancel Project JobTool to cancel a pending job for a project.
Check Project StatusTool to check the status of a DataRobot project.
Clone Application TemplateTool to clone an application template into a codespace.
Clone Files CollectionTool to create a duplicate files collection in DataRobot.
Copy File or FolderTool to copy a file or folder within the same DataRobot catalog item.
Create Access RoleCreate a custom Access Role for an organization.
Create and Start Notebook CodespaceTool to create and start a new notebook codespace in DataRobot.
Create Automated DocumentTool to request generation of automated compliance documents in DataRobot.
Create Batch Monitoring JobTool to create a DataRobot Batch Monitoring job for tracking model performance and data drift on batch predictions.
Create Batch Monitoring Job DefinitionTool to create a Batch Monitoring job definition for tracking deployment performance and data drift.
Create Batch Prediction Job DefinitionTool to create a Batch Prediction job definition.
Create Batch PredictionsTool to create a new batch predictions job in DataRobot.
Create Batch Predictions From ExistingTool to create a new Batch Prediction job based on an existing job's configuration.
Create Batch Predictions From Job DefinitionTool to launch a Batch Prediction job from a job definition.
Create Calendar from File UploadTool to create a DataRobot calendar by uploading a CSV or XLSX file containing date events.
Create Calendars from Country CodeInitialize generation of preloaded calendars from a country code.
Create Calendar from DatasetTool to create a calendar from a dataset in DataRobot.
Create Change RequestCreate a Change Request for a DataRobot deployment to enable governance workflows.
Create Code SnippetsTool to generate code snippets for DataRobot models, predictions, or workloads.
Download Code SnippetTool to download code snippets for DataRobot deployments, models, or workloads.
Create CommentTool to create a comment on a DataRobot entity (deployment, use case, model, catalog, etc.
Create Compliance Doc TemplateTool to create a new compliance documentation template in DataRobot.
Create CredentialsStore a new set of credentials in DataRobot for use with data sources and connections.
Create Custom Application SourceTool to create a custom application source in DataRobot.
Create Custom Application Source VersionCreate a new custom application source version in DataRobot.
Create Custom Application Source From TemplateTool to create a custom application source from a custom template.
Create Custom JobTool to create a new DataRobot custom job.
Create Custom Jobs CleanupTool to permanently delete a custom job.
Create Custom Job Hosted Custom Metric TemplateTool to create a hosted custom metric template for a DataRobot custom job.
Create Custom ModelTool to create a new DataRobot custom model for training or inference.
Get Custom Model Deployment LogsTool to request logs from a deployed custom model.
Clone Custom ModelTool to clone an existing custom model in DataRobot.
Create Custom Model VersionTool to create a new version for an existing DataRobot custom model.
Create Data Disparity InsightsTool to start data disparity insight calculations for a DataRobot model.
Create Data Engine Workspace StateCreate a Data Engine workspace state by executing a SQL query against DataRobot datasets.
Create Dataset DefinitionCreate a dataset definition in DataRobot to define how to access and use a dataset from the AI Catalog.
Create Dataset Chunk DefinitionTool to create a chunk definition for a dataset definition in DataRobot.
Create Dataset Feature ListTool to create a custom feature list within a dataset.
Create Dataset from FileTool to create a DataRobot dataset by uploading a file (CSV, Excel, etc.
Create Dataset Feature TransformTool to create a feature transform on a DataRobot dataset.
Create Dataset from Data SourceTool to create a dataset from an external data source connector (database, S3, etc.
Create Dataset from RecipeTool to create a dataset from a DataRobot wrangling recipe.
Create Dataset from URLTool to create a DataRobot dataset from a publicly accessible URL.
Create Dataset Refresh JobTool to schedule a dataset refresh job in DataRobot.
Create Dataset RelationshipsTool to create dataset relationships in DataRobot by linking features between two datasets.
Create Dataset Version from Latest VersionTool to create a new dataset version from the latest version of its data source.
Create Dataset Version from URLTool to create a new version for an existing DataRobot dataset from a publicly accessible URL.
Create Dataset Version from Previous VersionTool to create a new dataset version from a specific previous version of its data source.
Create Dataset Version from FileTool to create a new version of an existing DataRobot dataset by uploading a file (CSV, Excel, etc.
Create Data SliceCreate a new data slice in a DataRobot project to define a subset of data based on feature filters.
Create Data Slice Size ComputationTool to compute the number of rows available after applying a data slice to a dataset subset.
Create Data StageTool to create a data stage in DataRobot.
Create DeploymentCreate a DataRobot deployment from a model package to enable real-time or batch predictions.
Create Deployment Custom Metric from Custom JobTool to create a deployment custom metric from an existing custom job in DataRobot.
Create Deployments Actuals Data ExportsTool to create a deployment actuals data export for a specified time period.
Create Deployment Actuals from DatasetTool to submit actuals values from AI Catalog dataset for deployment monitoring.
Create Deployment Actuals from JSONSubmit actual values for predictions to a DataRobot deployment.
Create Deployment Custom MetricCreate a deployment custom metric for monitoring deployment or model-specific metrics.
Create Deployment Custom Metrics Bulk UploadBulk upload custom metric values to a DataRobot deployment.
Upload Custom Metric Values from DatasetTool to upload custom metric values from a dataset to a deployment.
Upload Custom Metrics from JSONTool to upload custom metric values from JSON for a DataRobot deployment.
Create Deployment Monitoring BatchTool to create a monitoring batch for a DataRobot deployment.
Delete Deployment Monitoring DataTool to delete deployment monitoring data for a specific model within a time period.
Create Deployment Prediction Data ExportTool to create a prediction data export for a deployment.
Create Deployment Retraining PolicyTool to create a deployment retraining policy in DataRobot.
Create Deployment Training Data ExportTool to create a deployment training data export in DataRobot.
Create Entity Notification ChannelTool to create an entity notification channel in DataRobot.
Create Entity Notification PolicyTool to create an entity notification policy in DataRobot.
Create Entity Notification Policy from TemplateTool to create an entity notification policy from a template.
Create Entity Notification Policy TemplateTool to create an entity notification policy template in DataRobot.
Create Entity TagTool to create a new entity tag in DataRobot.
Create Execution Environment VersionTool to create a new version for an existing DataRobot execution environment.
Request Execution Environment Version BuildTool to request an on-demand image build for an execution environment version.
Create External Data Store Standard UDFTool to start a job that creates a standard user-defined function in an external data store.
Create External Data SourcesTool to create a new external data source in DataRobot.
Create External Data StoreTool to create a new external data store in DataRobot.
Get External Data Store ColumnsTool to retrieve column metadata from an external data store.
Get Data Store Columns InfoTool to retrieve column metadata for a table in an external data store.
Retrieve Data Store SchemasTool to retrieve data store schemas.
List External Data Store TablesTool to retrieve database tables and views from a DataRobot external data store.
Create External OAuth ProviderCreate an external OAuth provider configuration in DataRobot for integrating with external services.
Create Empty Files Catalog ItemTool to create an empty files catalog item in DataRobot.
Copy Files in BatchTool to copy multiple files or folders in a batch operation within DataRobot's data registry.
Create Files DownloadsTool to generate a temporary download URL for a DataRobot catalog file.
Add Files to Existing Catalog ItemTool to add file(s) to an existing DataRobot files catalog item.
Create Files From StageTool to apply staged files to a catalog item.
Create Files From URLTool to create a files catalog item in DataRobot from a URL.
Add Files to Catalog Item from URLTool to add file(s) into an existing files catalog item from a URL.
Create Files Download LinksTool to generate temporary download URLs for files in a catalog item.
Create File StageTool to create an empty stage for File Registry files upload.
Stage File for Batch UploadTool to stage a file for a batch upload in DataRobot.
Create File Version Download LinksTool to generate temporary download URLs for catalog file versions.
Create GenAI Chat PromptsTool to create a GenAI chat prompt and execute it with an LLM.
Create GenAI Comparison ChatTool to create a new GenAI comparison chat within a playground for comparing different LLM responses.
Create GenAI Custom Model Embedding ValidationsTool to create and run validation tests for GenAI custom model embeddings.
Create GenAI Custom Model LLM ValidationsTool to create a GenAI custom model LLM validation.
Create GenAI Custom Model Vector Database ValidationCreate a validation for a GenAI custom model deployment's vector database capabilities.
Create GenAI Custom Model VersionTool to create a GenAI custom model version from an LLM blueprint in DataRobot.
Create GenAI Evaluation Dataset ConfigurationTool to create a GenAI evaluation dataset configuration for testing LLM applications.
Create GenAI LLM BlueprintTool to create a new GenAI LLM blueprint for generative AI applications.
Create GenAI LLM Blueprints from Chat PromptTool to create a GenAI LLM blueprint from an existing chat prompt.
Create GenAI LLM Blueprint from LLM BlueprintTool to create a new GenAI LLM Blueprint by copying from an existing LLM Blueprint.
Create GenAI LLM Test ConfigurationTool to create a GenAI LLM test configuration for AI robustness testing.
Create GenAI LLM Test SuiteTool to create a new GenAI LLM test suite for evaluating LLM applications.
Create GenAI PlaygroundTool to create a new GenAI playground for experimenting with LLM applications.
Create GenAI Playground OOTB Metric ConfigurationsTool to create OOTB (Out-of-the-Box) metric configurations for a GenAI playground.
Create GenAI Playground Trace DatasetTool to create a trace dataset for a GenAI playground.
Create Guard Configurations to New Custom Model VersionTool to apply guard configurations to a new custom model version in DataRobot.
Create Guard ConfigurationCreate a guard configuration to monitor and control AI model behavior during prompt and response stages.
Create Insights Lift ChartTool to request calculation of a Lift chart insight in DataRobot.
Create ROC Curve InsightRequest calculation of ROC curve insights for a DataRobot model, custom model, or vector database.
Create Insights SHAP DistributionsTool to request calculation of SHAP Distributions in DataRobot.
Create SHAP Impact InsightTool to request calculation of SHAP Impact insights in DataRobot.
Create Insights SHAP MatrixTool to request calculation of a SHAP Matrix insight in DataRobot.
Create SHAP Preview InsightsRequest calculation of SHAP Preview insights with an optional data slice.
Create Modeling FeaturelistTool to create a new modeling featurelist in a DataRobot project.
Create Model PackageTool to create a model package from a DataRobot Leaderboard model.
Create Model Package from JSONTool to create a DataRobot model package from JSON metadata.
Create NotebookTool to create a new notebook in DataRobot Workbench for interactive code development.
Create Notebook Environment VariablesTool to create one or more environment variables for a specific notebook.
Expose Notebook PortTool to expose a port on a DataRobot notebook execution environment.
Clone Notebook from RevisionTool to clone a notebook from an existing revision, creating a new notebook as a copy.
Create Notebook JobTool to create a scheduled notebook job in DataRobot.
Create Notebook Job Manual RunTool to manually trigger a notebook job run in DataRobot.
Create Notebook RevisionTool to create a new revision for a DataRobot notebook.
Create Notebook Runtime Client ActivityTool to record client activity for a running notebook session.
Batch Create Notebook CellsTool to batch create multiple cells in a DataRobot notebook.
Batch Delete Notebook CellsTool to batch delete multiple cells from a DataRobot notebook.
Create Notebook from FileTool to create a new notebook in DataRobot by uploading a notebook file.
Create Notification Channel TemplateTool to create a notification channel template in DataRobot.
Send Email Verification Code for Notification ChannelTool to send a 6-digit verification code to a user's email address for setting up a notification channel.
Create Notification Webhook Channel TestTool to test webhook notification channel configuration by creating a test notification.
Verify Email Channel CodeVerify the notification email channel verification code.
Create OCR Job ResourceTool to create an OCR (Optical Character Recognition) job resource in DataRobot.
Create OpenTelemetry Metric ConfigurationTool to create an OpenTelemetry metric configuration for a DataRobot entity (deployment, use case, etc.
Get OTEL Metrics Values Over Time SegmentsTool to get OpenTelemetry metric values for a specified entity, grouped by multiple attributes.
Create Prediction Dataset from Data SourceUpload a prediction dataset from a DataSource for making predictions on a DataRobot project.
Initialize Prediction ExplanationsTool to initialize prediction explanations for a DataRobot model.
Create DataRobot ProjectCreate a new DataRobot project from a dataset URL, existing dataset ID, or data source connector.
Clone DataRobot ProjectTool to clone an existing DataRobot project.
Create Projects AutopilotsTool to start Autopilot on a DataRobot project with a specific feature list.
Batch Transform Feature TypesCreate multiple new features by transforming existing features to a different variable type.
Create Bias Mitigation Feature InfoTool to submit a job to create bias mitigation data quality information for a given project and feature.
Create Blender ModelTool to create a blender model from multiple existing models in a DataRobot project.
Validate Cross-Series PropertiesTool to validate columns for potential use as the group-by column for cross-series functionality in a DataRobot project.
Create Datetime Model From ModelTool to retrain an existing datetime model with specified parameters.
Prepare Model for DeploymentTool to prepare a DataRobot model for deployment by marking it as deployment-ready.
Create Feature Association MatrixTool to compute a feature association matrix for a DataRobot project using a specific featurelist.
Create Feature ListTool to create a new featurelist in a DataRobot project.
Create Frozen ModelTrain a new frozen model with parameters from an existing model.
Train DataRobot ModelTool to train a new model in a DataRobot project using a specific blueprint.
Create Advanced-Tuned ModelSubmit a job to create a new version of a DataRobot model with different advanced tuning parameters.
Create Feature EffectsTool to request Feature Effects calculation for a DataRobot model.
Calculate Model Feature ImpactTool to add a request to calculate feature impact for a DataRobot model to the job queue.
Retrain Model From Existing ModelTool to retrain an existing model with specified parameters.
Create Projects Payoff MatricesTool to create a payoff matrix for a binary classification project in DataRobot.
Create Prediction Dataset UploadTool to create a prediction dataset upload for a DataRobot project.
Create Prediction Dataset from URLTool to upload a prediction dataset from a URL to a DataRobot project.
Create Project PredictionsTool to create new predictions for a dataset using a trained model within a DataRobot project.
Create Project Rating TableTool to upload a modified rating table file to a DataRobot project.
Create Project Training PredictionsTool to create training predictions for a DataRobot project.
Transform Feature TypeCreate a new feature by changing the type of an existing feature in a DataRobot project.
Create Recipe from RecipeTool to clone an existing wrangling recipe in DataRobot.
Create Recipe from DatasetTool to create a DataRobot recipe from a dataset.
Create Recipe from Data StoreCreate a recipe and data source from a DataRobot data store.
Create Recipe PreviewTool to start the job that generates a preview of the data after applying a wrangling recipe.
Create Recipe SQL QueryTool to build SQL query for a DataRobot recipe.
Create Relationships ConfigurationTool to create a relationships configuration in DataRobot for connecting multiple datasets for feature engineering.
Create Remote EventsTool to post a remote deployment event to DataRobot.
Create Secure ConfigurationTool to create a secure configuration for storing credentials and sensitive data.
Create String EncryptionTool to encrypt a string which DataRobot can decrypt when needed.
Create Usage Data ExportsTool to create a customer usage data artifact request in DataRobot.
Create Use CaseTool to create a new DataRobot use case.
Link Multiple Entities to Use CaseTool to link multiple entities to a DataRobot use case in bulk.
Create User Blueprint From BlueprintClone a blueprint from a project to create a user blueprint.
Validate User Blueprints in BulkTool to validate multiple user blueprints in bulk and check their configuration correctness.
Add User Blueprints to ProjectAdd user blueprints to a DataRobot project's repository.
Validate User Blueprint Task ParametersTool to validate task parameters for custom tasks in DataRobot User Blueprints.
Create User GroupTool to create a new user group.
Create Value TrackerTool to create a new DataRobot value tracker for tracking ML project value and lifecycle stages.
Create Value Tracker AttachmentsTool to attach resources to a DataRobot value tracker.
Delete Access RoleTool to delete a custom Access Role.
Delete Batch JobTool to cancel a DataRobot batch job.
Delete Batch Monitoring Job DefinitionTool to delete a Batch Prediction job definition.
Delete Batch PredictionTool to cancel a DataRobot Batch Prediction job.
Delete Batch Prediction Job DefinitionTool to delete a Batch Prediction job definition.
Delete CalendarTool to delete a DataRobot calendar.
Delete CommentTool to delete a DataRobot comment.
Delete Compliance Doc TemplateTool to delete a compliance documentation template.
Delete CredentialsTool to delete a credentials set.
Delete Custom ApplicationTool to delete a DataRobot custom application.
Delete Custom Application SourceTool to delete a custom application source with all its versions.
Delete Custom Application Source VersionTool to delete a custom application source version if it is still mutable.
Delete Custom JobTool to delete a DataRobot custom job.
Delete Custom ModelTool to delete a custom model in DataRobot.
Delete DatasetTool to delete a dataset from DataRobot.
Delete Dataset DefinitionTool to soft delete a dataset definition by ID.
Delete Chunk DefinitionTool to soft delete a chunk definition by ID.
Delete Dataset FeaturelistTool to delete a dataset featurelist.
Delete Dataset Refresh JobTool to delete a DataRobot dataset refresh job.
Delete Dataset RelationshipTool to delete a dataset relationship.
Delete Dataset VersionTool to delete a specific version of a dataset from DataRobot's catalog.
Delete Data SlicesTool to delete multiple data slices in bulk.
Delete Data SliceTool to delete a data slice.
Delete DeploymentTool to delete a DataRobot deployment.
Delete Actuals Data ExportTool to delete an actuals data export job from a deployment.
Delete Deployment Custom MetricTool to delete a custom metric from a deployment.
Delete Deployment Monitoring BatchTool to delete a monitoring batch from a deployment.
Delete Deployment Retraining PolicyTool to delete a retraining policy from a deployment.
Delete Entity Notification ChannelTool to delete an entity notification channel.
Delete Entity Notification PolicyTool to delete an entity notification policy.
Delete Entity TagTool to delete an entity tag.
Delete External Data SourceTool to delete an external data source.
Delete External Data StoreTool to delete an external data store.
Delete External OAuth ProviderTool to delete an external OAuth provider from DataRobot.
Delete FilesTool to delete a file from DataRobot.
Delete Files from CatalogTool to delete files or folders from a DataRobot catalog item.
Delete GenAI Cost Metric ConfigurationTool to delete a GenAI cost metric configuration.
Delete GenAI Custom Model LLM ValidationTool to delete a custom model LLM validation from DataRobot.
Delete GenAI Custom Model Vector Database ValidationsTool to delete a custom model vector database validation in DataRobot.
Delete GenAI LLM BlueprintsTool to delete a GenAI LLM blueprint in DataRobot.
Delete GenAI LLM Test ConfigurationsTool to delete an LLM test configuration in DataRobot.
Delete GenAI OOTB Metric ConfigurationTool to delete a GenAI out-of-the-box (OOTB) metric configuration.
Delete GenAI PlaygroundTool to delete a GenAI playground.
Delete Multiple GroupsTool to delete multiple user groups by their IDs in a single request.
Remove Users from GroupTool to remove one or more users from a DataRobot user group by groupId.
Delete Modeling FeaturelistTool to delete a specified modeling featurelist from a DataRobot project.
Delete Notebook CellTool to delete a specific cell from a DataRobot notebook.
Delete Notebook Environment VariableTool to delete a notebook environment variable by ID.
Delete Notebook Execution Environment PortTool to delete an exposed port from a notebook execution environment in DataRobot.
Delete Notebook JobTool to delete a notebook job in DataRobot.
Delete Notebook RevisionTool to delete a specific notebook revision by its ID.
Delete NotebookTool to delete a notebook in DataRobot.
Delete Notification Channel TemplateTool to delete a notification channel template.
Delete OpenTelemetry Metric ConfigurationTool to delete an OpenTelemetry metric configuration for a specified entity.
Delete Payoff MatrixTool to delete a payoff matrix from a DataRobot project.
Delete Prediction Explanations InitializationTool to delete an existing prediction explanations initialization for a model.
Delete ProjectTool to delete a DataRobot project.
Delete Project ModelTool to delete a model from a DataRobot project's leaderboard.
Delete Project Model JobTool to cancel a modeling job for a DataRobot project.
Delete Feature ListTool to delete a specified featurelist from a DataRobot project.
Delete Projects Prediction DatasetsTool to delete a prediction dataset that was uploaded for a DataRobot project.
Delete QuotaTool to delete a quota by its ID.
Delete RecipeTool to delete a wrangling recipe.
Delete Registered ModelTool to archive a registered model in DataRobot.
Delete Secure ConfigurationTool to delete a secure configuration and its values.
Delete Spark SessionsTool to stop a DataRobot Spark wrangling session by size.
Delete StatusTool to delete a task status object.
Delete Use CaseTool to delete a DataRobot use case.
Delete Use Case ReferenceTool to remove a related entity from a DataRobot use case.
Delete User BlueprintTool to delete a user-owned blueprint in DataRobot.
Delete User BlueprintsTool to delete one or more user blueprints by their IDs.
Delete User GroupTool to delete a user group by its ID.
Delete User NotificationTool to delete a user notification by ID.
Delete All User NotificationsTool to delete all user notifications in DataRobot.
Delete Value TrackerTool to delete a DataRobot value tracker.
Detect External Data Store UDFsTool to start the job that detects standard user-defined functions for an external data store.
Download Custom ModelTool to download the latest custom model version content from DataRobot.
Download Custom Model VersionTool to download custom model version content from DataRobot as a file archive.
Download FileTool to download file data from a DataRobot catalog item by streaming it.
Download Scoring CodeTool to download scoring code for a DataRobot deployment.
Evaluate EntitlementsTool to evaluate which entitlements are enabled for the authenticated user's DataRobot account.
Export Tenant UsageExport tenant resource usage data for billing and cost analysis.
Finalize Batch Predictions CSV UploadTool to finalize a multipart CSV upload for batch predictions.
Get Access RoleTool to retrieve details for a specific Access Role by ID.
Get Account InfoRetrieves account information for the currently authenticated user, including user ID (uid), email, name, tenant ID, and organization ID (orgId).
Get Accuracy Metrics ConfigTool to retrieve which accuracy metrics are displayed and their order for a deployment.
Get Accuracy Over TimeTool to retrieve baseline and accuracy metric values over time buckets for a deployment.
Get Batch JobTool to retrieve a DataRobot batch job by ID.
Get Batch Prediction Job DefinitionTool to retrieve a Batch Prediction job definition by ID.
Get Portable Batch Prediction Job DefinitionTool to retrieve a portable batch prediction job definition snippet.
Get Batch PredictionsTool to retrieve a Batch Prediction job by ID.
Get CalendarTool to retrieve information about a calendar by ID.
Get Catalog ItemTool to retrieve catalog item details by ID.
Get Change RequestTool to retrieve a DataRobot change request by ID.
Get Compliance Doc TemplateTool to retrieve a compliance documentation template by ID.
Get CredentialsTool to retrieve the credentials set by ID.
Get Credentials AssociationsTool to list credentials associated with a specific object.
Get Custom Application SourceTool to retrieve a custom application source by ID.
Get Custom JobTool to retrieve a custom job by ID.
Get Custom Job ItemTool to retrieve custom job file content by custom job ID and item ID.
Get Custom ModelTool to retrieve a DataRobot custom model by ID.
Get Custom Model VersionTool to retrieve a specific custom model version in DataRobot.
Get Custom TemplateTool to retrieve a single custom template by ID from DataRobot.
Get Custom Templates FilesTool to retrieve a single custom template file by its ID.
Get Data Engine Workspace StateTool to retrieve a data engine workspace state by ID.
Get DatasetTool to retrieve a dataset by ID from DataRobot's catalog.
Get Dataset DefinitionTool to retrieve a dataset definition by ID.
Get Dataset FeaturelistTool to retrieve a specific featurelist from a dataset.
Get Dataset FileTool to download the original dataset file from DataRobot.
Get Dataset Feature HistogramTool to retrieve histogram data for a specific feature in a dataset.
Get Dataset Feature TransformTool to retrieve a feature transform for a specific dataset feature.
Get Dataset Refresh JobTool to retrieve a scheduled dataset refresh job by ID.
Get Dataset VersionTool to retrieve detailed information about a specific dataset version.
Get Dataset Feature HistogramTool to retrieve dataset feature histogram from DataRobot.
Get Dataset Version FeaturelistTool to retrieve a specific featurelist from a dataset version.
Get Data SliceTool to retrieve a Data Slice by ID.
Get Datetime PartitioningTool to retrieve datetime partitioning configuration for a project.
Get DeploymentTool to retrieve a deployment by ID.
Get Deployment AccuracyTool to retrieve accuracy metrics for a deployment over a time period.
Get Deployment CapabilitiesTool to retrieve the capabilities for a deployment.
Get Deployment Champion Model PackageTool to retrieve the champion model package for a deployment.
Get Deployment FeaturesTool to retrieve features in the universe dataset associated with a deployment.
Get Deployment Actuals Data ExportTool to retrieve a single actuals data export for a deployment.
Get Deployment Custom MetricTool to retrieve a single custom metric metadata for a deployment.
Get Deployment Settings ChecklistTool to return a checklist of deployment settings and their configuration state.
Get Deployment Monitoring BatchTool to retrieve a monitoring batch in a deployment.
Get Deployment Prediction Data ExportTool to retrieve a single prediction data export for a deployment.
Get Deployment Training Data ExportTool to retrieve a single training data export for a deployment.
Get Entity Notification Channel By IDTool to retrieve an entity notification channel by ID for a specific deployment or custom job.
Get Entity Notification ChannelsTool to list notification channels related to a specific entity.
Get Entity Notification PoliciesTool to list entity notification policies for deployments or custom jobs.
Get Entity Notification Policy by IDTool to retrieve an entity notification policy by ID.
Get Entity Notification Policy TemplatesTool to list entity notification policy templates for a specific entity type.
Get Entity Notification Policy Template by IDTool to retrieve a specific entity notification policy template by ID.
Get Execution EnvironmentTool to retrieve details about a specific execution environment by its ID.
Get Execution Environment VersionTool to retrieve a specific execution environment version by ID.
Get External Data DriverTool to retrieve external data driver details by driver ID.
Get External Data SourceTool to retrieve external data source details by ID.
Get External Data StoreTool to retrieve external data store details by ID.
Get External Driver ConfigurationsTool to retrieve external driver configuration details by ID.
Get External OAuth ProviderTool to retrieve an external OAuth provider by ID.
Get Feature ListTool to retrieve a specific feature list by ID.
Get Frozen ModelTool to retrieve a frozen model from a DataRobot project.
Get GenAI ChatTool to retrieve a GenAI chat by ID.
Get GenAI Chat PromptTool to retrieve a GenAI chat prompt by ID.
Get GenAI Comparison ChatTool to retrieve a GenAI comparison chat by ID.
Get GenAI Custom Model LLM ValidationsTool to retrieve a GenAI custom model LLM validation by ID.
Get GenAI LLMTool to retrieve details for a specific GenAI LLM by its ID.
Get GenAI LLM Test SuiteTool to retrieve a GenAI LLM test suite by ID.
Get GenAI OOTB Metric ConfigurationTool to retrieve a GenAI OOTB metric configuration by ID.
Get GenAI PlaygroundTool to retrieve a DataRobot GenAI playground by ID.
Get Guard ConfigurationTool to retrieve a DataRobot guard configuration by ID.
Get Guard TemplateTool to retrieve information about a guard template by ID.
Get Data Quality Report for Dataset Version FeatureTool to retrieve data quality report for a feature of a dataset version.
Get Data Quality Summary for DatasetTool to retrieve data quality summary for a dataset.
Get Data Quality Summary for Dataset VersionTool to retrieve data quality summary for a dataset version.
Get Model Feature Effects InsightsTool to retrieve feature effects insights for a DataRobot model.
Get Feature Impact InsightsTool to retrieve feature impact insights for a DataRobot model.
Get Lift Chart InsightsTool to retrieve Lift chart insights for a DataRobot model.
Get ROC Curve Insights for ModelTool to retrieve paginated ROC curve insights for a specific model.
Get SHAP Distributions InsightsTool to retrieve SHAP (SHapley Additive exPlanations) distribution insights for a DataRobot model.
Get SHAP Impact Insights for ModelTool to retrieve paginated SHAP Impact insights for a specific model.
Get SHAP Preview Insights for ModelTool to retrieve SHAP Preview insights for a DataRobot model.
Get MLOps Compute BundleTool to retrieve a specific MLOps compute bundle by ID.
Get Model Number of Iterations TrainedTool to retrieve the number of iterations trained for a DataRobot model.
Get Model PackageTool to retrieve a model package by ID.
Get Model Word CloudTool to retrieve word cloud data for a DataRobot text-based model.
Get NotebookTool to retrieve a specific DataRobot notebook by ID.
Get Notebook Code SnippetTool to retrieve a notebook code snippet by ID.
Get Notebook Environment VariablesTool to retrieve notebook environment variables by ID.
Get Notebook Execution EnvironmentTool to retrieve a notebook execution environment by ID.
Get Notebook Execution StatusTool to retrieve the execution status of a DataRobot notebook.
Get Notebook JobTool to retrieve a DataRobot notebook job by ID.
Get Notebook Revision by IDTool to retrieve a specific notebook revision by notebook ID and revision ID.
Get Notification Channel TemplateTool to retrieve a specific notification channel template by ID.
Get Notification Webhook Channel TestsTool to retrieve the status of a notification webhook channel test.
Get OpenTelemetry Metric ConfigurationTool to retrieve an OpenTelemetry metric configuration for a specific entity.
Get OTel Metrics Values Over Time SegmentsTool to get OpenTelemetry metric values grouped by segments attribute.
Get OTel Metrics Values by SegmentsTool to retrieve OpenTelemetry metric values grouped by a segment attribute over a time period.
Get OpenTelemetry TracesTool to retrieve OpenTelemetry traces for monitoring and debugging.
Get ProjectTool to retrieve a DataRobot project by ID.
Get Project Access ControlTool to list users with their roles on a project.
Get Project BlueprintTool to retrieve a blueprint by its ID.
Get Project Blueprint JSONTool to retrieve the JSON representation of a DataRobot blueprint.
Get Project JobTool to retrieve details for an in-progress project job.
Get Project Multicategorical Invalid Format FileTool to retrieve a file with format errors of potential multicategorical features for a DataRobot project.
Get Batch Type Transform Features ResultTool to retrieve the result of a batch variable type transformation.
Get Projects Duplicate ImagesTool to get a list of duplicate images containing the number of occurrences of each image.
Get Project Feature HistogramsTool to retrieve feature histogram data for a specific feature in a DataRobot project.
Get Project FeatureTool to retrieve detailed information about a specific feature in a DataRobot project.
Get Modeling FeaturelistTool to retrieve a single modeling featurelist by ID.
Get Project Modeling FeatureTool to retrieve detailed information about a specific modeling feature in a DataRobot project.
Get Project ModelTool to retrieve a model from a DataRobot project.
Get Lift ChartTool to retrieve lift chart data from a single source for a project model.
Get Model ROC CurveTool to retrieve ROC curve data for a specific model from a single data source.
Get Projects Prediction DatasetsTool to get metadata of a specific prediction dataset in a DataRobot project.
Get Project Rating Table ModelTool to retrieve a rating table model from a DataRobot project.
Get Project Rating TableTool to retrieve rating table information from a DataRobot project.
Get Project Training PredictionsTool to retrieve training predictions for a project.
Get QuotasTool to retrieve a specific quota by ID.
Get Quota TemplateTool to retrieve a specific quota template by ID.
Get Recipe OperationTool to retrieve details of a specific operation from a wrangling recipe.
Get RecipeTool to retrieve a DataRobot wrangling recipe by ID.
Get Recommended SettingsTool to retrieve configured recommended settings for an entity type.
Get Registered ModelTool to retrieve a registered model by ID.
Get Registered Model VersionTool to retrieve a specific version of a registered model.
Get Secure ConfigTool to retrieve a secure configuration by ID.
Get Secure Config SchemaTool to retrieve a secure configuration schema by ID.
Get Tenant Active UsersRetrieve active users in a tenant over a date range.
Get Use CaseTool to retrieve a DataRobot use case by ID.
Get Use Cases DatasetsTool to get dataset details in the scope of a use case.
Get User GroupTool to retrieve a user group by its ID.
Get Value TrackerTool to retrieve a value tracker by ID.
Get Value Tracker Value TemplatesTool to retrieve a value tracker value template by its type.
Initialize Model Compliance DocumentationTool to initialize compliance documentation pre-processing for a model or model package.
Invite UsersTool to invite multiple users by email to join the DataRobot platform.
Link Entity to Use CaseTool to link a single entity to a DataRobot use case.
Link Notebooks to Use Case (Bulk)Tool to bulk link multiple DataRobot notebooks to a use case.
List Access RolesTool to list access roles.
List Access Role UsersTool to list users assigned to an Access Role.
List ApplicationsTool to list DataRobot applications created by the authenticated user.
List Application TemplatesTool to list application templates the user has access to.
List Application Templates MediaTool to retrieve an application template image from DataRobot.
List Approval Policy MatchTool to find approval policy ID matching the query.
List Approval Policy TriggersTool to get a list of available approval policy triggers.
List Automated Document OptionsTool to list all available automated document types and locales.
List Automated DocumentsTool to list all automated documents in DataRobot.
List Batch JobsTool to list DataRobot batch jobs.
List Batch Monitoring Job DefinitionsTool to list DataRobot batch monitoring job definitions.
List Batch Prediction Job DefinitionsTool to list batch prediction job definitions.
List Batch PredictionsTool to list DataRobot batch prediction jobs.
List Calendar Country CodesTool to retrieve the list of allowed country codes for preloaded calendar generation.
List CalendarsTool to list all available calendars for a user in DataRobot.
List Calendars Access ControlTool to list users with their roles on a calendar.
List Catalog ItemsTool to list all catalog items accessible by the user.
List Change RequestsTool to list change requests in DataRobot.
List Code SnippetsTool to retrieve available code snippets from DataRobot.
List Compliance Doc TemplatesTool to list compliance documentation templates in DataRobot.
List Compliance Doc Templates DefaultTool to retrieve the default compliance documentation template from DataRobot.
List CredentialsTool to list all available credentials.
List Credentials AssociationsTool to list all objects associated with specific credentials.
List Custom ApplicationsTool to list custom applications created by the authenticated user.
List Custom Application SourcesTool to list custom application sources created by the authenticated user.
List Custom Application Sources VersionsTool to list custom application source versions of a specified application source.
List Custom Job LimitsTool to retrieve custom job limits from DataRobot.
List Custom JobsTool to list custom jobs in DataRobot.
List Custom Jobs Custom MetricsTool to list all custom metrics associated with a custom job in DataRobot.
List Custom Job RunsTool to list custom job runs for a specific custom job in DataRobot.
List Custom Jobs Shared RolesTool to get the access control list for a custom job.
List Custom Model DeploymentsTool to list custom model deployments in DataRobot.
List Custom Model Deployments LogsTool to retrieve custom model deployment logs.
List Custom Model LimitsTool to get custom model resource limits in DataRobot.
List Custom ModelsTool to list custom models in DataRobot.
List Custom Models Access ControlTool to list users with their roles on a custom model.
List Custom Models VersionsTool to list custom model versions in DataRobot.
List Custom Model TestsTool to list custom model tests for a specific custom model in DataRobot.
List Custom TemplatesTool to retrieve a list of custom templates from DataRobot.
List Custom Training BlueprintsTool to list custom training blueprints in DataRobot.
List Dataset DefinitionsTool to list all dataset definitions for the user.
List Dataset Definition VersionsTool to list all dataset definition versions for a given dataset definition.
List Dataset PermissionsTool to retrieve permissions for a specific dataset in DataRobot.
List DatasetsTool to list all datasets in the DataRobot global catalog.
List Datasets Access ControlTool to list users with their roles on a dataset.
List Dataset All Features DetailsTool to retrieve detailed information about all features in a DataRobot dataset.
List Dataset FeaturelistsTool to retrieve featurelists associated with a specific dataset in DataRobot.
List Dataset Feature TransformsTool to list all feature transforms applied to a dataset.
List Dataset ProjectsTool to list all projects associated with a specific dataset.
List Dataset Refresh JobsTool to list scheduled refresh jobs for a specific dataset.
List Dataset Refresh Job Execution ResultsTool to list execution results of a dataset refresh job.
List Dataset RelationshipsTool to list related datasets for a specific dataset.
List Datasets Shared RolesTool to list shared roles for a dataset.
List Dataset VersionsTool to list all versions of a specific dataset in DataRobot.
List Dataset Version All Features DetailsTool to retrieve detailed information about all features in a specific DataRobot dataset version.
List Dataset Version Feature ListsTool to retrieve feature lists for a specific dataset version in DataRobot.
Download Dataset Version FileTool to download original dataset data from a specific dataset version.
List Dataset Version ProjectsTool to list all projects that use a specific dataset version.
List Data Slices Slice SizesTool to retrieve the number of rows available after applying a data slice to a specified dataset subset.
List Data SourcesTool to list all available data sources.
List Deleted Custom JobsTool to list all deleted custom jobs in DataRobot.
List Deployment ChallengersTool to list challenger models for a deployment.
List Deployment Runtime ParametersTool to list runtime parameters for a deployment.
List DeploymentsTool to list deployments a user can view.
List Deployments Accuracy Over BatchTool to retrieve accuracy metrics over batches for a deployment.
List Deployments Actuals Data ExportsTool to retrieve a list of asynchronous actuals data exports for a deployment.
List Deployments Batch Service StatsTool to retrieve service health metrics for a deployment's batch predictions.
List Deployments Challenger Replay SettingsTool to retrieve challenger replay settings for a deployment.
List Deployment Custom MetricsTool to retrieve a list of custom metrics for a deployment.
List Deployments Custom Metrics Batch SummaryTool to retrieve the summary of deployment batch custom metric.
List Deployment Custom Metrics SummaryTool to retrieve the summary of a deployment custom metric.
List Deployments Custom Metrics SummaryTool to retrieve the bulk summary of deployment custom metrics.
List Deployment Custom Metrics Values Over BatchTool to retrieve custom metric values over batch for a deployment.
List Deployment Custom Metrics Values Over TimeTool to retrieve custom metric values over time for a deployment.
List Deployment Segment AttributesTool to retrieve segment attributes for a deployment based on monitoring type.
List Deployments Feature DriftTool to retrieve feature drift scores for a deployment over a time period.
List Deployments Feature Drift Over BatchTool to retrieve drift over batch info for features of a deployment.
List Deployment Health SettingsTool to retrieve deployment health settings.
List Deployments Health Settings DefaultsTool to retrieve default deployment health settings for a deployment.
List Deployments Humility StatsTool to retrieve humility stats for a deployment.
List Deployments Humility Stats Over TimeTool to retrieve humility statistics over time for a deployment.
List Deployments Model HistoryTool to retrieve champion model history for a deployment.
List Deployment Monitoring BatchesTool to list monitoring batches for a deployment in DataRobot.
List Deployment Monitoring Batch ModelsTool to list information about models that have data in a monitoring batch.
Get Deployment Monitoring Batch LimitsTool to retrieve the limits related to monitoring batches for a deployment.
List Prediction Data ExportsTool to list prediction data exports for a deployment.
List Deployments Prediction ResultsTool to retrieve predictions results for a deployment.
List Deployment Predictions Over BatchTool to retrieve prediction metadata over batches for a deployment.
List Deployment Predictions Over TimeTool to retrieve metrics about predictions over time for a deployment.
List Predictions vs Actuals Over BatchTool to retrieve metrics about predictions and actuals over a specific set of batches.
List Deployments Predictions Vs Actuals Over TimeTool to retrieve predictions vs actuals over time for a deployment.
List Deployment Quota ConsumersTool to retrieve deployment quota consumers.
List Deployment Retraining PoliciesTool to list retraining policies for a deployment.
List Deployment Retraining SettingsTool to fetch deployment retraining settings.
List Deployments Segment ValuesTool to retrieve deployment segment values for monitoring and analytics.
List Deployments Service StatsTool to retrieve service health metrics for a deployment.
List Deployment Service Stats Over BatchTool to retrieve service health metrics over batch for a deployment.
List Deployments Service Stats Over TimeTool to retrieve service health metrics over time for a deployment.
List Deployment SettingsTool to retrieve deployment settings.
List Deployments Shared RolesTool to get a model deployment's access control list.
List Deployments Target DriftTool to retrieve target drift for a deployment over a specified time period.
List Training Data ExportsTool to list training data exports for a deployment.
List Deployments Model Secondary Dataset Configuration HistoryTool to list the secondary datasets configuration history for a deployment.
List Entitlement Set LeasesTool to retrieve entitlement set leases from DataRobot.
List Entity Notification Policy Templates Related PoliciesTool to retrieve all policies created from a notification policy template.
List Entity Notification Policy Templates Shared RolesTool to list shared roles for an entity notification policy template.
List Entity TagsTool to retrieve a list of entity tags from DataRobot.
List Event LogsTool to retrieve audit log records from DataRobot.
List Event Logs Prediction UsageTool to retrieve prediction usage event logs from DataRobot.
List Execution EnvironmentsTool to list execution environments in DataRobot.
List Execution Environments Versions Build LogTool to download execution environment build log.
Download Execution Environment VersionTool to download execution environment version files from DataRobot.
List Execution Environment VersionsTool to list all versions of an execution environment in DataRobot.
List External Data Store Standard User-Defined FunctionsTool to retrieve detected standard user-defined functions for a given external data store.
List External Data Driver ConfigurationTool to retrieve external data driver configuration details by driver ID.
List External Data DriversTool to list all available external data drivers in DataRobot.
List External Data Sources Access ControlTool to list users with their roles on an external data source.
List External Data Sources PermissionsTool to list permissions for the current user on an external data source.
List External Data Sources Shared RolesTool to get an external data source's access control list.
List External Data StoresTool to list external data stores in DataRobot.
List External Data Store CredentialsTool to list credentials associated with a specified external data store.
List External Data Stores PermissionsTool to retrieve permissions for an external data store.
List External Data Stores Shared RolesTool to list access control entries (shared roles) for an external data store.
List External Driver ConfigurationsTool to list available external driver configurations in DataRobot.
List External OAuth ProvidersTool to list external OAuth providers configured in DataRobot.
List Feature Association FeaturelistsTool to list all featurelists with feature association matrix availability flags for a project.
List Feature ListsTool to list all feature lists for a project.
List All Files for Catalog ItemTool to list all files associated with a catalog item in DataRobot.
List GenAI Chat PromptsTool to list GenAI chat prompts in DataRobot.
List GenAI ChatsTool to list GenAI chats available to the user.
List GenAI Comparison PromptsTool to list GenAI comparison prompts filtered by comparison chat ID or LLM blueprint IDs.
List GenAI Custom Model Embedding ValidationsTool to list GenAI custom model embedding validations.
List GenAI Custom Model Vector Database ValidationsTool to list custom model vector database validations for GenAI use cases.
List GenAI Evaluation Dataset ConfigurationsTool to list GenAI evaluation dataset configurations.
List GenAI LLM BlueprintsTool to list LLM blueprints for building generative AI applications with various large language models.
List GenAI LLMsTool to list all available GenAI LLMs in DataRobot.
List GenAI LLM Test Configurations Supported InsightsTool to list supported insights for LLM test configurations in DataRobot.
List GenAI LLM Test ConfigurationsTool to list GenAI LLM test configurations.
List GenAI LLM Test Configurations OOTB DatasetsTool to list out-of-the-box (OOTB) datasets for GenAI LLM test configurations.
List GenAI LLM Test ResultsTool to list GenAI LLM test results filtered by use case and playground.
List GenAI LLM Test SuitesTool to list GenAI LLM test suites in DataRobot.
List GenAI PlaygroundsTool to list all GenAI playgrounds accessible by the user.
List GenAI Playgrounds OOTB Metric ConfigurationsTool to list OOTB metric configurations for a GenAI playground.
List GenAI Playgrounds Supported InsightsTool to list supported insights for a GenAI playground.
List GenAI Playground TracesTool to list all prompt traces for a GenAI playground.
List GenAI Sidecar Model Metric ValidationsTool to list GenAI sidecar model metric validations from DataRobot.
List GenAI User Limits LLM API CallsTool to retrieve the count of LLM API calls made by the authenticated user.
List GenAI User Limits Vector DatabasesTool to retrieve the number of vector databases the user has created which count towards the usage limit.
List GenAI Vector DatabasesTool to list all GenAI vector databases used for RAG (Retrieval Augmented Generation) applications.
List GenAI Vector Databases Supported Retrieval SettingsTool to list all supported retrieval settings for GenAI vector databases.
List GenAI Vector Databases Supported Text ChunkingsTool to list all supported text chunking configurations for GenAI vector databases.
List Guard Configurations Prediction Environments In UseTool to show prediction environments in use for moderation by a specific custom model version.
List Guard ConfigurationsTool to list guard configurations for a specific entity in DataRobot.
List Guard TemplatesTool to list guard templates in DataRobot.
List Image Augmentation ListsTool to list image augmentation lists for a DataRobot project.
List MLOps Compute BundlesTool to list resource bundles for MLOps compute.
List Model FeaturesTool to retrieve the list of features used in a specific DataRobot model.
List Model Package FeaturesTool to retrieve features in a model package.
List Model PackagesTool to list model packages.
List Model Package CapabilitiesTool to retrieve capabilities of a model package.
List Model Package Model LogsTool to list model logs for a specific model package.
List Model Packages Shared RolesTool to get model package's access control list.
List Model RecordsRetrieve a paginated list of trained model records from a DataRobot project.
List Model Supported CapabilitiesTool to retrieve supported capabilities for a model.
List Moderation Supported LLMsTool to list supported LLMs for moderation in DataRobot.
List Multilabel Insights Pairwise Manual SelectionsTool to retrieve all manually selected label lists for pairwise multilabel insights analysis.
List Notebook Code SnippetsTool to retrieve all available notebook code snippets from DataRobot.
List Notebook Code Snippets TagsTool to retrieve available tags for notebook code snippets from DataRobot.
List Notebook Execution Environment NotebooksTool to list notebooks that use a specific execution environment in DataRobot.
List Notebook Execution EnvironmentsTool to list all notebook execution environments in DataRobot.
List Notebook Execution Environments MachinesTool to list available machine types for notebook execution environments.
List Notebook Execution Environment PortsTool to list exposed ports for a notebook execution environment.
List Notebook Execution Environment VersionsTool to list all versions of a notebook execution environment.
List Notebook Jobs Run HistoryTool to list notebook job run history in DataRobot.
List NotebooksTool to list Jupyter notebooks in DataRobot workspace.
List Notebook CellsTool to retrieve all cells from a DataRobot notebook.
List Notebook Sessions TerminalsTool to list all terminals in a DataRobot notebook session.
List Notebooks Filter OptionsTool to retrieve available filter options for notebooks, including tags and owners.
List Notebooks Shared RolesTool to get access control lists for multiple notebooks.
List Notification Channel TemplatesTool to list notification channel templates in DataRobot.
List Notification Channel Templates Shared RolesTool to get channel template's access control list.
List Notification EventsTool to list notification event types and groups available for notification policies.
List Notification Channel Templates Policy TemplatesTool to retrieve list of all policy templates using a specific notification channel template.
List Notification Channel Templates Related PoliciesTool to retrieve all policies created from a notification channel template.
List OCR Job ResourcesTool to retrieve user's OCR job resources from DataRobot.
List Organization UsersTool to list memberships (users) in an organization.
List OpenTelemetry LogsTool to retrieve OpenTelemetry logs for a specified entity.
List OpenTelemetry Metrics Autocollected ValuesTool to get aggregated values of OpenTelemetry metrics that DataRobot automatically collects for a specified entity.
List OpenTelemetry Metrics ConfigsTool to list OpenTelemetry metric configurations for a specified entity.
List OpenTelemetry Metrics Pod InfoTool to list pods and containers found in OpenTelemetry metrics of the specified entity.
List OpenTelemetry Metrics SummaryTool to list reported OpenTelemetry metrics of the specified entity.
List OpenTelemetry Metrics Value Over TimeTool to get a single OpenTelemetry metric value of the specified entity over time.
List OpenTelemetry Metrics ValuesTool to get OpenTelemetry metrics values for a specific entity over a single time period.
List OTel Metrics Values Over TimeTool to retrieve OpenTelemetry configured metrics values for a specific entity over time.
List OpenTelemetry TracesTool to list OpenTelemetry traces for a specified entity (deployment, use case, etc.
List Overall Moderation ConfigurationTool to get overall moderation configuration for an entity.
List Pinned Use CasesTool to list all pinned use cases in DataRobot.
List Prediction EnvironmentsLists all available prediction environments.
List Prediction ServersTool to list prediction servers available to the user.
List Project ImagesTool to retrieve image metadata for a DataRobot project.
List Project JobsTool to list all jobs for a given DataRobot project.
List Project ModelsTool to list models for a DataRobot project.
List ProjectsTool to list all available DataRobot projects.
List Bias Mitigated Models for ProjectTool to list bias-mitigated models for a DataRobot project.
List Projects Bias Mitigation Feature InfoTool to get bias mitigation data quality information for a given project and feature.
List Projects Bias vs Accuracy InsightsTool to list bias vs accuracy insights for a DataRobot project.
List Project Blender ModelsTool to list all blender models in a DataRobot project.
List Project BlueprintsTool to list all blueprints available in a DataRobot project.
Get Blueprint ChartTool to retrieve a blueprint chart by blueprint ID.
List Projects Blueprints Blueprint DocsTool to retrieve blueprint tasks documentation.
List Projects Combined ModelsTool to retrieve all existing combined models for a DataRobot project.
List Projects Data SlicesTool to list paginated data slices for a specific DataRobot project.
List Projects Datetime ModelsTool to list datetime partitioned models in a DataRobot project.
List Projects Document Text Extraction SamplesTool to list metadata on all computed document text extraction samples in a DataRobot project across all models.
List Project Document ThumbnailsTool to list document thumbnail metadata for a DataRobot project.
List Projects Datetime Models Accuracy Over Time PlotsTool to retrieve metadata for Accuracy over Time insights for datetime models.
List Projects External ScoresTool to list external scores on prediction datasets for a DataRobot project.
List Projects Feature Association MatrixTool to retrieve pairwise feature association statistics for a DataRobot project.
List Projects Feature Association Matrix DetailsTool to retrieve feature association matrix details between two features in a DataRobot project.
List Project FeaturesTool to list all features in a DataRobot project.
List Project Feature Frequent ValuesTool to retrieve frequent values information for a feature in a DataRobot project.
List Project Features MetricsTool to retrieve available metrics for a specific feature in a DataRobot project.
List Feature Multiseries PropertiesTool to retrieve potential multiseries ID columns to use with a particular datetime partition column.
List Projects Frozen ModelsTool to list all frozen models from a DataRobot project.
List Project Image Activation MapsTool to list all image activation maps for a DataRobot project.
List Project Image EmbeddingsTool to list all image embeddings for a DataRobot project.
List Project Modeling FeaturelistsTool to list all modeling featurelists from a DataRobot project.
List Projects Modeling FeaturesTool to list all modeling features for a DataRobot project.
List Projects Model JobsTool to list modeling jobs for a given DataRobot project.
List Projects Models Advanced Tuning ParametersTool to retrieve information about all advanced tuning parameters available for a specified model.
Get Model Blueprint ChartTool to retrieve a reduced model blueprint chart by model ID.
List Projects Models Blueprint DocsTool to retrieve task documentation for a reduced model blueprint.
List Projects Models Cross Class Accuracy ScoresTool to list cross-class accuracy scores for a specific model in a project.
List Projects Models Cross Validation ScoresTool to retrieve cross-validation scores for each partition in a DataRobot model.
List Projects Models Data Disparity InsightsTool to retrieve Cross Class Data Disparity insights for a DataRobot model.
List Lift Charts for Model DatasetsTool to retrieve list of lift chart data on prediction datasets for a project model.
List Feature EffectsTool to retrieve Feature Effects for a DataRobot model.
List Model Feature Effects MetadataTool to retrieve Feature Effects metadata for a model.
List Model Feature ImpactTool to retrieve feature impact scores for features in a DataRobot model.
List Grid Search ScoresTool to retrieve grid search scores for a specific model in a DataRobot project.
List Model Lift ChartsTool to retrieve all available lift charts for a DataRobot model.
Get Model Missing Values ReportTool to retrieve a summary of how a model's subtasks handle missing values.
List Prime RulesetsTool to list DataRobot Prime rulesets that approximate a specific model.
List Model ROC CurvesTool to retrieve all available ROC curves for a binary classification model.
Retrieve Model Scoring CodeTool to retrieve scoring code JAR file for a specific DataRobot model.
List Projects Multicategorical Invalid FormatTool to retrieve multicategorical data quality log for a DataRobot project.
List Projects Multiseries NamesTool to list the names of a multiseries project.
List Optimized Datetime PartitioningsTool to list all created optimized datetime partitioning configurations for a project.
List Projects Payoff MatricesTool to list all payoff matrices for a DataRobot project.
List Project Prediction DatasetsTool to list prediction datasets uploaded to a DataRobot project.
List Project Prediction Explanations RecordsTool to list prediction explanations records for a DataRobot project.
List Project PredictionsTool to list prediction records for a DataRobot project.
List Project Predict JobsTool to list all prediction jobs for a given DataRobot project.
List Projects Prime FilesTool to list Prime files available in a DataRobot project.
List Prime ModelsTool to list all Prime models in a DataRobot project.
List Projects Rating Table ModelsTool to list rating table models for a DataRobot project.
List Projects Rating TablesTool to list rating tables for a DataRobot project.
Retrieve Rating Table FileTool to retrieve a rating table file from a DataRobot project.
List Project Recommended ModelsTool to list recommended models for a DataRobot project.
List Project RuleFit FilesTool to list RuleFit code files for a DataRobot project.
List Project Secondary Datasets ConfigurationsTool to list all secondary dataset configurations for a DataRobot project.
List Projects Training PredictionsTool to list training prediction jobs for a specific DataRobot project.
List QuotasTool to list all quotas configured in DataRobot.
List Quota TemplatesTool to list quota templates in DataRobot.
List Recipe InputsTool to get inputs on a recipe.
List Recipe InsightsTool to retrieve recipe insights for a specific recipe.
List Recipes PreviewTool to retrieve a wrangling recipe preview.
List Recommended Settings ChoicesTool to retrieve available setting choices list for an entity type.
List Registered ModelsTool to list registered models from DataRobot.
List Registered Model DeploymentsTool to list deployments associated with a registered model.
List Registered Models Shared RolesTool to get a registered model's access control list.
List Registered Model Version DeploymentsTool to list all deployments associated with a registered model version.
List Registered Model VersionsTool to list versions of a registered model.
List Scheduled JobsTool to list scheduled deployment batch prediction jobs a user can view.
List Seat License AllocationsTool to list seat license allocations.
List Secure ConfigurationsTool to retrieve a list of secure configurations in DataRobot.
List Secure Config SchemasTool to retrieve a list of secure configuration schemas.
List Secure Config ValuesTool to retrieve a list of values for a secure configuration.
List Status JobsTool to list currently running async status jobs in DataRobot.
List Tenants Resource CategoriesTool to retrieve available resource categories for a specific tenant.
List Tenant Usage Resources CategoriesTool to get available resource categories for tenant usage.
List Usage Data Exports Supported EventsTool to list supported audit events for usage data export filtering.
List Use Case Custom ApplicationsTool to list custom applications referenced by a DataRobot use case.
List Use Case DatasetsTool to get a list of datasets associated with a DataRobot Use Case.
List Use Case DeploymentsTool to get deployments associated with a use case.
List Use Case Notebooks By IDTool to get a list of notebooks associated with a specific Use Case by ID.
List Use Case Registered ModelsTool to get registered models associated with a use case.
List Use CasesTool to retrieve a list of Use Cases from DataRobot.
List Use Cases All ResourcesTool to get a list of references associated with all Use Cases.
List Use Cases ApplicationsTool to list applications associated with a DataRobot Use Case.
List Use Case DataTool to retrieve a list of datasets and recipes from a specific DataRobot Use Case.
List Use Cases FilesTool to list catalog files associated with a specific Use Case.
Get Use Case Filter MetadataTool to retrieve filtering metadata for a DataRobot Use Case.
List Use Case Models for ComparisonTool to get models from projects in a Use Case for comparison.
List Use Cases NotebooksTool to get a list of notebooks from all Use Cases.
List Use Cases PlaygroundsTool to list playgrounds associated with a Use Case.
List Use Case ProjectsTool to get a list of projects associated with a use case.
List Use Case ResourcesTool to get a list of the references associated with a DataRobot use case.
List Use Cases Shared RolesTool to get a use case's access control list.
List Use Case Vector DatabasesTool to retrieve a list of vector databases associated with a DataRobot Use Case.
List Use Cases With Shortened InfoTool to retrieve a list of Use Cases with abbreviated content from DataRobot.
List User BlueprintsTool to list user blueprints from DataRobot.
List User Blueprints Input TypesTool to retrieve available input types for user blueprints.
List User Blueprints TasksTool to retrieve tasks for blueprint construction in DataRobot.
List User GroupsTool to list user groups.
List User NotificationsTool to list user notifications in DataRobot.
List Users in GroupLists users in a specific DataRobot user group.
List Value TrackersTool to list Value Trackers that the requesting user has access to in DataRobot.
List Value Trackers ActivitiesTool to retrieve the activities of a value tracker.
List Value Tracker AttachmentsTool to list resources attached to a DataRobot value tracker.
List Value Trackers Shared RolesTool to get a value tracker's access control list.
List Value Tracker Value TemplatesTool to list available value tracker templates in DataRobot.
List Value Tracker Value Templates CalculationTool to calculate value of template with given template parameters.
List VCS Git ProvidersTool to list all VCS Git providers configured in DataRobot.
List VersionTool to retrieve DataRobot API version information.
List Wrangling RecipesTool to list all available wrangling recipes.
Move Notebook Filesystem ObjectTool to move a file or directory within a DataRobot notebook session filesystem.
Pause or Unpause AutopilotTool to pause or unpause Autopilot for a project.
Restore Notebook RevisionTool to restore a DataRobot notebook to a specific revision.
Start DataRobot AutopilotTool to start the data modeling (Autopilot) process for a DataRobot project.
Start OCR Job ResourceStart an OCR job resource for optical character recognition processing.
Stop Notebook RuntimeTool to stop a running DataRobot notebook runtime.
Test External Data Store ConnectionTool to test an external data store connection in DataRobot.
Update Access RoleTool to update a custom Access Role.
Update Accuracy Metrics ConfigTool to update which accuracy metrics are returned by the accuracy endpoint for a deployment.
Upload CSV Data to Batch JobTool to stream CSV data to a DataRobot batch job.
Update Batch Monitoring Job DefinitionTool to update a Batch Monitoring job definition.
Update Batch Prediction Job DefinitionTool to update an existing Batch Prediction job definition.
Update Batch PredictionsTool to update a Batch Prediction job in DataRobot.
Upload Batch Prediction CSV PartTool to upload CSV data in multiple parts for batch predictions.
Update CalendarTool to update a calendar's name in DataRobot.
Update Calendars Access ControlTool to update access control for a calendar.
Update Catalog ItemTool to update a catalog item's name, description, or tags.
Update CommentTool to update an existing comment in DataRobot.
Update Compliance Doc TemplatesTool to update an existing model compliance documentation template in DataRobot.
Update Compliance Doc Template Shared RolesTool to update shared roles for a compliance document template.
Update CredentialsTool to update existing DataRobot credentials.
Update Credentials AssociationsTool to add or remove objects associated with credentials.
Update Credentials Associations by IDTool to set default credentials for a data connection or batch prediction job.
Update Custom Application SourceTool to update a custom application source's name.
Update Custom Application Source VersionTool to update a custom application source version with new label, base environment, or files.
Update Custom JobTool to update an existing DataRobot custom job.
Update Custom Job Shared RolesTool to update shared roles for a custom job.
Update Custom ModelTool to update a DataRobot custom model.
Update Custom Model Access ControlTool to update access control for a custom model.
Update Custom Model VersionTool to update custom model version files and configuration in DataRobot.
Update Custom Model VersionTool to update a custom model version in DataRobot.
Update Custom Models Versions With Training DataTool to add or replace training and holdout data for a custom model version.
Update Custom Task VersionTool to update a DataRobot custom task version by uploading new files and/or updating configuration.
Update Custom Task VersionTool to update a custom task version in DataRobot.
Update Dataset Definitions Chunk DefinitionsTool to update a chunk definition in a dataset definition.
Update Dataset FeaturelistTool to update a dataset featurelist's name or description.
Update Dataset RelationshipTool to update an existing dataset relationship in DataRobot.
Update Datasets (Bulk Action)Tool to execute bulk actions on multiple datasets.
Update Dataset Access ControlTool to update access control for a dataset.
Update Dataset by IDTool to update a dataset's name or categories in DataRobot's global catalog.
Recover Deleted DatasetTool to recover a deleted dataset in DataRobot.
Update Dataset Refresh JobTool to update a dataset refresh job configuration.
Update Dataset Shared RolesTool to modify dataset shared roles in DataRobot.
Recover Deleted Dataset VersionTool to recover a deleted dataset version in DataRobot.
Update Data Stage PartTool to upload a part file to a DataRobot data stage.
Update DeploymentTool to update an existing DataRobot deployment's metadata.
Update Challenger Replay SettingsTool to update challenger replay settings for a deployment.
Update Deployment Custom MetricTool to update settings for a deployment's custom metric.
Update Deployment SettingsTool to update deployment settings for a DataRobot deployment.
Update Deployment Health SettingsTool to update health settings for a DataRobot deployment.
Update Deployment ModelTool to replace a deployment's champion model with a different model or model package.
Update Deployment Monitoring BatchTool to update a monitoring batch in a deployment.
Update Deployments Retraining SettingsTool to update deployment retraining settings.
Update Deployment Shared RolesTool to modify deployment shared roles in DataRobot.
Update Deployment StatusTool to update deployment status.
Update Entity Notification ChannelTool to update an entity notification channel for a deployment or custom job.
Update Entity Notification PoliciesTool to update entity notification policy.
Update Entity Notification Policy TemplateTool to update an entity notification policy template in DataRobot.
Update Notification Policy Template Shared RolesTool to update entity notification policy template shared roles in DataRobot.
Update Entity TagTool to update an entity tag's name in DataRobot.
Update Execution EnvironmentTool to update a DataRobot execution environment.
Update External Data SourceTool to update an external data source's canonical name or configuration parameters.
Update External Data Sources Access ControlTool to update access control roles for an external data source.
Update External Data Source Shared RolesTool to modify external data source shared roles in DataRobot.
Update External Data StoreTool to update an external data store configuration.
Update External Data Store Access ControlTool to update access control settings for an external data store.
Update External Data Store Shared RolesTool to modify external data store shared roles in DataRobot.
Update External OAuth ProviderTool to update an external OAuth provider configuration.
Rename File or Folder in CatalogTool to rename a file or folder within a DataRobot catalog item.
Recover Deleted FileTool to recover a deleted file from DataRobot.
Update GenAI Comparison ChatTool to update a GenAI comparison chat name.
Update GenAI Custom Model LLM ValidationsTool to update a GenAI custom model LLM validation.
Update GenAI Custom Model Vector Database ValidationsTool to update a GenAI custom model vector database validation.
Update GenAI PlaygroundTool to update a GenAI playground.
Update User GroupTool to update a user group by its ID.
Update Guard ConfigurationTool to update a DataRobot guard configuration.
Update Image Augmentation ListsTool to update an existing image augmentation list in DataRobot.
Update Modeling FeaturelistTool to update a modeling featurelist's name or description.
Update NotebookTool to update a DataRobot notebook's name or description.
Update Notebook Execution EnvironmentTool to update a notebook execution environment configuration.
Update Notebook JobTool to update an existing notebook job in DataRobot.
Update Notebook Cell OutputTool to update the output of a specific cell in a DataRobot notebook.
Update Notebook StateTool to update notebook state and retrieve current cell information.
Update Notification Channel TemplateTool to update a notification channel template in DataRobot.
Update Notification Channel Template Shared RolesTool to update notification channel template shared roles in DataRobot.
Update OpenTelemetry Metrics ConfigurationsTool to set all OpenTelemetry metric configurations for a specified entity.
Update OTEL Metrics Config By IDTool to update an OpenTelemetry metric configuration for a specified entity.
Update Overall Moderation ConfigurationTool to update overall moderation configuration for a custom model, custom model version, or playground.
Update Pinned Use CasesTool to add or remove pinned use cases in DataRobot.
Update ProjectTool to update a DataRobot project's name, description, worker settings, or unlock holdout.
Update Project Access ControlTool to update access control settings for a DataRobot project.
Update Feature ListTool to update an existing feature list in a DataRobot project.
Update Projects ModelsTool to update a model's attributes in a DataRobot project.
Update Payoff MatrixTool to update a payoff matrix for a DataRobot project.
Update QuotasTool to update quota configuration for a DataRobot deployment resource.
Update Wrangling RecipeTool to update a DataRobot wrangling recipe.
Update Recipe DownsamplingTool to update the downsampling configuration in a DataRobot recipe.
Update Recipe InputsTool to update the input data sources for a DataRobot recipe.
Update Recipe OperationsTool to update the operations in a DataRobot recipe.
Update Recipe SettingsTool to update recipe settings that are reusable at the modeling stage.
Update Recommended SettingsTool to update recommended settings for a DataRobot entity (currently only deployments).
Update Registered ModelTool to update a registered model in DataRobot.
Update Registered Model Shared RolesTool to modify registered model shared roles in DataRobot.
Update Secure ConfigurationTool to update a secure configuration.
Update Secure Config Shared RolesTool to share a secure configuration with users, groups, or organizations.
Update Use CaseTool to update an existing DataRobot use case's metadata.
Move Entity Between Use CasesTool to move an entity (project, dataset, notebook, etc.
Update Use Case Shared RolesTool to update a use case's access control list.
Update User NotificationTool to mark a DataRobot user notification as read.
Update User NotificationsTool to mark all user notifications as read.
Update Value TrackerTool to update a DataRobot value tracker.
Update Value Tracker Shared RolesTool to update shared roles for a DataRobot value tracker.
Validate Deployment Model ReplacementTool to validate whether a model can replace the current champion model in a deployment.
Validate User BlueprintTool to validate a user-defined blueprint (custom modeling pipeline) in DataRobot.
Verify External Data Store SQLTool to verify a SQL query for an external data store.

What is the Composio tool router, and how does it fit here?

What is Composio SDK?

Composio's Composio SDK helps agents find the right tools for a task at runtime. You can plug in multiple toolkits (like Gmail, HubSpot, and GitHub), and the agent will identify the relevant app and action to complete multi-step workflows. This can reduce token usage and improve the reliability of tool calls. Read more here: Getting started with Composio SDK

The tool router generates a secure MCP URL that your agents can access to perform actions.

How the Composio SDK works

The Composio SDK follows a three-phase workflow:

  1. Discovery: Searches for tools matching your task and returns relevant toolkits with their details.
  2. Authentication: Checks for active connections. If missing, creates an auth config and returns a connection URL via Auth Link.
  3. Execution: Executes the action using the authenticated connection.

Step-by-step Guide

Prerequisites

Before you begin, make sure you have:
  • Python 3.8/Node 16 or higher installed
  • A Composio account with the API key
  • An OpenAI API key
  • A Datarobot account and project
  • Basic familiarity with async Python/Typescript

Getting API Keys for OpenAI, Composio, and Datarobot

OpenAI API key (OPENAI_API_KEY)
  • Go to the OpenAI dashboard
  • Create an API key if you don't have one
  • Assign it to OPENAI_API_KEY in .env
Composio API key and user ID
  • Log into the Composio dashboard
  • Copy your API key from Settings
    • Use this as COMPOSIO_API_KEY
  • Pick a stable user identifier (email or ID)
    • Use this as COMPOSIO_USER_ID

Installing dependencies

pip install composio-llamaindex llama-index llama-index-llms-openai llama-index-tools-mcp python-dotenv

Create a new Python project and install the necessary dependencies:

  • composio-llamaindex: Composio's LlamaIndex integration
  • llama-index: Core LlamaIndex framework
  • llama-index-llms-openai: OpenAI LLM integration
  • llama-index-tools-mcp: MCP client for LlamaIndex
  • python-dotenv: Environment variable management

Set environment variables

bash
OPENAI_API_KEY=your-openai-api-key
COMPOSIO_API_KEY=your-composio-api-key
COMPOSIO_USER_ID=your-user-id

Create a .env file in your project root:

These credentials will be used to:

  • Authenticate with OpenAI's GPT-5 model
  • Connect to Composio's Tool Router
  • Identify your Composio user session for Datarobot access

Import modules

import asyncio
import os
import dotenv

from composio import Composio
from composio_llamaindex import LlamaIndexProvider
from llama_index.core.agent.workflow import ReActAgent
from llama_index.core.workflow import Context
from llama_index.llms.openai import OpenAI
from llama_index.tools.mcp import BasicMCPClient, McpToolSpec

dotenv.load_dotenv()

Create a new file called datarobot_llamaindex_agent.py and import the required modules:

Key imports:

  • asyncio: For async/await support
  • Composio: Main client for Composio services
  • LlamaIndexProvider: Adapts Composio tools for LlamaIndex
  • ReActAgent: LlamaIndex's reasoning and action agent
  • BasicMCPClient: Connects to MCP endpoints
  • McpToolSpec: Converts MCP tools to LlamaIndex format

Load environment variables and initialize Composio

OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
COMPOSIO_API_KEY = os.getenv("COMPOSIO_API_KEY")
COMPOSIO_USER_ID = os.getenv("COMPOSIO_USER_ID")

if not OPENAI_API_KEY:
    raise ValueError("OPENAI_API_KEY is not set in the environment")
if not COMPOSIO_API_KEY:
    raise ValueError("COMPOSIO_API_KEY is not set in the environment")
if not COMPOSIO_USER_ID:
    raise ValueError("COMPOSIO_USER_ID is not set in the environment")

What's happening:

This ensures missing credentials cause early, clear errors before the agent attempts to initialise.

Create a Tool Router session and build the agent function

async def build_agent() -> ReActAgent:
    composio_client = Composio(
        api_key=COMPOSIO_API_KEY,
        provider=LlamaIndexProvider(),
    )

    session = composio_client.create(
        user_id=COMPOSIO_USER_ID,
        toolkits=["datarobot"],
    )

    mcp_url = session.mcp.url
    print(f"Composio MCP URL: {mcp_url}")

    mcp_client = BasicMCPClient(mcp_url, headers={"x-api-key": COMPOSIO_API_KEY})
    mcp_tool_spec = McpToolSpec(client=mcp_client)
    tools = await mcp_tool_spec.to_tool_list_async()

    llm = OpenAI(model="gpt-5")

    description = "An agent that uses Composio Tool Router MCP tools to perform Datarobot actions."
    system_prompt = """
    You are a helpful assistant connected to Composio Tool Router.
    Use the available tools to answer user queries and perform Datarobot actions.
    """
    return ReActAgent(tools=tools, llm=llm, description=description, system_prompt=system_prompt, verbose=True)

What's happening here:

  • We create a Composio client using your API key and configure it with the LlamaIndex provider
  • We then create a tool router MCP session for your user, specifying the toolkits we want to use (in this case, datarobot)
  • The session returns an MCP HTTP endpoint URL that acts as a gateway to all your configured tools
  • LlamaIndex will connect to this endpoint to dynamically discover and use the available Datarobot tools.
  • The MCP tools are mapped to LlamaIndex-compatible tools and plug them into the Agent.

Create an interactive chat loop

async def chat_loop(agent: ReActAgent) -> None:
    ctx = Context(agent)
    print("Type 'quit', 'exit', or Ctrl+C to stop.")

    while True:
        try:
            user_input = input("\nYou: ").strip()
        except (KeyboardInterrupt, EOFError):
            print("\nBye!")
            break

        if not user_input or user_input.lower() in {"quit", "exit"}:
            print("Bye!")
            break

        try:
            print("Agent: ", end="", flush=True)
            handler = agent.run(user_input, ctx=ctx)

            async for event in handler.stream_events():
                # Stream token-by-token from LLM responses
                if hasattr(event, "delta") and event.delta:
                    print(event.delta, end="", flush=True)
                # Show tool calls as they happen
                elif hasattr(event, "tool_name"):
                    print(f"\n[Using tool: {event.tool_name}]", flush=True)

            # Get final response
            response = await handler
            print()  # Newline after streaming
        except KeyboardInterrupt:
            print("\n[Interrupted]")
            continue
        except Exception as e:
            print(f"\nError: {e}")

What's happening here:

  • We're creating a direct terminal interface to chat with your Datarobot database
  • The LLM's responses are streamed to the CLI for faster interaction.
  • The agent uses context to maintain conversation history
  • You can type 'quit' or 'exit' to stop the chat loop gracefully
  • Agent responses and any errors are displayed in a clear, readable format

Define the main entry point

async def main() -> None:
    agent = await build_agent()
    await chat_loop(agent)

if __name__ == "__main__":
    # Handle Ctrl+C gracefully
    signal.signal(signal.SIGINT, lambda s, f: (print("\nBye!"), exit(0)))
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\nBye!")

What's happening here:

  • We're orchestrating the entire application flow
  • The agent gets built with proper error handling
  • Then we kick off the interactive chat loop so you can start talking to Datarobot

Run the agent

npx ts-node llamaindex-agent.ts

When prompted, authenticate and authorise your agent with Datarobot, then start asking questions.

Complete Code

Here's the complete code to get you started with Datarobot and LlamaIndex:

import asyncio
import os
import signal
import dotenv

from composio import Composio
from composio_llamaindex import LlamaIndexProvider
from llama_index.core.agent.workflow import ReActAgent
from llama_index.core.workflow import Context
from llama_index.llms.openai import OpenAI
from llama_index.tools.mcp import BasicMCPClient, McpToolSpec

dotenv.load_dotenv()

OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
COMPOSIO_API_KEY = os.getenv("COMPOSIO_API_KEY")
COMPOSIO_USER_ID = os.getenv("COMPOSIO_USER_ID")

if not OPENAI_API_KEY:
    raise ValueError("OPENAI_API_KEY is not set")
if not COMPOSIO_API_KEY:
    raise ValueError("COMPOSIO_API_KEY is not set")
if not COMPOSIO_USER_ID:
    raise ValueError("COMPOSIO_USER_ID is not set")

async def build_agent() -> ReActAgent:
    composio_client = Composio(
        api_key=COMPOSIO_API_KEY,
        provider=LlamaIndexProvider(),
    )

    session = composio_client.create(
        user_id=COMPOSIO_USER_ID,
        toolkits=["datarobot"],
    )

    mcp_url = session.mcp.url
    print(f"Composio MCP URL: {mcp_url}")

    mcp_client = BasicMCPClient(mcp_url, headers={"x-api-key": COMPOSIO_API_KEY})
    mcp_tool_spec = McpToolSpec(client=mcp_client)
    tools = await mcp_tool_spec.to_tool_list_async()

    llm = OpenAI(model="gpt-5")
    description = "An agent that uses Composio Tool Router MCP tools to perform Datarobot actions."
    system_prompt = """
    You are a helpful assistant connected to Composio Tool Router.
    Use the available tools to answer user queries and perform Datarobot actions.
    """
    return ReActAgent(
        tools=tools,
        llm=llm,
        description=description,
        system_prompt=system_prompt,
        verbose=True,
    );

async def chat_loop(agent: ReActAgent) -> None:
    ctx = Context(agent)
    print("Type 'quit', 'exit', or Ctrl+C to stop.")

    while True:
        try:
            user_input = input("\nYou: ").strip()
        except (KeyboardInterrupt, EOFError):
            print("\nBye!")
            break

        if not user_input or user_input.lower() in {"quit", "exit"}:
            print("Bye!")
            break

        try:
            print("Agent: ", end="", flush=True)
            handler = agent.run(user_input, ctx=ctx)

            async for event in handler.stream_events():
                # Stream token-by-token from LLM responses
                if hasattr(event, "delta") and event.delta:
                    print(event.delta, end="", flush=True)
                # Show tool calls as they happen
                elif hasattr(event, "tool_name"):
                    print(f"\n[Using tool: {event.tool_name}]", flush=True)

            # Get final response
            response = await handler
            print()  # Newline after streaming
        except KeyboardInterrupt:
            print("\n[Interrupted]")
            continue
        except Exception as e:
            print(f"\nError: {e}")

async def main() -> None:
    agent = await build_agent()
    await chat_loop(agent)

if __name__ == "__main__":
    # Handle Ctrl+C gracefully
    signal.signal(signal.SIGINT, lambda s, f: (print("\nBye!"), exit(0)))
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\nBye!")

Conclusion

You've successfully connected Datarobot to LlamaIndex through Composio's Tool Router MCP layer. Key takeaways:
  • Tool Router dynamically exposes Datarobot tools through an MCP endpoint
  • LlamaIndex's ReActAgent handles reasoning and orchestration; Composio handles integrations
  • The agent becomes more capable without increasing prompt size
  • Async Python provides clean, efficient execution of agent workflows
You can easily extend this to other toolkits like Gmail, Notion, Stripe, GitHub, and more by adding them to the toolkits parameter.

How to build Datarobot MCP Agent with another framework

FAQ

What are the differences in Tool Router MCP and Datarobot MCP?

With a standalone Datarobot MCP server, the agents and LLMs can only access a fixed set of Datarobot tools tied to that server. However, with the Composio Tool Router, agents can dynamically load tools from Datarobot and many other apps based on the task at hand, all through a single MCP endpoint.

Can I use Tool Router MCP with LlamaIndex?

Yes, you can. LlamaIndex fully supports MCP integration. You get structured tool calling, message history handling, and model orchestration while Tool Router takes care of discovering and serving the right Datarobot tools.

Can I manage the permissions and scopes for Datarobot while using Tool Router?

Yes, absolutely. You can configure which Datarobot scopes and actions are allowed when connecting your account to Composio. You can also bring your own OAuth credentials or API configuration so you keep full control over what the agent can do.

How safe is my data with Composio Tool Router?

All sensitive data such as tokens, keys, and configuration is fully encrypted at rest and in transit. Composio is SOC 2 Type 2 compliant and follows strict security practices so your Datarobot data and credentials are handled as safely as possible.

Used by agents from

Context
Letta
glean
HubSpot
Agent.ai
Altera
DataStax
Entelligence
Rolai
Context
Letta
glean
HubSpot
Agent.ai
Altera
DataStax
Entelligence
Rolai
Context
Letta
glean
HubSpot
Agent.ai
Altera
DataStax
Entelligence
Rolai

Never worry about agent reliability

We handle tool reliability, observability, and security so you never have to second-guess an agent action.