Sachin-Hansaka commited on
Commit
939829a
·
verified ·
1 Parent(s): 74a0ec1
src/langgraphagenticai/graph/graph_builder.py CHANGED
@@ -94,14 +94,13 @@ class GraphBuilder:
94
 
95
  self.graph_builder.add_node("fetch_news",ai_news_node.fetch_news)
96
  self.graph_builder.add_node("summarize_news",ai_news_node.summarize_news)
97
- self.graph_builder.add_node("save_result",ai_news_node.save_result)
98
 
99
  #added the edges
100
 
101
  self.graph_builder.set_entry_point("fetch_news")
102
  self.graph_builder.add_edge("fetch_news","summarize_news")
103
- self.graph_builder.add_edge("summarize_news","save_result")
104
- self.graph_builder.add_edge("save_result", END)
105
 
106
  def setup_graph(self, usecase: str):
107
  """
 
94
 
95
  self.graph_builder.add_node("fetch_news",ai_news_node.fetch_news)
96
  self.graph_builder.add_node("summarize_news",ai_news_node.summarize_news)
97
+ # self.graph_builder.add_node("save_result",ai_news_node.save_result)
98
 
99
  #added the edges
100
 
101
  self.graph_builder.set_entry_point("fetch_news")
102
  self.graph_builder.add_edge("fetch_news","summarize_news")
103
+ self.graph_builder.add_edge("summarize_news",END)
 
104
 
105
  def setup_graph(self, usecase: str):
106
  """