JjxdZddlmZddlZddlmZdZehdZdd Z dd Z dd Z ddZ ddZ ddZdS)zGateway response filtering helpers. These helpers operate at the gateway boundary: they decide whether a completed agent turn should be delivered to the chat, not what should be persisted in the conversation history. ) annotationsN)AnyNO_REPLY>NO REPLY[SILENT]SILENTrtextstrreturncd|S)N )joinstripuppersplit)r s >/home/rurouni/.hermes/hermes-agent/gateway/response_filters.py_canonical_silence_candidaters4 88DJJLL&&((..00 1 11cpd}t|}||kry||dvrotj||drB|dz }||kr7||dvr-tj||dB||kr||dz dvrxtj||dz drH|dz}||kr=||dz dvr0tj||dz dH|||S)zStrip stray edge punctuation without erasing marker structure. Models sometimes emit ``.NO_REPLY`` or ``*NO_REPLY*`` instead of the exact marker. Keep square brackets structural so malformed ``[SILENT`` does not become ``SILENT``. rz[]P)len unicodedatacategory startswithr)r startends r_strip_edge_silence_punctuationrsD E d))C #++$u+T11k6J4PU;6W6W6b6bcf6g6g1   #++$u+T11k6J4PU;6W6W6b6bcf6g6g1 ++$sQw-t33 8LTRUXYRY]8[8[8f8fgj8k8k3 q ++$sQw-t33 8LTRUXYRY]8[8[8f8fgj8k8k3 c ? " ""rtuple[str, ...]ct|}t|}||kr|fSt|}||fSN)rrr)r exactstrippedfallbacks r_canonical_silence_candidatesr%/sV ( . .E.tzz||<z2is_intentional_silence_response..Fs(qqIy77qqqqqqr) isinstancer rranyr%)r&r#s ris_intentional_silence_responser18st h $ $u~~H u 8}}ru qqIfgoIpIpqqq q qqr agent_result dict | Nonec|t|tsdS|drdSt|S)zBSilence markers suppress delivery only for successful agent turns.Ffailed)r/dictgetr1)r2r&s r#is_intentional_silence_agent_resultr8IsB lD ) )u!!u *8 4 44rct|tsdS|}|rt|dkrdSt |D]'r#t fdt DrdS(dS)uaReturn True while ``text`` could still resolve to a silence marker. The streaming path accumulates the reply delta-by-delta and must decide, before the whole response is known, whether to show what it has so far. A buffer whose canonical form is a non-empty *prefix* of a silence marker (e.g. ``"NO"`` on the way to ``"NO_REPLY"``, or an exact marker that has not yet been terminated by stream-end) is held back so a raw marker is never edited onto the screen and then belatedly retracted. Anything that has already diverged from every marker (ordinary prose) — and anything longer than the marker cap — returns False so normal streaming resumes immediately. This is the streaming counterpart to :func:`is_intentional_silence_response`, sharing the same marker set and canonicalization so the two never drift. Fr)c3BK|]}|VdSr!)r)r,markerr-s rr.z,is_partial_silence_marker..hs1ddfV..y99ddddddrT)r/r rrr%r0r+)r r#r-s @ris_partial_silence_markerr<Rs dC uzz||H s8}}r))u28<<  ddddHcddddd 44 5r)r r r r )r r r r)r&rr r')r2r3r&rr r')r rr r')__doc__ __future__rrtypingrSILENT_REPLY_TOKEN frozensetr+rrr%r1r8r<rrrCs#"""""  (i)))2222 # # # # rrrr"5555r