`gj[JUdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZmZej"eZdddd dd d dddd d d ddddddddddd d d dddddddddddd d d d ddd!dd"d#d$d%d&d d'd d(d)dd*dd+d,d-d.dd/d d d0 d1ddd2dd3d4dddd'd'd d5Zd6ed7<d8ZdJd:ZdKd;ZdLd<ZdMd=Z dNd>Zdad9ed?<ej:ZdOd@Zda d9edA<da!d9edB<ej:Z"dCZ#dDZ$dPdEZ%dQdFZ&GdGdHeZ'dRdIZ(y)Su FAL.ai video generation backend. User-facing surface: pick a **model family** (e.g. "Pixverse v6", "Veo 3.1", "Seedance 2.0", "Kling v3 4K", "LTX 2.3", "Happy Horse"). The plugin auto-routes to the family's text-to-video endpoint when called without ``image_url``, and to its image-to-video endpoint when ``image_url`` is provided. The agent never sees the routing — it just calls ``video_generate(prompt=..., image_url=...)``. Model families (each with t2v + i2v endpoints): Cheap tier: ltx-2.3 fal-ai/ltx-2.3-22b/text-to-video / fal-ai/ltx-2.3-22b/image-to-video pixverse-v6 fal-ai/pixverse/v6/text-to-video / fal-ai/pixverse/v6/image-to-video Premium tier: veo3.1 fal-ai/veo3.1 / fal-ai/veo3.1/image-to-video seedance-2.0 bytedance/seedance-2.0/text-to-video / bytedance/seedance-2.0/image-to-video kling-v3-4k fal-ai/kling-video/v3/4k/text-to-video / fal-ai/kling-video/v3/4k/image-to-video happy-horse alibaba/happy-horse/text-to-video / alibaba/happy-horse/image-to-video Selection precedence for the active family: 1. ``model=`` arg from the tool call 2. ``FAL_VIDEO_MODEL`` env var 3. ``video_gen.fal.model`` in ``config.yaml`` 4. ``video_gen.model`` in ``config.yaml`` (when it's one of our family IDs) 5. ``DEFAULT_MODEL`` Authentication via ``FAL_KEY`` or the managed Nous gateway. Output is an HTTPS URL from FAL's CDN; the gateway downloads and delivers it. ) annotationsN)AnyDictListOptionalTuple)VideoGenProvidererror_responsesuccess_responsez LTX 2.3 (22B)z~30-60scheapz322B model with native audio generation. Affordable.z fal-ai/ltx-2.3-22b/text-to-videoz!fal-ai/ltx-2.3-22b/image-to-videoT) displayspeedprice strengthstier text_endpointimage_endpoint aspect_ratios resolutions durationsaudionegativez Pixverse v6z~30-90sz.Affordable. Negative prompts. 1-15s durations.z fal-ai/pixverse/v6/text-to-videoz!fal-ai/pixverse/v6/image-to-video360p540p720p1080p)zVeo 3.1z~60-120spremiumzBGoogle DeepMind. Cinematic, native audio, strong prompt adherence.z fal-ai/veo3.1zfal-ai/veo3.1/image-to-video)16:99:16)rr4k)s) r rrrrrrrrrduration_suffixrrz Seedance 2.0z;ByteDance. Cinematic, synchronized audio + lip-sync, 4-15s.z$bytedance/seedance-2.0/text-to-videoz%bytedance/seedance-2.0/image-to-video)z21:9r!z4:31:1z3:4r")480prr)r$rFz Kling v3 4Kz ~120-300sz14K output, native audio (Chinese/English), 3-15s.z&fal-ai/kling-video/v3/4k/text-to-videoz'fal-ai/kling-video/v3/4k/image-to-videostart_image_urlr!r"r))r) r rrrrrrimage_param_keyrrrrrzHappy Horse 1.0uAAlibaba. New model, sparse public docs — conservative defaults.z!alibaba/happy-horse/text-to-videoz"alibaba/happy-horse/image-to-video)zltx-2.3 pixverse-v6zveo3.1z seedance-2.0z kling-v3-4kz happy-horsezDict[str, Dict[str, Any]] FAL_FAMILIESr/rct|trt|dk7rytd|Dsy|d|dz dkDS)zIHeuristic: a 2-tuple of ints with a gap > 1 is treated as ``(min, max)``.Fc3<K|]}t|tywN) isinstanceint).0ds Q/root/.hermes/venv/lib/python3.12/site-packages/plugins/video_gen/fal/__init__.py z%_is_duration_range..s5az!S!5srr)r5tuplelenall)rs r9_is_duration_ranger>sC i '3y>Q+> 595 5 Q<)A, & **c|jd}|sSt|rdS|dSt|r|\}}t|t|S|vrSt|fdS)Nrrc t|z Sr4)abs)r8durations r9z!_clamp_duration..sAL(9r?)key)getr>maxmin)familyrCrlohis ` r9_clamp_durationrLsw ;'I  *)4tF)A,F)$B2s2x())9 y9 ::r?c ddlm}|}t|tr|j dnd}t|tr|SiS#t $r"}t jd|icYd}~Sd}~wwxYw)Nr) load_config video_genz#Could not load video_gen config: %s)hermes_cli.configrNr5dictrF Exceptionloggerdebug)rNcfgsectionexcs r9_load_video_gen_sectionrXsb1m*4S$*?#''+&T$Wd3w;;  :C@ s$AAA A1A,&A1,A1cg}|j||jtjjdt }t |jdt r|jdni}t |t r |j|jd|jd}t |tr|j||D]W}t |ts|js%|jtvs<|j}|t|fcStttfS)z>Decide which FAL family to use. Returns ``(family_id, meta)``.FAL_VIDEO_MODELfalmodel) appendosenvironrFrXr5rQstrstripr0 DEFAULT_MODEL)explicit candidatesrUfal_cfgtopcfids r9_resolve_familyris&(Jhbjjnn%678 ! #C *3775>4 @cggenbG'4 '++g./ ''' C#s# * a !'') \0I'')C S)) )* ,}5 55r?ci} |r|| d<|r|jdxsd} || | <||| d<|jdr ||dvr|| d<|jdr ||dvr|| d<t||} | ,|jd r|jd d } | | | d <|jd r|t|| d<|jdr|r|| d<| S)zJBuild a family-specific payload, dropping keys the family doesn't declare.promptr. image_urlseedr aspect_ratior resolutionrr(rCrgenerate_audiornegative_prompt)rFrLbool) rIrkrlrCrnrorrrrmpayloadrEclampedsuffixs r9_build_payloadrws!G "jj*+:{    zz/" 6/2 2&2GN #zz-  . .$.GL !fh/Gvzz+6-r2!( &2  zz'u0$(K ! zz*/%4!" Nr? _fal_clientcttSt5ttcdddSddlm}|atcdddS#1swYyxYw)aoLazy-load the ``fal_client`` SDK and cache it on this module. Delegates the actual import to :func:`tools.fal_common.import_fal_client` so the ``lazy_deps`` ensure-install handling stays in one place. Thread-safe via double-checked locking: concurrent first calls import the SDK exactly once instead of each racing thread re-running the import. Nrimport_fal_client)rx_fal_client_locktools.fal_commonr{rzs r9_load_fal_clientr~-sK   " 7')  s AAA_managed_fal_video_client _managed_fal_video_client_configcNddlm}m}|r |dsyddlm}|dS)zsReturn managed fal-queue gateway config when the user prefers the gateway or direct FAL credentials are absent.r)fal_key_is_configuredprefers_gatewayrON)resolve_managed_tool_gatewayz fal-queue)tools.tool_backend_helpersrrtools.managed_tool_gatewayr)rrrs r9"_resolve_managed_fal_video_gatewayrJs&R{'CG ' 44r?c:ddlm}|jjd|jf}t 5t t|k(rt cdddSt|t|j|ja|at cdddS#1swYyxYw)zQReuse the managed FAL client so its internal httpx.Client is not leaked per call.r)_ManagedFalSyncClient/N)rEqueue_run_origin) r}rgateway_originrstripnous_user_token_managed_fal_video_client_lockrrr~rx)managed_gatewayr client_configs r9_get_managed_fal_video_clientrVs7 &&--c2''M ( ) $ 05UYf5f, ) ) $9 //,;;% ! ,9(( ) ) )sB4BBcbtdttji}t }|t j |||St|} |j |||S#t$r7}ddl m }||}|!d|cxkrdkrntd|d |d |d}~wwxYw) zSubmit a FAL video request using direct credentials or the managed queue gateway. Returns a request handle whose ``.get()`` blocks until the result is ready. zx-idempotency-keyN) argumentsheadersr)_extract_http_statusiiz-Nous Subscription gateway rejected endpoint 'z' (HTTP u). This model may not yet be enabled on the Nous Portal's FAL proxy. Either: • Set FAL_KEY in your environment to use FAL.ai directly, or • Pick a different model via `hermes tools` → Video Generation.) r~r`uuiduuid4rrxsubmitrrRr}r ValueError)endpointrrequest_headersrmanaged_clientrWrstatuss r9_submit_fal_video_requestrms *C ,=>O8:O!!(i!YY2?CN$$ #%   9%c*  #"5#"5 ?zJ!XY     sA.. B.72B))B.c6ddlm}|rytduS)zNTrue if the FAL.ai video backend is reachable (direct key or managed gateway).rrTN)rrrrs r9_check_fal_video_availablers@ - /t ;;r?c eZdZdZeddZeddZddZddZddZ ddZ ddZ d d d d d d d d d d dd Z y )FALVideoGenProviderzFAL.ai multi-family video generation backend. Routes between text-to-video and image-to-video endpoints automatically based on whether ``image_url`` was provided. cy)Nr[selfs r9namezFALVideoGenProvider.namer?cy)NFALrrs r9 display_namez FALVideoGenProvider.display_namerr?c6 tS#t$rYywxYw)NF)rrRrs r9 is_availablez FALVideoGenProvider.is_availables" -/ /  s c 8g}tjD]\}}g}|jdr|jd|jdr|jd|j||d|d|d|d|jd d |d |S) Nrtextrimager rrrrr )idr rrrr modalities)r0itemsrFr])routrhmetars r9 list_modelszFALVideoGenProvider.list_modelss$&%++- IC$&Jxx(!!&)xx()!!'* JJ ?g!+.g3(    r?ctSr4)rbrs r9 default_modelz!FALVideoGenProvider.default_modelsr?cdddddddgdS) Nrpaidu^LTX, Pixverse, Veo 3.1, Seedance 2.0, Kling 4K, Happy Horse — text-to-video & image-to-videoFAL_KEYzFAL.ai API keyzhttps://fal.ai/dashboard/keys)rErkurl)rbadgetagenv_varsrrs r9get_setup_schemaz$FALVideoGenProvider.get_setup_schemas*s%.:  r?c $ddggdgddddddd S) Nrrr,rrrTr)rrr max_duration min_durationsupports_audiosupports_negative_promptmax_reference_imagesrrs r9 capabilitiesz FALVideoGenProvider.capabilitiess)!7+4<"(,$%  r?Nr!r) r\rlreference_image_urlsrCrnrorrrrmc  tstddd|S t|xsdj }t |\} } |xsdj xsd}|r)| j d}d }|s=td | d d d| | S| j d}d}|std | dd d| | S|stddd| | St| ||||||| |  } t||}|j }t|tr|xsij dnd}d}t|tr|j d}nt|tr|}|stddd| | Sd|i}t|tr2|j dr|d|d<|j d r|d |d <t|| ||d!|vr|ndd"|vr(t!dj#d#|d"Dxsd$nd%d|&S#t$rtddd|cYSwxYw#t$r8}tjd| ||dtd|dd| ||cYd}~Sd}~wwxYw)'NuNo FAL backend available. Either set FAL_KEY (run `hermes tools` → Video Generation → FAL to configure) or sign in to Nous (`hermes setup`) for managed gateway access. auth_requiredr[)error error_typeproviderrkz@fal_client Python package not installed (pip install fal-client)missing_dependencyrprrz FAL family us has no image-to-video endpoint. Pick a family with image-to-video support via `hermes tools` → Video Generation.modality_unsupported)rrrr\rkrrzq has no text-to-video endpoint. Pass an image_url to use its image-to-video endpoint, or pick a different family.zprompt is required.missing_prompt)rkrlrCrnrorrrrmz1FAL video gen failed (family=%s, endpoint=%s): %sT)exc_infozFAL video generation failed: api_error)rrrr\rkrnvideorz%FAL returned no video URL in responseempty_responser file_size content_typernrCc3BK|]}|js|ywr4)isdigit)r7rgs r9r:z/FALVideoGenProvider.generate..cs Oq199; Os0r)rr\rkmodalityrnrCrextra)rr r~ ImportErrorrarirFrwrrRrSwarningr5rQr`r r6join)rrkr\rlrrCrnrorrrrmkwargs family_idrIimage_url_normr modality_usedrthandleresultrWrrrs r9generatezFALVideoGenProvider.generates*+!V+     ,B%%'+E2 6$/r002:d zz"23H#M%%i[1CD 6")Fzz/2H"M%%i[1OP 6")F!++i  ! $%!+   .xAFZZ\F0:&$/G2""7+T! eT "))E"C s #C!=+i  ",X 6 eT "yy%%*;%7k"yy((-n(=n%")77)B[eip[pS OGJ,? OOVSVWvw   !X/   |  NNC8S4  "5cU;&i)   s/ G0H0H  H  I-I I I)returnr`rrs)rzList[Dict[str, Any]])r Optional[str]rDict[str, Any])rkr`r\rrlrrzOptional[List[str]]rC Optional[int]rnr`ror`rrrrOptional[bool]rmrrrrr) __name__ __module__ __qualname____doc__propertyrrrrrrrrrr?r9rrs  &    $#'48"&" )- $"E E  E ! E 2 E  E E E 'E E E E  E r?rc6|jty)uFPlugin entry point — wire ``FALVideoGenProvider`` into the registry.N)register_video_gen_providerr)ctxs r9registerrns##$7$9:r?)rrrrs)rIrrCrrrr)rcrrzTuple[str, Dict[str, Any]])rIrrkr`rlrrCrrnr`ror`rrrrrrmrrr)rr)rr`rrr)rNone))r __future__rloggingr^ threadingrtypingrrrrragent.video_gen_providerr r r getLoggerrrSr0__annotations__rbr>rLrXrirwrxLockr|r~rrrrrrrrrrr?r9rsX@#  33   8 $.#J;="!E;=8  Y(8). "R?AG0"!HAC-0$%X<>ca+ 'aF +;0 661 1 1 1  1  11#1 1 11p S!9>>#0"&3%(, #,!/!1 5).B<J *J d;r?