Upload mythos/message.py

mythos/message.py

@@ -0,0 +1,48 @@
+"""Message passing system for Mythos agents."""
+
+from __future__ import annotations
+
+import uuid
+from datetime import datetime, timezone
+from enum import Enum
+from typing import Any, Optional
+
+from pydantic import BaseModel, Field
+
+
+class MessageType(str, Enum):
+    """Types of messages agents can exchange."""
+
+    TASK = "task"           # A task to be performed
+    RESPONSE = "response"   # Result of a task
+    QUERY = "query"         # Information request
+    BROADCAST = "broadcast" # Message to all agents
+    SYSTEM = "system"       # System-level message
+    MEMORY = "memory"       # Memory update/retrieval
+
+
+class Message(BaseModel):
+    """A typed message exchanged between agents in a Pantheon."""
+
+    id: str = Field(default_factory=lambda: str(uuid.uuid4()))
+    type: MessageType
+    sender: str                           # Agent name or "system"
+    recipient: str                        # Agent name or "broadcast"
+    content: str
+    metadata: dict[str, Any] = Field(default_factory=dict)
+    timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    parent_id: Optional[str] = None       # For threading/replies
+    priority: int = 5                     # 1 (highest) to 10 (lowest)
+
+    def reply(self, content: str, msg_type: MessageType = MessageType.RESPONSE) -> Message:
+        """Create a reply message."""
+        return Message(
+            type=msg_type,
+            sender="",  # Will be set by the agent
+            recipient=self.sender,
+            content=content,
+            parent_id=self.id,
+        )
+
+    def __str__(self) -> str:
+        return f"[{self.type.value.upper()}] {self.sender} -> {self.recipient}: {self.content[:80]}"