
    iXT                        d Z ddlmZ ddlmZmZ ddlmZ ddlm	Z	 ddl
mZmZmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZm Z  erddl!m"Z"  G d d      Z#y)a  Base Provider class for dynamic MCP components.

This module provides the `Provider` abstraction for providing tools,
resources, and prompts dynamically at runtime.

Example:
    ```python
    from fastmcp import FastMCP
    from fastmcp.server.providers import Provider
    from fastmcp.tools import Tool

    class DatabaseProvider(Provider):
        def __init__(self, db_url: str):
            super().__init__()
            self.db = Database(db_url)

        async def _list_tools(self) -> list[Tool]:
            rows = await self.db.fetch("SELECT * FROM tools")
            return [self._make_tool(row) for row in rows]

        async def _get_tool(self, name: str) -> Tool | None:
            row = await self.db.fetchone("SELECT * FROM tools WHERE name = ?", name)
            return self._make_tool(row) if row else None

    mcp = FastMCP("Server", providers=[DatabaseProvider(db_url)])
    ```
    )annotations)AsyncIteratorSequence)asynccontextmanager)partial)TYPE_CHECKINGLiteralcast)Self)Prompt)Resource)ResourceTemplate)
Visibility)Tool)gather)FastMCPComponent)VersionSpecversion_sort_key)	Transformc                     e Zd ZdZddZd dZed!d       Zd"dZd#dZ	d$dZ
	 d%	 	 	 	 	 d&d	Zd'd
Z	 d%	 	 	 	 	 d(dZd)dZ	 d%	 	 	 	 	 d*dZd+dZ	 d%	 	 	 	 	 d,dZd$dZ	 d%	 	 	 	 	 d&dZd'dZ	 d%	 	 	 	 	 d(dZd)dZ	 d%	 	 	 	 	 d*dZd+dZ	 d%	 	 	 	 	 d,dZd-dZed.d       Zddddddd	 	 	 	 	 	 	 	 	 	 	 	 	 d/dZdddddd	 	 	 	 	 	 	 	 	 	 	 d0dZy)1Providera"  Base class for dynamic component providers.

    Subclass and override whichever methods you need. Default implementations
    return empty lists / None, so you only need to implement what your provider
    supports.

    Provider semantics:
        - Return `None` from `get_*` methods to indicate "I don't have it" (search continues)
        - Static components (registered via decorators) always take precedence over providers
        - Providers are queried in registration order; first non-None wins
        - Components execute themselves via run()/read()/render() - providers just source them

    Error handling:
        - `list_*` methods: Errors are logged and the provider returns empty (graceful degradation).
          This allows other providers to still contribute their components.
    c                    g | _         y N)_transformsselfs    m/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/providers/base.py__init__zProvider.__init__E   s
    ,.    c                4    | j                   j                   dS )Nz())	__class____name__r   s    r   __repr__zProvider.__repr__H   s    ..))*"--r   c                ,    t        | j                        S )z8All transforms applied to components from this provider.)listr   r   s    r   
transformszProvider.transformsK   s     D$$%%r   c                :    | j                   j                  |       y)a   Add a transform to this provider.

        Transforms modify components (tools, resources, prompts) as they flow
        through the provider. They're applied in order - first added is innermost.

        Args:
            transform: The transform to add.

        Example:
            ```python
            from fastmcp.server.transforms import Namespace

            provider = MyProvider()
            provider.add_transform(Namespace("api"))
            # Tools become "api_toolname"
            ```
        N)r   append)r   	transforms     r   add_transformzProvider.add_transformP   s    $ 		*r   c                     ddl m}  || |      S )a}  Return a new provider with this transform applied (immutable).

        Unlike add_transform() which mutates this provider, wrap_transform()
        returns a new provider that wraps this one. The original provider
        is unchanged.

        This is useful when you want to apply transforms without side effects,
        such as adding the same provider to multiple aggregators with different
        namespaces.

        Args:
            transform: The transform to apply.

        Returns:
            A new provider that wraps this one with the transform applied.

        Example:
            ```python
            from fastmcp.server.transforms import Namespace

            provider = MyProvider()
            namespaced = provider.wrap_transform(Namespace("api"))
            # provider is unchanged
            # namespaced returns tools as "api_toolname"
            ```
        r   )_WrappedProvider))fastmcp.server.providers.wrapped_providerr,   )r   r)   r,   s      r   wrap_transformzProvider.wrap_transformd   s    8 	Oi00r   c                   K   | j                          d{   }| j                  D ]  }|j                  |       d{   } |S 7 07 
w)u  List tools with all transforms applied.

        Applies transforms sequentially: base → transforms (in order).
        Each transform receives the result from the previous transform.
        Components may be marked as disabled but are NOT filtered here -
        filtering happens at the server level to allow session transforms to override.

        Returns:
            Transformed sequence of tools (including disabled ones).
        N)_list_toolsr&   
