datarobot-genai 0.15.4

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), 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.
• Integrations: CrewAI, LangGraph, LlamaIndex, NAT, MCP.

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 dep 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. It also requires an entry to CHANGELOG.md. Changes breaking backward compatibility requires a minor version change.

Test pypi

To build and publish a dev version of a package, comment /build on a PR.

Publishing

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

Links

• Home: https://github.com/datarobot-oss/datarobot-genai
• PyPI: https://pypi.org/project/datarobot-genai/
• TestPyPI: https://test.pypi.org/project/datarobot-genai/

License

Apache-2.0