߲j$UdZddlmZddlZddlmZddlmZmZddl m Z m Z m Z m Z ejeZdZdZd Zd Zd Zd ZeeeehZd ed<eGddZddZddZddZ d dZd!dZ d"dZ d#dZ d$dZ!Gdde"Z#d%dZ$eZ%eZ& d!dZ' d#dZ(y)&zHermes middleware contract helpers. Observer hooks report what happened. Middleware can change what happens by rewriting a request or wrapping the actual execution callback. Agent-loop call sites and plugins share this vocabulary. ) annotationsN)deepcopy) dataclassfield)AnyCallableDictListzhermes.observer.v1zhermes.middleware.v1 tool_requesttool_execution llm_request llm_executionzset[str]VALID_MIDDLEWAREcPeZdZUdZded<ded<dZded<eeZd ed <y ) RequestMiddlewareResultz;Result of applying request middleware to a mutable payload.rpayloadoriginal_payloadFboolchanged)default_factoryList[Dict[str, Any]]traceN) __name__ __module__ __qualname____doc____annotations__rrlistr:/home/ubuntu/.hermes/hermes-agent/hermes_cli/middleware.pyrrs*E LGT"'"=E =r rc 2|jdt|S)Ntelemetry_schema_version) setdefaultOBSERVER_SCHEMA_VERSIONkwargss r!observer_payloadr('s 02IJ Mr c ^|jdt|jdt|S)Nr#middleware_schema_version)r$r%MIDDLEWARE_SCHEMA_VERSIONr&s r!middleware_payloadr,,s+ 02IJ 13LM Mr c t|S#t$r=}tjd|t |t r t |n|cYd}~Sd}~wwxYw)zDeep-copy a request payload, tolerating non-deepcopyable members. An LLM request can carry clients/callbacks/file handles; a hard ``deepcopy`` failure would otherwise abort the whole request-middleware pass. z,H,HI 4&$' zz+. ,- \*4 6::c?2%C8U J   U2x23 4 #(DKu   s#-Cc ddlm}|ts t||St |}t tdg|ft ||d|S)z]Apply registered LLM request middleware; ``{"request": {...}}`` replaces the provider kwargs.rhas_middleware)rrrequestrMoriginal_request)r<rLLLM_REQUEST_MIDDLEWARErr4rI)rMcontextrLrOs r!apply_llm_request_middlewarerRXsV1 0 1&wQQ!'*  2/? +,?O SZ r c t|}t|}g}t|jdxsd}t|j dd}|r?|s=ddlm}|j|||} | |k7rt| }|jdd idd l m } | tst|s|n||t|| Sttd ||f|||d |S)zApply registered tool request middleware; ``{"args": {...}}`` replaces the effective tool arguments before hooks, guardrails, approvals, and execution see them. session_id skip_relayFr) relay_runtime)rT tool_nameargsr7 nemo_relayrKr;rYrXrY original_args)r4r>r=rpopagentrWapply_tool_request_interceptsr?r<rLTOOL_REQUEST_MIDDLEWARErrI) rXrYrQr\ current_argsrrTrVrW relay_argsrLs r!apply_tool_request_middlewarercfs t$Mm,L"$EW[[.4"5Jgkk,67J*'"@@!Y\AK  %%j1L LL(L1 21 1 2& %D<$Ku     ,m OV r c Jtt|f||jd|d|S)zCRun provider execution through registered LLM execution middleware.rOrN)_run_execution_chainLLM_EXECUTION_MIDDLEWAREr]rM next_callrQs r!run_llm_execution_middlewareris8  ) _'++6H'*R _V] __r c Ltt|f|||jd|d|S)z@Run tool execution through registered tool execution middleware.r\r[)reTOOL_EXECUTION_MIDDLEWAREr])rXrYrhrQs r!run_tool_execution_middlewarerls: !9 e$gkk/SW6X e\c eer c$eZdZdZdfd ZxZS)_DownstreamExecutionErrorzMarks an exception raised BELOW a middleware frame so the frame's own failure handling (skip-and-continue) doesn't swallow it.cDt|t|||_yN)super__init__r>rB)selfrB __class__s r!rrz"_DownstreamExecutionError.__init__s X'  r )rB BaseExceptionreturnNone)rrrrrr __classcell__)rts@r!rnrns/!!r rnc ddlm}dvrdnd|tjj gs Sdfd dS)Nr)_delivery_managerrMrYc bt k\rS ddddd  fd }td i }|<||d< d i|S#t$r}|jd}~wt$r8} j ||drcYd}~Sr dzcYd}~Sd}~wwxYw) NFc r&tddtdtdd dz| n|dS#t$r}t ||d}~wwxYw)Nz Middleware 'z ' callback rzF called next_call() more than once; downstream execution is single-useT) RuntimeErrorgetattrreprr.rn) rEr3call_atcallbackindexr@ next_called next_resultnext_succeededrs r!rhz8_run_execution_chain..call_at..next_calls""4& xT(^DEFUU K >%eaiL/4#= >sA A AArh Middleware)surfacer}rp)rErrvrr)lenr,rnrBr._report_hook_failure)rrrh call_kwargsr3rrrrr callbacksr@r'managerrA terminal_calls`` @@@@r!rz%_run_execution_chain..call_ats C N " ) )U#   > >$)262 #* K #, K  /*k* *( ,,  /  ( (xcS_ ( `""519g. . /s6 A B. A** B.6B)B.B)#B.)B.)rintrrrvr)r<rzr _middlewarer=)r@rr'rzrrrrAs``` @@@@r!reresk4(F2)K!GW((,,T267I VK011*/*/X 1f[) **r c t|fi|S)z7Compatibility wrapper for older ``api_request`` naming.)rR)rMrQs r!apply_api_request_middlewarers ( ;7 ;;r c t||fi|S)z9Compatibility wrapper for older ``api_execution`` naming.)rirgs r!run_api_execution_middlewarers ( Fg FFr )r'rrvDict[str, Any])rrrvr) r@r>rAr>rrrBrr'rrvr)rMrrQrrvr)rXr>rYrrQrrvr)rMrrhCallable[[Dict[str, Any]], Any]rQrrvr) rXr>rYrrhrrQrrvr)r@r>rzCallable[[Any], Any]r'rrvr))r __future__rloggingcopyr dataclassesrrtypingrrr r getLoggerrr/r%r+r`rkrPrfrrrr(r,r4rIrRrcrirlr.rnreAPI_EXECUTION_MIDDLEWAREAPI_REQUEST_MIDDLEWARErrrr r!rs #(,,   8 $.2(,&*68NPh(  >> >  G (<HKWZ2 (58B_ _(G_TW_\__ee(e5Teadee! !5+z4/< <<<G G.GG Gr