yj;)DUdZddlmZddlZddlmZddlmZmZddl m Z m Z m Z m Z ejeZdZdZd Zd Zd Zd ZeZeZeeeehZd ed<eGddZd3dZd3dZd4dZd5dZd6dZ d5d Z!d7d#Z"d8d$Z#d7d%Z$d9d(Z%d:d*Z&d;d,Z'daHermes middleware contract helpers. Observer hooks report what happened. Middleware can change what happens by rewriting a request or wrapping the actual execution callback. Keep the small contract helpers here so agent-loop call sites and plugins share one vocabulary. ) annotationsN)deepcopy) dataclassfield)AnyCallableDictListzhermes.observer.v1zhermes.middleware.v1 tool_requesttool_execution llm_request llm_executionzset[str]VALID_MIDDLEWAREcXeZdZUdZded<ded<dZded<eeZd ed <d S) RequestMiddlewareResultz;Result of applying request middleware to a mutable payload.rpayloadoriginal_payloadFboolchanged)default_factoryzList[Dict[str, Any]]traceN) __name__ __module__ __qualname____doc____annotations__rrlistr;/home/ragecks/.hermes/hermes-agent/hermes_cli/middleware.pyrr%s\EELLLG"'%"="="=E======rrkwargsrreturnDict[str, Any]c <|dt|S)Ntelemetry_schema_version) setdefaultOBSERVER_SCHEMA_VERSIONr!s r observer_payloadr)/s 02IJJJ Mrc r|dt|dt|S)Nr%middleware_schema_version)r&r'MIDDLEWARE_SCHEMA_VERSIONr(s r middleware_payloadr-4s7 02IJJJ 13LMMM Mrrc t|S#t$rP}td|t |t rt |cYd}~S|cYd}~Sd}~wwxYw)aDeep-copy a request payload, tolerating non-deepcopyable members. Request payloads are normally plain JSON-shaped dicts, but an LLM request can occasionally carry non-deepcopyable objects (clients, callbacks, file handles). A hard ``deepcopy`` failure there would otherwise abort the whole request-middleware pass. Fall back to a shallow ``dict`` copy so middleware still runs and the original nested objects are shared by reference rather than corrupting the live payload. zA&A+A& A+&A+requestcontextc ttst||dgSt|}t|}g}t tf||d|D]t}t |t s|d}t |t sCt|}|t|ut||t||S)zApply registered LLM request middleware. Middleware may return ``{"request": {...}}`` to replace the effective provider kwargs before Hermes sends them. Frrrrr6original_requestr6) _has_middlewareLLM_REQUEST_MIDDLEWARErr5_invoke_middlewarer2r3getappend _trace_entryr)r6r7r;current_requestrresult next_requests r apply_llm_request_middlewarerEMs! 1 2 2 &$     "'** !122O"$E$)   + + &$''  zz),, ,--  $\22 \&))**** ")U     r tool_namestrargsc 2t|}t|}g}t|dpd}t|dd}|rL|sJddlm}||||} | |kr&t| }|dd itts$t|s|n||t|| Sttf|||d |D]t} t| ts| d } t| tsCt| }|t| ut||t|| S) zApply registered tool request middleware. Middleware may return ``{"args": {...}}`` to replace the effective tool arguments before hooks, guardrails, approvals, and execution see them. session_id skip_relayFr) relay_runtime)rJrFrHsource nemo_relayr9rFrH original_argsrH)r5rGr?rpopagentrMapply_tool_request_interceptsr@r<TOOL_REQUEST_MIDDLEWARErr>r2r3rA) rFrHr7rQ current_argsrrJrLrM relay_argsrC next_argss r apply_tool_request_middlewarerYxst$$Mm,,L"$EW[[..4"55Jgkk,6677J 3* 3''''''"@@!A   % %%j11L LL(L1 2 2 2 2 3 3 & %7DD<!KK     % #    + +&$''  JJv&& )T**  !),,  \&))**** "&U     rc t|fi|S)z7Compatibility wrapper for older ``api_request`` naming.)rE)r6r7s r apply_api_request_middlewarer[s ( ; ;7 ; ;;r next_callCallable[[Dict[str, Any]], Any]c tt}|s ||Stt||f||d|d|S)zCRun provider execution through registered LLM execution middleware.r;r:)_get_middleware_callbacksLLM_EXECUTION_MIDDLEWARE_run_execution_chainrR)r6r\r7 callbackss r run_llm_execution_middlewarercsp **BCCI "y!!!    %7AA       rc tt}|s ||Stt||f|||d|d|S)z@Run tool execution through registered tool execution middleware.rQrP)r_TOOL_EXECUTION_MIDDLEWARErarR)rFrHr\r7rbs r run_tool_execution_middlewarerfsp**CDDI y !  kk/488      rc t||fi|S)z9Compatibility wrapper for older ``api_execution`` naming.)rc)r6r\r7s r run_api_execution_middlewarerhs ( F Fg F FFrkind List[Any]c 4ddlm}||fitdi|S)Nr)invoke_middlewarer)hermes_cli.pluginsrlr-)rir!rls r r>r>s<444444  T B B%7%A%A&%A%A B BBrrc$ddlm}||S)Nr)has_middleware)rmro)riros r r<r<s$111111 >$  rList[Callable]cnddlm}t|j|gS)Nr)get_plugin_manager)rmrrr _middlewarer?)rirrs r r_r_s?555555 ""$$044T2>> ? ??rrb terminal_callCallable[[Any], Any]c dvrdndGddtd fd d S) Nr6rHc eZdZdfd ZxZS)7_run_execution_chain.._DownstreamExecutionErrororiginal BaseExceptionr"Nonecrtt|||_dSN)super__init__rGry)selfry __class__s r rz@_run_execution_chain.._DownstreamExecutionError.__init__s, GG  S]] + + +$DMMMr)ryrzr"r{)rrrr __classcell__)rs@r _DownstreamExecutionErrorrxs= % % % % % % % % % %rrindexintrrr"c t kr S dddd d  f d }td i }|<||d< d i|S# $r }|jd}~wt$r^}td t dt|rcYd}~Sr d zcYd}~Sd}~wwxYw) NF next_payloadrr"c  r2tddtdtdd dz| n|d S#t$r}||d}~wwxYw)Nz Middleware 'z ' callback rzF called next_call() more than once; downstream execution is single-useT) RuntimeErrorgetattrreprr/) rr4rcall_atcallbackrri next_called next_resultnext_succeededrs r r\z8_run_execution_chain..call_at..next_calls  "U4UUxT(^^DDUUU K >%geaiL > >//44#= >sA A+ A&&A+r\z&Middleware '%s' callback %s raised: %srrr})rrr"rr)lenr-ryr/r0warningrr)rrr\ call_kwargsr4rrrrrrrbrir! payload_keyrts`` @@@@r rz%_run_execution_chain..call_at s C NN " " =)) )U#   > > > > > > > > > > > > > > >()22622 #* K #, K  /8**k** *(   ,  / / / NN8*d8nn==      #""""""" 7519g.. . . . . . . /s6A""C'A.. C;r<r_rarArrr rsR#"""""((((((((,,,,,,,,,,,,  8 $ $.2(,&*03  >>>>>>> >  &((((V8888v<<<<&*GGGGCCCC     @@@@ >+>+>+>+Br