@danhachuel/thunderbolt 0.5.27 → 0.5.29

package/app/main.py

@@ -78,7 +78,7 @@ from hermes_ui.mcp import detect_local_service, install_skill_locally, load_inte
 from hermes_ui.mcp_server import server_status, start_server, stop_server
 from hermes_ui.material_sources import apply_material_source_cards_to_settings, ensure_material_source_cards, material_source_catalog, material_source_definition, new_material_card, normalize_material_card, selected_material_source
 from hermes_ui.llm_providers import LLM_CARDS_KEY, LLM_PROVIDER_CATALOG, apply_llm_cards_to_settings, ensure_llm_provider_cards, new_llm_card, normalize_llm_card, provider_definition, test_llm_provider_card, stamp_test_result
-from hermes_ui.media_providers import FULL_IA_VIDEO_PROVIDER_CODES, MEDIA_CARDS_KEY, MEDIA_IMAGE_ACTIVE_CARD_KEY, MEDIA_VIDEO_ACTIVE_CARD_KEY, apply_media_provider_cards_to_settings, ensure_media_provider_cards, media_cards_for_pool, media_provider_catalog, media_provider_definition, new_media_card, normalize_media_card
+from hermes_ui.media_providers import FULL_IA_VIDEO_PROVIDER_CODES, KIE_MEDIA_MODEL_CATALOG, MEDIA_CARDS_KEY, MEDIA_IMAGE_ACTIVE_CARD_KEY, MEDIA_VIDEO_ACTIVE_CARD_KEY, apply_media_provider_cards_to_settings, ensure_media_provider_cards, media_cards_for_pool, media_provider_catalog, media_provider_definition, new_media_card, normalize_media_card
 from hermes_ui.music import create_music_task, list_music_files, list_music_tasks, materialize_suno_audio, request_suno_generation, run_music_task, store_music_file, store_voiceover_file, transition_music_task
 from hermes_ui.music_generation import MUSIC_GENRES, MUSIC_VOCAL_OPTIONS, generate_music_fields
 from hermes_ui.media_downloader import AUDIO_FORMATS, VIDEO_CONTAINERS, VIDEO_QUALITY_OPTIONS, MediaDownloadError, build_download_options, clear_media_download_history, dependency_status, download_media, list_media_downloads, media_download_file
