Create agents/base_agent.py

agents/base_agent.py

@@ -0,0 +1,11 @@
+# agents/base_agent.py
+
+class BaseAgent:
+    def __init__(self, name, role_description):
+        self.name = name
+        self.role = role_description
+
+    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)