Langgraph

[ThangQT2606]

tôi thực hiện đoạn mã langgraph như dưới đây nhưng kết quả nhận được thì không như mong đợi, mn có thể hỗ trợ tôi được không. Cảm ơn mọi người!
kết quả: Tôi không thể trả lời câu hỏi đó. Tôi không có khả năng thực hiện phép tính toán học.
đoạn mã của tôi:

llm = ChatGoogleGenerativeAI(
model="gemini-1.5-flash",
# model='gemini-2.0-flash-exp',
api_key=google_api_key,
temperature=0.2,
max_output_tokens=512,
convert_system_message_to_human=True
)

@tool
def add_numbers(input_text: str) -> str:
"""Use this tool to add two numbers."""
numbers = [int(s) for s in input_text.split() if s.isdigit()]
return sum(numbers)

from typing import Annotated
from typing_extensions import TypedDict

from langgraph.graph import StateGraph, START, END
from langgraph.graph.message import add_messages
from langgraph.checkpoint.memory import MemorySaver
from langgraph.prebuilt import ToolNode, tools_condition

class State(TypedDict):
# Messages have the type "list". The add_messages function
# in the annotation defines how this state key should be updated
# (in this case, it appends messages to the list, rather than overwriting them)
messages: Annotated[list, add_messages]

memory = MemorySaver()
graph_builder = StateGraph(State)
llm_with_tools = llm.bind_tools([add_numbers])

print(llm_with_tools.invoke("thư viện onnx runtime"))

def chatbot(state: State):
message = llm_with_tools.invoke(state["messages"])
# return {"messages": agent.invoke(state["messages"])['output']}
return {"messages": [message]}

graph_builder.add_node("chatbot", chatbot)

tool_node = ToolNode(tools=[add_numbers])
graph_builder.add_node("tools", tool_node)

graph_builder.add_conditional_edges(
"chatbot",
tools_condition,
{"tools": "tools", END: END},
)
graph_builder.add_edge(START, "chatbot")
graph_builder.add_edge("tools", "chatbot")

graph_builder.add_edge("chatbot", END)

graph = graph_builder.compile(checkpointer=memory)

dưới đây là ảnh đồ thị

user_input = "2 + 2 bằng mấy?"
config = {"configurable": {"thread_id": "1"}}

events = graph.stream(
{"messages": [{"role": "user", "content": user_input}]},
config,
stream_mode="values",
)

for event in events:
print("Event:", event)
if "messages" in event:
event["messages"][-1].pretty_print()