yj UdZddlmZddlZddlmZmZmZddlm Z ddl m Z m Z m Z ddlmZmZd3d Zd3d Zd4dZd5dZd6dZd7dZdddddddddddddddgd d!Zd"ed#<d$d%ddd&ddddd'd(ddddd)ddgd d!Zd"ed*<d8d-Zd8d.Zejdd/eeegd0d12ejd$d/eeegd0d12dS)9z1xAI-specific Imagine video edit and extend tools.) annotationsN)AnyDictOptional) load_config)has_xai_video_credentialsrun_xai_video_editrun_xai_video_extend)registry tool_errorreturnboolc t}n#t$rYdSwxYwt|tr|dnd}t|to|ddkS)NF video_genproviderxai)r Exception isinstancedictget)cfgsections ;/home/ragecks/.hermes/hermes-agent/tools/xai_video_tools.py_configured_for_xai_videorsmm uu&0d&;&;Ecggk"""G gt $ $ IZ)@)@E)IIs  c:to tSN)rrr_check_xai_video_requirementsrs $ & & F+D+F+FFrvaluer Optional[str]ct|tr(|r|SdSr)rstrstripr s r _clean_stringr&s6%%++--{{}} 4r Optional[int]c|dSt|trdS t|S#ttf$rYdSwxYwr)rrint TypeError ValueErrorr%s r _coerce_intr,%s[ }t%t5zz z "tts,AAr#c4tjdddddS)NFzzxAI video edit/extend tools require `video_gen.provider` to be configured as `xai` via `hermes tools` -> Video Generation.provider_not_configuredr)successerror error_typer)jsondumpsrrr_provider_not_configured_errorr40s1 : J0  r video_urlc~t|}|sdS|dr|SdS)z9Require a public HTTPS MP4 URL (``http``/``https`` only).N)zhttp://zhttps://)r&lower startswith)r5cleaneds r_normalize_public_video_urlr:<sEI&&G t}}!!"9:: 4rxai_video_editzEdit an existing video with xAI Imagine. This is separate from `video_generate` because video editing is provider-specific. `video_url` must be the public HTTPS MP4 URL from a prior Imagine result (`video` or `public_url` on files-cdn).objectstringz7Instruction for how xAI should modify the source video.)type descriptionuiPublic HTTPS MP4 URL of the source video — the `video` or `public_url` from a prior xAI Imagine result.z$Optional xAI Imagine model override.promptr5modelrA)r> propertiesrequired)namer? parametersDict[str, Any]XAI_VIDEO_EDIT_SCHEMAxai_video_extendzExtend an existing video with xAI Imagine. This is separate from `video_generate` because video extension is provider-specific. `video_url` must be the public HTTPS MP4 URL from a prior Imagine result (`video` or `public_url` on files-cdn).z9Instruction for how xAI should continue the source video.integerzNDesired extension duration in seconds. xAI clamps this to its supported range.rAr5durationrBXAI_VIDEO_EXTEND_SCHEMAargs_kwc t|d}t|d}t|d}|stdS|stdSt st St |||}tj|S)NrAr5rBz%prompt is required for xAI video edit_video_url must be a public HTTPS MP4 URL (the `video`/`public_url` from a prior Imagine result)r@) r&rr:r rr4r r2r3)rNrOrAr5rBresults r_handle_xai_video_editrSs 488H-- . .F+DHH[,A,ABBI $((7++ , ,E CABBB   +    % & &0-/// F :f  rc t|d}t|d}t|d}t|d}|st dS|st dSt st St||||}tj |S)NrAr5rBrLz'prompt is required for xAI video extendrQrK) r&rr:r,r rr4r r2r3)rNrOrAr5rBrLrRs r_handle_xai_video_extendrUs 488H-- . .F+DHH[,A,ABBI $((7++ , ,E488J//00H ECDDD   +    % & &0-/// ! F :f  rrFvideo)rEtoolsetschemahandlercheck_fn requires_envis_asyncemoji)r r)r rr r!)r rr r')r r#)r5rr r!)rNrGrOrr r#)__doc__ __future__rr2typingrrrhermes_cli.configrplugins.video_gen.xairr r tools.registryr r rrr&r,r4r:rH__annotations__rMrSrUregisterrrrrfs777"""""" &&&&&&&&&&)))))) 0///////JJJJGGGG       9 !X !D!E  "{+'))B  9 !Z !D".!E'  0{+5$+$+$$$$N.2   " *         " $ *        r