Поделиться через

Инструменты агента ИИ для структурированного поиска

  • Статья

Важный

Эта функция находится в общедоступной предварительной версии .

В этой статье показано, как создать агенты ИИ для получения структурированных данных с помощью Платформы агента ИИ Мозаики. Структурированные извлекатели позволяют агентам запрашивать структурированные источники данных, такие как таблицы SQL.

Дополнительные сведения о средствах агента см. в статье средства агента ИИ.

Средство запроса таблиц

В следующем примере создается средство, позволяющее агенту запрашивать структурированные данные клиента из таблицы каталога Unity.

Он определяет функцию UC с именем lookup_customer_info, которая позволяет агенту ИИ извлекать структурированные данные из гипотетической таблицы customer_data .

Выполните следующий код в редакторе SQL.

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 particular customer given the customer name, including the customer's email and ID. The
customer ID can be used for other queries.'
RETURN SELECT CONCAT(
    'Customer ID: ', customer_id, ', ',
    'Customer Email: ', customer_email
  )
  FROM main.default.customer_data
  WHERE customer_name = customer_name
  LIMIT 1;

Дальнейшие действия

После создания инструмента агента добавьте его в агент ИИ. См. статью Добавление средств каталога Unity в агенты.