Tạo công cụ tùy chỉnh
Giờ bạn đã có một hàm để trích xuất dữ liệu khách hàng từ DataFrame customers, hãy chuyển hàm này thành một công cụ tương thích với các agent của LangChain.
Bài tập này là một phần của khóa học
Phát triển ứng dụng LLM với LangChain
Hướng dẫn bài tập
- Thêm bộ trang trí
@tooltrước định nghĩa hàm để chuyển nó thành một công cụ của LangChain. - In ra các đối số của công cụ bằng thuộc tính
.argstrên công cụ (ví dụ:tool_name.args).
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Convert the retrieve_customer_info function into a tool
____
def retrieve_customer_info(name: str) -> str:
"""Retrieve customer information based on their name."""
customer_info = customers[customers['name'] == name]
return customer_info.to_string()
# Print the tool's arguments
print(retrieve_customer_info.____)