Update crew_utils.py

crew_utils.py

@@ -1,9 +1,20 @@
 # crew_utils.py
+import asyncio
+from typing import Any, List
+
 from crewai import Agent, Crew, Process, Task
 from crewai_tools import SerperDevTool
 from crewai import LLM
 from pydantic import BaseModel, Field
-from typing import List,Any
+
+# --- Installation des dépendances (si nécessaire) ---
+# Assure-toi que les packages nécessaires sont installés.
+# Tu peux exécuter `crewai install` si tu as utilisé le CLI de CrewAI pour initialiser le projet.
+# Sinon, installe-les manuellement avec :
+# pip install crewai crewai-tools langchain-google-genai reportlab
+
+# --- Outil de génération de PDF (exemple avec reportlab) ---
+# Crée un fichier `pdf_tool.py` dans le même répertoire que ce script, par exemple.
 from pdf_tool import PDFTool
 
 # --- Définition des outils ---
@@ -12,8 +23,6 @@ pdf_tool = PDFTool()
 
 # --- Définition du LLM (Gemini) ---
 # Remplace par ta clé API
-
-
 GEMINI_API_KEY = "AIzaSyD6yZxfVOnh63GXBJjakAupk9aP4CZrgrQ"
 llm = LLM(
     model="gemini/gemini-1.5-flash",
@@ -100,7 +109,7 @@ class SectionContent(BaseModel):
 
 # Tâches pour les Rédacteurs (exemple pour 3 sections, à adapter)
 async def write_section_task(
-        researcher: Agent, section_title: str, context: List[Any]
+        researcher: Agent, section_title: str, context: List[Task]
     ) -> Task:
     return Task(
         description=f"""
@@ -159,7 +168,7 @@ class ExposeCrew(Crew):
     async def kickoff(self, inputs: dict = {}) -> str:
         # Exécuter la tâche de création du plan
         plan = await create_plan_task.execute(context=inputs)
-        print(f"Plan de l'exposé : {plan}")
+        self.log.info(f"Plan de l'exposé : {plan}")
 
         # Créer et exécuter les tâches de rédaction de section en parallèle
         section_tasks = await process_section_tasks(
@@ -167,10 +176,11 @@ class ExposeCrew(Crew):
             plan=plan,
             researchers=[researcher_intro, researcher_dev, researcher_conclusion],
         )
+        
+        sections_results = await asyncio.gather(*[task.execute(context=[create_plan_task]) for task in section_tasks])
+        section_outputs = [result.content for result in sections_results]
 
-        section_outputs = await asyncio.gather(*[task.execute() for task in section_tasks])
-
-        print(f"Sections rédigées : {section_outputs}")
+        self.log.info(f"Sections rédigées : {section_outputs}")
 
         # Mettre à jour la description de la tâche de compilation avec les sections rédigées
         compile_report_task.description = f"""
@@ -184,5 +194,5 @@ class ExposeCrew(Crew):
         compile_report_task.context = section_tasks
 
         # Exécuter la tâche de compilation
-        result = await compile_report_task.execute(context=section_outputs)
+        result = await compile_report_task.execute()
         return result