Modularised the agents 6

crew.py

@@ -91,6 +91,15 @@ def get_tools_for(agent_name: str):
 class GAIACrew():
     tasks: List[Task]
 
+    def __init__(self):
+        self.agents_config = self._load_yaml("config/agents.yaml")
+        self.tasks_config = self._load_yaml("config/tasks.yaml")
+
+    def _load_yaml(self, path):
+        import yaml
+        with open(path, "r") as f:
+            return yaml.safe_load(f)
+
     @property
     def agents(self) -> List[Agent]:
         return [
@@ -102,27 +111,9 @@ class GAIACrew():
                 tools=get_tools_for(name),
                 verbose=True
             )
-            for name in self.agents_config.keys()
+            for name in self.agents_config
         ]
 
-    @task
-    def manager_task(self) -> Task:
-        task = Task(
-            config=self.tasks_config["manager_task"]
-        )
-        task.agent = next(agent for agent in self.agents if agent.config["name"] == "manager_agent")
-        return task
-
-
-
-    @crew
-    def crew(self) -> Crew:
-        return Crew(
-            agents=self.agents,
-            tasks=[self.manager_task()],
-            verbose=True
-        )
-
 
 def run_crew(question, file_path):
     final_question = question