Agent async iteration skips userprompt node?

[diego898]

Question

In the async iteration of an agent's graph [docs] why dont we actually get to the first UserPromptNode?

When we use async iteration directly, seems we skip the first node:

 print(nodes)
    """
    [
        ModelRequestNode(

When we use next manually, [docs] we do see the entire agent graph.:

print(all_nodes)
        """
        [
            UserPromptNode(

[diego898]

Just to clarify - this isnt jsut an issue with the docs - running the code also gives this

[dmontagu]

Hi @diego898, thanks for reporting. The issue is that when we do iteration, we always yield the next node. But we don't start by yielding the first one, we grab the next one and yield it. I think we can tweak it so that it yields the first one, just need to figure out how to tweak it to make that happen. Should be an easy fix.

[diego898]

Thanks @dmontagu - figured it was something like that.