Criar ferramentas de agente de IA

Importante

Esta funcionalidade está em Pré-visualização Pública.

Este artigo fornece uma visão geral da criação de ferramentas de agente de IA usando o Mosaic AI Agent Framework.

O Agent Framework ajuda os desenvolvedores a criar ferramentas que os agentes de IA podem usar para executar ações além da geração de linguagem, como recuperar dados estruturados ou não estruturados ou executar código.

Para uma introdução aos agentes de IA, consulte O que são sistemas compostos de IA e agentes de IA?.

Ferramentas de função do Unity Catalog vs. ferramentas de código do agente

Para criar ferramentas e adicioná-las a agentes com o Mosaic AI Agent Framework, você pode usar qualquer combinação dos seguintes métodos:

• funções do Catálogo Unity: funções do Catálogo Unity são definidas e gerenciadas dentro do Catálogo Unity, oferecendo recursos integrados de segurança e conformidade. Escrever sua ferramenta como uma função do Catálogo Unity concede mais facilidade de descoberta, governança e reutilização. As funções do Unity Catalog funcionam especialmente bem para aplicar transformações e agregações em grandes conjuntos de dados.
• Ferramentas de código do Agent: Essas ferramentas são definidas no mesmo código que define o agente de IA. Essa abordagem é útil ao chamar APIs REST, usar código arbitrário ou bibliotecas ou executar ferramentas de baixa latência. No entanto, essa abordagem não tem a capacidade de descoberta e a governança internas fornecidas pelas funções do Unity Catalog.

Ambos os métodos são compatíveis com agentes escritos em código Python personalizado ou usando bibliotecas de criação de agentes como LangGraph.

Para ver exemplos de ferramentas de função do Unity Catalog e ferramentas de programação de agente, consulte Exemplos de ferramentas de agente

Melhore a chamada de ferramentas com documentação

Uma documentação clara e detalhada ajuda os agentes de IA a entender quando e como usar as ferramentas que você fornece. Ao criar ferramentas, documente os parâmetros da ferramenta e retorne valores cuidadosamente para garantir que seu agente de IA use as ferramentas corretamente e no momento certo:

Para funções do Unity Catalog, use COMMENT para descrever a ferramenta e os parâmetros.

Exemplo de documentação de ferramentas eficazes

O exemplo a seguir mostra cadeias de caracteres de COMMENT efetivas para uma ferramenta de função Unity Catalog que consulta uma tabela estruturada.

CREATE OR REPLACE FUNCTION main.default.lookup_customer_info(
  customer_name STRING COMMENT 'Name of the customer whose info to look up.'
)
RETURNS STRING
COMMENT 'Returns metadata about a specific customer including their email and ID.'
RETURN SELECT CONCAT(
    'Customer ID: ', customer_id, ', ',
    'Customer Email: ', customer_email
  )
  FROM main.default.customer_data
  WHERE customer_name = customer_name
  LIMIT 1;

Exemplo de documentação de ferramenta ineficaz

O exemplo a seguir mostra cadeias de caracteres de COMMENT ineficazes que perdem informações de chave, como os valores de retorno.

CREATE OR REPLACE FUNCTION main.default.lookup_customer_info(
  customer_name STRING COMMENT 'Name of the customer.'
)
RETURNS STRING
COMMENT 'Returns info about a customer.'
RETURN SELECT CONCAT(
    'Customer ID: ', customer_id, ', ',
    'Customer Email: ', customer_email
  )
  FROM main.default.customer_data
  WHERE customer_name = customer_name
  LIMIT 1;

Exemplos da ferramenta Agent

Consulte os seguintes artigos para obter exemplos de ferramentas de agente:

• Ferramentas de interpretação de código permitem que os agentes executem código arbitrário, como Python.
• Ferramentas de recuperação de dados estruturados permitem que o seu agente consulte fontes de dados estruturados, como tabelas SQL.
• Ferramentas de recuperação de dados não estruturados permitem que o seu agente consulte fontes de dados não estruturadas, como um corpus de texto, para executar uma geração aumentada por recuperação.

Adicionar ferramentas do Unity Catalog aos agentes

Depois de criar as ferramentas do Catálogo Unity, adicione-as ao seu agente. Os agentes LangChain podem aproveitar o UCFunctionToolkit para incorporar ferramentas de UC.

Exporte agentes de chamada de ferramentas do AI Playground

O AI Playground fornece uma maneira conveniente de adicionar ferramentas do Unity Catalog a um LLM, testar o agente e exportar seu código.

Para usar o AI Playground para exportar agentes, seu espaço de trabalho deve atender aos seguintes requisitos:

• O Catálogo Unity deve estar habilitado.
• A computação sem servidor deve estar habilitada.
• Os modelos de base Pay-per-token ou Modelos externos devem ser habilitados.

Use as seguintes etapas para exportar o código dos agentes chamadores de ferramentas:

1. No AI Playground, selecione um modelo com o rótulo Ferramentas ativado.

   

2. Selecione Ferramentas e clique em Adicionar uma ferramenta.

3. No menu suspenso, selecione uma função do Catálogo Unity:

   

4. Use o Playground para conversar e testar a combinação atual de LLM, ferramentas e prompt do sistema. Experimente variações para ter uma ideia de como funciona a configuração atual.

   

   Depois de adicionar ferramentas, exporte o agente para blocos de anotações Python:

5. Clique em Exportar para gerar blocos de anotações Python que definem e implantam o agente.

   Depois de exportar o código do agente, você verá três arquivos salvos em seu espaço de trabalho:

   • agent notebook: Contém código Python que define seu agente usando LangChain.
   • driver notebook: Contém código Python para registrar, rastrear, registrar e implantar o agente de IA usando o Mosaic AI Agent Framework.
   • config.yml: Contém informações de configuração sobre seu agente, incluindo definições de ferramentas.

6. Abra o agent bloco de anotações para ver o código LangChain que define seu agente. Use este bloco de anotações para testar e iterar no agente programaticamente, como definir mais ferramentas.

7. Quando estiver satisfeito com as saídas do agente, execute o notebook driver para registar e implantar o agente num endpoint de serviço de fornecimento de modelo.

Próximos passos

• Rastrear um agente.