list_tools)r   toolsr)   s      r   r1   zProvider.list_tools   sH      &&((I#..u55E ) )5   AA'AA		A	ANc                    K   dd fd}|} j                   D ]  }t        |j                  |      }  |||       d{   S 7 w)a  Get tool by transformed name with all transforms applied.

        Note: This method does NOT filter disabled components. The Server
        (FastMCP) performs enabled filtering after all transforms complete,
        allowing session-level transforms to override provider-level disables.

        Args:
            name: The transformed tool name to look up.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The tool if found (may be marked disabled), None if not found.
        Nc                D   K   j                  | |       d {   S 7 wr   )	_get_toolnversionr   s     r   basezProvider.get_tool.<locals>.base   s     73333     	call_nextr9   r   )r8   strr9   VersionSpec | NonereturnTool | None)r&   r   get_toolr   namer9   r:   chainr)   s   `     r   rC   zProvider.get_tool   sH     "	4 II..%@E ) 41111   ?A	AA	c                   K   | j                          d{   }| j                  D ]  }|j                  |       d{   } |S 7 07 
w)z}List resources with all transforms applied.

        Components may be marked as disabled but are NOT filtered here.
        N)_list_resourcesr&   list_resources)r   	resourcesr)   s      r   rJ   zProvider.list_resources   sI     
 ..00	I'66yAAI ) 1Ar3   c                    K   dd fd}|} j                   D ]  }t        |j                  |      }  |||       d{   S 7 w)a  Get resource by transformed URI with all transforms applied.

        Note: This method does NOT filter disabled components. The Server
        (FastMCP) performs enabled filtering after all transforms complete.

        Args:
            uri: The transformed resource URI to look up.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The resource if found (may be marked disabled), None if not found.
        Nc                D   K   j                  | |       d {   S 7 wr   )_get_resourceur9   r   s     r   r:   z#Provider.get_resource.<locals>.base   s      ++Aw7777r;   r<   r>   r   )rP   r?   r9   r@   rA   Resource | None)r&   r   get_resourcer   urir9   r:   rF   r)   s   `     r   rR   zProvider.get_resource   sH      	8 II22eDE ) 30000rG   c                   K   | j                          d{   }| j                  D ]  }|j                  |       d{   } |S 7 07 
w)zList resource templates with all transforms applied.

        Components may be marked as disabled but are NOT filtered here.
        N)_list_resource_templatesr&   list_resource_templates)r   	templatesr)   s      r   rW   z Provider.list_resource_templates   sI     
 7799	I'??	JJI ) :Jr3   c                    K   	 d	 	 	 	 	 d fd}|} j                   D ]  }t        |j                  |      }  |||       d{   S 7 w)a  Get resource template by transformed URI with all transforms applied.

        Note: This method does NOT filter disabled components. The Server
        (FastMCP) performs enabled filtering after all transforms complete.

        Args:
            uri: The transformed template URI to look up.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The template if found (may be marked disabled), None if not found.
        Nc                D   K   j                  | |       d {   S 7 wr   )_get_resource_templaterO   s     r   r:   z,Provider.get_resource_template.<locals>.base   s"      44Q@@@@r;   r<   r>   r   )rP   r?   r9   r@   rA   ResourceTemplate | None)r&   r   get_resource_templaterS   s   `     r   r]   zProvider.get_resource_template   si     " 37	A	A/	A$	A
 II;;uME ) 30000s   AAA	Ac                   K   | j                          d{   }| j                  D ]  }|j                  |       d{   } |S 7 07 
