Update modules/orchestrator.py

modules/orchestrator.py

@@ -8,6 +8,7 @@ import signal
 import time
 import psutil
 import threading
+import functools
 from abc import ABC, abstractmethod
 from datetime import datetime
 from concurrent.futures import ProcessPoolExecutor, ThreadPoolExecutor
@@ -83,8 +84,6 @@ def monitor_system_resources(func: Callable):
         return result
     return wrapper
 
-import functools # Re-importing inside for safety if used as standalone
-
 # ==========================================
 # 5. CORE INTERFACES (CONTRACTS)
 # ==========================================
@@ -516,8 +515,9 @@ if __name__ == "__main__":
         pass
     except Exception as e:
         print(f"FATAL ERROR: {e}")
+
 # ==========================================
-# REGISTER MODULE FUNCTION
+# 12. REGISTER MODULE FUNCTION (FIXED LINE 539)
 # ==========================================
 def register_module(app, client, username):
     """Register orchestrator module with FastAPI app"""
@@ -531,15 +531,14 @@ def register_module(app, client, username):
             "module": "orchestrator",
             "status": "active",
             "version": "2.0.1-Stable",
-            "endpoints": ["/orchestrator/health", "/system/task", "/system/status"]
+            "endpoints": ["/orchestrator/health", "/orchestrator/diagnostics", "/system/task", "/system/status"]
         }
     
-    # Register existing endpoints
-    from .orchestrator import get_system_diagnostics
-    
     @router.get("/orchestrator/diagnostics")
     async def module_diagnostics():
+        """Diagnostics endpoint - FIXED: No circular import"""
+        # DIRECTLY CALL THE FUNCTION DEFINED ABOVE (line ~526)
         return await get_system_diagnostics()
     
     app.include_router(router)
-    print("✅ Orchestrator module registered with FastAPI")        
+    print("✅ Orchestrator module registered with FastAPI")