Create agents/base_agent.py

agents/base_agent.py

@@ -0,0 +1,41 @@
+# agents/base_agent.py
+
+from typing import Dict
+
+class ACPMessage:
+    def __init__(self, sender: str, receiver: str, performative: str, content: str):
+        self.sender = sender
+        self.receiver = receiver
+        self.performative = performative  # "inform", "request", etc.
+        self.content = content
+
+    def to_dict(self) -> Dict:
+        return {
+            "sender": self.sender,
+            "receiver": self.receiver,
+            "performative": self.performative,
+            "content": self.content
+        }
+
+    def __str__(self):
+        return f"[{self.performative.upper()}] {self.sender} → {self.receiver}: {self.content}"
+
+
+class BaseAgent:
+    def __init__(self, name: str, role: str):
+        self.name = name
+        self.role = role
+
+    def receive_message(self, message: ACPMessage) -> ACPMessage:
+        """
+        Override this method to define agent-specific reply logic.
+        """
+        raise NotImplementedError("You must override receive_message() in your agent.")
+
+    def create_message(self, receiver: str, performative: str, content: str) -> ACPMessage:
+        return ACPMessage(
+            sender=self.name,
+            receiver=receiver,
+            performative=performative,
+            content=content
+        )