Membuat Agen dengan Alat Kustom

Agen AI dengan Hugging Face smolagents

Adel Nehme

VP of AI Curriculum, DataCamp

Manfaat Alat Kustom

Keandalan: Tulis dan uji logika secara eksplisit daripada mengandalkan agen untuk menebak
Dapat digunakan kembali: Gunakan alat di berbagai proyek dan agen
Konsistensi: Dapatkan perilaku yang dapat diprediksi di setiap pengulangan (bagus untuk debugging)
Akses terkontrol: Hanya buka yang Anda inginkan (file, API, basis data, dll.)

Skenario: Anda Menjalankan Toko Ritel

Data inventaris disimpan dalam file CSV (ukuran, warna, jumlah, dan harga)
Agen kode dapat menulis kode untuk membaca CSV
Namun, mereka tidak memiliki akses ke file secara default
Anda perlu membungkus akses file dalam alat kustom

Anatomi Alat Kustom

from smolagents import tool
import pandas as pd

@tool
def check_inventory(product_name: str) -> int:
    """
    Periksa jumlah produk yang tersedia dalam CSV inventaris.

    Args:
        product_name (str): Nama produk yang akan dicari.

    Returns:
        int: Jumlah dalam stok. Mengembalikan 0 jika produk tidak ditemukan.
    """
    df = pd.read_csv("store_inventory.csv")
    match = df[df["product_name"] == product_name]
    stock_quantity = int(match.iloc[0]["quantity"]) if not match.empty else 0
    return stock_quantity

Praktik Terbaik untuk Alat Kustom

Apakah kita memiliki kaos dalam stok?

Mendaftarkan Alat Kustom dengan Agen Anda

from smolagents import CodeAgent

agent = CodeAgent(
    tools=[check_inventory], # Tambahkan alat kustom
    model=InferenceClientModel(),
    additional_authorized_imports=["pandas"]  # Izinkan paket eksternal
)

agent.run("Apakah kita memiliki kaos besar dalam stok?")

Ya, kita memiliki 8 kaos besar dalam stok.

Alat Kustom dalam Proyek Produksi

Ayo berlatih!

Agen AI dengan Hugging Face smolagents