w)z{List prompts with all transforms applied.

        Components may be marked as disabled but are NOT filtered here.
        N)_list_promptsr&   list_prompts)r   promptsr)   s      r   r`   zProvider.list_prompts   sH     
 **,,I%227;;G ) -;r3   c                    K   dd fd}|} j                   D ]  }t        |j                  |      }  |||       d{   S 7 w)a  Get prompt by transformed name with all transforms applied.

        Note: This method does NOT filter disabled components. The Server
        (FastMCP) performs enabled filtering after all transforms complete.

        Args:
            name: The transformed prompt name to look up.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The prompt if found (may be marked disabled), None if not found.
        Nc                D   K   j                  | |       d {   S 7 wr   )_get_promptr7   s     r   r:   z!Provider.get_prompt.<locals>.base  s      ))!W5555r;   r<   r>   r   )r8   r?   r9   r@   rA   Prompt | None)r&   r   
get_promptrD   s   `     r   rf   zProvider.get_prompt  sH      	6 II00EBE ) 41111rG   c                   K   g S w)zReturn all available tools.

        Override to provide tools dynamically. Returns ALL versions of all tools.
        The server handles deduplication to show one tool per name.
         r   s    r   r0   zProvider._list_tools!        	   c                  K   | j                          d{   }|D cg c]  }|j                  |k(  s| }}|r+|D cg c]   }|j                  |j                        s|" }}|syt	        |t
              S 7 dc c}w c c}w w)a  Get a specific tool by name.

        Default implementation filters _list_tools() and picks the highest version
        that matches the spec.

        Args:
            name: The tool name.
            version: Optional version filter. If None, returns highest version.
                     If specified, returns highest version matching the spec.

        Returns:
            The Tool if found, or None to continue searching other providers.
        Nkey)r0   rE   matchesr9   maxr   )r   rE   r9   r2   tmatchings         r   r6   zProvider._get_tool)  s|       &&(($7u!$Au7#+J8awqyy/I8HJ8!122 )7J6   BA;BA=A=B  B!B%B=
Bc                   K   g S w)zReturn all available resources.

        Override to provide resources dynamically. Returns ALL versions of all resources.
        The server handles deduplication to show one resource per URI.
        rh   r   s    r   rI   zProvider._list_resourcesA  ri   rj   c                $  K   | j                          d{   }|D cg c]  }t        |j                        |k(  s| }}|r+|D cg c]   }|j                  |j                        s|" }}|syt        |t              S 7 mc c}w c c}w w)a  Get a specific resource by URI.

        Default implementation filters _list_resources() and returns highest
        version matching the spec.

        Args:
            uri: The resource URI.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The Resource if found, or None to continue searching other providers.
        Nrl   )rI   r?   rT   rn   r9   ro   r   )r   rT   r9   rK   rrq   s         r   rN   zProvider._get_resourceI  s      ..00	(>y!CJ#,=Ay>#+J8awqyy/I8HJ8!122 1>Js7   BBBBBB	 B*B.B
Bc                   K   g S w)zReturn all available resource templates.

        Override to provide resource templates dynamically. Returns ALL versions.
        The server handles deduplication.
        rh   r   s    r   rV   z!Provider._list_resource_templates`  ri   rj   c                  K   | j                          d{   }|D cg c]  }|j                  |      | }}|r+|D cg c]   }|j                  |j                        s|" }}|syt        |t              S 7 fc c}w c c}w w)a  Get a resource template that matches the given URI.

        Default implementation lists all templates, finds those whose pattern
        matches the URI, and returns the highest version matching the spec.

        Args:
            uri: The URI to match against templates.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The ResourceTemplate if a matching one is found, or None to continue searching.
        Nrl   )rV   rn   r9   ro   r   )r   rT   r9   rX   rp   rq   s         r   r[   zProvider._get_resource_templateh  s      7799	(Gy!AIIcN,FAyG#+J8awqyy/I8HJ8!122 :GJs6   B	A=B	A?A?B	 B#B'B	?
B	c                   K   g S w)zReturn all available prompts.

        Override to provide prompts dynamically. Returns ALL versions of all prompts.
        The server handles deduplication to show one prompt per name.
        rh   r   s    r   r_   zProvider._list_prompts  ri   rj   c                  K   | j                          d{   }|D cg c]  }|j                  |k(  s| }}|r+|D cg c]   }|j                  |j                        s|" }}|syt	        |t
              S 7 dc c}w c c}w w)a|  Get a specific prompt by name.

        Default implementation filters _list_prompts() and picks the highest version
        matching the spec.

        Args:
            name: The prompt name.
            version: Optional version filter. If None, returns highest version.

        Returns:
            The Prompt if found, or None to continue searching other providers.
        Nrl   )r_   rE   rn   r9   ro   r   )r   rE   r9   ra   prq   s         r   rd   zProvider._get_prompt  s|      **,,&9w!!&&D.Aw9#+J8awqyy/I8HJ8!122 -9Jrr   c                  K   t        | j                         | j                         | j                         | j	                                d{   }t        t        t           |d         }t        t        t           |d         }t        t        t           |d         }t        t        t           |d         }| j                  D ]f  }|j                  |       d{   }|j                  |       d{   }|j                  |       d{   }|j                  |       d{   }h g ||||D cg c]  }|j                   j#                         r|  c}S 7 7 7 o7 X7 Ac c}w w)af  Return components that should be registered as background tasks.

        Override to customize which components are task-eligible.
        Default calls list_* methods, applies provider transforms, and filters
        for components with task_config.mode != 'forbidden'.

        Used by the server during startup to register functions with Docket.
        Nr            )r   r0   rI   rV   r_   r
   r   r   r   r   r   r&   r1   rJ   rW   r`   task_configsupports_tasks)r   resultsr2   rK   rX   ra   r)   cs           r   	get_taskszProvider.get_tasks  sq       "))+ 	
 
 Xd^WQZ0(+WQZ8	"23WQZ@	x'4 I#..u55E'66yAAI'??	JJI%227;;G	 )  		
 }}++- 	
 		
