Modularised the agents

crew.py

@@ -67,157 +67,44 @@ CODE_TOOLS = [
     AITools.code_execution_tool
 ]
 
+#Get specific tools
+def get_tools_for(agent_name: str):
+    if "document" in agent_name or "translation" in agent_name or "summarization" in agent_name:
+        return DOCUMENT_TOOLS
+    elif any(keyword in agent_name for keyword in ["image", "audio", "video", "youtube"]):
+        return MEDIA_TOOLS
+    elif "web_search" in agent_name or "web_browser" in agent_name:
+        return WEB_TOOLS
+    elif "code_generation" in agent_name or "code_execution" in agent_name:
+        return CODE_TOOLS
+    elif "arithmetic" in agent_name:
+        return ARITHMETIC_TOOLS
+    elif "manager" in agent_name:
+        return []
+    else:
+        return []
 
 
 #CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 
 @CrewBase
 class GAIACrew():
-    agents: List[BaseAgent]
     tasks: List[Task]
 
-    @agent
-    def web_search_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["web_search_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=WEB_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def web_browser_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["web_browser_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=3,
-            tools=WEB_TOOLS,
-            verbose=True
-        )
-    
-    @agent
-    def image_analysis_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["image_analysis_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=MEDIA_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def audio_analysis_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["audio_analysis_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=[AITools.audio_analysis_tool],
-            verbose=True
-        )
-
-    @agent
-    def video_analysis_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["video_analysis_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=MEDIA_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def youtube_analysis_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["youtube_analysis_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=MEDIA_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def document_analysis_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["document_analysis_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=DOCUMENT_TOOLS,
-            verbose=True
-        )
+    @property
+    def agents(self) -> List[Agent]:
+        return [
+            Agent(
+                config=self.agents_config[name],
+                allow_delegation=("manager" in name),
+                llm=MANAGER_MODEL if "manager" in name else AGENT_MODEL,
+                max_iter=5 if "manager" in name else 2,
+                tools=get_tools_for(name),
+                verbose=True
+            )
+            for name in self.agents_config.keys()
+        ]
 
-    @agent
-    def arithmetic_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["document_analysis_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=ARITHMETIC_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def code_generation_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["code_generation_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=3,
-            tools=CODE_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def code_execution_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["code_execution_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=3,
-            tools=CODE_TOOLS,
-            verbose=True
-        )
-    
-    @agent
-    def translation_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["translation_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=DOCUMENT_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def summarization_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["summarization_agent"],
-            allow_delegation=False,
-            llm=AGENT_MODEL,
-            max_iter=2,
-            tools=DOCUMENT_TOOLS,
-            verbose=True
-        )
-
-    @agent
-    def manager_agent(self) -> Agent:
-        return Agent(
-            config=self.agents_config["manager_agent"],
-            allow_delegation=True,
-            llm=MANAGER_MODEL,
-            max_iter=5,
-            verbose=True
-        )
-    
     @task
     def manager_task(self) -> Task:
         return Task(
@@ -228,10 +115,11 @@ class GAIACrew():
     def crew(self) -> Crew:
         return Crew(
             agents=self.agents,
-            tasks=self.tasks,
+            tasks=[self.manager_task()],
             verbose=True
         )
 
+
 def run_crew(question, file_path):
     final_question = question