Định nghĩa một hàm dừng chatbot
Giờ khi trạng thái của chatbot đã được thiết lập, bạn cần xây dựng các hàm để quản lý luồng công việc trong đồ thị của chatbot. Bắt đầu bằng cách xây dựng hàm should_continue() để kiểm tra lời gọi công cụ trong tin nhắn cuối cùng của chatbot bên trong state đã được nạp sẵn cho bạn. Nếu không có lời gọi công cụ, chatbot sẽ dừng lại. Nếu có công cụ được gọi, chatbot sẽ chuyển sang tác vụ tiếp theo. Để quản lý các tin nhắn, các mô-đun sau đã được nhập sẵn cho bạn.
from langgraph.graph import MessagesState, START, END
Bài tập này là một phần của khóa học
Thiết kế Hệ thống Agentic với LangChain
Hướng dẫn bài tập
- Chỉ định kiểu đầu vào cho tham số
statebằngMessagesState. - Truy cập tin nhắn cuối cùng từ
statebằng"messages"để kiểm tra lời gọi công cụ. - Kiểm tra xem tin nhắn cuối có chứa
tool_callshay không và chỉ định giá trị trả về là"tools"nếu có. - Chỉ định giá trị trả về là
ENDđể kết thúc hội thoại khi không cótool_calls.
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.
# Use MessagesState to define the state of the function
def should_continue(state: ____):
# Get the last message from the state
last_message = ____["____"][____]
# Check if the last message includes tool calls
if ____.____:
return "____"
# End the conversation if no tool calls are present
return ____