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

phenoml/tools/client.py

@@ -4,14 +4,11 @@ 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
-from .types.fhir_client_config import FhirClientConfig
-from .types.lang2fhir_and_create_request_provider import Lang2FhirAndCreateRequestProvider
 from .types.lang2fhir_and_create_request_resource import Lang2FhirAndCreateRequestResource
 from .types.lang2fhir_and_create_response import Lang2FhirAndCreateResponse
-from .types.lang2fhir_and_search_request_provider import Lang2FhirAndSearchRequestProvider
 from .types.lang2fhir_and_search_response import Lang2FhirAndSearchResponse
 
 # this is used as the default value for optional parameters
@@ -21,6 +18,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:
@@ -38,8 +36,7 @@ class ToolsClient:
         *,
         resource: Lang2FhirAndCreateRequestResource,
         text: str,
-        provider: typing.Optional[Lang2FhirAndCreateRequestProvider] = OMIT,
-        meta: typing.Optional[FhirClientConfig] = OMIT,
+        provider: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Lang2FhirAndCreateResponse:
         """
@@ -53,10 +50,8 @@ class ToolsClient:
         text : str
             Natural language text to convert to FHIR resource
 
-        provider : typing.Optional[Lang2FhirAndCreateRequestProvider]
-            FHIR provider to use for storing the resource
-
-        meta : typing.Optional[FhirClientConfig]
+        provider : typing.Optional[str]
+            FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -79,7 +74,7 @@ class ToolsClient:
         )
         """
         _response = self._raw_client.create_fhir_resource(
-            resource=resource, text=text, provider=provider, meta=meta, request_options=request_options
+            resource=resource, text=text, provider=provider, request_options=request_options
         )
         return _response.data
 
@@ -90,8 +85,7 @@ class ToolsClient:
         patient_id: typing.Optional[str] = OMIT,
         practitioner_id: typing.Optional[str] = OMIT,
         count: typing.Optional[int] = OMIT,
-        provider: typing.Optional[Lang2FhirAndSearchRequestProvider] = OMIT,
-        meta: typing.Optional[FhirClientConfig] = OMIT,
+        provider: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Lang2FhirAndSearchResponse:
         """
@@ -111,10 +105,8 @@ class ToolsClient:
         count : typing.Optional[int]
             Maximum number of results to return
 
-        provider : typing.Optional[Lang2FhirAndSearchRequestProvider]
-            FHIR provider to use for searching
-
-        meta : typing.Optional[FhirClientConfig]
+        provider : typing.Optional[str]
+            FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -141,18 +133,12 @@ class ToolsClient:
             practitioner_id=practitioner_id,
             count=count,
             provider=provider,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
 
     def analyze_cohort(
-        self,
-        *,
-        text: str,
-        provider: CohortRequestProvider,
-        meta: typing.Optional[FhirClientConfig] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, text: str, provider: str, request_options: typing.Optional[RequestOptions] = None
     ) -> CohortResponse:
         """
         Uses LLM to extract search concepts from natural language and builds patient cohorts with inclusion/exclusion criteria
@@ -162,10 +148,8 @@ class ToolsClient:
         text : str
             Natural language text describing the patient cohort criteria
 
-        provider : CohortRequestProvider
-            FHIR provider to use for searching
-
-        meta : typing.Optional[FhirClientConfig]
+        provider : str
+            FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -184,18 +168,17 @@ class ToolsClient:
         )
         client.tools.analyze_cohort(
             text="female patients over 20 with diabetes but not hypertension",
-            provider="medplum",
+            provider="550e8400-e29b-41d4-a716-446655440000",
         )
         """
-        _response = self._raw_client.analyze_cohort(
-            text=text, provider=provider, meta=meta, request_options=request_options
-        )
+        _response = self._raw_client.analyze_cohort(text=text, provider=provider, request_options=request_options)
         return _response.data
 
 
 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:
@@ -213,8 +196,7 @@ class AsyncToolsClient:
         *,
         resource: Lang2FhirAndCreateRequestResource,
         text: str,
-        provider: typing.Optional[Lang2FhirAndCreateRequestProvider] = OMIT,
-        meta: typing.Optional[FhirClientConfig] = OMIT,
+        provider: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Lang2FhirAndCreateResponse:
         """
@@ -228,10 +210,8 @@ class AsyncToolsClient:
         text : str
             Natural language text to convert to FHIR resource
 
-        provider : typing.Optional[Lang2FhirAndCreateRequestProvider]
-            FHIR provider to use for storing the resource
-
-        meta : typing.Optional[FhirClientConfig]
+        provider : typing.Optional[str]
+            FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -262,7 +242,7 @@ class AsyncToolsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.create_fhir_resource(
-            resource=resource, text=text, provider=provider, meta=meta, request_options=request_options
+            resource=resource, text=text, provider=provider, request_options=request_options
         )
         return _response.data
 
@@ -273,8 +253,7 @@ class AsyncToolsClient:
         patient_id: typing.Optional[str] = OMIT,
         practitioner_id: typing.Optional[str] = OMIT,
         count: typing.Optional[int] = OMIT,
-        provider: typing.Optional[Lang2FhirAndSearchRequestProvider] = OMIT,
-        meta: typing.Optional[FhirClientConfig] = OMIT,
+        provider: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Lang2FhirAndSearchResponse:
         """
@@ -294,10 +273,8 @@ class AsyncToolsClient:
         count : typing.Optional[int]
             Maximum number of results to return
 
-        provider : typing.Optional[Lang2FhirAndSearchRequestProvider]
-            FHIR provider to use for searching
-
-        meta : typing.Optional[FhirClientConfig]
+        provider : typing.Optional[str]
+            FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -332,18 +309,12 @@ class AsyncToolsClient:
             practitioner_id=practitioner_id,
             count=count,
             provider=provider,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
 
     async def analyze_cohort(
-        self,
-        *,
-        text: str,
-        provider: CohortRequestProvider,
-        meta: typing.Optional[FhirClientConfig] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, text: str, provider: str, request_options: typing.Optional[RequestOptions] = None
     ) -> CohortResponse:
         """
         Uses LLM to extract search concepts from natural language and builds patient cohorts with inclusion/exclusion criteria
@@ -353,10 +324,8 @@ class AsyncToolsClient:
         text : str
             Natural language text describing the patient cohort criteria
 
-        provider : CohortRequestProvider
-            FHIR provider to use for searching
-
-        meta : typing.Optional[FhirClientConfig]
+        provider : str
+            FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -380,13 +349,11 @@ class AsyncToolsClient:
         async def main() -> None:
             await client.tools.analyze_cohort(
                 text="female patients over 20 with diabetes but not hypertension",
-                provider="medplum",
+                provider="550e8400-e29b-41d4-a716-446655440000",
             )
 
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.analyze_cohort(
-            text=text, provider=provider, meta=meta, request_options=request_options
-        )
+        _response = await self._raw_client.analyze_cohort(text=text, provider=provider, request_options=request_options)
         return _response.data

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