@danhachuel/thunderbolt 0.5.48 → 0.5.50

package/hermes_ui/growth_youtube.py

@@ -161,7 +161,9 @@ def _vision_prompt(video: Mapping[str, Any]) -> str:
 def analyse_thumbnail_with_paligemma(path: str, card: Mapping[str, Any]) -> dict[str, Any]:
     key = str(card.get("api_key") or "").strip()
     base = str(card.get("base_url") or "https://integrate.api.nvidia.com/v1").rstrip("/")
-    endpoint = base if base.endswith("/chat/completions") else f"{base}/chat/completions"
+    if "integrate.api.nvidia.com" in base:
+        base = base.replace("integrate.api.nvidia.com", "ai.api.nvidia.com")
+    endpoint = base if "/vlm/google/paligemma" in base else f"{base}/vlm/google/paligemma"
     model = str(card.get("model") or "google/paligemma").strip()
     if not key:
         return {"score": 50, "diagnosis": "API key NVIDIA NIM Paligemma não configurada.", "status": "sem_dados"}

package/hermes_ui/llm_providers.py

@@ -13,6 +13,7 @@ from integrations.openai_model_discovery import (
     DEFAULT_NVIDIA_NIM_BASE_URL,
     OpenAICompatibleAPIError,
     fetch_openai_compatible_models,
+    validate_paligemma_api_key,
     validate_openai_compatible_api_key,
     validate_openrouter_api_key,
 )
@@ -431,7 +432,9 @@ def test_llm_provider_card(card: Mapping[str, Any]) -> dict[str, Any]:
     if not model:
         return {"ok": False, "status": "error", "message": "Missing configuration — introduza o modelo antes do teste."}
     try:
-        if normalized["provider"] == "openrouter":
+        if normalized["provider"] == "nvidia_nim_paligemma":
+            validate_paligemma_api_key(api_key, base_url)
+        elif normalized["provider"] == "openrouter":
             validate_openrouter_api_key(api_key, base_url, model)
         else:
             validate_openai_compatible_api_key(api_key, base_url, model)

package/integrations/openai_model_discovery.py

@@ -258,3 +258,39 @@ def fetch_replicate_models(
     if not models:
         raise ModelDiscoveryError("O endpoint Replicate não devolveu modelos com versão disponível.")
     return sorted(models, key=str.casefold)[:MAX_MODEL_IDS]
+
+
+def validate_paligemma_api_key(
+    api_key: str,
+    base_url: str,
+    *,
+    timeout: float = DEFAULT_TIMEOUT_SECONDS,
+) -> None:
+    """Validate the Paligemma VLM route instead of the text-only chat route."""
+    token = str(api_key or "").strip()
+    if not token:
+        raise OpenAICompatibleAPIError("Informe a API key antes do teste.")
+    configured = str(base_url or DEFAULT_NVIDIA_NIM_BASE_URL).rstrip("/")
+    if configured.endswith("/chat/completions"):
+        configured = configured.removesuffix("/chat/completions")
+    if "integrate.api.nvidia.com" in configured:
+        configured = configured.replace("integrate.api.nvidia.com", "ai.api.nvidia.com")
+    endpoint = configured if "/vlm/google/paligemma" in configured else f"{configured}/vlm/google/paligemma"
+    payload = {
+        "messages": [{"role": "user", "content": "Describe this image in one word."}],
+        "max_tokens": 8,
+        "temperature": 0,
+    }
+    try:
+        response = requests.post(
+            endpoint,
+            headers={"Accept": "application/json", "Authorization": f"Bearer {token}", "Content-Type": "application/json"},
+            json=payload,
+            timeout=timeout,
+        )
+    except requests.RequestException as exc:
+        raise OpenAICompatibleAPIError(f"Não foi possível testar o endpoint Paligemma: {exc}") from exc
+    if response.status_code >= 400:
+        if response.status_code in (401, 403):
+            raise OpenAICompatibleAPIError(f"O endpoint Paligemma recusou a API key (HTTP {response.status_code}).")
+        raise OpenAICompatibleAPIError(f"O endpoint Paligemma devolveu HTTP {response.status_code}.")

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@danhachuel/thunderbolt",
-  "version": "0.5.48",
+  "version": "0.5.50",
   "description": "Thunderbolt — interface local para operação de canais faceless e motor MoneyPrinterTurbo",
   "license": "MIT",
   "main": "scripts/cli.mjs",

package/scripts/streamlit_bootstrap.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import io
 import os
+import signal
 import sys
 
 # Este ficheiro é o primeiro módulo Python executado pelo launcher. As
@@ -28,6 +29,26 @@ def _utf8_stream(stream: object) -> object:
 sys.stdout = _utf8_stream(sys.stdout)
 sys.stderr = _utf8_stream(sys.stderr)
 
+
+def _custom_sigint_handler(signum: int, frame: object) -> None:
+    """Encerrar sem executar ``server.stop``/Click do handler do Streamlit."""
+    print("\nEncerrando servidor...", file=sys.stderr, flush=True)
+    raise SystemExit(0)
+
+
+_original_signal = signal.signal
+
+
+def _protected_signal(signum: signal.Signals, handler: object) -> object:
+    """Não permitir que Streamlit substitua o handler seguro de SIGINT."""
+    if signum == signal.SIGINT:
+        return _original_signal(signal.SIGINT, _custom_sigint_handler)
+    return _original_signal(signum, handler)
+
+
+_original_signal(signal.SIGINT, _custom_sigint_handler)
+signal.signal = _protected_signal
+
 from streamlit.web.cli import main  # noqa: E402
 
 if __name__ == "__main__":