PL jo'$UdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZmZejeZdZded<d Zd Zded <d ZGd dejZd0dZdddd1dZdddd2dZdddddd3d*Zd+ddddd,d4d/ZdS)5u Video Generation Provider ABC ============================= Defines the pluggable-backend interface for video generation. Providers register instances via ``PluginContext.register_video_gen_provider()``; the active one (selected via ``video_gen.provider`` in ``config.yaml``) services every ``video_generate`` tool call. Providers live in ``/plugins/video_gen//`` (built-in, auto-loaded as ``kind: backend``) or ``~/.hermes/plugins/video_gen//`` (user, opt-in via ``plugins.enabled``). Mirrors the ``image_gen`` provider design (``agent/image_gen_provider.py``) so the two surfaces stay learnable together. Unified surface --------------- One tool — ``video_generate`` — covers **text-to-video** and **image-to-video**. The router is the presence of ``image_url``: if it's set, the provider routes to its image-to-video endpoint; if it's omitted, the provider routes to text-to-video. Users pick one **model family** (e.g. Pixverse v6, Veo 3.1, Kling O3 Standard); the provider handles which underlying FAL/xAI endpoint to hit. Video edit and video extend are intentionally NOT exposed in this surface — the inconsistency across backends is too large for one unified tool. If those use cases warrant attention later they can ship as separate tools. Response shape -------------- All providers return a dict built by :func:`success_response` / :func:`error_response`. Keys: success bool video str | None URL or absolute file path model str provider-specific model identifier prompt str echoed prompt modality str "text" | "image" (which mode was used) aspect_ratio str provider-native (e.g. "16:9") or "" duration int seconds (0 if not applicable) provider str provider name (for diagnostics) error str only when success=False error_type str only when success=False ) annotationsN)Path)AnyDictListOptionalTuple)16:9z9:16z1:1z4:3z3:4z3:2z2:3zTuple[str, ...]COMMON_ASPECT_RATIOSr )480p540p720p1080pCOMMON_RESOLUTIONSrc eZdZdZeejd!dZed!dZd"dZ d#d Z d$d Z d%d Z d$dZ ejddddeedddd d&d ZdS)'VideoGenProvideruAbstract base class for a video generation backend. Subclasses must implement :meth:`generate`. Everything else has sane defaults — override only what your provider needs. returnstrcdS)zStable short identifier used in ``video_gen.provider`` config. Lowercase, no spaces. Examples: ``xai``, ``fal``, ``google``. Nselfs __.``. %Y%m%d_%H%M%SN_.) base64 b64decodedatetimenowstrftimeuuiduuid4hexrc write_bytes)rhrfrgrawtsshortrbs rsave_b64_videorzs  8 $ $C     ) )/ : :B JLL RaR E   F!E!ER!E!E%!E!E)!E!E EDS Krrwbytesc tjd}tjjdd}t |d|d|d|z }|||S)z@Write raw video bytes (e.g. an HTTP download body) to the cache.rjNrkrlrm)rprqrrrsrtrurcrv)rwrfrgrxryrbs rsave_bytes_videor}s     ) )/ : :B JLL RaR E   F!E!ER!E!E%!E!E)!E!E EDS Krr4r()modalityrErDextrarArJr~rErDintproviderrOptional[Dict[str, Any]]r&c d||||||rt|nd|d}|r0|D]\} } || | |S)uBuild a uniform success response dict. ``video`` may be an HTTP URL or an absolute filesystem path. ``modality`` is ``"text"`` (text-to-video) or ``"image"`` (image-to-video) — indicates which endpoint was actually hit, useful for diagnostics. Tr)successrdrArJr~rErDr)ritems setdefault) rdrArJr~rErDrrpayloadkvs rsuccess_responsersz$$%-4CMMM1  G %KKMM % %DAq   q! $ $ $ $ Nrprovider_error) error_typerrArJrEerrorrc dd||||||dS)z$Build a uniform error response dict.FN)rrdrrrArJrErr)rrrrArJrEs rerror_responsers+ $   r)rr)rhrrfrrgrrr)rwr{rfrrgrrr)rdrrArrJrr~rrErrDrrrrrrr&)rrrrrrrArrJrrErrr&)rT __future__rrVrnrploggingrspathlibrtypingrrrrr getLoggerrQloggerr __annotations__rXrrYABCrrcrzr}rrrrrrs,,,\#"""""  33333333333333  8 $ $ )\[[[[&GGGGGy y y y y swy y y B .      &&*H'r