Make the models work in parallel 4

crew.py

@@ -146,8 +146,6 @@ class GAIACrew():
 
 import concurrent.futures
 
-import concurrent.futures
-
 def run_parallel_crew(question: str, file_path: str):
     """
     1) Prepares the prompt (including file data if any).
@@ -172,16 +170,15 @@ def run_parallel_crew(question: str, file_path: str):
     agents = crew_instance.agents
 
     # Build (name, agent) pairs
-    pairs = list(zip(names, agents))
+    pairs   = list(zip(names, agents))
     workers = [agent for name, agent in pairs if name != "manager_agent"]
     manager = next(agent for name, agent in pairs if name == "manager_agent")
 
     # 3) Run workers in parallel
-    inputs = {"question": final_question}
     results = {}
     with concurrent.futures.ThreadPoolExecutor(max_workers=len(workers)) as pool:
         futures = {
-            pool.submit(lambda ag: ag.kickoff(inputs), ag): name
+            pool.submit(lambda ag: ag.kickoff({"question": final_question}), ag): name
             for name, ag in pairs if name != "manager_agent"
         }
         for fut in concurrent.futures.as_completed(futures):
@@ -202,11 +199,10 @@ def run_parallel_crew(question: str, file_path: str):
     )
 
     # 5) Run the manager for the final answer
-    final = manager.kickoff(inputs={"question": manager_prompt})
+    final = manager.kickoff({"question": manager_prompt})
     return get_final_answer(FINAL_ANSWER_MODEL, question, str(final))
 
 
-
 def get_final_answer(model, question, answer):
     prompt_template = """
         You are an expert question answering assistant. Given a question and an initial answer, your task is to provide the final answer.