Agentes de registo (serviços de agentes)

Importante

Os serviços de agentes estão em Beta. O Unity AI Gateway está geralmente disponível, mas as suas capacidades beta estão ativadas separadamente. Um administrador de conta deve ativar as funcionalidades beta do Unity AI Gateway na página de Pré-visualizações da consola de contas. Ver Gerir as pré-visualizações de Azure Databricks.

Registe agentes no Unity Catalog para que todos os agentes de cada equipa vivam num só lugar. Quando regista um agente como serviço de agente, pode navegá-lo e descobri-lo juntamente com as suas tabelas, modelos e funções, e definir permissões com as mesmas concessões que protegem os seus outros ativos do Unity Catalog.

Descubra agentes no Explorador de Catálogos

Depois de um serviço de agente ser registado, ele aparece no Explorador de Catálogos no esquema onde foi criado. Para ver um serviço agente, o utilizador precisa de um privilégio explícito sobre ele, como EXECUTE. Selecione um serviço de agente para visualizar o seu nome, descrição, ligação e permissões atuais.

Note

Conceder EXECUTE também permite ao utilizador invocar o serviço agente, não apenas descobri-lo. Conceda-o apenas a utilizadores que devem poder usar o agente.

Use o comment campo para adicionar uma descrição quando criar ou atualizar um serviço de agente. Esta descrição aparece no Explorador de Catálogos e ajuda os membros da equipa a compreender o que o agente faz ao explorá-lo.

Isto dá à sua organização um único local para ver que agentes existem, quem os detém e quem tem acesso, sem ter de os rastrear em sistemas ou documentação separadas.

Requirements

  • As funcionalidades beta do Unity AI Gateway ativadas para a sua conta. Ver Gerir as pré-visualizações de Azure Databricks.
  • Para criar um serviço de agente, tem de ter USE CATALOG e USE SCHEMA no catálogo e esquema principais, CREATE SERVICE no esquema e USE CONNECTION na conexão a que o serviço de agente faz referência.
  • Para gerir permissões num serviço de agente, deve ser o proprietário do serviço de agente ou ter esse MANAGE privilégio.

Criar um serviço de agente

Um serviço de agente faz referência a uma ligação existente ao Unity Catalog que contém o host e as credenciais do agente. Crie a ligação antes de criar o serviço de agente.

O exemplo seguinte cria um serviço agente nomeado support_agent no main.default esquema:

databricks api post \
  "/api/2.1/unity-catalog/agent-services?parent=schemas/main.default&agent_service_id=support_agent" \
  --json '{
    "agent_service_type": "AGENT_SERVICE_TYPE_EXTERNAL",
    "comment": "Support agent for the customer team",
    "config": {
      "source_connection": {
        "name": "connections/main.default.my_agent_connection"
      },
      "base_path": "/v1/chat",
      "system_prompt": "You are a helpful support assistant."
    }
  }'

A resposta inclui o nome do recurso do serviço de agente:

{
  "name": "agent-services/main.default.support_agent",
  "agent_service_type": "AGENT_SERVICE_TYPE_EXTERNAL",
  "created_by": "you@company.com",
  "config": {
    "source_connection": { "name": "connections/main.default.my_agent_connection" },
    "base_path": "/v1/chat",
    "system_prompt": "You are a helpful support assistant."
  }
}

Obter e listar serviços do agente

Recupere um serviço de agente único pelo seu nome completo:

databricks api get "/api/2.1/unity-catalog/agent-services/main.default.support_agent"

Liste todos os serviços do agente num esquema:

databricks api get "/api/2.1/unity-catalog/agent-services?parent=schemas/main.default"

Para listar os serviços do agente em todo o metaarmazenamento, omita o filtro de esquema:

databricks api get "/api/2.1/unity-catalog/agent-services"

Atualizar um serviço de agente

Para atualizar campos específicos sem sobrescrever outros, use um PATCH pedido e especifique quais os campos a alterar. O exemplo seguinte atualiza o prompt do sistema mantendo todos os outros campos inalterados:

databricks api patch \
  "/api/2.1/unity-catalog/agent-services/main.default.support_agent?update_mask=config.system_prompt" \
  --json '{ "config": { "system_prompt": "You are a concise support assistant." } }'

Campos que pode atualizar: comment, config.system_prompt, config.base_path.

Conceder acesso

Defina permissões num serviço de agente com o mesmo modelo de concessão que protege os seus outros objetos seguros do Unity Catalog. Subsídio EXECUTE a utilizadores e principais de serviço:

databricks api patch \
  "/api/2.1/unity-catalog/permissions/AGENT_SERVICE/main.default.support_agent" \
  --json '{
    "changes": [
      { "principal": "teammate@company.com", "add": ["EXECUTE"] }
    ]
  }'

Inspecionar as subvenções atuais:

databricks api get "/api/2.1/unity-catalog/permissions/AGENT_SERVICE/main.default.support_agent"

Revogar acesso:

databricks api patch \
  "/api/2.1/unity-catalog/permissions/AGENT_SERVICE/main.default.support_agent" \
  --json '{
    "changes": [
      { "principal": "teammate@company.com", "remove": ["EXECUTE"] }
    ]
  }'

Privilégios atribuíbles: EXECUTE, READ METADATA, MANAGE, ALL PRIVILEGES.

Eliminar um serviço de agente

databricks api delete "/api/2.1/unity-catalog/agent-services/main.default.support_agent"

Limitações

Durante a beta, aplicam-se as seguintes limitações:

  • A invocação em tempo de execução não está disponível. Os agentes não podem ser contactados através de um serviço de agentes registados. A gestão de registos e permissões está disponível agora.
  • As políticas de serviço e os limites de taxa não são suportados pela API. Tentar definir estes campos retorna um erro.
  • SQL DDL para serviços de agente não está disponível. Criar e gerir serviços de agente com a API REST.
  • Os campos full_name e owner retornam null nas respostas GET nesta versão.
  • A Pesquisa Global do Catálogo Unity não disponibiliza serviços de agentes.
  • O BROWSE privilégio não é apoiado. Para ver um serviço agente, o utilizador precisa de um privilégio explícito sobre ele, como EXECUTE.

Passos seguintes