datarobot-genai 0.15.11

Generic helpers for GenAI

DataRobot GenAI Library

Homepage · PyPI · Support

Features

• AG-UI integration—agents expose a standard AG-UI event stream (RunAgentInput in, lifecycle + text + tool-call events out), so UIs and the DataRobot platform can render runs consistently without bespoke adapters per framework.
• Multi-agent systems out of the box—first-class patterns for planner/writer crews, LangGraph multi-node graphs, and LlamaIndex AgentWorkflow handoffs; wrap them with one helper and keep the same streaming contract.
• Unified LLM layer (DataRobot-compatible)—one get_llm() entry point per integration (LangGraph, LlamaIndex, CrewAI, NAT), all backed by the same LiteLLM-based routing to the DataRobot LLM Gateway, LLM deployments, NIM, or external providers—driven by the same environment and Config, so every component speaks to DataRobot consistently.
• Utilities for common GenAI workflows.
• Orchestration—build agents from universal pieces in the low-code workflow.yaml interface. Combine and reuse LLMs, tools, agents, and evaluators. The design is compatible with and inspired by NeMo Agentic Toolkit.
• Evaluating and serving with DRAgent—use a front-end server to plug your agent into a real-world application. It supports distributed tracing, generation and evaluation endpoints, async generations, and two-way communication over WebSockets.

User-facing walkthrough: docs/README.md.

Installation

• Requires Python 3.11–3.13.
• Install:

pip install --upgrade pip
pip install "datarobot-genai"

• Optional extras:

pip install "datarobot-genai[crewai]"
pip install "datarobot-genai[langgraph]"
pip install "datarobot-genai[llamaindex]"
# Multiple extras
pip install "datarobot-genai[crewai,langgraph,llamaindex]"

Available extras include: crewai, langgraph, llamaindex, nat, drmcp, pydanticai.

Excluded Dependencies

Some transitive dependencies are excluded via exclude-dependencies in pyproject.toml because they are unused by this project. Do not re-add them.

Package	Pulled in by	Reason for exclusion
uv	build tooling	Not a runtime dependency.
langchain-milvus	langchain ecosystem	Unused vector store integration.
pymilvus	langchain-milvus	Transitive dependency of langchain-milvus.
flask	nvidia-nat-core 1.6.0	Only used in NAT examples, not core library code (ref).

Development

Prerequisites: Python 3.11–3.14, uv, Task CLI, pre-commit.

uv sync --all-extras --dev
pre-commit install
task test

Semantic versioning

Every change requires a patch version change and an entry in CHANGELOG.md. Changes that break backward compatibility require a minor version bump.

TestPyPI

Comment /build on a PR to build and publish a dev version of the package to TestPyPI.

Publishing

• Same-repo PRs—comment /build to publish dev builds to TestPyPI (.devN).
• Merge to main—creates tag v{version} and publishes to PyPI automatically.
• Version tags—pushing a v* tag also triggers PyPI publish.
• Local release—optional task release:tag-and-push creates and pushes v{version} locally.

Links

• Repository—source and issues.
• PyPI—released packages.
• TestPyPI—dev builds.

License

Apache-2.0—see LICENSE.