Tags and Metadata for Nodes

[kevinqmcdonald]

Is there any way to add tags and metadata to individual nodes in a graph? I want to be able to filter events returned by the streamEvents() method by tag, but I can't find a good way to tag a specific node (and all its children). I tried filtering by name, but that doesn't stream events for the node's child runs.

To illustrate what I'm trying to do, here is a screenshot of my recent run in Langsmith:

In this trace, I want to filter the events streamed back so that I only receive events from the worker_loop node. I tried filtering by name, but I don't get any of the child runs (RunnableLambda, ChannelWrite, etc.). I can't seem to find a way to tag the node so that I can filter by tags, other than this hacky way that I came up with below:

const builder = new StateGraph<GraphState>({
  channels: graphStateChannels
})
  .addNode(PLANNER_NODE_NAME, planner)
  .addNode(WORKER_LOOP_NODE_NAME, (state: GraphState, config?: GraphConfig) => {
    const node = new RunnableLambda({
      func: workerLoop
    }).withConfig({
      ...config,
      tags: [WORKER_LOOP_NODE_NAME],
    });
    return node.invoke(state);
  })
  .addNode(WAIT_NODE_NAME, wait)
  .addNode(SOLVER_NODE_NAME, solver)
  .addEdge(START, PLANNER_NODE_NAME)
  .addEdge(PLANNER_NODE_NAME, WORKER_LOOP_NODE_NAME)
  .addConditionalEdges(WORKER_LOOP_NODE_NAME, (state) =>
    state.paused ? WAIT_NODE_NAME : SOLVER_NODE_NAME
  )
  .addEdge(WAIT_NODE_NAME, WORKER_LOOP_NODE_NAME)
  .addEdge(SOLVER_NODE_NAME, END);

const checkpointer = new MemorySaver();

const plannerGraph = builder.compile({
  checkpointer,
  interruptBefore: [WAIT_NODE_NAME]
});

Does anyone know of a better way to do this?

[jacoblee93]

Hey @kevinqmcdonald, will add this!

[kevinqmcdonald]

Hi @jacoblee93, any updates on this item?