Update crew.py

crew.py

@@ -52,129 +52,6 @@ tracer_provider = register(
 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_search_tool],
-            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=[image_analysis_tool],
-            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=[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=[video_analysis_tool],
-            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=[youtube_analysis_tool],
-            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_analysis_tool],
-            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=[arithmetic_tool],
-            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_execution_tool],
-            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_generation_tool],
-            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(
-            config=self.tasks_config["manager_task"]
-        )
-
-    @crew
-    def crew(self) -> Crew:
-        return Crew(
-            agents=self.agents,
-            tasks=self.tasks,
-            verbose=True
-        )
 
     @tool("Web Search Tool")
     def web_search_tool(question: str) -> str:
@@ -438,6 +315,129 @@ class GAIACrew():
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
+    @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_search_tool],
+            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=[image_analysis_tool],
+            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=[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=[video_analysis_tool],
+            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=[youtube_analysis_tool],
+            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_analysis_tool],
+            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=[arithmetic_tool],
+            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_execution_tool],
+            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_generation_tool],
+            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(
+            config=self.tasks_config["manager_task"]
+        )
+
+    @crew
+    def crew(self) -> Crew:
+        return Crew(
+            agents=self.agents,
+            tasks=self.tasks,
+            verbose=True
+        )
+
 def run_crew(question, file_path):
     final_question = question