Create an AI agent
This article introduces the process of creating AI agents on Azure Databricks and outlines the available methods for creating agents.
To learn more about agents, see What are compound AI systems and AI agents?.
AI Builder provides a simple, no-code approach to build and optimize domain-specific, high-quality AI agent systems for common AI use cases. Specify your use case and data, and AI Builder will automatically build out several AI agent systems for you that you can further refine. See AI Builder.
Mosaic AI Agent Framework and MLflow provide tools to help you author enterprise-ready agents in Python.
Databricks supports authoring agents using third-party agent authoring libraries like LangGraph/LangChain, LlamaIndex, or custom Python implementations.
To learn how to create AI agents on Databricks, see Author AI agents in code.
The AI Playground is the easiest way to create an agent on Azure Databricks. AI Playground lets you select from various LLMs and quickly add tools to the LLM using a low-code UI. You can then chat with the agent to test its responses and then export the agent to code for deployment or further development.
See Prototype tool-calling agents in AI Playground.
Databricks uses MLflow Model Signatures to define agents' input and output schema. Product features like the AI Playground assume that your agent has one of a set of supported model signatures.
If you follow the recommended approach to authoring agents, MLflow will automatically infer a signature for your agent that is compatible with Databricks product features, with no additional work required on your part.
Otherwise, you must ensure that your agent adheres to one of the other signatures in Legacy input and output agent schema, to ensure compatibility with Databricks features.