ensure json output

tasks/create_article_pitch.py

@@ -1,7 +1,7 @@
 from crewai import Task
 from agents.curiosity_catalyst import curiosity_catalyst
 from tools.scrape_website import scrape_tool
-from utils.recommended_article import RecommendedArticle
+from utils.recommended_article import RecommendedArticles
 
 article_pitch_task = Task(
   description=(
@@ -14,7 +14,7 @@ article_pitch_task = Task(
   expected_output=(
     "Json array of all the artilces that have passed evaluation phase. Each article should have the following keys title, article_url, pitch and reason behind the recommedation"
   ),
-  # output_json=RecommendedArticle,
+  output_json=RecommendedArticles,
   tools=[scrape_tool],
   agent=curiosity_catalyst
 )

utils/recommended_article.py

@@ -1,7 +1,11 @@
 from pydantic import BaseModel
+from typing import List
 
 class RecommendedArticle(BaseModel):
   title: str
   url: str
   pitch: str
   reason_for_recommendation: str
+
+class RecommendedArticles(BaseModel):
+  articles: List[RecommendedArticle]