Tworzenie narzędzi agenta sztucznej inteligencji

Ważne

Ta funkcja jest dostępna w publicznej wersji zapoznawczej.

Ten artykuł zawiera omówienie tworzenia narzędzi agenta sztucznej inteligencji przy użyciu struktury agentów Mozaiki AI.

Struktura agentów ułatwia deweloperom tworzenie narzędzi, których agenci sztucznej inteligencji mogą używać do wykonywania akcji poza generowaniem języka, takich jak pobieranie ustrukturyzowanych lub nieustrukturyzowanych danych lub wykonywanie kodu.

Aby zapoznać się z wprowadzeniem do agentów sztucznej inteligencji, zobacz Co to są złożone systemy sztucznej inteligencji i agenci sztucznej inteligencji?.

Narzędzia funkcji Unity Catalog a narzędzia kodu agenta

Aby utworzyć narzędzia i dodać je do agentów za pomocą programu Mosaic AI Agent Framework, możesz użyć dowolnej kombinacji następujących metod:

• funkcje wykazu aparatu Unity: funkcje katalogu aparatu Unity są definiowane i zarządzane w wykazie aparatu Unity, oferując wbudowane funkcje zabezpieczeń i zgodności. Tworzenie narzędzia jako funkcji Unity Catalog umożliwia łatwiejsze odnajdywanie, zarządzanie i ponowne użycie. Funkcje wykazu aparatu Unity działają szczególnie dobrze w przypadku stosowania przekształceń i agregacji w dużych zestawach danych.
• Narzędzia kodu agenta: Te narzędzia są zdefiniowane w tym samym kodzie, który definiuje agenta sztucznej inteligencji. Takie podejście jest przydatne podczas wywoływania interfejsów API REST, używania dowolnego kodu lub bibliotek lub wykonywania narzędzi o małych opóźnieniach. Jednak to podejście nie ma wbudowanej możliwości odnajdywania i zapewniania ładu zapewnianego przez funkcje wykazu aparatu Unity.

Obie metody są zgodne z agentami napisanymi w niestandardowym kodzie języka Python lub przy użyciu bibliotek tworzenia agentów, takich jak LangGraph.

Aby zapoznać się z przykładami narzędzi funkcji Unity Catalog i narzędzi kodu agenta, zobacz przykłady narzędzi agenta

Ulepszanie wywoływania narzędzi przy użyciu dokumentacji

Jasna i szczegółowa dokumentacja pomaga agentom sztucznej inteligencji zrozumieć, kiedy i jak korzystać z podanych narzędzi. Podczas tworzenia narzędzi należy dokładnie udokumentować parametry narzędzi i wartości zwracane, aby upewnić się, że agent AI używa ich prawidłowo i we właściwym czasie.

W przypadku funkcji Unity Catalog użyj COMMENT, aby opisać narzędzie i parametry.

Przykład efektywnej dokumentacji narzędzi

W poniższym przykładzie pokazano efektywne ciągi COMMENT dla narzędzia funkcji w Unity Catalog, które wysyła zapytanie do tabeli ustrukturyzowanej.

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 specific customer including their email and ID.'
RETURN SELECT CONCAT(
    'Customer ID: ', customer_id, ', ',
    'Customer Email: ', customer_email
  )
  FROM main.default.customer_data
  WHERE customer_name = customer_name
  LIMIT 1;

Przykład nieskutecznej dokumentacji narzędzi

W poniższym przykładzie pokazano nieskuteczne ciągi COMMENT, które nie zawierają kluczowych informacji, takich jak zwracane wartości.

CREATE OR REPLACE FUNCTION main.default.lookup_customer_info(
  customer_name STRING COMMENT 'Name of the customer.'
)
RETURNS STRING
COMMENT 'Returns info about a customer.'
RETURN SELECT CONCAT(
    'Customer ID: ', customer_id, ', ',
    'Customer Email: ', customer_email
  )
  FROM main.default.customer_data
  WHERE customer_name = customer_name
  LIMIT 1;

Przykłady narzędzi Agent

Zobacz następujące artykuły, aby zapoznać się z przykładami narzędzi agenta:

• narzędzia interpretera kodu umożliwiają agentom wykonywanie dowolnego kodu, takiego jak Python.
• narzędzia do pobierania danych ustrukturyzowanych umożliwiają agentowi wykonywanie zapytań o ustrukturyzowane źródła danych, takie jak tabele SQL.
• Narzędzia do pobierania danych nieustrukturyzowanych umożliwiają agentowi wykonywanie zapytań o źródła danych nieustrukturyzowanych, takie jak korpus tekstowy, w celu wspomaganego generowania.

Dodaj narzędzia Unity Catalog do agentów

Po utworzeniu narzędzi katalogu Unity dodaj je do agenta. Agenci LangChain mogą wykorzystywać UCFunctionToolkit do integracji narzędzi UC.

Eksportowanie agentów wywołujących narzędzia z placu zabaw dla sztucznej inteligencji

AI Playground zapewnia wygodny sposób dodawania narzędzi katalogu Unity do modelu językowego LLM, testowania agenta i eksportowania jego kodu.

Aby wyeksportować agentów za pomocą narzędzia AI Playground, obszar roboczy musi spełniać następujące wymagania:

• Unity Catalog musi być włączony.
• Przetwarzanie bezserwerowe musi być włączone.
• Należy włączyć albo modele bazowe płatności za token albo modele zewnętrzne.

Aby wyeksportować kod agentów wywołujących narzędzia, wykonaj następujące czynności:

1. Na placu zabaw dla sztucznej inteligencji wybierz model z etykietą Tools, która ma włączoną funkcję.

   

2. Wybierz pozycję Narzędzia i kliknij pozycję Dodaj narzędzie.

3. W menu rozwijanym wybierz funkcję Wykazu aparatu Unity:

   

4. Użyj placu zabaw, aby porozmawiać i przetestować bieżącą kombinację llM, narzędzi i monitu systemowego. Wypróbuj odmiany, aby dowiedzieć się, jak działa bieżąca konfiguracja.

   

   Po dodaniu narzędzi wyeksportuj agenta do notesów języka Python:

5. Kliknij pozycję Eksportuj , aby wygenerować notesy języka Python, które definiują i wdrażają agenta.

   Po wyeksportowaniu kodu agenta zostaną wyświetlone trzy pliki zapisane w obszarze roboczym:

   • agent notes: zawiera kod języka Python definiujący agenta przy użyciu biblioteki LangChain.
   • driver notes: zawiera kod języka Python do rejestrowania, śledzenia, rejestrowania i wdrażania agenta sztucznej inteligencji przy użyciu struktury agenta Mozaika AI.
   • config.yml: zawiera informacje o konfiguracji agenta, w tym definicje narzędzi.

6. Otwórz notes, agent aby zobaczyć kod LangChain definiujący agenta. Ten notes służy do testowania i iterowania agenta programowo, na przykład definiowania większej liczby narzędzi.

7. Jeśli jesteś zadowolony z wyników agenta, uruchom notes driver, aby zarejestrować i wdrożyć agenta w punkcie końcowym świadczenia modelu.

Następne kroki

• Śledź agenta.