@danhachuel/thunderbolt 0.5.28 → 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,8 @@ 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()}"}

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:
@@ -254,7 +259,9 @@ def _image_endpoint(card: Mapping[str, Any]) -> str:
         return explicit
     if style == "openrouter":
         return f"{base}/images"
-    if style in {"openai_compatible", "huggingface", "agnes", "kie"}:
+    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"
@@ -301,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,
@@ -425,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)
 
 
@@ -491,7 +507,9 @@ def _video_endpoint(card: Mapping[str, Any]) -> str:
         return f"{base}/{_model(card).lstrip('/')}"
     if style == "openrouter":
         return f"{base}/videos"
-    if style in {"openai_compatible", "agnes", "kie"}:
+    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"
@@ -518,6 +536,15 @@ def _video_request(card: dict[str, Any], prompt: str, image_url: str = "") -> An
         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:
@@ -570,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
@@ -648,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

@@ -25,6 +25,19 @@ FULL_IA_VIDEO_PROVIDER_CODES = (
     "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.
 INTERNAL_IMAGE_ASPECT_RATIO = "16:9"
 INTERNAL_IMAGE_SIZE = "1K"

package/package.json

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