通过

结构化检索 AI 代理工具

  • 项目

重要说明

此功能目前以公共预览版提供。

本文介绍如何使用马赛克 AI 代理框架为结构化数据检索创建 AI 代理。 结构化检索器使代理能够查询结构化数据源，例如 SQL 表。

若要详细了解代理工具，请参阅 AI 代理工具。

表查询工具

以下示例创建一个工具，允许代理从 Unity 目录表查询结构化客户数据。

它定义了一个名为 lookup_customer_info的 UC 函数，该函数允许 AI 代理从假设 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;

后续步骤

创建代理工具后，将该工具添加到 AI 代理。 请参阅将 Unity Catalog 工具添加到代理。