$jJdZddlmZddlmZddlmZmZmZm Z ddl Z ddl m Z ddlmZddlmZmZmZmZmZdd lmZerdd lmZdd lmZdd lmZdd lmZe ddZ dZ!dZ"GddeZ#d5dZ$erdd lmZd6dZ%ddd7d"Z&d8d$Z'd9d%Z(dddddd&d'd:d/Z)dddddd&d'd:d0Z*d;d1Z+e d2dZ,dd,Z ddd?d/Z d@d2ZdddAd5ZdS)B VisibilityaSets visibility state on matching components. Does NOT filter inline - just marks components with visibility state. Later transforms in the chain can override earlier marks. Final filtering happens at the Provider level after all transforms run. Example: ```python # Disable components tagged "internal" Visibility(False, tags={"internal"}) # Re-enable specific tool (override earlier disable) Visibility(True, names={"safe_tool"}) # Allowlist via composition: Visibility(False, match_all=True) # disable everything Visibility(True, tags={"public"}) # enable public ``` NFnameskeysversiontags components match_allenabledboolrset[str] | NonerrVersionSpec | Nonerr =set[Literal['tool', 'resource', 'template', 'prompt']] | Noner!returnNonech||_||_||_||_||_||_||_dS)axInitialize a visibility marker. Args: enabled: If True, mark matching as enabled; if False, mark as disabled. names: Component names or URIs to match. keys: Component keys to match (e.g., {"tool:my_tool@v1"}). version: Component version spec to match. Unversioned components (version=None) will NOT match a version spec. tags: Tags to match (component must have at least one). components: Component types to match (e.g., {"tool", "prompt"}). match_all: If True, matches all components regardless of other criteria. N)_enabledrrrrr r!)selfr"rrrrr r!s l/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/transforms/visibility.py__init__zVisibility.__init__<s90      $"strcX|jrdnd}|jr d|jdSg}|jr*|dt |j|jr*|dt |j|jr|d|j|jr*|dt |j|jr*|d t |j|rd|d d |d Sd|d S) Nenabledisablez Visibility(z, match_all=True)znames=zkeys=zversion=z components=ztags=z, )) r*r!rappendsetrrr rjoin)r+actionpartss r,__repr__zVisibility.__repr__\sP!]9  > BAAAA A : 5 LL3#dj//33 4 4 4 9 3 LL1TY11 2 2 2 < 6 LL4DL44 5 5 5 ? ? LL=s4?';';== > > > 9 3 LL1TY11 2 2 2  ?>>>499U+;+;>>> >&V&&&&r. componentrcp|jrdS|j|j|j|j |jdS|j+|jdd}||jvrdS|j|j|jvrdS|jj|j|jv}d}t|trt|j |jv}n#t|tr|j|jv}|s|sdS|j#|j|jdsdS|jdupt!|j|jzS)anCheck if this transform applies to the component. All specified criteria must match (intersection semantics). An empty rule (no criteria) matches nothing. Use match_all=True to match everything. Args: component: Component to check. Returns: True if this transform should mark the component. TNF:r) match_none)r!rrrr rkeysplitname isinstancer r/urir uri_templatematchesr#)r+r:component_type matches_name matches_uris r,_matcheszVisibility._matchesosu > 4 J  ! $' !5 ? &&]0055NT_44u 9 }DI--u : !$>TZ7LK)X.. C!)-00DJ> I'788 C'4 B   K u < #DL,@,@  %-A- - #5yD DD$))C$D$DDr.rcl||s|S|jttd|jiii}ng|jti}|ti}i|d|ji}i|t|i}i|jt|i}|d|iS)zSet visibility state in component metadata if rule matches. Returns a copy of the component with updated metadata to avoid mutating shared objects cached in providers. N visibilitymeta)update)rHrK _FASTMCP_KEY _INTERNAL_KEYr*get model_copy)r+r:new_meta old_fastmcp old_internal new_internal new_fastmcps r,_mark_componentzVisibility._mark_components }}Y''   > !$}|T]6S&TUHH#.,,\2>>K&??="==LHlHL$-HHLF[F-FFKD).D, DDH##FH+=#>>>r.toolsSequence[Tool]c$Kfd|DS)zMark tools by visibility state.c:g|]}|SrV.0tr+s r, z)Visibility.list_tools..s'777A$$Q''777r.r[)r+rWs` r, list_toolszVisibility.list_toolss!77777777r.rr@ call_nextr Tool | Nonec^K|||d{V}|dS||S)zMark tool if found.rbNr\)r+r@rcrtools r,get_toolzVisibility.get_toolsLYtW555555555 <4##D)))r. resourcesSequence[Resource]c$Kfd|DS)z#Mark resources by visibility state.c:g|]}|Sr[r\)r^rr+s r,r`z-Visibility.list_resources..';;;A$$Q'';;;r.r[)r+rhs` r,list_resourceszVisibility.list_resourcess!;;;;;;;;r.rBr Resource | Nonec^K|||d{V}|dS||S)zMark resource if found.rbNr\)r+rBrcrresources r, get_resourcezVisibility.get_resourceM#3888888888  4##H---r. templatesSequence[ResourceTemplate]c$Kfd|DS)z,Mark resource templates by visibility state.c:g|]}|Sr[r\r]s r,r`z6Visibility.list_resource_templates..rmr.r[)r+rts` r,list_resource_templatesz"Visibility.list_resource_templatess#<;;;;;;;r.r ResourceTemplate | Nonec^K|||d{V}|dS||S)z Mark resource template if found.rbNr\)r+rBrcrtemplates r,get_resource_templatez Visibility.get_resource_templatersr.promptsSequence[Prompt]c$Kfd|DS)z!Mark prompts by visibility state.c:g|]}|Sr[r\)r^pr+s r,r`z+Visibility.list_prompts..s'999A$$Q''999r.r[)r+r}s` r, list_promptszVisibility.list_promptss!99999999r.r Prompt | Nonec^K|||d{V}|dS||S)zMark prompt if found.rbNr\)r+r@rcrprompts r, get_promptzVisibility.get_promptsL!yw777777777 >4##F+++r.)r"r#rr$rr$rr%rr$r r&r!r#r'r()r'r/r:rr'r#)r:rr'r)rWrXr'rX)r@r/rcrrr%r'rd)rhrir'ri)rBr/rcr rr%r'ro)rtrur'ru)rBr/rcr rr%r'ry)r}r~r'r~)r@r/rcr rr%r'r)__name__ __module__ __qualname____doc__r-r9rHrVrargrnrrrxr|rrr[r.r,rr'sr0"& $&* $######@''''&<E<E<E<E|????.8888 SW******<<<<'+ . . . . . ."<<<<'+ . . . . . .":::: UY,,,,,,,,r.rr:r'r#c|jpi}|ti}|ti}|ddS)a<Check if component is enabled. Returns True if: - No visibility mark exists (default is enabled) - Visibility mark is True Returns False if visibility mark is False. Args: component: Component to check. Returns: True if component should be enabled/visible to clients. rJT)rKrOrMrN)r:rKrinternals r, is_enabledrsJ > RDhh|R((G{{="--H << d + ++r.contextrlist[dict[str, Any]]c@K|dd{VpgS)z.Load visibility rule dicts from session state._visibility_rulesN) get_staters r,get_visibility_rulesr,s2""#677 7 7 7 7 7 7 =2=r.r rulesr r&r(cK|d|d{V|d|vr7|tjd{V|d|vsd|vr7|tjd{V|d|vr9|tjd{VdSdS)aSave visibility rule dicts to session state and send notifications. Args: context: The context to save rules for. rules: The visibility rules to save. components: Optional hint about which component types are affected. If None, sends notifications for all types (safe default). If provided, only sends notifications for specified types. rNrfrqr{r) set_statesend_notificationmcptypesToolListChangedNotificationResourceListChangedNotificationPromptListChangedNotification)rrr s r,save_visibility_rulesr1s   / 7 77777777Vz11'' (M(M(O(OPPPPPPPPPZ:55z9Q9Q'' (Q(Q(S(STTTTTTTTTX33'' (O(O(Q(QRRRRRRRRRRR43r.list[Visibility]cg}|D]X}d}|drS|d}t|d|d|d}|t|d|drt |dnd|d rt |d nd||d rt |d nd|d rt |d nd|d d Z|S)z,Convert rule dicts to Visibility transforms.Nrgtelteqrrrr"rrrr r!Fr)rOrr4rr5)r transformsparamsr version_dicts r,create_visibility_transformsrLsjJ   ::i  !),L! $$U++##D))##D))G  y!.4jj.A.AKc&/***t,2JJv,>,>HS(((D,2JJv,>,>HS(((D17L1I1ISC|,---t **[%88     r.cK |j}n#t$rgcYSwxYwt|d{V}t|S)zd:&&&$  D LL : F F FFFFFFFFFFr.cdK|r|nd}t|d{V}d|rt|nd|rt|nd|r|j|j|jdnd|rt|nd|rt|nd|d}||t |||d{VdS)aFDisable components matching criteria for this session only. Session rules override global transforms. Rules accumulate - each call adds a new rule to the session. Later marks override earlier ones (Visibility transform semantics). Sends notifications to this session only: ToolListChangedNotification, ResourceListChangedNotification, and PromptListChangedNotification. Args: context: The context for this session. names: Component names or URIs to match. keys: Component keys to match (e.g., {"tool:my_tool@v1"}). version: Component version spec to match. tags: Tags to match (component must have at least one). components: Component types to match (e.g., {"tool", "prompt"}). match_all: If True, matches all components regardless of other criteria. NFrrrrrs r,disable_componentsrs: *3tJ'w// / / / / / /E %/e4",T  GKwz D D D",T *4>d:&&&$  D LL : F F FFFFFFFFFFr.c6Kt|gd{VdS)a9Clear all session visibility rules. Use this to reset session visibility back to global defaults. Sends notifications to this session only: ToolListChangedNotification, ResourceListChangedNotification, and PromptListChangedNotification. Args: context: The context for this session. N)rrs r,reset_visibilityrs0  , ,,,,,,,,,,r. ComponentTSequence[ComponentT]cKddlm}|}||St|d{V}|s|St |}|D]fd|D}|S)aApply session-specific visibility transforms to components. This helper applies session-level enable/disable rules by marking components with their visibility state. Session transforms override global transforms due to mark-based semantics (later marks win). Args: components: The components to apply session transforms to. Returns: The components with session transforms applied. r)_current_contextNc:g|]}|Sr[r\)r^c transforms r,r`z,apply_session_transforms.. s'???1)++A..???r.)fastmcp.server.contextrrOrr)r r current_ctxsession_transformsresultrs @r,apply_session_transformsrs877777"&&((K5kBBBBBBBB *  F'@@ ??????? Mr.r)rrr'r)rrrrr r&r'r()rrr'r)rrr'r)rrrr$rr$rr%rr$r r&r!r#r'r()rrr'r()r rr'r).r __future__rcollections.abcrtypingrrrr mcp.typesrfastmcp.resources.baser fastmcp.resources.templater fastmcp.server.transformsr r r rrfastmcp.utilities.versionsrfastmcp.prompts.baserrrfastmcp.tools.baserfastmcp.utilities.componentsrrrMrNrrrrrrrrrrrr[r.r,rs#"""""$$$$$$777777777777++++++777777322222>++++++......''''''====== GC)***  e,e,e,e,e,e,e,e,P,,,,2/......>>>>QU SSSSSS68 / / / /" "& PT3G3G3G3G3G3Gr" "& PT3G3G3G3G3G3Gl - - - -W\); < < < r.