@@ -7087,6 +7087,26 @@ def _persist_media_cards(settings: dict[str, Any], cards: list[dict[str, Any]],
 def _fetch_media_models(card: dict[str, Any]) -> list[str]:
     """Consultar modelos do provider, incluindo o catálogo nativo Gemini."""
     provider = str(card.get("provider") or "").strip().lower()
+    if provider == "kie_ai":
+        return list(KIE_MEDIA_MODEL_CATALOG)
+    if provider == "openrouter":
+        base_url = str(card.get("base_url") or "https://openrouter.ai/api/v1").rstrip("/")
+        headers = {"Accept": "application/json", "Authorization": f"Bearer {str(card.get('api_key') or '').strip()}"}
+        discovered: set[str] = set()
+        failures: list[str] = []
+        for suffix in ("/images/models", "/videos/models"):
+            response = requests.get(f"{base_url}{suffix}", headers=headers, timeout=12)
+            if response.status_code >= 400:
+                failures.append(f"HTTP {response.status_code}")
+                continue
+            payload = response.json()
+            entries = payload.get("data") if isinstance(payload, dict) else []
+            if isinstance(entries, list):
+                discovered.update(str(item.get("id") or "").strip() for item in entries if isinstance(item, dict) and str(item.get("id") or "").strip())
+        if not discovered:
+            detail = f" ({', '.join(failures)})" if failures else ""
+            raise ValueError(f"O OpenRouter não devolveu modelos de imagem/vídeo{detail}.")
+        return sorted(discovered, key=str.casefold)
     if provider == "nano_banana":
         base_url = str(card.get("base_url") or "https://generativelanguage.googleapis.com/v1beta").rstrip("/")
         response = requests.get(

package/hermes_ui/api_key_tests.py

@@ -15,6 +15,7 @@ from urllib.parse import quote, urlsplit, urlunsplit
 import requests
 
 from app.modules.niche_finder.apify import APIFY_API_BASE
+from integrations.openai_model_discovery import OpenAICompatibleAPIError, validate_openrouter_api_key
 
 DEFAULT_TIMEOUT = 20
 
@@ -176,6 +177,12 @@ def test_media_provider_card(card: Mapping[str, Any]) -> dict[str, Any]:
         return _get(f"{base_url}/v3/users/me", headers={"X-Api-Key": api_key})
     if not model and provider not in {"inferenceport", "cloudflare_workers_ai", "heygen"}:
         return _result("missing", "Complete o modelo antes de testar.")
+    if provider == "openrouter":
+        try:
+            validate_openrouter_api_key(api_key, base_url, model)
+        except OpenAICompatibleAPIError as exc:
+            return _result("error", str(exc)[:240])
+        return _result("success", "OpenRouter API Key e modelo OK")
     headers = {"Authorization": f"Bearer {api_key}"} if api_key else {}
     if provider == "cloudflare_workers_ai":
         account_id = str(source.get("account_id") or "").strip()

package/hermes_ui/media_generation.py

@@ -191,6 +191,11 @@ def _image_value(payload: Any) -> tuple[bytes | None, str]:
 def _image_request_id(payload: Any) -> str:
     if not isinstance(payload, Mapping):
         return ""
+    nested = payload.get("data")
+    if isinstance(nested, Mapping):
+        task_id = str(nested.get("taskId") or "").strip()
+        if task_id:
+            return task_id
     for key in ("id", "prediction_id", "request_id", "task_id", "job_id"):
         value = str(payload.get(key) or "").strip()
         if value:
@@ -252,7 +257,11 @@ def _image_endpoint(card: Mapping[str, Any]) -> str:
     explicit = str(card.get("image_endpoint") or "").strip()
     if explicit:
         return explicit
-    if style in {"openai_compatible", "huggingface", "agnes", "kie"}:
+    if style == "openrouter":
+        return f"{base}/images"
+    if style == "kie":
+        return f"{base}/jobs/createTask"
+    if style in {"openai_compatible", "huggingface", "agnes"}:
         return f"{base}/images/generations"
     if style == "replicate":
         return f"{base}/predictions"
@@ -299,6 +308,12 @@ def _image_request(card: dict[str, Any], prompt: str, *, topic: str = "", letter
     if style == "replicate":
         body = {"version": _model(card), "input": {"prompt": constrained_prompt}}
         return requests.post(endpoint, headers=_headers(card), json=body, timeout=180)
+    if style == "kie":
+        body = {
+            "model": _model(card),
+            "input": {"prompt": constrained_prompt, "aspect_ratio": "16:9", "resolution": "1K"},
+        }
+        return requests.post(endpoint, headers=_headers(card), json=body, timeout=180)
     if provider == "agnes":
         body = {
             "model": _model(card) or AGNES_IMAGE_MODEL,
@@ -309,6 +324,13 @@ def _image_request(card: dict[str, Any], prompt: str, *, topic: str = "", letter
             "extra_body": {"response_format": "b64_json"},
         }
         return requests.post(endpoint, headers=_headers(card), json=body, timeout=AGNES_IMAGE_TIMEOUT_SECONDS)
+    if style == "openrouter":
+        return requests.post(
+            endpoint,
+            headers=_headers(card),
+            json={"model": _model(card), "prompt": constrained_prompt, "n": 1, "aspect_ratio": "16:9"},
+            timeout=180,
+        )
     body = {"model": _model(card), "prompt": constrained_prompt, "n": 1, "response_format": "b64_json"}
     if provider == "pollinations":
         body["size"] = "1792x1024"
@@ -416,6 +438,9 @@ def generate_image_for_card(
         style = str(routed.card.get("api_style") or media_provider_definition(routed.card.get("provider")).api_style)
         if request_id and style == "replicate":
             image_bytes, url = _poll_image(routed.card, request_id)
+        elif request_id and style == "kie":
+            urls = _poll_kie_task(routed.card, request_id, endpoint="/jobs/recordInfo", veo=False)
+            url = urls[0] if urls else ""
     return _download_or_write(image_bytes, url, destination, routed.card)
 
 
@@ -480,7 +505,11 @@ def _video_endpoint(card: Mapping[str, Any]) -> str:
         if not _model(card):
             raise MediaGenerationError("FAL AI requer o identificador da rota/modelo para gerar vídeo.")
         return f"{base}/{_model(card).lstrip('/')}"
-    if style in {"openai_compatible", "agnes", "kie"}:
+    if style == "openrouter":
+        return f"{base}/videos"
+    if style == "kie":
+        return f"{base}/jobs/createTask"
+    if style in {"openai_compatible", "agnes"}:
         return f"{base}/videos/generations"
     if style == "replicate":
         return f"{base}/predictions"
@@ -503,6 +532,19 @@ def _video_request(card: dict[str, Any], prompt: str, image_url: str = "") -> An
     }
     if image_url:
         body["image_url"] = image_url
+    if style == "openrouter":
+        body["aspect_ratio"] = str(card.get("aspect_ratio") or INTERNAL_VIDEO_ASPECT_RATIO)
+        body["resolution"] = str(card.get("video_size") or INTERNAL_VIDEO_SIZE)
+        return requests.post(endpoint, headers=_headers(card), json=body, timeout=180)
+    if style == "kie":
+        input_payload: dict[str, Any] = {
+            "prompt": body["prompt"],
+            "aspect_ratio": str(card.get("aspect_ratio") or INTERNAL_VIDEO_ASPECT_RATIO),
+            "resolution": str(card.get("video_size") or INTERNAL_VIDEO_SIZE),
+        }
+        if image_url:
+            input_payload["imageUrls"] = [image_url]
+        return requests.post(endpoint, headers=_headers(card), json={"model": _model(card), "input": input_payload}, timeout=180)
     if style == "heygen":
         avatar_id = str(card.get("avatar_id") or "").strip()
         if not avatar_id:
@@ -534,7 +576,11 @@ def _video_result(payload: Mapping[str, Any]) -> tuple[str, str]:
     raw_output = payload.get("output")
     if isinstance(raw_output, list) and raw_output:
         raw_output = raw_output[0]
-    direct = str(payload.get("video_url") or payload.get("url") or raw_output or "").strip()
+    unsigned_urls = payload.get("unsigned_urls")
+    if isinstance(unsigned_urls, list) and unsigned_urls:
+        direct = str(unsigned_urls[0] or "").strip()
+    else:
+        direct = str(payload.get("content_url") or payload.get("video_url") or payload.get("url") or raw_output or "").strip()
     if direct.startswith(("http://", "https://")):
         return direct, ""
     result = payload.get("result")
@@ -551,7 +597,7 @@ def _video_result(payload: Mapping[str, Any]) -> tuple[str, str]:
         direct = str(data.get("video_url") or data.get("url") or "").strip()
         if direct.startswith(("http://", "https://")):
             return direct, ""
-        for key in ("video_id", "id", "request_id", "task_id", "job_id"):
+        for key in ("video_id", "id", "request_id", "task_id", "taskId", "job_id"):
             value = str(data.get(key) or "").strip()
             if value:
                 return "", value
@@ -576,6 +622,8 @@ def _poll_video(card: Mapping[str, Any], request_id: str, *, attempts: int = 24,
         endpoint = f"{base}/predictions/{request_id}"
     elif style == "heygen":
         endpoint = f"{base}/v3/videos/{request_id}"
+    elif style == "openrouter":
+        endpoint = f"{base}/videos/{request_id}"
     else:
         endpoint = f"{base}/videos/{request_id}"
     headers = _headers(card, fal=style == "fal_queue")
@@ -627,7 +675,11 @@ def generate_video_for_card(
     url, request_id = _video_result(routed.payload)
     if not url and request_id:
         try:
-            url = _poll_video(routed.card, request_id)
+            if str(routed.card.get("api_style") or media_provider_definition(routed.card.get("provider")).api_style) == "kie":
+                urls = _poll_kie_task(routed.card, request_id, endpoint="/jobs/recordInfo", veo=False)
+                url = urls[0] if urls else ""
+            else:
+                url = _poll_video(routed.card, request_id)
         except ProviderCallError as exc:
             raise MediaGenerationError(str(exc)) from exc
     if not url:

package/hermes_ui/media_providers.py

@@ -22,6 +22,20 @@ FULL_IA_VIDEO_PROVIDER_CODES = (
     "huggingface",
     "inferenceport",
     "heygen",
+    "openrouter",
+)
+
+# A KIE disponibiliza o Market por páginas de modelo, sem um endpoint global
+# documentado para listar modelos. Estes slugs são os modelos multimédia
+# suportados pelo contrato unificado jobs/createTask.
+KIE_MEDIA_MODEL_CATALOG = (
+    "bytedance/seedream-v4-text-to-image",
+    "seedream/4.5-text-to-image",
+    "grok-imagine/text-to-image",
+    "kling-2.6/text-to-video",
+    "grok-imagine/image-to-video",
+    "veo3_fast",
+    "veo3",
 )
 
 # Mantidos internamente para preservar compatibilidade e controlar o prompt dos providers.
@@ -165,6 +179,16 @@ MEDIA_PROVIDER_CATALOG: tuple[MediaProviderDefinition, ...] = (
         extra_fields=("avatar_id", "voice_id"),
         description="Vídeos com avatar, script ou áudio pela API V3 da HeyGen.",
     ),
+    MediaProviderDefinition(
+        "openrouter",
+        "OpenRouter",
+        default_base_url="https://openrouter.ai/api/v1",
+        supports_image=True,
+        supports_video=True,
+        supports_text=True,
+        api_style="openrouter",
+        description="Gateway unificado para modelos de geração de imagem e vídeo via APIs dedicadas do OpenRouter.",
+    ),
 )
 
 _MEDIA_BY_CODE = {item.code: item for item in MEDIA_PROVIDER_CATALOG}

package/package.json

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