Fij)&UdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZmZejeZdZded<d ZGd d ejZd6dZd7dZdddd8dZddddddZdddd d9d&Zdd'd:d/Zd0d1d1d1ed2d;d5ZdS)/plugins/image_gen//`` (built-in, auto-loaded as ``kind: backend``) or ``~/.hermes/plugins/image_gen//`` (user, opt-in via ``plugins.enabled``). Response shape -------------- All providers return a dict that :func:`success_response` / :func:`error_response` produce. The tool wrapper JSON-serializes it. Keys: success bool image str | None URL or absolute file path model str provider-specific model identifier prompt str echoed prompt aspect_ratio str "landscape" | "square" | "portrait" provider str provider name (for diagnostics) error str only when success=False error_type str only when success=False ) annotationsN)Path)AnyDictListOptionalTuple) landscapesquareportraitzTuple[str, ...]VALID_ASPECT_RATIOSr ceZdZdZeejddZeddZddZ dd Z dd Z dd Z eje fddZdS)ImageGenProvideruAbstract base class for an image generation backend. Subclasses must implement :meth:`generate`. Everything else has sane defaults — override only what your provider needs. returnstrcdS)zStable short identifier used in ``image_gen.provider`` config. Lowercase, no spaces. Examples: ``fal``, ``openai``, ``replicate``. Nselfs =/home/ubuntu/.hermes/hermes-agent/agent/image_gen_provider.pynamezImageGenProvider.name:c4|jS)zMHuman-readable label shown in ``hermes tools``. Defaults to ``name.title()``.)rtitlers r display_namezImageGenProvider.display_nameBsy   rboolcdS)zReturn True when this provider can service calls. Typically checks for a required API key. Default: True (providers with no external dependencies are always available). Trrs r is_availablezImageGenProvider.is_availableGs trList[Dict[str, Any]]cgS)a Return catalog entries for ``hermes tools`` model picker. Each entry:: { "id": "gpt-image-1.5", # required "display": "GPT Image 1.5", # optional; defaults to id "speed": "~10s", # optional "strengths": "...", # optional "price": "$...", # optional } Default: empty list (provider has no user-selectable models). rrs r list_modelszImageGenProvider.list_modelsOs  rDict[str, Any]c|jddgdS)a5Return provider metadata for the ``hermes tools`` picker. Used by ``tools_config.py`` to inject this provider as a row in the Image Generation provider list. Shape:: { "name": "OpenAI", # picker label "badge": "paid", # optional short tag "tag": "One-line description...", # optional subtitle "env_vars": [ # keys to prompt for {"key": "OPENAI_API_KEY", "prompt": "OpenAI API key", "url": "https://platform.openai.com/api-keys"}, ], } Default: minimal entry derived from ``display_name``. Override to expose API key prompts and custom badges. )rbadgetagenv_vars)rrs rget_setup_schemaz!ImageGenProvider.get_setup_schema`s"*%    r Optional[str]ch|}|r|ddSdS)z7Return the default model id, or None if not applicable.ridN)r"get)rmodelss r default_modelzImageGenProvider.default_model{s6!!##  '!9==&& &trprompt aspect_ratiokwargsrc dS)u'Generate an image. Implementations should return the dict from :func:`success_response` or :func:`error_response`. ``kwargs`` may contain forward-compat parameters future versions of the schema will expose — implementations should ignore unknown keys. Nr)rr0r1r2s rgeneratezImageGenProvider.generaterrN)rr)rr)rr )rr#)rr*)r0rr1rr2rrr#)__name__ __module__ __qualname____doc__propertyabcabstractmethodrrrr"r)r/DEFAULT_ASPECT_RATIOr4rrrrr3s    X !!!X!"    6 1         rrvaluer*rrct|tstS|}|t vr|StS)zClamp an aspect_ratio value to the valid set, defaulting to landscape. Invalid values are coerced rather than rejected so the tool surface is forgiving of agent mistakes. ) isinstancerr<striplowerr )r=vs rresolve_aspect_ratiorCsL eS ! !$## A  rrc`ddlm}|dz dz }|dd|S)zBReturn ``$HERMES_HOME/cache/images/``, creating parents as needed.r)get_hermes_homecacheimagesT)parentsexist_ok)hermes_constantsrEmkdir)rEpaths r_images_cache_dirrMsF000000 ?  w & 1DJJtdJ+++ Krimagepng)prefix extensionb64_datarPrQc2tj|}tjd}t jjdd}t|d|d|d|z }| ||S)zDecode base64 image data and write it under ``$HERMES_HOME/cache/images/``. Returns the absolute :class:`Path` to the saved file. Filename format: ``__.``. %Y%m%d_%H%M%SN_.) base64 b64decodedatetimenowstrftimeuuiduuid4hexrM write_bytes)rRrPrQrawtsshortrLs rsave_b64_imagerds  8 $ $C     ) )/ : :B JLL RaR E   F!E!ER!E!E%!E!E)!E!E EDS Krjpgwebpgif)z image/pngz image/jpegz image/jpgz image/webpz image/gifgN@i)rPtimeout max_bytesurlrhfloatriintc ddl}|||d}||jdpdddd}t|}|W|d dd}d D]&} |d | r | d krd n| }n'|d}tj d} tj jdd} t|d| d| d |z } d} | d5}|dD]}|s| t%|z } | |krS| | n#t*$rYnwxYwt-d|d|dzd|| dddn #1swxYwY| dkr9 | n#t*$rYnwxYwt-d|d| S)uDownload an image URL and write it under ``$HERMES_HOME/cache/images/``. Used by providers (xAI, fallback OpenAI) whose API returns an *ephemeral* URL instead of inline base64 — those URLs frequently expire before a downstream consumer (Telegram ``send_photo``, browser fetch) can resolve them, so we materialise the bytes locally at tool-completion time. Mirrors :func:`save_b64_image`'s shape so providers can swap in one line. Returns the absolute :class:`Path` to the saved file. Raises on any network / HTTP / oversize / non-image-content-type error so callers can fall back to returning the bare URL with a clear error message. rNT)rhstreamz Content-Typer%;?)rOrejpegrfrgrWrrrerOrTrUrVwbi) chunk_sizez Image at z exceeds izMB cap; refusing to cache.z% returned 0 bytes; refusing to cache.)requestsr-raise_for_statusheaderssplitr@rA_URL_IMAGE_CONTENT_TYPESendswithrZr[r\r]r^r_rMopen iter_contentlencloseunlinkOSError ValueErrorwrite)rjrPrhriruresponse content_typerQurl_pathextrbrcrL bytes_writtenfhchunks rsave_url_imagers &OOO||C|>>H  $((88>BEEc1MMaPVVXX^^``L(,,\::I99S!$$Q'--//8  C  S++ %(F]]EE       ) )/ : :B JLL RaR E   F!E!ER!E!E%!E!E)!E!E EDM 4B**i*@@  E  SZZ 'My(( KKMMMMD dddiK.Hddd HHUOOOO    KKMMMM    D OSOOOPPP KsIAH0G$#H0$ G1.H00G112H00H47H4I I$#I$)extramodelr0r1providerrOptional[Dict[str, Any]]r#c|d|||||d}|r0|D]\}}||||S)zBuild a uniform success response dict. ``image`` may be an HTTP URL or an absolute filesystem path (for b64 providers like OpenAI). Callers that need to pass through additional backend-specific fields can supply ``extra``. T)successrNrr0r1r)items setdefault) rNrr0r1rrpayloadkrBs rsuccess_responserse $ G %KKMM % %DAq   q! $ $ $ $ Nrprovider_errorr%) error_typerrr0r1errorrc dd||||||dS)z$Build a uniform error response dict.FN)rrNrrrr0r1rr)rrrrr0r1s rerror_responser1s+ $   r)r=r*rr)rr)rRrrPrrQrrr) rjrrPrrhrkrirlrr)rNrrrr0rr1rrrrrrr#)rrrrrrrrr0rr1rrr#)r8 __future__rr:rXrZloggingr]pathlibrtypingrrrrr getLoggerr5loggerr __annotations__r<ABCrrCrMrdryrrrrrrrs8#"""""  33333333333333  8 $ $(KJJJJ"\ \ \ \ \ sw\ \ \ H      2 % BBBBBBX'+@',r