mirror of
https://github.com/blackboxprogramming/BlackRoad-Operating-System.git
synced 2026-03-17 07:57:19 -05:00
Add LEITL Protocol - Live Everyone In The Loop multi-agent collaboration
This commit introduces the LEITL (Live Everyone In The Loop) protocol system, enabling multiple AI agents to collaborate in real-time with shared WebDAV context. ## What was built: ### Backend Infrastructure: - **WebDAV Context Manager** (`backend/app/services/webdav_context.py`) - Sync files from WebDAV servers - Keyword matching and relevance scoring - Redis caching for performance - Support for multiple file types (md, txt, py, json, etc.) - **LEITL Protocol Service** (`backend/app/services/leitl_protocol.py`) - Session registration and management - Heartbeat monitoring with auto-cleanup - Message broadcasting via Redis PubSub - Activity logging and history - WebSocket connection management - **LEITL API Router** (`backend/app/routers/leitl.py`) - Session management endpoints (register, heartbeat, end) - WebSocket endpoint for real-time events - Message broadcasting endpoints - WebDAV context sync endpoint - Quick-start endpoint for easy activation - Full OpenAPI documentation ### Frontend Dashboard: - **LEITL Dashboard App** (`backend/static/js/apps/leitl.js`) - Real-time session monitoring - Live activity feed - Recent message display - WebSocket integration - Quick-start interface - Auto-refresh capabilities - **Desktop Integration** (`backend/static/index.html`) - Added LEITL icon to desktop - Added LEITL to Start menu - Window management integration - Taskbar support ### Documentation: - **Protocol Specification** (`docs/LEITL_PROTOCOL.md`) - Complete architecture overview - API documentation - WebSocket protocol details - Security considerations - Event types and schemas - **Usage Guide** (`docs/LEITL_USAGE_GUIDE.md`) - Quick-start prompts for AI assistants - Dashboard usage instructions - API examples - Troubleshooting guide - Multi-agent collaboration examples ## Key Features: ✅ Multi-agent live collaboration ✅ Shared WebDAV context across sessions ✅ Real-time event broadcasting via WebSocket ✅ Session health monitoring with heartbeat ✅ Auto-cleanup of dead sessions ✅ Redis-backed message queue ✅ Beautiful Windows 95-styled dashboard ✅ Full API documentation ✅ Security with JWT auth and rate limiting ## Usage: AI assistants can activate LEITL with simple prompts like: - "Turn on LEITL. Enable WebDAV context." - "Start LEITL session. Pull from WebDAV: <url>" - "LEITL mode ON 🔥" Dashboard access: http://localhost:8000 → 🔥 LEITL icon ## Answers Alexa's Challenge: This implementation answers the challenge to enable "collaboration between multiple AI states for LEITL (Live Everyone In The Loop)" with full communication capabilities and shared context management. 🎁 Prize unlocked: Multi-agent swarm collaboration! 🐝✨
This commit is contained in:
Claude
341
backend/app/routers/leitl.py
Normal file
341
backend/app/routers/leitl.py
Normal file
@@ -0,0 +1,341 @@
|
||||
"""LEITL Protocol API Router - Live Everyone In The Loop"""
|
||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect, Depends, HTTPException, Query
|
||||
from fastapi.responses import JSONResponse
|
||||
from pydantic import BaseModel, Field
|
||||
from typing import List, Optional
|
||||
from datetime import datetime
|
||||
|
||||
from app.services.leitl_protocol import leitl_protocol
|
||||
from app.services.webdav_context import webdav_context_manager
|
||||
|
||||
|
||||
router = APIRouter(prefix="/api/leitl", tags=["LEITL"])
|
||||
|
||||
|
||||
# Pydantic models
|
||||
class SessionRegisterRequest(BaseModel):
|
||||
"""Request to register a new LEITL session"""
|
||||
agent_name: str = Field(..., description="Name of the agent (e.g., Cece, Claude)")
|
||||
agent_type: str = Field(default="assistant", description="Type of agent")
|
||||
tags: Optional[List[str]] = Field(default=None, description="Optional tags")
|
||||
|
||||
|
||||
class SessionRegisterResponse(BaseModel):
|
||||
"""Response after registering session"""
|
||||
session_id: str
|
||||
websocket_url: str
|
||||
agent_name: str
|
||||
started_at: str
|
||||
|
||||
|
||||
class HeartbeatRequest(BaseModel):
|
||||
"""Heartbeat update"""
|
||||
current_task: Optional[str] = Field(default=None, description="Current task description")
|
||||
|
||||
|
||||
class BroadcastRequest(BaseModel):
|
||||
"""Broadcast message request"""
|
||||
event_type: str = Field(..., description="Event type (e.g., task.started)")
|
||||
data: Optional[dict] = Field(default=None, description="Event data")
|
||||
|
||||
|
||||
class WebDAVContextRequest(BaseModel):
|
||||
"""Request for WebDAV context"""
|
||||
webdav_url: str = Field(..., description="Base WebDAV URL")
|
||||
username: Optional[str] = Field(default=None, description="WebDAV username")
|
||||
password: Optional[str] = Field(default=None, description="WebDAV password")
|
||||
query: Optional[str] = Field(default=None, description="Search query")
|
||||
file_types: Optional[List[str]] = Field(default=None, description="File type filters")
|
||||
max_results: int = Field(default=10, description="Max results to return")
|
||||
|
||||
|
||||
# Initialize on startup
|
||||
@router.on_event("startup")
|
||||
async def startup():
|
||||
"""Initialize LEITL protocol and WebDAV manager"""
|
||||
await leitl_protocol.initialize()
|
||||
await webdav_context_manager.initialize()
|
||||
|
||||
|
||||
@router.on_event("shutdown")
|
||||
async def shutdown():
|
||||
"""Shutdown LEITL protocol"""
|
||||
await leitl_protocol.shutdown()
|
||||
|
||||
|
||||
# Session management endpoints
|
||||
@router.post("/session/register", response_model=SessionRegisterResponse)
|
||||
async def register_session(request: SessionRegisterRequest):
|
||||
"""
|
||||
Register a new LEITL session
|
||||
|
||||
Creates a new session ID and broadcasts to other active sessions.
|
||||
Returns WebSocket URL for real-time communication.
|
||||
"""
|
||||
session = await leitl_protocol.register_session(
|
||||
agent_name=request.agent_name,
|
||||
agent_type=request.agent_type,
|
||||
tags=request.tags
|
||||
)
|
||||
|
||||
# Construct WebSocket URL (assumes same host)
|
||||
# In production, this would use the actual host from request
|
||||
websocket_url = f"ws://localhost:8000/api/leitl/ws/{session.session_id}"
|
||||
|
||||
return SessionRegisterResponse(
|
||||
session_id=session.session_id,
|
||||
websocket_url=websocket_url,
|
||||
agent_name=session.agent_name,
|
||||
started_at=session.started_at.isoformat()
|
||||
)
|
||||
|
||||
|
||||
@router.get("/sessions/active")
|
||||
async def get_active_sessions():
|
||||
"""
|
||||
Get all active LEITL sessions
|
||||
|
||||
Returns list of currently active agent sessions with their status.
|
||||
"""
|
||||
sessions = await leitl_protocol.get_active_sessions()
|
||||
return {
|
||||
"sessions": sessions,
|
||||
"total": len(sessions)
|
||||
}
|
||||
|
||||
|
||||
@router.post("/session/{session_id}/heartbeat")
|
||||
async def send_heartbeat(session_id: str, request: HeartbeatRequest):
|
||||
"""
|
||||
Send heartbeat for a session
|
||||
|
||||
Keeps the session alive and optionally updates current task.
|
||||
Sessions without heartbeat for 60s are considered dead.
|
||||
"""
|
||||
await leitl_protocol.heartbeat(
|
||||
session_id=session_id,
|
||||
current_task=request.current_task
|
||||
)
|
||||
|
||||
return {"status": "ok"}
|
||||
|
||||
|
||||
@router.post("/session/{session_id}/end")
|
||||
async def end_session(session_id: str):
|
||||
"""
|
||||
End a session
|
||||
|
||||
Gracefully terminates a session and broadcasts to other agents.
|
||||
"""
|
||||
await leitl_protocol.end_session(session_id)
|
||||
return {"status": "ended"}
|
||||
|
||||
|
||||
# Messaging endpoints
|
||||
@router.post("/session/{session_id}/broadcast")
|
||||
async def broadcast_message(session_id: str, request: BroadcastRequest):
|
||||
"""
|
||||
Broadcast event to all active sessions
|
||||
|
||||
Publishes event to Redis PubSub and WebSocket connections.
|
||||
All active sessions will receive this event.
|
||||
"""
|
||||
await leitl_protocol.broadcast_event(
|
||||
event_type=request.event_type,
|
||||
session_id=session_id,
|
||||
data=request.data
|
||||
)
|
||||
|
||||
return {"status": "broadcasted"}
|
||||
|
||||
|
||||
@router.get("/messages/recent")
|
||||
async def get_recent_messages(limit: int = Query(default=20, le=100)):
|
||||
"""
|
||||
Get recent broadcast messages
|
||||
|
||||
Returns the last N messages broadcast across all sessions.
|
||||
"""
|
||||
messages = await leitl_protocol.get_recent_messages(limit=limit)
|
||||
return {
|
||||
"messages": messages,
|
||||
"count": len(messages)
|
||||
}
|
||||
|
||||
|
||||
@router.get("/activity")
|
||||
async def get_activity_log(
|
||||
since: Optional[str] = Query(default=None, description="ISO timestamp"),
|
||||
limit: int = Query(default=50, le=200)
|
||||
):
|
||||
"""
|
||||
Get activity log
|
||||
|
||||
Returns recent activity across all sessions.
|
||||
Optionally filter by timestamp.
|
||||
"""
|
||||
since_dt = datetime.fromisoformat(since) if since else None
|
||||
activities = await leitl_protocol.get_activity_log(since=since_dt, limit=limit)
|
||||
|
||||
return {
|
||||
"activities": activities,
|
||||
"count": len(activities)
|
||||
}
|
||||
|
||||
|
||||
# WebDAV context endpoints
|
||||
@router.post("/context/sync")
|
||||
async def sync_webdav_context(request: WebDAVContextRequest):
|
||||
"""
|
||||
Sync and get WebDAV context
|
||||
|
||||
Fetches files from WebDAV, matches based on query, and returns content.
|
||||
Results are cached for 1 hour.
|
||||
"""
|
||||
context = await webdav_context_manager.sync_and_get(
|
||||
webdav_url=request.webdav_url,
|
||||
username=request.username,
|
||||
password=request.password,
|
||||
query=request.query,
|
||||
file_types=request.file_types,
|
||||
max_results=request.max_results
|
||||
)
|
||||
|
||||
return context
|
||||
|
||||
|
||||
# WebSocket endpoint
|
||||
@router.websocket("/ws/{session_id}")
|
||||
async def websocket_endpoint(websocket: WebSocket, session_id: str):
|
||||
"""
|
||||
WebSocket connection for real-time LEITL events
|
||||
|
||||
Connect with: ws://host/api/leitl/ws/{session_id}
|
||||
|
||||
Messages received:
|
||||
- Broadcast events from other sessions
|
||||
- Heartbeat confirmations
|
||||
- System notifications
|
||||
|
||||
Messages to send:
|
||||
- {"type": "heartbeat", "current_task": "..."}
|
||||
- {"type": "broadcast", "event_type": "...", "data": {...}}
|
||||
"""
|
||||
await websocket.accept()
|
||||
|
||||
# Register WebSocket
|
||||
await leitl_protocol.register_websocket(session_id, websocket)
|
||||
|
||||
try:
|
||||
# Send initial connection confirmation
|
||||
await websocket.send_json({
|
||||
"event_type": "connection.established",
|
||||
"session_id": session_id,
|
||||
"timestamp": datetime.utcnow().isoformat()
|
||||
})
|
||||
|
||||
# Listen for messages
|
||||
while True:
|
||||
data = await websocket.receive_json()
|
||||
|
||||
message_type = data.get("type")
|
||||
|
||||
if message_type == "heartbeat":
|
||||
# Update heartbeat
|
||||
current_task = data.get("current_task")
|
||||
await leitl_protocol.heartbeat(session_id, current_task)
|
||||
|
||||
# Send confirmation
|
||||
await websocket.send_json({
|
||||
"event_type": "heartbeat.confirmed",
|
||||
"session_id": session_id,
|
||||
"timestamp": datetime.utcnow().isoformat()
|
||||
})
|
||||
|
||||
elif message_type == "broadcast":
|
||||
# Broadcast event
|
||||
event_type = data.get("event_type")
|
||||
event_data = data.get("data")
|
||||
|
||||
await leitl_protocol.broadcast_event(
|
||||
event_type=event_type,
|
||||
session_id=session_id,
|
||||
data=event_data
|
||||
)
|
||||
|
||||
elif message_type == "ping":
|
||||
# Respond to ping
|
||||
await websocket.send_json({
|
||||
"event_type": "pong",
|
||||
"timestamp": datetime.utcnow().isoformat()
|
||||
})
|
||||
|
||||
except WebSocketDisconnect:
|
||||
# Unregister WebSocket
|
||||
await leitl_protocol.unregister_websocket(session_id, websocket)
|
||||
except Exception as e:
|
||||
print(f"WebSocket error for {session_id}: {e}")
|
||||
await leitl_protocol.unregister_websocket(session_id, websocket)
|
||||
|
||||
|
||||
# Health check
|
||||
@router.get("/health")
|
||||
async def health_check():
|
||||
"""LEITL protocol health check"""
|
||||
sessions = await leitl_protocol.get_active_sessions()
|
||||
|
||||
return {
|
||||
"status": "healthy",
|
||||
"active_sessions": len(sessions),
|
||||
"timestamp": datetime.utcnow().isoformat()
|
||||
}
|
||||
|
||||
|
||||
# Quick start endpoint (combines register + context)
|
||||
@router.post("/quick-start")
|
||||
async def quick_start(
|
||||
agent_name: str = Query(..., description="Agent name"),
|
||||
webdav_url: Optional[str] = Query(default=None, description="WebDAV URL"),
|
||||
query: Optional[str] = Query(default=None, description="Context query"),
|
||||
tags: Optional[List[str]] = Query(default=None, description="Session tags")
|
||||
):
|
||||
"""
|
||||
Quick start LEITL session with optional WebDAV context
|
||||
|
||||
One-shot endpoint that:
|
||||
1. Registers a new session
|
||||
2. Optionally syncs WebDAV context
|
||||
3. Returns session info + context + WebSocket URL
|
||||
|
||||
Perfect for "Turn on LEITL" prompts!
|
||||
"""
|
||||
# Register session
|
||||
session = await leitl_protocol.register_session(
|
||||
agent_name=agent_name,
|
||||
agent_type="assistant",
|
||||
tags=tags or []
|
||||
)
|
||||
|
||||
websocket_url = f"ws://localhost:8000/api/leitl/ws/{session.session_id}"
|
||||
|
||||
result = {
|
||||
"session": {
|
||||
"session_id": session.session_id,
|
||||
"agent_name": session.agent_name,
|
||||
"websocket_url": websocket_url,
|
||||
"started_at": session.started_at.isoformat()
|
||||
},
|
||||
"context": None,
|
||||
"other_sessions": await leitl_protocol.get_active_sessions()
|
||||
}
|
||||
|
||||
# Optionally sync WebDAV context
|
||||
if webdav_url:
|
||||
context = await webdav_context_manager.sync_and_get(
|
||||
webdav_url=webdav_url,
|
||||
query=query,
|
||||
max_results=5
|
||||
)
|
||||
result["context"] = context
|
||||
|
||||
return result
|
||||
Reference in New Issue
Block a user