Protótipo de ferramenta - chamando agentes no AI Playground
Este artigo mostra como criar um protótipo de um AI agente de chamada de ferramenta com AI Playground o.
Use o AI Playground para criar rapidamente um agente de chamadas de ferramentas e conversar com ele ao vivo para ver como ele se comporta. Em seguida, exporte o agente para implantação ou desenvolvimento adicional no código Python.
Para criar agentes usando uma abordagem de código primeiro, consulte Criar agentes AI em código.
Requisitos
O site workspace deve ter o seguinte recurso habilitado para criar protótipos de agentes usando AI Playground:
Protótipo de ferramenta - chamando agentes no AI Playground
Para criar o protótipo de um agente de chamada de ferramentas:
-
No Playground, selecione um modelo com o rótulo Tools enabled (Ferramentas ativadas ).
-
Clique em Ferramentas > + Adicionar ferramenta e selecione as ferramentas para fornecer ao agente. Você pode escolher até 20 ferramentas. As opções de ferramentas incluem:
- Função hospedada : Selecione uma função do Unity Catalog para seu agente usar.
- Definição da função : defina uma função personalizada para seu agente chamar.
- Pesquisa vetorial : especifique um índice de pesquisa vetorial para seu agente usar como uma ferramenta para ajudar a responder às consultas. Se seu agente usar um índice de pesquisa vetorial, sua resposta citará as fontes usadas.
Para este guia, selecione a função integrada do Unity Catalog,
system.ai.python_exec
. Essa função dá ao seu agente a capacidade de executar código Python arbitrário. Para saber como criar ferramentas de agente, consulte AI ferramentas de agente.Você também pode selecionar um índice de pesquisa vetorial, que permite que seu agente consulte o índice para ajudar a responder às consultas.
-
Converse para testar a combinação atual de LLM, ferramentas e avisos do sistema e tente variações. O LLM seleciona a ferramenta apropriada a ser usada para gerar uma resposta.
Ao fazer uma pergunta relacionada a informações no índice de pesquisa vetorial, o LLM consulta as informações de que precisa e cita todos os documentos de origem usados em sua resposta.
Exportação e implantado AI Playground agentes
Depois de criar o protótipo do agente AI em AI Playground, exporte-o para o notebook Python para implantá-lo em um modelo de serviço endpoint.
-
Clique em Exportar para gerar o Notebook que define e implanta o agente AI.
Depois de exportar o código do agente, uma pasta com um Notebook do driver é salva no site workspace. Esse driver define uma ferramenta chamada LangGraph ChatAgent, testa o agente localmente, usa logging baseado em código, registra e implanta o agente AI usando o Mosaic AI Agent Framework.
-
Resolver todas as tarefas pendentes no Notebook.
O código exportado pode se comportar de forma diferente da sua sessão do AI Playground. Databricks recomenda a execução do Notebook exportado para iterar e depurar mais, avaliar a qualidade do agente e, em seguida, implantar o agente para compartilhar com outras pessoas.
Desenvolva agentes em código
Use o Notebook exportado para testar e iterar programaticamente. Use o Notebook para fazer coisas como adicionar ferramentas ou ajustar os parâmetros do agente.
No desenvolvimento programático, os agentes devem atender a requisitos específicos para serem compatíveis com outros recursos do agente Databricks. Para saber como criar agentes usando uma abordagem de código primeiro, consulte Criar agentes AI em código