datarobot-genai
Generic helpers for GenAI
Decision gist · record as of 2026-08-14
Yes, if you are building agents on DataRobot or need unified LLM routing across multiple agentic frameworks. The library is actively maintained, has no known vulnerabilities, installs cleanly with low friction, and is permissively licensed. Install one of the framework-specific extras (crewai, langgraph, llamaindex, nat) matching your choice of agent framework. Requires a DataRobot account for production features.AI-flagged interpretation of the facts on this page — verify before relying
Before you install
- Requires Python 3.11–3.13.
- DataRobot-backed features require a DataRobot account and DATAROBOT_API_TOKEN environment variable set.
- Active maintenance with a release on 2026-08-14.
License · maintenance · safety
Apache-2.0 (permissive) — Licensed under Apache-2.0 (permissive), allowing commercial use, modification, and distribution with minimal restrictions—suitable for most production and proprietary projects.
last release 2026-08-14 (0 days) · last repo commit 2026-08-14 · 16 stars
0 known vulnerabilities (OSV.dev, 2026-08-14) · 164,813 downloads/mo, #10,538 on PyPI
Alternatives
Verify before relying
pip install "datarobot-genai[langgraph]"
import datarobot_genai
llm = datarobot_genai.get_llm("langgraph")- Whether the library's multi-agent patterns (planner/writer crews, LangGraph multi-node graphs, LlamaIndex AgentWorkflow) are production-ready or still experimental.
- Performance characteristics and latency overhead of the unified LLM layer routing through DataRobot LLM Gateway.
- Compatibility and stability of the AG-UI event stream contract across different framework versions.
What it is and what it does
DataRobot GenAI is a framework for building AI agents that integrates with DataRobot's LLM Gateway and deployment infrastructure. It provides a unified entry point (`get_llm()`) that routes requests to DataRobot LLM deployments, external LLM providers, or NVIDIA NIM, backed by LiteLLM. The library supports multiple agentic frameworks—LangGraph, LlamaIndex, CrewAI, and NVIDIA NAT—through a single abstraction layer, eliminating the need to write framework-specific integration code.
Beyond LLM routing, it includes a library of agentic tools (`drtools`), a DataRobot-compatible MCP server (`drmcp`), and orchestration via a low-code `workflow.yaml` interface. Agents expose a standard AG-UI event stream for consistent rendering across the DataRobot platform and custom UIs. The package also provides DRAgent, a serving layer that handles distributed tracing, async generations, and WebSocket communication for real-world deployment.
Use it for
- Build a LangGraph agent that routes LLM calls through DataRobot's gateway without rewriting framework-specific code.
- Orchestrate multi-agent systems (planner/writer crews) using a shared `workflow.yaml` configuration across different frameworks.
- Deploy an agent with DRAgent to serve it as a web application with distributed tracing and async generation support.
- Integrate agentic tools via `drtools` to give agents first-class capabilities for interacting with external systems.
- Expose a custom MCP server in DataRobot using `drmcp` for standardized tool discovery and invocation.
- Evaluate agent performance using built-in evaluation utilities backed by the NeMo Evaluator launcher.
Worth the install?
AI-flagged interpretation of the facts on this page. Verify before relying on it.
Yes, if you are building agents on DataRobot or need unified LLM routing across multiple agentic frameworks.
The library is actively maintained, has no known vulnerabilities, installs cleanly with low friction, and is permissively licensed. Install one of the framework-specific extras (crewai, langgraph, llamaindex, nat) matching your choice of agent framework. Requires a DataRobot account for production features.
Install
datarobot-genai on PyPI
Before you install
Active maintenance with a release on 2026-08-14. Requires Python 3.11–3.13 and installs as a pure wheel with no runtime dependencies in the base package; framework-specific extras (crewai, langgraph, llamaindex, nat) pull in their respective ecosystems. Install friction is low.
Requires Python 3.11–3.13. DataRobot-backed features require a DataRobot account and DATAROBOT_API_TOKEN environment variable set.
License in practice
Licensed under Apache-2.0 (permissive), allowing commercial use, modification, and distribution with minimal restrictions—suitable for most production and proprietary projects.
Quickstart
pip install "datarobot-genai[langgraph]"
import datarobot_genai
llm = datarobot_genai.get_llm("langgraph")Verify before relying
- Whether the library's multi-agent patterns (planner/writer crews, LangGraph multi-node graphs, LlamaIndex AgentWorkflow) are production-ready or still experimental.
- Performance characteristics and latency overhead of the unified LLM layer routing through DataRobot LLM Gateway.
- Compatibility and stability of the AG-UI event stream contract across different framework versions.
Package facts
| License | Apache-2.0 permissive |
| Python support | Supports the current Python release <3.14,>=3.11 |
| Install friction | Low. Pure-Python wheel |
| Runtime dependencies | None |
| Maintenance | Actively maintained 0 days since the last release |
| Last repo commit | |
| First released | |
| Downloads | 164,813 / month, #10,538 on PyPI 30-day window, as of 2026-08-14 |
| Known vulnerabilities | None known OSV.dev, checked 2026-08-14 |
Evidence: datarobot_genai-0.27.14-py3-none-any.whl
Tags
Let your AI agent find packages like this
Example. Real query, live index.
You found this page by searching. An agent finds it by wishing: SkillFed indexes 14,416 PyPI packages by what they can do, searchable in plain language.
wish › “datarobot agent framework”
- datarobot-genaiA toolkit for building and deploying AI agents on DataRobot,…
- datarobot-predictProvides a unified Python interface for making predictions using…
- datarobotA Python client library for programmatically interacting with the…
Give your agent the search over MCP, or paste the wish link into any chat.
More Artificial Intelligence packages
LiteLLM provides a unified Python interface to call 100+ LLM providers (OpenAI, Anthropic, Gemini, Bedrock, Azure, and others) using OpenAI-compatible API format, available as both a Python SDK and a self-hosted AI Gateway proxy server.
Install it if you need to work with multiple LLM providers or want to centralize LLM routing in your organization.
Client library and CLI tool for downloading, uploading, and managing models, datasets, and repositories on the Hugging Face Hub platform.
Install it if you work with Hugging Face Hub models or datasets.
LangChain provides a framework for building agents and LLM-powered applications by composing language models, tools, and memory through a unified API that abstracts over multiple model providers.
hf-xet provides chunk-based deduplication and efficient file transfer for the Hugging Face Hub, enabling faster uploads and downloads of large files with local disk caching.
Tokenizers converts raw text into token sequences for NLP models, with support for training custom vocabularies and using pre-built tokenizers (BPE, WordPiece) optimized for speed via Rust.
Transformers provides a unified framework for loading, fine-tuning, and running state-of-the-art pretrained models across text, vision, audio, video, and multimodal tasks using PyTorch, JAX, or TensorFlow.
Install it if you need to run or train any transformer-based model for NLP, vision, audio, or multimodal tasks.
See also llama-index-agent-openai · datarobot · datarobot-predict · datarobot-moderations · datarobot-drum · crewai · leptonai · deepagents · langgraph · tooluniverse