feat: implement AITBC mesh network deployment infrastructure
✅ 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
This commit is contained in:
@@ -0,0 +1,113 @@
|
||||
"""
|
||||
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
|
||||
Reference in New Issue
Block a user