PL j' UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl Z ddl m Z mZmZmZmZmZddlmZmZejeZddd d d d d d dZded<dZddddddddZddhZdZd)dZd*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_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 B/home/kuhnn/.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)Nz xAI (Grok)rHrIs r1 display_namez XAIImageGenProvider.display_names|r>boolcbt}t|dS)Napi_key)rrNr))rJcredss r1 is_availablez XAIImageGenProvider.is_availables',..EIIi(()))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!cdddgddS)NzxAI Grok Imagine (image)paiduHgrok-imagine-image — text-to-image; uses xAI Grok OAuth or XAI_API_KEYxai_grok)rKbadgetagenv_vars post_setuprHrIs r1get_setup_schemaz$XAIImageGenProvider.get_setup_schemas" /]$    r>prompt aspect_ratiokwargsrc t}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|}n|r|}nt d,d%d||| Sd-| i}t7|||| d|.S)/z1Generate an image using xAI's grok-imagine-image.rPrVproviderr%zSNo xAI credentials found. Configure xAI OAuth in `hermes model` or set XAI_API_KEY.missing_api_key)error error_typerjrgr)r6rfrgr@zBearer zapplication/json) Authorizationz Content-Typez User-Agentbase_urlzhttps://api.x.ai/v1/z/images/generationsx)headersjsontimeoutNrrlmessagei,zxAI image gen failed (%d): %szxAI image generation failed (z): api_error)rlrmrjr6rfrgz%xAI image generation timed out (120s)rtzxAI connection error: connection_errorzxAI returned invalid JSON: invalid_responsedatazxAI returned no image dataempty_responseb64_jsonurlxai_)prefixzCould not save image to cache: io_errorz/xAI response contained neither b64_json nor URLr@)imager6rfrgrjextra)rr9r)stripr r=r _XAI_ASPECT_RATIOSrDrArBrrstriprequestspostraise_for_status HTTPErrorresponse status_coderstextr*r+rlTimeoutConnectionErrorr r)rJrfrgrhrQrP provider_namerYrZaspectxai_arr@xai_respayloadrrrorr0statuserr_msgresultryfirstb64r| saved_path image_refrs r1generatezXAIImageGenProvider.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  II!G+#   '!       s/FJ5I&A G0/I0*HIH9IJ5'J5J5J0*J50J59K K:K5/K:5K:M44 N >NN N Nr r9)r rN)r rSr r!)rfr9rgr9rhrr r!) __name__ __module__ __qualname____doc__propertyrKrMrRr]rer rrHr>r1rFrFys-- XX****          1R R R R R R R r>rFctxrNonecH|tdS)z3Register this provider with the image gen registry.N)register_image_gen_providerrF)rs r1registerr:s###$7$9$9:::::r>r)r r3r)rrr r)$r __future__rloggingr7typingrrrrrragent.image_gen_providerr r r r r rtools.xai_httprr getLoggerrr+r__annotations__r:rrArBr2r=rDrFrrHr>r1rs"#""""" 33333333333333ONNNNNNN  8 $ $() 2P## & &    %     $<     1 1 1 1y y y y y *y y y B;;;;;;r>