LangGraph ToolNode 사용 방법

---

• ToolNode 란
• Define Tools
• ToolNode 수동으로 호출하기

ToolNode 란

ToolNode 란 도구 호출을 위한 LangGraph 의 pre-built 된 함수이다. ToolNod 는 메시지 목록이 포함된 그래프 상태를 입력으로 받아 도구 호출 결과로 상태를 업데이트하는 LangChain Runnable 이다.

StateGraph 와 함께 사용하는 경우 적절한 리듀서가 있는 messages 키를 포함하고 있어야한다.

Define Tools

from langchain_core.messages import AIMessage
from langchain_core.tools import tool

from langgraph.prebuilt import ToolNode

@tool
def get_weather(location: str):
    """Call to get the current weather."""
    if location.lower() in ["sf", "san francisco"]:
        return "It's 60 degrees and foggy."
    else:
        return "It's 90 degrees and sunny."


@tool
def get_coolest_cities():
    """Get a list of coolest cities"""
    return "nyc, sf"

tools = [get_weather, get_coolest_cities]
tool_node = ToolNode(tools)

ToolNode 수동으로 호출하기

• ToolNode 는 메시지 목록을 포함한 그래프 상태에서 작동한다. 메시지 목록의 마지막 메시지는 AIMessage 여야 하며, tool_calls 매개변수를 포함하고 있어야 한다.

message_with_single_tool_call = AIMessage(
	content='',
	tool_calls=[
		{
			'name': 'get_weather',
			'args': {'location': 'sf'},
			'id': 'tool_call_id',
			'type': 'tool_call'
		}
	]
)

tool_node.invoke({'messages': [message_with_single_tool_call]})