Update agents/base_agent.py

agents/base_agent.py

@@ -1,12 +1,19 @@
-# agents/base_agent.py
+# base_agent.py
 
-class BaseAgent:
-    def __init__(self, name="Agent", role="Generic Assistant"):
+from abc import ABC, abstractmethod
+from multi_inference import multi_query
+
+class ACPMessage:
+    def __init__(self, role, content):
+        self.role = role
+        self.content = content
+
+class BaseAgent(ABC):
+    def __init__(self, name, role):
         self.name = name
         self.role = role
+        self.memory = []
 
-    def run(self, prompt, query_func):
-        """
-        Should be overridden by child classes.
-        """
-        raise NotImplementedError("Each agent must implement its own `run` method.")
+    @abstractmethod
+    def generate(self, messages):
+        pass