Update agents/base_agent.py

agents/base_agent.py

@@ -1,11 +1,12 @@
 # agents/base_agent.py
 
 class BaseAgent:
-    def __init__(self, name, role_description):
+    def __init__(self, name="Agent", role="Generic Assistant"):
         self.name = name
-        self.role = role_description
+        self.role = role
 
-    def generate_response(self, prompt, inference_fn):
-        system_prompt = f"You are {self.name}, a {self.role}. Respond uniquely and in your voice."
-        full_prompt = f"{system_prompt}\nUser: {prompt}\n{self.name}:"
-        return inference_fn(full_prompt)
+    def run(self, prompt, query_func):
+        """
+        Should be overridden by child classes.
+        """
+        raise NotImplementedError("Each agent must implement its own `run` method.")