iK%UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl Z ddl m Z mZmZmZmZmZddlmZejeZdZddd d d iZd ed <dZddddddddZdddZdZd'dZd(dZ d)d Z!Gd!d"eZ"d*d&Z#dS)+axAI image generation backend. Exposes xAI's ``grok-imagine-image`` model as an :class:`ImageGenProvider` implementation. Features: - Text-to-image generation - Multiple aspect ratios (1:1, 16:9, 9:16, etc.) - Multiple resolutions (1K, 2K) - Base64 output saved to cache Selection precedence (first hit wins): 1. ``XAI_IMAGE_MODEL`` env var 2. ``image_gen.xai.model`` in ``config.yaml`` 3. :data:`DEFAULT_MODEL` ) annotationsN)AnyDictListOptionalTuple)DEFAULT_ASPECT_RATIOImageGenProvidererror_responseresolve_aspect_ratiosave_b64_imagesuccess_response)hermes_xai_user_agentzgrok-imagine-imagezGrok Imagine Imagez~5-10szFast, high-quality)displayspeed strengthszDict[str, Dict[str, Any]]_MODELSz16:91:1z9:164:33:43:22:3) landscapesquareportraitrrrr10242048)1k2krreturnDict[str, Any]cn ddlm}|}t|tr|dnd}t|tr|dnd}t|tr|niS#t $r'}t d|icYd}~Sd}~wwxYw)z(Read ``image_gen.xai`` from config.yaml.r) load_config image_genNxaiz'Could not load image_gen.xai config: %s)hermes_cli.configr# isinstancedictget Exceptionloggerdebug)r#cfgsection xai_sectionexcs C/home/ubuntu/.hermes/hermes-agent/plugins/image_gen/xai/__init__.py_load_xai_configr2Os 111111kmm*4S$*?*?I#''+&&&T,6w,E,EOgkk%(((4 (d;;C{{C  >DDD sBB B4 B/)B4/B4Tuple[str, Dict[str, Any]]cttjd}|r|tvr|t|fSt }t |dt r|dnd}|r|tvr|t|fStttfS)z:Decide which model to use and return ``(model_id, meta)``.XAI_IMAGE_MODELmodelN)osenvironr)rr2r'str DEFAULT_MODEL) env_overrider- candidates r1_resolve_modelr=]s:>>"344L3 //W\222   C$.swww/?/?$E$EO   4I-Y'))'),,, '-0 00r9ct}t|dtr|dnd}|r |tvr|St S)zGet configured resolution. resolutionN)r2r'r)r9_XAI_RESOLUTIONSDEFAULT_RESOLUTION)r-ress r1_resolve_resolutionrDks\   C#-cggl.C.CS#I#I S#'',   tC s&&& r>cfeZdZdZeddZeddZddZdd Zdd Z e fddZ dS)XAIImageGenProviderz#xAI ``grok-imagine-image`` backend.r r9cdS)Nr%selfs r1namezXAIImageGenProvider.name|sur>cdS)N xAI (Grok)rHrIs r1 display_namez XAIImageGenProvider.display_names|r>boolcDttjdS)N XAI_API_KEY)rOr7getenvrIs r1 is_availablez XAIImageGenProvider.is_availablesBIm,,---r>List[Dict[str, Any]]cHdtDS)Nc g|]H\}}||d||dd|dddIS)rrr)idrrr)r)).0model_idmetas r1 z3XAIImageGenProvider.list_models..sk   $ 88Ix88'2..!XXk266      r>)ritemsrIs r1 list_modelszXAIImageGenProvider.list_modelss-  #*--//    r>r!cdddddddgdS) NrMpaidz2Native xAI image generation via grok-imagine-imagerQz xAI API keyzhttps://console.x.ai/)keyprompturl)rKbadgetagenv_varsrHrIs r1get_setup_schemaz$XAIImageGenProvider.get_setup_schemas4 G)+2   r>rb aspect_ratiokwargsrc Ptjdd}|stddd|St \}}t |}t |d}t} t| d} t||| d } d |d td } tjd pd d} tj| d| | d}|nK#tj$r}|jr |jjnd} |jdid|jjdd}n:#t*$r-|jr|jjddnt-|}YnwxYwt.d||td|d|dd|||cYd}~Sd}~wtj$rtddd|||cYStj$r"}td|d d|||cYd}~Sd}~wwxYw |}n/#t*$r"}td!|d"d|||cYd}~Sd}~wwxYw|d#g}|std$d%d|||S|d}|d&}|d'}|rU t7|d(|)}n/#t*$r"}td*|d+d|||cYd}~Sd}~wwxYwt-|}n|r|}ntd,d%d|||Sd-| i}t9||||d|.S)/z1Generate an image using xAI's grok-imagine-image.rQrWz5XAI_API_KEY not set. Get one at https://console.x.ai/missing_api_keyr%)error error_typeproviderrhrr)r6rbrhr@zBearer zapplication/json) Authorizationz Content-Typez User-Agent XAI_BASE_URLzhttps://api.x.ai/v1/z/images/generationsx)headersjsontimeoutrrlmessageNi,zxAI image gen failed (%d): %szxAI image generation failed (z): api_error)rlrmrnr6rbrhz%xAI image generation timed out (120s)ruzxAI connection error: connection_errorzxAI returned invalid JSON: invalid_responsedatazxAI returned no image dataempty_responseb64_jsonrcxai_)prefixzCould not save image to cache: io_errorz/xAI response contained neither b64_json nor URLr@)imager6rbrhrnextra)r7rRstripr r=r _XAI_ASPECT_RATIOSr)rDrA API_MODELrrstriprequestspostraise_for_status HTTPErrorresponse status_coderttextr*r9r+rlTimeoutConnectionErrorr r)rJrbrhriapi_keyrZr[aspectxai_arr@xai_respayloadrsbase_urlrr0statuserr_msgresultrzfirstb64rc saved_path image_refrs r1generatezXAIImageGenProvider.generates>)M2..4466 !M,)  ())$%l33#''66(** "&&z6::"! # # 1w00./11   In--F1FMMOOVVWZ[[( }000 H  % % ' ' ' '!   14DS\--1F P,++--11'2>>BB9clN_`dad`dNeff P P P58\O#,+DSD11s3xx P LL8&' J J J!JfJJJJ&#          !=$#    '   !4s44-#         ]]__FF   !9C99-#        zz&"%% !2+#  Qii ##ii   +C8Ix8I8IJJJ    %ACAA)""!!'  JII  II!G+#   '!       s>/D//I7>HAF('H(4GHG9HI7'I7I7I2,I72I7;J J<J71J<7J<!L66 M"MM"M"Nr r9)r rO)r rTr r!)rbr9rhr9rirr r!) __name__ __module__ __qualname____doc__propertyrKrNrSr^rgr rrHr>r1rFrFys-- XX....          "1O O O O O O O r>rFctxrNonecH|tdS)z3Register this provider with the image gen registry.N)register_image_gen_providerrF)rs r1registerr7s###$7$9$9:::::r>r)r r3r)rrr r)$r __future__rloggingr7typingrrrrrragent.image_gen_providerr r r r r rtools.xai_httpr getLoggerrr+rr__annotations__r:rrArBr2r=rDrFrrHr>r1rs"#""""" 33333333333333100000  8 $ $ ! ')&%             1 1 1 1v v v v v *v v v |;;;;;;r>