🔥 refactor(chat.py): remove unnecessary code for building the root node in the graph

The code for building the root node in the graph has been removed as it is no longer necessary. Previously, the root node was built separately due to certain nodes requiring parameters that were not connected to it. However, this approach has been reconsidered and the tools are now connected to the ZeroShotPrompt, eliminating the need for separate building of the root node.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-08 11:40:56 -03:00
commit 72b4a2d943

View file

@ -115,13 +115,6 @@ async def stream_build(flow_id: str):
number_of_nodes = len(graph.nodes)
flow_data_store[flow_id]["status"] = BuildStatus.IN_PROGRESS
# To deal with the ZeroShotAgent case
# we need to build the root node first
# and then the rest of the graph
# This is a big problem because certain nodes require
# params that are not connected to it.
# We should consider connecting the tools to the ZeroShotPrompt
graph.build()
for i, vertex in enumerate(graph.generator_build(), 1):
try: