wip: refactor query pipeline agent to use stateful function components

[jerryjliu]

this should be a cleaner UX than the old guide which made users use AgentInputComponent and AgentFnComponent to define an agentic query pipeline.

Here you can define a query pipeline through a single abstraction StatefulFnComponent:

• no more two separate component classes (which didn't make sense)
• single state dictionary passed around (instead of both task and step state)
• use same abstractions as LoopComponent

When you wrap a query pipeline with an agent worker, the task and step.step_state are injected in the state dictionary of every stateful component.

This also makes it easier to test the query pipeline used for an agent in isolation. Just initialize the state dictionary of the query pipeline with task and step.step_state, and do pipeline.run()

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB