God Mode+ v3 fix: main.py

main.py

@@ -1,7 +1,6 @@
 """
-🚀 GOD MODE+ Autonomous AI Operating System
-Devin + Manus + Genspark Style — Production-Grade Backend
-Version: 3.0.0 — Full God Mode Upgrade
+🚀 GOD MODE+ Main — Fixed, Unified, Production-Ready
+All LLM calls go through LLMRouter (Cloudflare Gateway primary)
 """
 
 import asyncio
@@ -22,28 +21,31 @@ from slowapi import Limiter, _rate_limit_exceeded_handler
 from slowapi.util import get_remote_address
 from slowapi.errors import RateLimitExceeded
 
-from api.routes import tasks, chat, memory, github, health
+from api.routes import tasks, chat, memory, health
 from api.routes import connectors, agents as agents_router
 from api.websocket_manager import WebSocketManager
 from core.task_engine import TaskEngine
 from memory.db import init_db
 
-# ─── God Mode Agents ───────────────────────────────────────────────────────────
-from ai_router.router import AIRouter
-from agents.orchestrator import GodAgentOrchestrator
-from agents.chat_agent import ChatAgent
-from agents.planner_agent import PlannerAgent
-from agents.coding_agent import CodingAgent
-from agents.debug_agent import DebugAgent
-from agents.memory_agent import MemoryAgent
-from agents.connector_agent import ConnectorAgent
-from agents.deploy_agent import DeployAgent
-from agents.workflow_agent import WorkflowAgent
-from agents.sandbox_agent import SandboxAgent
-from agents.ui_agent import UIAgent
-from connectors.manager import ConnectorManager
+# ─── God Mode+ LLM Router ────────────────────────────────────────────────────
+from ai_router.router import LLMRouter
 
