߲j|UdZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZej"eZdZd ed <d ZGd d eZd#dZd$dZdddd%dZddddddZdddd d&dZddd d'dZdd d d ed! d(d"ZddlZddl Z ddl!Z!y))aImage generation provider ABC. Providers register via ``PluginContext.register_image_gen_provider()`` (from ``/plugins/image_gen//`` or ``~/.hermes/plugins/image_gen//``); the one selected by ``image_gen.provider`` services every ``image_generate`` call. One tool covers text-to-image and editing: ``image_url`` / ``reference_image_urls`` route to the provider's edit endpoint, otherwise text-to-image. Mirrors ``agent/video_gen_provider.py``. Response dicts come from :func:`success_response` / :func:`error_response` (``success, image, model, prompt, aspect_ratio, modality, provider`` + ``error, error_type`` on failure). ) annotationsN)Path)AnyDictListOptionalTuple)provider_media)CatalogProviderBase) landscapesquareportraitzTuple[str, ...]VALID_ASPECT_RATIOSr c`eZdZdZddZej efddd ddZy)ImageGenProviderzAbstract base class for an image generation backend. Subclasses implement :attr:`name` and :meth:`generate`; ``list_models`` entries may add ``speed`` / ``strengths`` / ``price`` for the picker.cdgddS)z``modalities`` (``"text"`` and/or ``"image"``) and ``max_reference_images``. Surfaced in the dynamic tool schema so the model knows when ``image_url`` is honored; the text-only default keeps non-overriding providers backward compatible.textr) modalitiesmax_reference_images)selfs =/home/ubuntu/.hermes/hermes-agent/agent/image_gen_provider.py capabilitieszImageGenProvider.capabilities#s &hBBN) image_urlreference_image_urlsc y)aGenerate an image, or edit ``image_url`` (``reference_image_urls`` are extra style/composition refs, clamped to ``max_reference_images``); any source image routes to the edit endpoint. Return :func:`success_response` / :func:`error_response`. Unknown ``kwargs`` MUST be ignored (forward compat); ``upscale`` (bool) is a post-generation high-res pass, reported as ``upscaled: True`` in ``extra``.Nr)rprompt aspect_ratiorrkwargss rgeneratezImageGenProvider.generate)sr)returnDict[str, Any]) rstrrr$r Optional[str]rOptional[List[str]]r rr"r#) __name__ __module__ __qualname____doc__rabcabstractmethodDEFAULT_ASPECT_RATIOr!rrrrrs|=C  1 W $(48 W W W ! W 2 W W  W Wrrct|tr|jjnd}|tvr|St S)zClamp to :data:`VALID_ASPECT_RATIOS`; invalid values coerce to landscape so the tool surface forgives agent mistakes instead of rejecting them.) isinstancer$striplowerrr-)valuevs rresolve_aspect_ratior5:s7",E3!7 RA((1B.BBrct|tr|g}t|ttfsy|Dcgc]4}t|ts|j s%|j 6c}xsdScc}w)zCoerce a str or list into a clean list of non-blank strings; ``None`` when nothing usable remains so providers treat "no refs" as one sentinel.N)r0r$listtupler1)r3items rnormalize_reference_imagesr:AsW% edE] +%* UTjs.C DJJL U ]Y]] UsA/A/A/imagepngprefix extensionc4tjd|||S)zNDecode base64 image data into ``$HERMES_HOME/cache/images/``; return the path.imagesr=)r save_b64)b64_datar>r?s rsave_b64_imagerDKs  " "8XfPY ZZrjpgwebpgif)z image/pngz image/jpegz image/jpgz image/webpz image/gifgN@i)r>timeout max_bytesc Jtjd||||dtdddd S)zDownload an (often ephemeral) image URL into ``$HERMES_HOME/cache/images/``. Raises on network / HTTP / oversize / empty errors so callers can fall back to the bare URL.rAi)r<rEjpegrFrGr<Imagez3Image at {url} returned 0 bytes; refusing to cache.) r>rHrI chunk_size content_typesurl_extensionsdefault_extensionlabel empty_error)r save_url_URL_IMAGE_CONTENT_TYPES)urlr>rHrIs rsave_url_imagerVUs4  " "#fg,Dr$r?r$r"r) rUr$r>r$rHfloatrIintr"r)r;r$r[r$rr$rr$r\r$rWr$rXzOptional[Dict[str, Any]]r"r#)rer$rcr$r\r$r[r$rr$rr$r"r#)"r* __future__rr+loggingpathlibrtypingrrrrr agentr agent.provider_baser getLoggerr'loggerr__annotations__r-rr5r:rDrTrVrarfbase64datetimeuuidrrrrus # 33 3   8 $(K_J"W*W8C^4;U[ e%vdi 'P`    16 JM   (&*         $.',           ( r