'
 6AJ;	
sm   A
E0E BE0E#E05E%6E0E'E0'E)(E0:#E+E0#E0%E0'E0)E0+E0c                  K   d yw)a   User-overridable lifespan for custom setup and teardown.

        Override this method to perform provider-specific initialization
        like opening database connections, setting up external resources,
        or other state management needed for the provider's lifetime.

        The lifespan scope matches the server's lifespan - code before yield
        runs at startup, code after yield runs at shutdown.

        Example:
            ```python
            @asynccontextmanager
            async def lifespan(self):
                # Setup
                self.db = await connect_database()
                try:
                    yield
                finally:
                    # Teardown
                    await self.db.close()
            ```
        Nrh   r   s    r   lifespanzProvider.lifespan  s     0 	s   	F)nameskeysr9   tags
componentsonlyc                   |r&| j                   j                  t        dd             | j                   j                  t        d||||rt        |      nd|rt        |      nd             | S )a  Enable components matching all specified criteria.

        Adds a visibility transform that marks matching components as enabled.
        Later transforms override earlier ones, so enable after disable makes
        the component enabled.

        With only=True, switches to allowlist mode - first disables everything,
        then enables matching components.

        Args:
            names: Component names or URIs to enable.
            keys: Component keys to enable (e.g., {"tool:my_tool@v1"}).
            version: Component version spec to enable (e.g., VersionSpec(eq="v1") or
                VersionSpec(gte="v2")). Unversioned components will not match.
            tags: Enable components with these tags.
            components: Component types to include (e.g., {"tool", "prompt"}).
            only: If True, ONLY enable matching components (allowlist mode).

        Returns:
            Self for method chaining.
        FT)	match_allNr   r   r9   r   r   r   r(   r   set)r   r   r   r9   r   r   r   s          r   enablezProvider.enable  sj    @  ##Ju$EF.83z?d"&SYD		
 r   )r   r   r9   r   r   c                   | j                   j                  t        d||||rt        |      nd|rt        |      nd             | S )a  Disable components matching all specified criteria.

        Adds a visibility transform that marks matching components as disabled.
        Components can be re-enabled by calling enable() with matching criteria
        (the later transform wins).

        Args:
            names: Component names or URIs to disable.
            keys: Component keys to disable (e.g., {"tool:my_tool@v1"}).
            version: Component version spec to disable (e.g., VersionSpec(eq="v1") or
                VersionSpec(gte="v2")). Unversioned components will not match.
            tags: Disable components with these tags.
            components: Component types to include (e.g., {"tool", "prompt"}).

        Returns:
            Self for method chaining.
        FNr   r   )r   r   r   r9   r   r   s         r   disablezProvider.disable  sI    6 	.83z?d"&SYD		
 r   )rA   None)rA   r?   )rA   zlist[Transform])r)   r   rA   r   )r)   r   rA   r   )rA   zSequence[Tool]r   )rE   r?   r9   r@   rA   rB   )rA   zSequence[Resource])rT   r?   r9   r@   rA   rQ   )rA   zSequence[ResourceTemplate])rT   r?   r9   r@   rA   r\   )rA   zSequence[Prompt])rE   r?   r9   r@   rA   re   )rA   zSequence[FastMCPComponent])rA   zAsyncIterator[None])r   set[str] | Noner   r   r9   r@   r   r   r   =set[Literal['tool', 'resource', 'template', 'prompt']] | Noner   boolrA   r   )r   r   r   r   r9   r@   r   r   r   r   rA   r   )r"   
__module____qualname____doc__r   r#   propertyr&   r*   r.   r1   rC   rJ   rR   rW   r]   r`   rf   r0   r6   rI   rN   rV   r[   r_   rd   r   r   r   r   r   rh   r   r   r   r   3   s5   "/. & &+(1H" 8<22"42	24 7;11!31	12 7;11!31	 16 8<22"42	2: 8<33"43	30 7;33!33	3. 7;33!33	 3. 8<33"43	36&
X  @ "& $&* $/ / 	/
 $/ // / 
/h "& $&* $% % 	%
 $% %% 
%r   r   N)$r   
__future__r   collections.abcr   r   
contextlibr   	functoolsr   typingr   r	   r
   typing_extensionsr   fastmcp.prompts.promptr   fastmcp.resources.resourcer   fastmcp.resources.templater   $fastmcp.server.transforms.visibilityr   fastmcp.tools.toolr   fastmcp.utilities.async_utilsr   fastmcp.utilities.componentsr   fastmcp.utilities.versionsr   r   fastmcp.server.transformsr   r   rh   r   r   <module>r      sK   8 # 3 *  / / " ) / 7 ; # 0 9 D3O Or   