Tạo các công cụ tùy chỉnh
Trong khi LangChain cung cấp một tập sâu rộng các công cụ tích hợp, thường có các kịch bản nơi bạn cần tạo các công
cụ tùy chỉnh để đáp ứng các yêu cầu cụ thể. Tạo các công cụ tùy chỉnh trong LangChain là một quá trình theo đường
thẳng cái đòi hỏi định nghĩa một hàm và gói nó với tool decorator phù hợp.
Trong ảnh dưới, hãy xem làm cách nào chúng ta có thể tạo một công cụ tùy chỉnh từng bước một.
1. Định nghĩa hàm công cụ: Tạo một hàm Python cái thực hiện hoạt động mong muốn. Hàm này nên lấy các tham số đầu
vào và trả về một kết quả.
2. Thêm các type hints: Sử dụng Python type hints để chỉ ra các dạng đầu vào và đầu ra của hàm của bạn. Cái này
giúp LangChain hiểu làm cách nào sử dụng công cụ.
3. Sử dụng @tool Decorator: Nhập khẩu @tool decorator từ LangChain và áp dụng nó vào hàm của bạn. Decorator này
chuyển dạng hàm của bạn vào một công cụ LangChain.
4. Cung cấp metadata: Trong decorator, chỉ ra tên và mô tả của công cụ của bạn. Thông tin này được sử dụng bởi
LangChain để quyết định khi nào và như thế nào sử dụng công cụ.
Sau đây là một ví dụ về tạo một công cụ tùy chỉnh cái tính toán square của một số:
from langchain.tools import tool
from typing import Union
@tool(name=”SquareCalculator”, description=”Calculates the square of a given number”)
def calculate_square(number: Union[int, float]) -> Union[int, float]:“””
Calculate the square of a given number.
Args:
number (Union[int, float]): The number to be squared.
Returns:
Union[int, float]: The square of the input number.
“””
return number**2output:
100.0
Code này định nghĩa một công cụ LangChain gọi là SquareCalculator cái tính toán square của một số, sử dụng các type
hints để hỗ trợ cả các đầu vào số nguyên và số dấu phẩy động, và cung cấp một docstring giải thích chức năng và các
tham sô của nó.