v2.5.32

phidatahq · Nov 14, 2024 · ef5f314 · ef5f314
1 parent 51866c6
commit ef5f314
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cookbook/workflows/news_article_streaming.py b/cookbook/workflows/news_article_streaming.py
@@ -202,11 +202,11 @@ def run(
                 and article_scraper_response.content
                 and isinstance(article_scraper_response.content, ScrapedArticle)
             ):
-                scraped_articles[article_scraper_response.content.url] = article_scraper_response.content.model_dump()
+                scraped_articles[article_scraper_response.content.url] = article_scraper_response.content
                 logger.info(f"Scraped article: {article_scraper_response.content.url}")
 
         # 2.3: Save the scraped_articles in the session state
-        self.session_state["scraped_articles"] = {k: v for k, v in scraped_articles.items()}
+        self.session_state["scraped_articles"] = {k: v.model_dump() for k, v in scraped_articles.items()}
 
         ####################################################
         # Step 3: Write a report