Partager via

Outils de l’agent IA

Cet article fournit une vue d’ensemble de la création d’outils d’agent IA à l’aide de Mosaic AI Agent Framework.

Les outils d’agent IA permettent aux agents d’effectuer des tâches au-delà de la génération de langage, telles que la récupération de données structurées ou non structurées et l’exécution de code personnalisé.

utils de fonction de Unity Catalog contre outils de code de l’agent

Il existe deux façons principales de créer des outils dans Agent Framework : définir l’outil en tant que fonction de catalogue Unity ou le définir directement dans le code de l’agent.

Votre agent peut utiliser n’importe quelle combinaison d’outils de fonction de catalogue Unity ou d’outils de code d’agent. Les deux types d’outils fonctionnent avec des agents créés à l’aide de Python natif ou à l’aide de bibliothèques de création d’IA de génération telles que LangGraph et openAI SDK.

Outil de catalogue de fonctions Unity Outil de code de l’agent
  • Défini en tant qu'UDFs du catalogue Unity
  • Géré dans le catalogue Unity en tant que registre central pour les outils
  • Fonctionnalités de sécurité et de conformité intégrées
  • Accorde une détectabilité et une réutilisation plus faciles
  • Adapté à l’application de transformations et d’agrégations sur des jeux de données volumineux
  • Défini directement dans le code de votre agent
  • Manque de gouvernance et de facilité de découverte des fonctions intégrées du catalogue Unity
  • Bon pour appeler des API REST, exécuter du code arbitraire ou exécuter des outils à faible latence

Créer des outils d’agent

Pour savoir comment créer des outils d’agent, consultez Créer des outils d’agent IA personnalisés avec des fonctions de catalogue Unity.

Les types courants d’outils d’agent sont les suivants :

Ajouter des outils de catalogue Unity aux agents

Contrairement aux outils de code de l’agent, qui sont définis directement dans le code de l’agent, les outils catalogue Unity doivent être ajoutés explicitement aux agents pour les rendre disponibles.

Databricks recommande d’utiliser UCFunctionToolkit pour intégrer des outils de catalogue Unity à des frameworks de création d’agents tels que LangChain, OpenAI ou d’autres kits SDK. Le kit de ressources garantit la cohérence entre différents frameworks et automatise les fonctionnalités utiles. Voir Créer des outils d’agent IA personnalisés avec des fonctions Unity Catalog.

Vous pouvez utiliser ai Playground pour ajouter rapidement des outils de catalogue Unity aux agents et prototyper leur comportement avant de les déployer. Consultez Prototype d’agent d’appel d’outil dans AI Playground.

Gérer les outils de catalogue Unity

Utilisez le client de fonction Databricks pour gérer les outils de catalogue Unity. Le client de fonction Databricks est basé sur le client de fonction de catalogue Unity open source, mais offre plusieurs améliorations uniques à Databricks.

Pour plus d’informations sur la gestion des fonctions de catalogue Unity, consultez la documentation du catalogue Unity - Client de fonction.

Protocole de contexte de modèle : normaliser l’accès aux outils

Le protocole MCP (Model Context Protocol) est une norme ouverte qui offre aux agents IA un moyen universel de se connecter à des outils, des données et des ressources. MCP agit comme un pont entre les agents et les systèmes externes avec lesquels ils ont besoin d’interagir.

Databricks fournit les options MCP suivantes :

  • Serveurs MCP managés : Databricks dispose de serveurs prêts à l’emploi qui permettent aux agents d’interroger des données et d’accéder aux outils dans le catalogue Unity.

  • Serveurs MCP personnalisés : hébergez en toute sécurité votre propre serveur MCP ou exécutez un serveur MCP tiers.

Consultez le protocole MCP (Model Context Protocol) sur Databricks.

Étapes suivantes