c876b0aa20
✅ Phase 0: Pre-implementation checklist completed - Environment configurations (dev/staging/production) - Directory structure setup (logs, backups, monitoring) - Virtual environment with dependencies ✅ Master deployment script created - Single command deployment with validation - Progress tracking and rollback capability - Health checks and deployment reporting ✅ Validation script created - Module import validation - Basic functionality testing - Configuration and script verification ✅ Implementation fixes - Fixed dataclass import in consensus keys - Fixed async function syntax in tests - Updated deployment script for virtual environment 🚀 Ready for deployment: ./scripts/deploy-mesh-network.sh dev
114 lines
3.6 KiB
Python
114 lines
3.6 KiB
Python
"""
|
|
Message Protocol for AITBC Agents
|
|
Handles message creation, routing, and delivery between agents
|
|
"""
|
|
|
|
import json
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Dict, Any, Optional, List
|
|
from enum import Enum
|
|
|
|
class MessageTypes(Enum):
|
|
"""Message type enumeration"""
|
|
TASK_REQUEST = "task_request"
|
|
TASK_RESPONSE = "task_response"
|
|
HEARTBEAT = "heartbeat"
|
|
STATUS_UPDATE = "status_update"
|
|
ERROR = "error"
|
|
DATA = "data"
|
|
|
|
class MessageProtocol:
|
|
"""Message protocol handler for agent communication"""
|
|
|
|
def __init__(self):
|
|
self.messages = []
|
|
self.message_handlers = {}
|
|
|
|
def create_message(
|
|
self,
|
|
sender_id: str,
|
|
receiver_id: str,
|
|
message_type: MessageTypes,
|
|
content: Dict[str, Any],
|
|
message_id: Optional[str] = None
|
|
) -> Dict[str, Any]:
|
|
"""Create a new message"""
|
|
if message_id is None:
|
|
message_id = str(uuid.uuid4())
|
|
|
|
message = {
|
|
"message_id": message_id,
|
|
"sender_id": sender_id,
|
|
"receiver_id": receiver_id,
|
|
"message_type": message_type.value,
|
|
"content": content,
|
|
"timestamp": datetime.utcnow().isoformat(),
|
|
"status": "pending"
|
|
}
|
|
|
|
self.messages.append(message)
|
|
return message
|
|
|
|
def send_message(self, message: Dict[str, Any]) -> bool:
|
|
"""Send a message to the receiver"""
|
|
try:
|
|
message["status"] = "sent"
|
|
message["sent_timestamp"] = datetime.utcnow().isoformat()
|
|
return True
|
|
except Exception:
|
|
message["status"] = "failed"
|
|
return False
|
|
|
|
def receive_message(self, message_id: str) -> Optional[Dict[str, Any]]:
|
|
"""Receive and process a message"""
|
|
for message in self.messages:
|
|
if message["message_id"] == message_id:
|
|
message["status"] = "received"
|
|
message["received_timestamp"] = datetime.utcnow().isoformat()
|
|
return message
|
|
return None
|
|
|
|
def get_messages_by_agent(self, agent_id: str) -> List[Dict[str, Any]]:
|
|
"""Get all messages for a specific agent"""
|
|
return [
|
|
msg for msg in self.messages
|
|
if msg["sender_id"] == agent_id or msg["receiver_id"] == agent_id
|
|
]
|
|
|
|
class AgentMessageClient:
|
|
"""Client for agent message communication"""
|
|
|
|
def __init__(self, agent_id: str, protocol: MessageProtocol):
|
|
self.agent_id = agent_id
|
|
self.protocol = protocol
|
|
self.received_messages = []
|
|
|
|
def send_message(
|
|
self,
|
|
receiver_id: str,
|
|
message_type: MessageTypes,
|
|
content: Dict[str, Any]
|
|
) -> Dict[str, Any]:
|
|
"""Send a message to another agent"""
|
|
message = self.protocol.create_message(
|
|
sender_id=self.agent_id,
|
|
receiver_id=receiver_id,
|
|
message_type=message_type,
|
|
content=content
|
|
)
|
|
self.protocol.send_message(message)
|
|
return message
|
|
|
|
def receive_messages(self) -> List[Dict[str, Any]]:
|
|
"""Receive all pending messages for this agent"""
|
|
messages = []
|
|
for message in self.protocol.messages:
|
|
if (message["receiver_id"] == self.agent_id and
|
|
message["status"] == "sent" and
|
|
message not in self.received_messages):
|
|
self.protocol.receive_message(message["message_id"])
|
|
self.received_messages.append(message)
|
|
messages.append(message)
|
|
return messages
|