sync: 141 file da Baida98/AI@35be4001 (2026-08-07 14:07 UTC)

api/state.py

@@ -5,8 +5,8 @@ TTL constants, prune helpers. Extracted from main.py — zero behaviour change.
 """
 import os, time, asyncio as _asyncio_mod, json as _json, re as _re
 import logging
-from typing import Optional, Any
-from fastapi import HTTPException, APIRouter, Request
+from typing import Optional, Any, AsyncIterator, List, Tuple
+from fastapi import HTTPException, APIRouter, Request, Body
 from pydantic import BaseModel, field_validator
 
 _logger = logging.getLogger("api.state")
@@ -32,7 +32,6 @@ _current_client_idx = 0
 
 try:
     from supabase import create_client
-    
     # S-FIX: Preferisce SERVICE_ROLE_KEY per bypassare RLS nelle operazioni di sistema
     def _get_key(p):
         return os.getenv(f"SUPABASE_SERVICE_ROLE_KEY_{p}") or os.getenv(f"SUPABASE_SERVICE_ROLE_{p}") or \
@@ -46,7 +45,6 @@ try:
         {"id": "D", "url": os.getenv("SUPABASE_URL_4") or os.getenv("SUPABASE_URL_D"), "key": _get_key("D")},
         {"id": "E", "url": os.getenv("SUPABASE_URL_5") or os.getenv("SUPABASE_URL_E"), "key": _get_key("E")},
     ]
-
     for cfg in PROJECT_CONFIGS:
         if cfg["url"] and cfg["key"]:
             try:
@@ -54,6 +52,126 @@ try:
                 _clients.append({"client": c, "id": cfg["id"], "status": "connected"})
                 _logger.info(f"BOOT: Supabase #{cfg['id']} connected OK")
             except Exception as e:
+                _logger.error(f"BOOT: Supabase #{cfg['id']} init failed: {e}")
+except ImportError:
+    _logger.error("BOOT: Supabase init module failed: create_client not found.")
+
+def _get_sb() -> Any:
+    """Ritorna il client Supabase corrente dal pool (round-robin)."""
+    global _current_client_idx
+    if not _clients: return None
+    # S-FIX: Salta i client marcati come "failed" (semplice circuit breaker)
+    for _ in range(len(_clients)):
+        entry = _clients[_current_client_idx]
+        _current_client_idx = (_current_client_idx + 1) % len(_clients)
+        if entry["status"] != "failed":
+            return entry["client"]
+    return _clients[0]["client"] if _clients else None
+
+_sb = _get_sb()
+
+@router.get("/health")
+async def health_check(request: Request):
+    health = {
+        "status": "ok",
+        "timestamp": time.time(),
+        "version": "1.5.5",
+        "database": "unknown",
+        "pool_size": len(_clients)
+    }
+    try:
+        if _sb:
+            try:
+                res = _sb.table("agent_memory").select("key").limit(1).execute()
+                health["database"] = "connected"
+            except Exception as inner_e:
+                for entry in _clients:
+                    if entry["client"] == _sb:
+                        entry["status"] = "failed"
+                        break
+                raise inner_e
+        else:
+            health["database"] = "disconnected"
+    except Exception as e:
         health["status"] = "degraded"
         health["database"] = f"RAW_ERROR: {str(e)}"
     return health
+
+# ── SENSITIVE keys set (Z-GAP-4) ──────────────────────────────────────────────
+SENSITIVE = {
+    'OPENROUTER_API_KEY', 'OPENAI_API_KEY', 'GEMINI_API_KEY', 'GROQ_API_KEY',
+    'HF_TOKEN', 'HUGGINGFACE_API_KEY', 'GH_TOKEN', 'GITHUB_TOKEN',
+    'QDRANT_API_KEY', 'DATABASE_URL', 'SESSION_SECRET', 'SECRET_KEY',
+    'RAILWAY_TOKEN', 'SUPABASE_KEY', 'SUPABASE_ANON_KEY',
+    'TELEGRAM_BOT_TOKEN', 'TELEGRAM_CHAT_ID',
+    'CF_API_TOKEN', 'CLOUDFLARE_API_TOKEN', 'CF_ACCOUNT_ID',
+    'CF_API_TOKEN_B', 'CF_ACCOUNT_ID_B',
+    'CEREBRAS_API_KEY', 'SAMBANOVA_API_KEY',
+    'VAULT_KEY', 'INTERNAL_TOKEN', 'DEPLOY_SECRET', 'WEBHOOK_TOKEN',
+    'TERMINAL_SECRET', 'EXEC_TOKEN', 'VITE_INTERNAL_TOKEN', 'VITE_TERMINAL_SECRET',
+    'VITE_OPENROUTER_API_KEY', 'VITE_HF_TOKEN', 'VITE_GROQ_API_KEY',
+    'GH_PAGES_TOKEN', 'VERCEL_TOKEN',
+}
+
+# ── In-memory stores ──────────────────────────────────────────────────────────
+_mem_fallback: dict[str, dict] = {}
+_agent_tasks: dict[str, dict] = {}
+_run_stream_tasks: dict[str, dict] = {}
+_loop_registry: dict[str, dict] = {}
+_LOOP_REGISTRY_TTL_S: float = 10 * 60
+_task_checkpoints: dict[str, dict] = {}
+_CHECKPOINT_TTL_MS = 2 * 60 * 60 * 1000
+_CHECKPOINT_MAX    = 100
+_AGENT_TASK_TTL_MS = 2 * 60 * 60 * 1000
+_AGENT_TASK_MAX    = 200
+
+# ── Telemetry & Health ────────────────────────────────────────────────────────
+_ai_health_cache: dict = {"data": None, "at": 0.0}
+_AI_HEALTH_TTL = 60.0
+_heartbeat_state: dict = {
+    "last_run_at":     None,
+    "next_run_at":     None,
+    "best_provider":   None,
+    "best_latency_ms": None,
+    "providers":       [],
+    "runs":            0,
+}
+
+# ── Singleton Getters ──────────────────────────────────────────���──────────────
+def get_supabase() -> Optional[Any]:
+    return _sb
+
+_mem_manager: Any = None
+_mem_manager_inited = False
+def _get_mem_manager() -> Any:
+    global _mem_manager, _mem_manager_inited
+    if _mem_manager_inited: return _mem_manager
+    try:
+        from memory.memory_manager import MemoryManager
+        _mem_manager = MemoryManager()
+        _mem_manager_inited = True
+    except Exception: _mem_manager = None
+    return _mem_manager
+
+_executor: Any = None
+def _get_executor() -> Any:
+    global _executor
+    if _executor is not None: return _executor
+    try:
+        from agents.executor import Executor
+        _executor = Executor(memory=_get_mem_manager())
+    except Exception: _executor = None
+    return _executor
+
+_ai_client: Any = None
+def _get_ai_client() -> Any:
+    global _ai_client
+    if _ai_client is not None: return _ai_client
+    try:
+        from models.ai_client import AIClient
+        _ai_client = AIClient()
+    except Exception: _ai_client = None
+    return _ai_client
+
+async def _get_mem_manager_async() -> Any:
+    return _get_mem_manager()