final fixed code

src/langgraphagenticai/nodes/ai_news_node.py

@@ -8,8 +8,6 @@ class AINewsNode:
         """
         self.tavily = TavilyClient()
         self.llm = llm
-        # this is used to capture various steps in this file so that later can be use for steps shown
-        self.state = {}
 
     def fetch_news(self, state: dict) -> dict:
         """
@@ -20,7 +18,6 @@ class AINewsNode:
             dict: Updated state with 'news_data' key containing fetched news.
         """
         frequency = state['messages'][0].content.lower()
-        self.state['frequency'] = frequency
         
         time_range_map = {'daily': 'd', 'weekly': 'w', 'monthly': 'm', 'year': 'y'}
         days_map = {'daily': 1, 'weekly': 7, 'monthly': 30, 'year': 366}
@@ -35,7 +32,8 @@ class AINewsNode:
         )
         
         state['news_data'] = response.get('results', [])
-        self.state['news_data'] = state['news_data']
+        state['frequency'] = frequency
+        
         return state
 
     def summarize_news(self, state: dict) -> dict:
@@ -46,7 +44,8 @@ class AINewsNode:
         Returns:
             dict: Updated state with 'summary' key containing the summarized news.
         """
-        news_items = self.state['news_data']
+        news_items = state.get('news_data', [])
+        frequency = state.get('frequency', 'daily')
         
         prompt_template = ChatPromptTemplate.from_messages([
             ("system", """Summarize AI news articles into markdown format. For each item include:
@@ -69,11 +68,10 @@ class AINewsNode:
         response = self.llm.invoke(prompt_template.format(articles=articles_str))
         
         # Create formatted summary with header
-        frequency = self.state['frequency']
         formatted_summary = f"# {frequency.capitalize()} AI News Summary\n\n{response.content}"
         
         state['summary'] = formatted_summary
-        self.state['summary'] = formatted_summary
         
+        # CRITICAL FIX: Return state, not self.state
         return state