PL jc: dZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl Z ddlmZmZddlmZmZejeZdZdZd Zd Zd ZdGdZdHdZdIdZdIdZdJdZ dKdZ!dLdZ"dMdZ#dNd Z$dOd#Z% dPdQd-Z&d.d/d0d1d2d3d4d5d1id6d7d4d5d1id8d7d1d9d3d1d:d3d;dd%d=d?d&gd@dAZ'dBZ(ej)d.d.e'e(e!dCgdDdEFdS)RuX Search tool backed by xAI's built-in ``x_search`` Responses API tool. Authentication -------------- The tool registers when **either** xAI credential path is available: * ``XAI_API_KEY`` is set in ``~/.hermes/.env`` or the process environment (paid xAI API key), OR * The user is signed in via xAI Grok OAuth — SuperGrok subscription — i.e. ``hermes auth add xai-oauth`` has been run and the stored refresh token still works. Credential preference at call time matches :func:`tools.xai_http.resolve_xai_http_credentials`: SuperGrok OAuth first, direct OAuth resolver second, ``XAI_API_KEY`` last. That helper also auto-refreshes the OAuth access token when it's within the refresh skew window, so a ``True`` from :func:`check_x_search_requirements` means the bearer is fetchable AND non-empty. Salvaged from PR #10786 (originally by @Jaaneek); credential resolution reworked to honor both auth modes per Teknium's design. ) annotationsN)AnyDictListOptionalTuple)registry tool_error)hermes_xai_user_agentresolve_xai_http_credentialszhttps://api.x.ai/v1zgrok-4.20-reasoning returnDict[str, Any]ct ddlm}|dipiS#t$ricYSwxYw)Nr load_configx_search)hermes_cli.configrget Exceptionrs 7/home/kuhnn/.hermes/hermes-agent/tools/x_search_tool.py_load_x_search_configr3s^111111{}}  R006B6  s %( 77strct}t|dpdptS)Nmodel)rrrstripDEFAULT_X_SEARCH_MODEL)cfgs r_get_x_search_modelr"<s>  ! !C   &B ' ' - - / / I3IJintct}|dt} tdt |S#t $r tcYSwxYw)Ntimeout_seconds)rr DEFAULT_X_SEARCH_TIMEOUT_SECONDSmaxr$rr! raw_values r_get_x_search_timeout_secondsr,Asd  ! !C)+KLLI02s9~~&&& 000////0AAAct}|dt} tdt |S#t $r tcYSwxYw)Nretriesr)rrDEFAULT_X_SEARCH_RETRIESr)r$rr*s r_get_x_search_retriesr1Jsc  ! !C #;<Ws ) * *E%))I&&,"--3355G   O   599Z((@,@AAGGIIPPQTUUH :&&/% 0 0F Hf $$r#boolc t}tt|dpdS#t $rYdSwxYw)a)Return True when xAI credentials are available AND valid. ``resolve_xai_http_credentials`` calls :func:`hermes_cli.auth.resolve_xai_oauth_runtime_credentials` which auto-refreshes the OAuth access token if it's expiring; a successful return therefore implies a usable bearer. r4rF)r r?rrrr)r<s rcheck_x_search_requirementsrAmsg,..C ),,23399;;<<< uusAA A! A!handlesOptional[List[str]] field_name List[str]cg}|pgD]O}t|pdd}|r||Pt |t krt |dt d|S)Nr@z supports at most z handles)rrlstripappendlen MAX_HANDLES ValueError)rBrDcleanedhandle normalizeds r_normalize_handlesrPsG-R''2&&,,..55c::  ' NN: & & & 7||k!!JOO+OOOPPP Nr#payloadc4t|dpd}|r|Sg}|dgpgD]}|ddkr|dgpgD]h}|d}|dvrMt|dpd}|r||id |S) N output_textroutputtypemessagecontent>textrSrXz )rrrrIjoin)rQrSpartsitemrWctyperXs r_extract_response_textr]s$gkk-006B77==??KE Hb))/R'' 88F  y ( ( xx 2..4" ' 'GKK''E///7;;v..4"55;;=='LL&&&  ' ;;u   # # % %%r#List[Dict[str, Any]]c g}|dgpgD]}|ddkr|dgpgD]}|dgpgD]}|ddkr||dd|d d|d |d d |S) NrTrUrVrWr url_citationurlrtitle start_index end_index)rarbrcrd)rrI)rQ citationsr[rW annotations r_extract_inline_citationsrgs&(I Hb))/R 88F  y ( ( xx 2..4"  G%kk-<<B   >>&))^;;  )~~eR88!+!r9rrPr"r,r1rangerequestspostr raise_for_status HTTPErrorrologgerwarningrstimesleepmin ReadTimeoutConnectionErrorrpr]listrrgdumpsrmrU__name__r)rtrurvrwrxryrzr4r5r=rhallowedexcludedtool_defrQr& max_retriesrkattempterdatarrers r x_search_toolrs < <:;;;$$7$9$9!66 $$$#c((########$A $%68KLL%&8:NOO  bx b`aa a$*J#7  4,3H( )  6-5H) * ??   6$-OO$5$5H[ ! ==?? 2")--//HY  % :59H1 2 % :59H1 2)**#${{}} Z   899+-- 04[1_--" :" :G! :#=+++)<7)<)<(:&;&=&= !+   ))+++% : : :%gaT&B&BMSWXX &+*;*;w+?U?UDaK!O'**   3sC7Q;$78899999999((*BC : : :k))EaK!O   3sC7Q;$78899999999 :  KLL L}}'--+..4"55 4T::z!%+" ) &$4             *A ===z !",Q//"1gg.              -q4 @@@z !"b9V9X9Xbbb"1gg.             *A ===z !"Q"1gg.         s: A*A%A*%A*.2M2!CM25AF:7M2:K BIM2K3AK M2 KB!M22R?AOR?R?0AQ R? R?AR:4R?:R?rzSearch X (Twitter) posts, profiles, and threads using xAI's built-in X Search tool. Use this for current discussion, reactions, or claims on X rather than general web pages. Available when xAI credentials are configured (SuperGrok OAuth or XAI_API_KEY).objectstringzWhat to look up on X.)rU descriptionarrayrUz;Optional list of X handles to include exclusively (max 10).)rUitemsrz/Optional list of X handles to exclude (max 10).z)Optional start date in YYYY-MM-DD format.z'Optional end date in YYYY-MM-DD format.booleanz?Whether xAI should analyze images attached to matching X posts.)rUrdefaultz?Whether xAI should analyze videos attached to matching X posts.rtrurvrwrxryrz)rU propertiesrequired)namer parametersc vt|dd|d|d|dd|ddt|ddt|d d S) NrtrrurvrwrxryFrzr)rrr?)argskws r_handle_x_searchrs hhw##((#67788$899((;++B''#'1Mu(U(U#V#V#'1Mu(U(U#V#V   r# XAI_API_KEYu🐦i)rtoolsetschemahandlercheck_fn requires_envemojimax_result_size_chars)rr)rr)rr$)rr2)rr?)rBrCrDrrrE)rQrrr)rQrrr^)rhrirr)NNrrFF)rtrrurCrvrCrwrrxrryr?rzr?rr)*__doc__ __future__rrploggingosrtypingrrrrrrtools.registryr r tools.xai_httpr r getLoggerrrr:r r(r0rKrr"r,r1r>rArPr]rgrsrX_SEARCH_SCHEMArregisterr#rrs.#"""""  33333333333333////////NNNNNNNN  8 $ $,.#&  KKKK 0000((((%%%%,    &&&&&$(<.2.2',',R R R R R l  ; !6   (+\""   (+P## !J !H "` ++ "` ++9! ! DII%%..b        ( !      r#