ComenzarEmpieza gratis

Añade una herramienta a un grafo

Ya tienes lista tu herramienta de Wikipedia. ¡Ahora toca incorporarla al flujo del grafo de tu chatbot! Vas a modificar los nodos y aristas del grafo para integrar la herramienta como un nodo adicional. Configurarás el grafo para que el chatbot use la herramienta solo cuando sea necesario según la consulta del usuario. Para gestionar tu nodo de herramienta y sus aristas asociadas, se han importado por ti los siguientes módulos. La wikipedia_tool que construiste también está disponible en tu entorno. Una vez que finalice la construcción del grafo, se visualizará como un diagrama de LangGraph.

from langgraph.prebuilt import ToolNode, tools_condition

Este ejercicio forma parte del curso

Diseño de sistemas agénticos con LangChain

Ver curso

Instrucciones del ejercicio

  • Usa .add_node() para añadir el nodo "chatbot" al grafo, enlazándolo con la función chatbot.
  • Crea un ToolNode() con wikipedia_tool como tools y usa .add_node() para agregarlo al grafo como "tools".
  • Aplica .add_conditional_edges() para enrutar desde "chatbot" según tools_condition.
  • Usa .add_edge() para conectar "tools" de vuelta a "chatbot", START a "chatbot" y "chatbot" a END para completar el flujo de trabajo.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# Add chatbot node to the graph
graph_builder.____("____", ____)

# Create a ToolNode to handle tool calls and add it to the graph
tool_node = ____(tools=[____])
graph_builder.____("tools", ____)

# Set up a condition to direct from chatbot to tool or END node
graph_builder.____("____", ____)

# Connect tools back to chatbot and connect START and END nodes
graph_builder.add_edge("____", "____")
graph_builder.add_edge(____, "chatbot")
graph_builder.add_edge("chatbot", ____)

graph = graph_builder.compile()

try:
    display(Image(graph.get_graph().draw_mermaid_png()))

except Exception:
    print("Plot generation failed... falling back to cached asset.")
    display_fallback()
Editar y ejecutar código