Fij?+$UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl Z ddl m Z mZmZmZmZmZmZddlmZmZejeZddd d d d d d dZded<dZddddddddZddhZdZd)dZ d*d Z!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_imagesave_url_imagesuccess_response)hermes_xai_user_agentresolve_xai_http_credentialszGrok Imagine Imagez~5-10szFast, high-quality)displayspeed strengthszGrok Imagine Image (Quality)z~10-20sz9Higher fidelity / detail; slower than the standard model.)grok-imagine-imagezgrok-imagine-image-qualityzDict[str, Dict[str, Any]]_MODELSrz16:91:1z9:164:33:43:22:3) landscapesquareportraitrrrr1k2kreturnDict[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_configr3Ps 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*rr3r(str DEFAULT_MODEL) env_overrider. candidates r2_resolve_modelr>^s:>>"344L3 //W\222   C$.swww/?/?$E$EO   4I-Y'))'),,, '-0 00r:ct}t|dtr|dnd}|r |tvr|St S)zGet configured resolution. resolutionN)r3r(r*r:_XAI_RESOLUTIONSDEFAULT_RESOLUTION)r.ress r2_resolve_resolutionrEls\   C#-cggl.C.CS#I#I S#'',   tC s&&& r?cfeZdZdZeddZeddZddZdd Zdd Z e fddZ dS)XAIImageGenProviderz#xAI ``grok-imagine-image`` backend.r!r:cdS)Nr&selfs r2namezXAIImageGenProvider.name}sur?cdS)Nz xAI (Grok)rIrJs r2 display_namez XAIImageGenProvider.display_names|r?boolcbt}t|dS)Napi_key)rrOr*)rKcredss r2 is_availablez XAIImageGenProvider.is_availables',..EIIi(()))r?List[Dict[str, Any]]cHdtDS)Nc g|]H\}}||d||dd|dddIS)rrr)idrrr)r*).0model_idmetas r2 z3XAIImageGenProvider.list_models..sk   $ 88Ix88'2..!XXk266      r?)ritemsrJs r2 list_modelszXAIImageGenProvider.list_modelss-  #*--//    r?r"cdddgddS)NzxAI Grok Imagine (image)paiduHgrok-imagine-image — text-to-image; uses xAI Grok OAuth or XAI_API_KEYxai_grok)rLbadgetagenv_vars post_setuprIrJs r2get_setup_schemaz$XAIImageGenProvider.get_setup_schemas" /]$    r?prompt aspect_ratiokwargsrc zt}t|dpd}t|dpdpd}|st dd||St \}}t |} t| d} t} | tvr| nt} ||| | d } d |d td }t|d pd d} tj|d|| d}|n4#tj$r}|j}||jnd} |did|jdd}n0#t*$r#||jddnt|}YnwxYwt,d||t d|d|d|||| cYd}~Sd}~wtj$rt dd|||| cYStj$r"}t d|d |||| cYd}~Sd}~wwxYw |}n/#t*$r"}t d!|d"|||| cYd}~Sd}~wwxYw|d#g}|st d$d%|||| S|d}|d&}|d'}|rU t5|d(|)}n/#t*$r"}t d*|d+d||| cYd}~Sd}~wwxYwt|}nq|rZ t7|d(|)}t|}nJ#t*$r(}t,d,|||}Yd}~nd}~wwxYwt d-d%d||| Sd.| i}t;|||| d|/S)0z1Generate an image using xAI's grok-imagine-image.rQrWproviderr&zSNo xAI credentials found. Configure xAI OAuth in `hermes model` or set XAI_API_KEY.missing_api_key)error error_typerkrhr)r7rgrhrAzBearer zapplication/json) Authorizationz Content-Typez User-Agentbase_urlzhttps://api.x.ai/v1/z/images/generationsx)headersjsontimeoutNrrmmessagei,zxAI image gen failed (%d): %szxAI image generation failed (z): api_error)rmrnrkr7rgrhz%xAI image generation timed out (120s)ruzxAI connection error: connection_errorzxAI returned invalid JSON: invalid_responsedatazxAI returned no image dataempty_responseb64_jsonurlxai_)prefixzCould not save image to cache: io_errorzDxAI image URL %s could not be cached (%s); falling back to bare URL.z/xAI response contained neither b64_json nor URLrA)imager7rgrhrkextra)rr:r*stripr r>r _XAI_ASPECT_RATIOSrErBrCrrstriprequestspostraise_for_status HTTPErrorresponse status_coderttextr+r,rmTimeoutConnectionErrorr rwarningr)rKrgrhrirRrQ provider_namerZr[aspectxai_arrAxai_respayloadrsrprr1statuserr_msgresultrzfirstb64r} saved_path image_refrs r2generatezXAIImageGenProvider.generates-..eii **0b117799EIIj11:U;;AACCLu  !k,&)  ())$%l33#''66(** *.> > >**DV"! # # 1w00./11   uyy,,E0EFFLLNNUUVYZZ) }000 H  % % ' ' ' '!   |H-5-AX))qF T"--//--gr::>>y(-X\Y\X\J]^^ T T T191E(---3s88 T LL8&' J J J!JfJJJJ&&#          !=$&#    '   !4s44-&#         ]]__FF   !9C99-&#        zz&"%% !2+&#  Qii ##ii (  +C8Ix8I8IJJJ    %ACAA)""!!'  JII   ,+C8Ix8I8IJJJ  OO    Z   "G+#   '!       s/FJ5I&A G0/I0*HIH9IJ5'J5J5J0*J50J59K K:K5/K:5K:M44 N >NN N 6O P $PP Nr!r:)r!rO)r!rTr!r")rgr:rhr:rirr!r") __name__ __module__ __qualname____doc__propertyrLrNrSr^rfr rrIr?r2rGrGzs-- XX****          1c c c c c c c r?rGctxrNonecH|tdS)z3Register this provider with the image gen registry.N)register_image_gen_providerrG)rs r2registerrLs###$7$9$9:::::r?r)r!r4r)rrr!r)%r __future__rloggingr8typingrrrrrragent.image_gen_providerr r r r r rrtools.xai_httprr getLoggerrr,r__annotations__r;rrBrCr3r>rErGrrIr?r2rs "#""""" 33333333333333ONNNNNNN  8 $ $() 2P## & &    %     $<     1 1 1 1J J J J J *J J J d;;;;;;r?