Update app.py

app.py

@@ -4,17 +4,26 @@ import streamlit as st
 from crewai import Agent, Crew, Process, Task
 from crewai_tools import SerperDevTool
 from langchain_google_genai import ChatGoogleGenerativeAI
-
+from crewai import LLM
 # Assurez-vous que vos clés API sont définies dans les variables d'environnement
 os.environ["GEMINI_API_KEY"] = "AIzaSyD6yZxfVOnh63GXBJjakAupk9aP4CZrgrQ"
 os.environ["SERPER_API_KEY"] = "9b90a274d9e704ff5b21c0367f9ae1161779b573"
 
+
+llm = LLM(
+    model="gemini/gemini-1.5-flash",
+    temperature=0.7,
+    timeout=120,
+    max_tokens=8000,
+)
+
+
 # Définition des agents
 project_manager = Agent(
     role="Chef de Projet",
     goal="Coordonner les autres agents pour assurer la cohérence et la qualité de l'exposé.",
     backstory="Un chef de projet expérimenté avec une expertise dans la gestion d'équipes et la coordination de projets complexes.",
-    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GEMINI_API_KEY"]),
+    llm=llm,
     verbose=True,
     allow_delegation=True,
 )
@@ -23,7 +32,7 @@ topic_analyst = Agent(
     role="Analyste de Thème",
     goal="Analyser le thème de l'exposé et générer un plan détaillé.",
     backstory="Un expert en recherche et en structuration de contenu, capable de décomposer des sujets complexes en plans clairs et concis.",
-    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GEMINI_API_KEY"]),
+    llm=llm,
     verbose=True,
     tools=[SerperDevTool()],
 )
@@ -32,7 +41,7 @@ content_writer = Agent(
     role="Rédacteur de Contenu",
     goal="Rédiger les sections de l'exposé en se basant sur le plan et les recherches.",
     backstory="Un rédacteur spécialisé dans la création de contenu clair, concis et engageant.",
-    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GEMINI_API_KEY"]),
+    llm=llm,
     verbose=True,
     tools=[SerperDevTool()],
 )
@@ -41,7 +50,7 @@ editor = Agent(
     role="Éditeur/Réviseur",
     goal="Réviser et peaufiner le contenu, assurer la cohérence du style et du ton, corriger les erreurs.",
     backstory="Un éditeur expérimenté avec un souci du détail et une excellente maîtrise de la langue.",
-    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GEMINI_API_KEY"]),
+    llm=llm,
     verbose=True,
 )
 
@@ -80,7 +89,7 @@ if st.button("Générer l'exposé"):
             agents=[project_manager, topic_analyst, content_writer, editor],
             tasks=assign_tasks(topic),
             process=Process.hierarchical,
-            manager_llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GEMINI_API_KEY"]),
+            manager_llm=llm,
             verbose=True,
         )