iJdZddlmZddlmZddlmZmZmZm Z ddl Z ddl m Z ddlmZddlmZmZmZmZmZdd lmZerdd lmZdd lmZdd lmZdd lmZe ddZ dZ!dZ"GddeZ#d"dZ$erdd lmZd#dZ%dd d$dZ&d%dZ'd&dZ(ddddddd d'dZ)ddddddd d'dZ*d(dZ+e d dZ, d)d!Z-y)*aVisibility transform for marking component visibility state. Each Visibility instance marks components via internal metadata. Multiple visibility transforms can be stacked - later transforms override earlier ones. Final filtering happens at the Provider level. ) annotations)Sequence) TYPE_CHECKINGAnyLiteralTypeVarN)Resource)ResourceTemplate) GetPromptNextGetResourceNextGetResourceTemplateNext GetToolNext Transform) VersionSpec)Prompt)Context)Tool)FastMCPComponentTr)boundfastmcp _internalceZdZdZddddddd ddZddZddZddZdd Zdd  dd Z dd Z dd  dd Z ddZ dd  ddZ ddZdd  ddZy) 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_allcf||_||_||_||_||_||_||_y)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!)selfenabledrrrrr r!s t/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/transforms/visibility.py__init__zVisibility.__init__<s50      $"c|jrdnd}|jrd|jdSg}|jr'|jdt |j|j r'|jdt |j |j r|jd|j |jr'|jdt |j|jr'|jd t |j|rd|d d j|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__\s !]]  >> /@A A :: LL6#djj/!23 4 99 LL5TYY 01 2 << LL8DLL#34 5 ?? LL;s4??';&<= > 99 LL5TYY 01 2  499U+;* >VHA&&r(c(|jry|j1|j%|j|j |j y|j-|j jdd}||jvry|j|j |jvry|jy|j|jv}d}t|tr"t|j|jv}n(t|tr|j|jv}|s|sy|j(|jj|jdsy|j duxs"t!|j |j zS)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 strurir uri_templatematchesbool)r$ componentcomponent_type matches_name matches_uris r&_matcheszVisibility._matchesosc >> JJ  ! $' ! ?? &&]]005NT__4 99 }}DII- :: !$>>TZZ7LK)X.!)--0DJJ> I'78'44 B  K << #DLL,@,@   %-A- yyD DD$)))C$DDr(cr|j|s|S|jttd|jiii}nj|jj ti}|j ti}i|d|ji}i|t|i}i|jt|i}|j 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. visibilitymeta)update)rCrF _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&TUH#..,,\2>K&??="=LHlHL$--HLF[F-FKD)..D, DH##FH+=#>>r(cRK|Dcgc]}|j|c}Scc}ww)zMark tools by visibility state.rQ)r$toolsts r& list_toolszVisibility.list_toolss)167A$$Q'777 '"'rcZK|||d{}|y|j|S7w)zMark tool if found.rXNrS)r$r8 call_nextrtools r&get_toolzVisibility.get_tools5tW55 <##D))6 +)+cRK|Dcgc]}|j|c}Scc}ww)z#Mark resources by visibility state.rS)r$ resourcesrs r&list_resourceszVisibility.list_resourcess)1:;A$$Q';;;rWcZK|||d{}|y|j|S7w)zMark resource if found.rXNrS)r$r;rZrresources r& get_resourcezVisibility.get_resource6#388  ##H--9r]cRK|Dcgc]}|j|c}Scc}ww)z,Mark resource templates by visibility state.rS)r$ templatesrUs r&list_resource_templatesz"Visibility.list_resource_templatess+2;;A$$Q';;;rWcZK|||d{}|y|j|S7w)z Mark resource template if found.rXNrS)r$r;rZrtemplates r&get_resource_templatez Visibility.get_resource_templaterer]cRK|Dcgc]}|j|c}Scc}ww)z!Mark prompts by visibility state.rS)r$promptsps r& list_promptszVisibility.list_promptss)189A$$Q'999rWcZK|||d{}|y|j|S7w)zMark prompt if found.rXNrS)r$r8rZrprompts r& get_promptzVisibility.get_prompts5!w77 >##F++8r])r%r>rset[str] | NonerrsrVersionSpec | Nonerrsr =set[Literal['tool', 'resource', 'template', 'prompt']] | Noner!r>returnNone)rvr:r?rrvr>)r?rrvr)rTSequence[Tool]rvry)r8r:rZrrrtrvz Tool | None)r_Sequence[Resource]rvrz)r;r:rZr rrtrvzResource | None)rgSequence[ResourceTemplate]rvr{)r;r:rZr rrtrvzResourceTemplate | None)rmSequence[Prompt]rvr|)r8r:rZr rrtrvz Prompt | None)__name__ __module__ __qualname____doc__r'r2rCrQrVr\rardrhrkrorrr(r&rr's\0"& $&* $## #  # $ #### #@'&<E|?.8 SW**$/*=O* *<'+ .  .# . $ .  ."<3< #<'+ .  .+ . $ . ! .": UY,,$1,?Q, ,r(rc|jxsi}|jti}|jti}|jddS)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. rET)rFrJrHrI)r?rFrinternals r& is_enabledrsC >> RDhh|R(G{{="-H << d ++r(cHK|jdd{xsgS7w)z.Load visibility rule dicts from session state._visibility_rulesN) get_statecontexts r&get_visibility_rulesr,s#""#67 7 =2= 7s "  "r cK|jd|d{|d|vr5|jtjj d{|d|vsd|vr5|jtjj d{|d|vr6|jtjj d{yy777E7 w)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. rNr[rcrjrq) set_statesend_notificationmcptypesToolListChangedNotificationResourceListChangedNotificationPromptListChangedNotification)rrulesr s r&save_visibility_rulesr1s   / 777Vz1'' (M(M(OPPPZ:5z9Q'' (Q(Q(STTTX3'' (O(O(QRRR48 QTRsECC:CC>CC:C CCCCCc"g}|D]}d}|jdr@|d}t|jd|jd|jd}|jt|d|jdrt |dnd|jd rt |d nd||jd rt |d nd|jd rt |d nd|jd d  |S)z,Convert rule dicts to Visibility transforms.Nrgtelteqrrrr%rrrr r!Fr)rJrr-rr.)r transformsparamsr version_dicts r&create_visibility_transformsrLsJ ::i !),L! $$U+##D)##D)G  y!.4jj.Ac&/*t,2JJv,>S(D,2JJv,>S(D17L1IC|,-t **[%8  , r(cK |j}t|d{}t|S#t$rgcYSwxYw7 w)z?1)++A.?( MC@s!)A-A&A-A( A-(A-rx)rrrvlist[dict[str, Any]])rrrrr rurvrw)rrrvlist[Visibility])rrrvr)rrrrsrrsrrtrrsr rur!r>rvrw)rrrvrw)r Sequence[ComponentT]rvr).r __future__rcollections.abcrtypingrrrr mcp.typesrfastmcp.resources.resourcer fastmcp.resources.templater fastmcp.server.transformsr r r rrfastmcp.utilities.versionsrfastmcp.prompts.promptrrrfastmcp.tools.toolrfastmcp.utilities.componentsrrrHrIrrrrrrrrrrrrr(r&rs#$77/73-.'= C)*  e,e,P,2.>QU S S SN S  S68 /" "& PT3G 3G 3G  3G  3G  3GN3G3G 3Gr" "& PT3G 3G 3G  3G  3G  3GN3G3G 3Gl -\); < $r(