phenoml 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

phenoml/tools/client.py

@@ -4,6 +4,7 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from .mcp_server.client import AsyncMcpServerClient, McpServerClient
 from .raw_client import AsyncRawToolsClient, RawToolsClient
 from .types.cohort_request_provider import CohortRequestProvider
 from .types.cohort_response import CohortResponse
@@ -21,6 +22,7 @@ OMIT = typing.cast(typing.Any, ...)
 class ToolsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._raw_client = RawToolsClient(client_wrapper=client_wrapper)
+        self.mcp_server = McpServerClient(client_wrapper=client_wrapper)
 
     @property
     def with_raw_response(self) -> RawToolsClient:
@@ -196,6 +198,7 @@ class ToolsClient:
 class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._raw_client = AsyncRawToolsClient(client_wrapper=client_wrapper)
+        self.mcp_server = AsyncMcpServerClient(client_wrapper=client_wrapper)
 
     @property
     def with_raw_response(self) -> AsyncRawToolsClient:

phenoml/tools/mcp_server/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from . import tools
+
+__all__ = ["tools"]

phenoml/tools/mcp_server/client.py

@@ -0,0 +1,336 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ..types.mcp_server_response import McpServerResponse
+from .raw_client import AsyncRawMcpServerClient, RawMcpServerClient
+from .tools.client import AsyncToolsClient, ToolsClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class McpServerClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawMcpServerClient(client_wrapper=client_wrapper)
+        self.tools = ToolsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawMcpServerClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawMcpServerClient
+        """
+        return self._raw_client
+
+    def create(
+        self, *, name: str, mcp_server_url: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> McpServerResponse:
+        """
+        Creates a new MCP server
+
+        Parameters
+        ----------
+        name : str
+            Name of the MCP server
+
+        mcp_server_url : str
+            URL of the MCP server
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully created MCP server
+
+        Examples
+        --------
+        from phenoml import phenoml
+
+        client = phenoml(
+            token="YOUR_TOKEN",
+        )
+        client.tools.mcp_server.create(
+            name="My MCP Server",
+            mcp_server_url="https://mcp.example.com",
+        )
+        """
+        _response = self._raw_client.create(name=name, mcp_server_url=mcp_server_url, request_options=request_options)
+        return _response.data
+
+    def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> McpServerResponse:
+        """
+        Lists all MCP servers for a specific user
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully listed MCP servers
+
+        Examples
+        --------
+        from phenoml import phenoml
+
+        client = phenoml(
+            token="YOUR_TOKEN",
+        )
+        client.tools.mcp_server.list()
+        """
+        _response = self._raw_client.list(request_options=request_options)
+        return _response.data
+
+    def get(self, mcp_server_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> McpServerResponse:
+        """
+        Gets a MCP server by ID
+
+        Parameters
+        ----------
+        mcp_server_id : str
+            ID of the MCP server to retrieve
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully retrieved MCP server
+
+        Examples
+        --------
+        from phenoml import phenoml
+
+        client = phenoml(
+            token="YOUR_TOKEN",
+        )
+        client.tools.mcp_server.get(
+            mcp_server_id="mcp_server_id",
+        )
+        """
+        _response = self._raw_client.get(mcp_server_id, request_options=request_options)
+        return _response.data
+
+    def delete(
+        self, mcp_server_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> McpServerResponse:
+        """
+        Deletes a MCP server by ID
+
+        Parameters
+        ----------
+        mcp_server_id : str
+            ID of the MCP server to delete
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully deleted MCP server
+
+        Examples
+        --------
+        from phenoml import phenoml
+
+        client = phenoml(
+            token="YOUR_TOKEN",
+        )
+        client.tools.mcp_server.delete(
+            mcp_server_id="mcp_server_id",
+        )
+        """
+        _response = self._raw_client.delete(mcp_server_id, request_options=request_options)
+        return _response.data
+
+
+class AsyncMcpServerClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawMcpServerClient(client_wrapper=client_wrapper)
+        self.tools = AsyncToolsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawMcpServerClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawMcpServerClient
+        """
+        return self._raw_client
+
+    async def create(
+        self, *, name: str, mcp_server_url: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> McpServerResponse:
+        """
+        Creates a new MCP server
+
+        Parameters
+        ----------
+        name : str
+            Name of the MCP server
+
+        mcp_server_url : str
+            URL of the MCP server
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully created MCP server
+
+        Examples
+        --------
+        import asyncio
+
+        from phenoml import Asyncphenoml
+
+        client = Asyncphenoml(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.mcp_server.create(
+                name="My MCP Server",
+                mcp_server_url="https://mcp.example.com",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create(
+            name=name, mcp_server_url=mcp_server_url, request_options=request_options
+        )
+        return _response.data
+
+    async def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> McpServerResponse:
+        """
+        Lists all MCP servers for a specific user
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully listed MCP servers
+
+        Examples
+        --------
+        import asyncio
+
+        from phenoml import Asyncphenoml
+
+        client = Asyncphenoml(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.mcp_server.list()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list(request_options=request_options)
+        return _response.data
+
+    async def get(
+        self, mcp_server_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> McpServerResponse:
+        """
+        Gets a MCP server by ID
+
+        Parameters
+        ----------
+        mcp_server_id : str
+            ID of the MCP server to retrieve
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully retrieved MCP server
+
+        Examples
+        --------
+        import asyncio
+
+        from phenoml import Asyncphenoml
+
+        client = Asyncphenoml(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.mcp_server.get(
+                mcp_server_id="mcp_server_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get(mcp_server_id, request_options=request_options)
+        return _response.data
+
+    async def delete(
+        self, mcp_server_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> McpServerResponse:
+        """
+        Deletes a MCP server by ID
+
+        Parameters
+        ----------
+        mcp_server_id : str
+            ID of the MCP server to delete
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        McpServerResponse
+            Successfully deleted MCP server
+
+        Examples
+        --------
+        import asyncio
+
+        from phenoml import Asyncphenoml
+
+        client = Asyncphenoml(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.mcp_server.delete(
+                mcp_server_id="mcp_server_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete(mcp_server_id, request_options=request_options)
+        return _response.data