Make the models work in parallel 3

crew.py

@@ -146,6 +146,8 @@ 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).
@@ -164,19 +166,23 @@ def run_parallel_crew(question: str, file_path: str):
         else:
             final_question = f"{question} File path: {file_path}."
 
-    # 2) Instantiate your crew and split manager vs workers
+    # 2) Instantiate your crew and split manager vs workers by zipping names → agents
     crew_instance = GAIACrew()
-    all_agents    = crew_instance.agents
-    workers       = [a for a in all_agents if a.config.get("name") != "manager_agent"]
-    manager       = next(a for a in all_agents if a.config.get("name") == "manager_agent")
+    names  = list(crew_instance.agents_config.keys())
+    agents = crew_instance.agents
+
+    # Build (name, agent) pairs
+    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): ag.config["name"]
-            for ag in workers
+            pool.submit(lambda ag: ag.kickoff(inputs), ag): name
+            for name, ag in pairs if name != "manager_agent"
         }
         for fut in concurrent.futures.as_completed(futures):
             name = futures[fut]
@@ -186,8 +192,8 @@ def run_parallel_crew(question: str, file_path: str):
                 results[name] = f"<error: {e}>"
 
     # 4) Compose a manager prompt that includes all worker outputs
-    combined = "\n\n".join(f"--- {name} output ---\n{out}"
-                           for name, out in results.items())
+    combined = "\n\n".join(f"--- {n} output ---\n{out}"
+                           for n, out in results.items())
     manager_prompt = (
         f"You have received these reports from your coworkers:\n\n"
         f"{combined}\n\n"
@@ -195,11 +201,12 @@ def run_parallel_crew(question: str, file_path: str):
         f"Original question: {question}"
     )
 
-    # 5) Run the manager agent for the final answer
+    # 5) Run the manager for the final answer
     final = manager.kickoff(inputs={"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.