구조적 검색 AI 에이전트 도구

이 문서에서는 Mosaic AI 에이전트 프레임워크를 사용하여 구조화된 데이터 검색을 위한 AI 에이전트 도구 만드는 방법을 보여줍니다. 에이전트가 SQL 테이블과 같은 구조적 데이터 원본을 쿼리할 수 있도록 하려면 다음 방법 중 하나를 사용할 수 있습니다.

• SQL Unity 카탈로그 함수: 에이전트가 채울 수 있는 매개 변수를 사용하여 SQL 쿼리를 정의합니다. 쿼리에 알려진 고정 형식이 있는 경우 이 메서드를 사용합니다.

Unity 카탈로그 SQL 함수 도구를 사용하여 데이터 쿼리

쿼리가 미리 알려지고 에이전트가 매개 변수를 제공하는 경우 Unity 카탈로그 SQL 함수를 사용하여 구조화된 검색 도구를 만듭니다.

다음 예제에서는 AI 에이전트가 가상 lookup_customer_info테이블에서 구조화된 데이터를 검색할 수 있도록 하는 customer_data 이라는 Unity 카탈로그 함수를 만듭니다.

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's 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 카탈로그 도구를 만든 후 에이전트에 추가합니다. 에이전트 도구 만들기를 참조하세요.

Genie 다중 에이전트 시스템

에이전트가 대답해야 하는 쿼리 종류를 모르고 여러 테이블을 쿼리할 수 있는 유연성이 필요한 경우 Genie 에이전트를 포함하는 다중 에이전트 시스템을 만듭니다.

자세한 내용은 다중 에이전트 시스템에서 Genie 사용을 참조하세요.