-# ─── Structured Logging ────────────────────────────────────────────────────────
+# ─── God Mode+ Agents ────────────────────────────────────────────────────────
+from agents.orchestrator    import GodAgentOrchestrator
+from agents.chat_agent      import ChatAgent
+from agents.planner_agent   import PlannerAgent
+from agents.coding_agent    import CodingAgent
+from agents.debug_agent     import DebugAgent
+from agents.memory_agent    import MemoryAgent
+from agents.connector_agent import ConnectorAgent
+from agents.deploy_agent    import DeployAgent
+from agents.workflow_agent  import WorkflowAgent
+from agents.sandbox_agent   import SandboxAgent
+from agents.ui_agent        import UIAgent
+from agents.reasoning_agent import ReasoningAgent
+from connectors.manager     import ConnectorManager
+
+# ─── Structured Logging ───────────────────────────────────────────────────────
 structlog.configure(
     processors=[
         structlog.processors.TimeStamper(fmt="iso"),
@@ -54,57 +56,58 @@ structlog.configure(
 )
 log = structlog.get_logger()
 
-# ─── Rate Limiter ──────────────────────────────────────────────────────────────
+# ─── Rate Limiter ─────────────────────────────────────────────────────────────
 limiter = Limiter(key_func=get_remote_address)
 
-# ─── Global Managers ──────────────────────────────────────────────────────────
-ws_manager = WebSocketManager()
-task_engine = TaskEngine(ws_manager)
-ai_router = AIRouter(ws_manager)
+# ─── Global Instances ─────────────────────────────────────────────────────────
+ws_manager        = WebSocketManager()
+task_engine       = TaskEngine(ws_manager)
+llm_router        = LLMRouter(ws_manager)
 connector_manager = ConnectorManager()
 
+
 # ─── Build God Agent Ecosystem ────────────────────────────────────────────────
 def build_orchestrator() -> GodAgentOrchestrator:
-    orchestrator = GodAgentOrchestrator(ws_manager=ws_manager, ai_router=ai_router)
-
-    # Register all specialized agents
-    orchestrator.register_agent("chat",      ChatAgent(ws_manager, ai_router))
-    orchestrator.register_agent("planner",   PlannerAgent(ws_manager, ai_router))
-    orchestrator.register_agent("coding",    CodingAgent(ws_manager, ai_router))
-    orchestrator.register_agent("debug",     DebugAgent(ws_manager, ai_router))
-    orchestrator.register_agent("memory",    MemoryAgent(ws_manager, ai_router))
-    orchestrator.register_agent("connector", ConnectorAgent(ws_manager, ai_router))
-    orchestrator.register_agent("deploy",    DeployAgent(ws_manager, ai_router))
-    orchestrator.register_agent("workflow",  WorkflowAgent(ws_manager, ai_router))
-    orchestrator.register_agent("sandbox",   SandboxAgent(ws_manager, ai_router))
-    orchestrator.register_agent("ui",        UIAgent(ws_manager, ai_router))
-
-    log.info("🤖 God Agent Ecosystem initialized", agents=10)
-    return orchestrator
+    orch = GodAgentOrchestrator(ws_manager=ws_manager, ai_router=llm_router)
+
+    orch.register_agent("chat",      ChatAgent(ws_manager, llm_router))
+    orch.register_agent("planner",   PlannerAgent(ws_manager, llm_router))
+    orch.register_agent("coding",    CodingAgent(ws_manager, llm_router))
+    orch.register_agent("debug",     DebugAgent(ws_manager, llm_router))
+    orch.register_agent("memory",    MemoryAgent(ws_manager, llm_router))
+    orch.register_agent("connector", ConnectorAgent(ws_manager, llm_router))
+    orch.register_agent("deploy",    DeployAgent(ws_manager, llm_router))
+    orch.register_agent("workflow",  WorkflowAgent(ws_manager, llm_router))
+    orch.register_agent("sandbox",   SandboxAgent(ws_manager, llm_router))
+    orch.register_agent("ui",        UIAgent(ws_manager, llm_router))
+    orch.register_agent("reasoning", ReasoningAgent(ws_manager, llm_router))
+
+    active = llm_router.get_active_provider()
+    log.info("🤖 God Agent Ecosystem initialized", agents=11, llm_provider=active)
+    return orch
+
 
 orchestrator = build_orchestrator()
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    """Startup + Shutdown lifecycle."""
-    log.info("🚀 Starting GOD MODE+ AI Operating System...")
+    log.info("🚀 Starting GOD MODE+ AI OS...")
     await init_db()
     await task_engine.start()
     asyncio.create_task(ws_manager.heartbeat_loop())
-    log.info("✅ GOD MODE+ Platform ready — All agents online")
-    log.info("🤖 Agents: Chat, Planner, Coding, Debug, Memory, Connector, Deploy, Workflow, Sandbox, UI")
-    log.info("🌐 AI Router: OpenAI → Groq → Cerebras → OpenRouter → Anthropic")
+    active = llm_router.get_active_provider()
+    log.info("✅ GOD MODE+ Platform ready", llm_provider=active, agents=11)
     yield
     log.info("🛑 Shutting down...")
     await task_engine.stop()
     log.info("✅ Shutdown complete")
 
 
-# ─── FastAPI App ───────────────────────────────────────────────────────────────
+# ─── FastAPI App ──────────────────────────────────────────────────────────────
 app = FastAPI(
     title="🤖 GOD MODE+ AI Operating System",
-    description="Devin + Manus + Genspark Autonomous AI Engineering Platform",
+    description="Devin + Manus + Genspark Autonomous AI Engineering Platform — v3.0",
     version="3.0.0",
     lifespan=lifespan,
     docs_url="/api/docs",
@@ -114,14 +117,14 @@ app = FastAPI(
 app.state.limiter = limiter
 app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
 
-# ─── Share state ───────────────────────────────────────────────────────────────
-app.state.ws_manager = ws_manager
-app.state.task_engine = task_engine
-app.state.ai_router = ai_router
-app.state.orchestrator = orchestrator
+# ─── Share state ──────────────────────────────────────────────────────────────
+app.state.ws_manager        = ws_manager
+app.state.task_engine       = task_engine
+app.state.ai_router         = llm_router
+app.state.orchestrator      = orchestrator
 app.state.connector_manager = connector_manager
 
-# ─── Middleware ────────────────────────────────────────────────────────────────
+# ─── Middleware ───────────────────────────────────────────────────────────────
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -134,31 +137,39 @@ app.add_middleware(GZipMiddleware, minimum_size=1000)
 
 @app.middleware("http")
 async def log_requests(request: Request, call_next):
-    start = time.time()
+    start    = time.time()
     response = await call_next(request)
-    duration = round((time.time() - start) * 1000, 2)
-    log.info("HTTP", method=request.method, path=request.url.path, status=response.status_code, ms=duration)
+    ms       = round((time.time() - start) * 1000, 2)
+    log.info("HTTP", method=request.method, path=request.url.path,
+             status=response.status_code, ms=ms)
     return response
 
 
-# ─── REST API Routers ──────────────────────────────────────────────────────────
-app.include_router(health.router,             prefix="/api/v1",              tags=["health"])
-app.include_router(tasks.router,              prefix="/api/v1/tasks",        tags=["tasks"])
-app.include_router(chat.router,               prefix="/api/v1",              tags=["chat"])
-app.include_router(memory.router,             prefix="/api/v1/memory",       tags=["memory"])
-app.include_router(github.router,             prefix="/api/v1/github",       tags=["github"])
-app.include_router(connectors.router,         prefix="/api/v1/connectors",   tags=["connectors"])
-app.include_router(agents_router.router,      prefix="/api/v1/agents",       tags=["agents"])
+# ─── REST Routers ─────────────────────────────────────────────────────────────
+app.include_router(health.router,          prefix="/api/v1",            tags=["health"])
+app.include_router(tasks.router,           prefix="/api/v1/tasks",      tags=["tasks"])
+app.include_router(chat.router,            prefix="/api/v1",            tags=["chat"])
+app.include_router(memory.router,          prefix="/api/v1/memory",     tags=["memory"])
+app.include_router(connectors.router,      prefix="/api/v1/connectors", tags=["connectors"])
+app.include_router(agents_router.router,   prefix="/api/v1/agents",     tags=["agents"])
+
+# GitHub route (conditional import — skip if missing)
+try:
+    from api.routes import github
+    app.include_router(github.router, prefix="/api/v1/github", tags=["github"])
+except ImportError:
+    pass
+
 
+# ─── WebSocket Endpoints ──────────────────────────────────────────────────────
 
-# ─── WebSocket Endpoints ───────────────────────────────────────────────────────
 @app.websocket("/ws/tasks/{task_id}")
 async def ws_task(websocket: WebSocket, task_id: str):
     await ws_manager.connect(websocket, room=f"task:{task_id}")
     try:
         while True:
             data = await websocket.receive_text()
-            msg = json.loads(data)
+            msg  = json.loads(data)
             if msg.get("type") == "ping":
                 await websocket.send_json({"type": "pong", "timestamp": time.time()})
     except WebSocketDisconnect:
@@ -171,7 +182,7 @@ async def ws_logs(websocket: WebSocket):
     try:
         while True:
             data = await websocket.receive_text()
-            msg = json.loads(data)
+            msg  = json.loads(data)
             if msg.get("type") == "ping":
                 await websocket.send_json({"type": "pong", "timestamp": time.time()})
     except WebSocketDisconnect:
@@ -184,11 +195,10 @@ async def ws_chat(websocket: WebSocket, session_id: str):
     try:
         while True:
             data = await websocket.receive_text()
-            msg = json.loads(data)
+            msg  = json.loads(data)
             if msg.get("type") == "ping":
                 await websocket.send_json({"type": "pong", "timestamp": time.time()})
             elif msg.get("type") == "chat_message":
-                # Route through God Agent Orchestrator
                 asyncio.create_task(
                     orchestrator.orchestrate(
                         user_message=msg.get("content", ""),
@@ -197,7 +207,6 @@ async def ws_chat(websocket: WebSocket, session_id: str):
                     )
                 )
             elif msg.get("type") == "task_message":
-                # Create autonomous task via task engine
                 from core.models import TaskCreateRequest
                 req = TaskCreateRequest(
                     goal=msg.get("content", ""),
@@ -214,7 +223,7 @@ async def ws_agent_status(websocket: WebSocket):
     try:
         while True:
             data = await websocket.receive_text()
-            msg = json.loads(data)
+            msg  = json.loads(data)
             if msg.get("type") == "ping":
                 await websocket.send_json({"type": "pong", "timestamp": time.time()})
             elif msg.get("type") == "get_status":
@@ -228,44 +237,45 @@ async def ws_agent_status(websocket: WebSocket):
 
 @app.websocket("/ws/sandbox/{session_id}")
 async def ws_sandbox(websocket: WebSocket, session_id: str):
-    """Live sandbox terminal stream."""
     await ws_manager.connect(websocket, room=f"sandbox:{session_id}")
     sandbox = orchestrator.get_agent("sandbox")
     try:
         while True:
             data = await websocket.receive_text()
-            msg = json.loads(data)
+            msg  = json.loads(data)
             if msg.get("type") == "ping":
                 await websocket.send_json({"type": "pong", "timestamp": time.time()})
             elif msg.get("type") == "execute" and sandbox:
-                cmd = msg.get("command", "")
+                cmd    = msg.get("command", "")
                 result = await sandbox.execute(cmd, session_id=session_id)
                 await websocket.send_json({
-                    "type": "terminal_output",
-                    "command": cmd,
-                    "output": result,
+                    "type":      "terminal_output",
+                    "command":   cmd,
+                    "output":    result,
                     "timestamp": time.time(),
                 })
     except WebSocketDisconnect:
         ws_manager.disconnect(websocket, room=f"sandbox:{session_id}")
 
 
-# ─── Root ──────────────────────────────────────────────────────────────────────
+# ─── Root ─────────────────────────────────────────────────────────────────────
 @app.get("/")
 async def root():
-    cs = connector_manager.get_summary()
+    cs     = connector_manager.get_summary()
+    active = llm_router.get_active_provider()
     return {
-        "name": "🤖 GOD MODE+ AI Operating System",
-        "version": "3.0.0",
-        "status": "operational",
-        "mode": "god_mode_plus",
-        "agents": orchestrator.get_status()["agents"],
+        "name":            "🤖 GOD MODE+ AI Operating System",
+        "version":         "3.0.0",
+        "status":          "operational",
+        "mode":            "god_mode_plus",
+        "llm_provider":    active,
+        "agents":          orchestrator.get_status()["agents"],
         "connectors": {
             "connected": cs["connected"],
-            "total": cs["total"],
-            "ai_ready": cs["ai_ready"],
+            "total":     cs["total"],
+            "ai_ready":  cs["ai_ready"],
         },
-        "docs": "/api/docs",
+        "docs":       "/api/docs",
         "websockets": [
             "/ws/tasks/{task_id}",
             "/ws/logs",
@@ -273,14 +283,14 @@ async def root():
             "/ws/agent/status",
             "/ws/sandbox/{session_id}",
         ],
-        "phases_complete": [
-            "Phase 1: God Agent Orchestrator",
-            "Phase 2: Sandbox Agent",
-            "Phase 3: Connector System",
-            "Phase 4: Autonomous Coding Engine",
-            "Phase 5: Memory System",
-            "Phase 6: Real-time Streaming",
-            "Phase 7: Workflow Factor OS",
-            "Phase 9: Multi-Model AI Router",
-        ],
+        "endpoints": {
+            "chat":        "POST /api/v1/chat",
+            "chat_stream": "POST /api/v1/chat/stream",
+            "tasks":       "POST /api/v1/tasks/create",
+            "execute":     "POST /api/v1/execute",
+            "health":      "GET  /api/v1/health",
+            "agents":      "POST /api/v1/agents/orchestrate",
+            "sandbox":     "POST /api/v1/agents/sandbox/execute",
+            "connectors":  "GET  /api/v1/connectors/",
+        },
     }