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

phenoml/agent/raw_client.py

@@ -16,12 +16,13 @@ from .errors.internal_server_error import InternalServerError
 from .errors.not_found_error import NotFoundError
 from .errors.unauthorized_error import UnauthorizedError
 from .types.agent_chat_response import AgentChatResponse
-from .types.agent_create_request_provider import AgentCreateRequestProvider
 from .types.agent_delete_response import AgentDeleteResponse
 from .types.agent_fhir_config import AgentFhirConfig
+from .types.agent_get_chat_messages_request_order import AgentGetChatMessagesRequestOrder
+from .types.agent_get_chat_messages_response import AgentGetChatMessagesResponse
 from .types.agent_list_response import AgentListResponse
+from .types.agent_provider import AgentProvider
 from .types.agent_response import AgentResponse
-from .types.agent_update_request_provider import AgentUpdateRequestProvider
 from .types.chat_fhir_client_config import ChatFhirClientConfig
 from .types.json_patch import JsonPatch
 
@@ -40,8 +41,9 @@ class RawAgentClient:
         prompts: typing.Sequence[str],
         is_active: bool,
         description: typing.Optional[str] = OMIT,
+        tools: typing.Optional[typing.Sequence[str]] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[AgentResponse]:
@@ -62,10 +64,13 @@ class RawAgentClient:
         description : typing.Optional[str]
             Agent description
 
+        tools : typing.Optional[typing.Sequence[str]]
+            Array of MCP server tool IDs to use for this agent
+
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentCreateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -85,10 +90,11 @@ class RawAgentClient:
                 "name": name,
                 "description": description,
                 "prompts": prompts,
+                "tools": tools,
                 "is_active": is_active,
                 "tags": tags,
                 "provider": convert_and_respect_annotation_metadata(
-                    object_=provider, annotation=AgentCreateRequestProvider, direction="write"
+                    object_=provider, annotation=AgentProvider, direction="write"
                 ),
                 "meta": convert_and_respect_annotation_metadata(
                     object_=meta, annotation=AgentFhirConfig, direction="write"
@@ -330,9 +336,10 @@ class RawAgentClient:
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
         prompts: typing.Optional[typing.Sequence[str]] = OMIT,
+        tools: typing.Optional[typing.Sequence[str]] = OMIT,
         is_active: typing.Optional[bool] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentUpdateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[AgentResponse]:
@@ -353,13 +360,16 @@ class RawAgentClient:
         prompts : typing.Optional[typing.Sequence[str]]
             Array of prompt IDs to use for this agent
 
+        tools : typing.Optional[typing.Sequence[str]]
+            Array of MCP server tool IDs to use for this agent
+
         is_active : typing.Optional[bool]
             Whether the agent is active
 
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentUpdateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -379,10 +389,11 @@ class RawAgentClient:
                 "name": name,
                 "description": description,
                 "prompts": prompts,
+                "tools": tools,
                 "is_active": is_active,
                 "tags": tags,
                 "provider": convert_and_respect_annotation_metadata(
-                    object_=provider, annotation=AgentUpdateRequestProvider, direction="write"
+                    object_=provider, annotation=AgentProvider, direction="write"
                 ),
                 "meta": convert_and_respect_annotation_metadata(
                     object_=meta, annotation=AgentFhirConfig, direction="write"
@@ -763,6 +774,99 @@ class RawAgentClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    def get_chat_messages(
+        self,
+        *,
+        chat_session_id: str,
+        num_messages: typing.Optional[int] = None,
+        role: typing.Optional[str] = None,
+        order: typing.Optional[AgentGetChatMessagesRequestOrder] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[AgentGetChatMessagesResponse]:
+        """
+        Retrieves a list of chat messages for a given chat session
+
+        Parameters
+        ----------
+        chat_session_id : str
+            Chat session ID
+
+        num_messages : typing.Optional[int]
+            Number of messages to return
+
+        role : typing.Optional[str]
+            Filter by role
+
+        order : typing.Optional[AgentGetChatMessagesRequestOrder]
+            Order of messages
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[AgentGetChatMessagesResponse]
+            Chat messages retrieved successfully
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "agent/chat/messages",
+            method="GET",
+            params={
+                "chat_session_id": chat_session_id,
+                "num_messages": num_messages,
+                "role": role,
+                "order": order,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AgentGetChatMessagesResponse,
+                    parse_obj_as(
+                        type_=AgentGetChatMessagesResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
 
 class AsyncRawAgentClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -775,8 +879,9 @@ class AsyncRawAgentClient:
         prompts: typing.Sequence[str],
         is_active: bool,
         description: typing.Optional[str] = OMIT,
+        tools: typing.Optional[typing.Sequence[str]] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[AgentResponse]:
@@ -797,10 +902,13 @@ class AsyncRawAgentClient:
         description : typing.Optional[str]
             Agent description
 
+        tools : typing.Optional[typing.Sequence[str]]
+            Array of MCP server tool IDs to use for this agent
+
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentCreateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -820,10 +928,11 @@ class AsyncRawAgentClient:
                 "name": name,
                 "description": description,
                 "prompts": prompts,
+                "tools": tools,
                 "is_active": is_active,
                 "tags": tags,
                 "provider": convert_and_respect_annotation_metadata(
-                    object_=provider, annotation=AgentCreateRequestProvider, direction="write"
+                    object_=provider, annotation=AgentProvider, direction="write"
                 ),
                 "meta": convert_and_respect_annotation_metadata(
                     object_=meta, annotation=AgentFhirConfig, direction="write"
@@ -1067,9 +1176,10 @@ class AsyncRawAgentClient:
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
         prompts: typing.Optional[typing.Sequence[str]] = OMIT,
+        tools: typing.Optional[typing.Sequence[str]] = OMIT,
         is_active: typing.Optional[bool] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentUpdateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[AgentResponse]:
@@ -1090,13 +1200,16 @@ class AsyncRawAgentClient:
         prompts : typing.Optional[typing.Sequence[str]]
             Array of prompt IDs to use for this agent
 
+        tools : typing.Optional[typing.Sequence[str]]
+            Array of MCP server tool IDs to use for this agent
+
         is_active : typing.Optional[bool]
             Whether the agent is active
 
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentUpdateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -1116,10 +1229,11 @@ class AsyncRawAgentClient:
                 "name": name,
                 "description": description,
                 "prompts": prompts,
+                "tools": tools,
                 "is_active": is_active,
                 "tags": tags,
                 "provider": convert_and_respect_annotation_metadata(
-                    object_=provider, annotation=AgentUpdateRequestProvider, direction="write"
+                    object_=provider, annotation=AgentProvider, direction="write"
                 ),
                 "meta": convert_and_respect_annotation_metadata(
                     object_=meta, annotation=AgentFhirConfig, direction="write"
@@ -1499,3 +1613,96 @@ class AsyncRawAgentClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def get_chat_messages(
+        self,
+        *,
+        chat_session_id: str,
+        num_messages: typing.Optional[int] = None,
+        role: typing.Optional[str] = None,
+        order: typing.Optional[AgentGetChatMessagesRequestOrder] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[AgentGetChatMessagesResponse]:
+        """
+        Retrieves a list of chat messages for a given chat session
+
+        Parameters
+        ----------
+        chat_session_id : str
+            Chat session ID
+
+        num_messages : typing.Optional[int]
+            Number of messages to return
+
+        role : typing.Optional[str]
+            Filter by role
+
+        order : typing.Optional[AgentGetChatMessagesRequestOrder]
+            Order of messages
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[AgentGetChatMessagesResponse]
+            Chat messages retrieved successfully
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "agent/chat/messages",
+            method="GET",
+            params={
+                "chat_session_id": chat_session_id,
+                "num_messages": num_messages,
+                "role": role,
+                "order": order,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AgentGetChatMessagesResponse,
+                    parse_obj_as(
+                        type_=AgentGetChatMessagesResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

phenoml/agent/types/__init__.py

@@ -3,43 +3,43 @@
 # isort: skip_file
 
 from .agent_chat_response import AgentChatResponse
-from .agent_create_request_provider import AgentCreateRequestProvider
-from .agent_create_request_provider_item import AgentCreateRequestProviderItem
 from .agent_delete_response import AgentDeleteResponse
 from .agent_fhir_config import AgentFhirConfig
+from .agent_get_chat_messages_request_order import AgentGetChatMessagesRequestOrder
+from .agent_get_chat_messages_response import AgentGetChatMessagesResponse
 from .agent_list_response import AgentListResponse
 from .agent_prompts_response import AgentPromptsResponse
+from .agent_provider import AgentProvider
 from .agent_response import AgentResponse
 from .agent_template import AgentTemplate
-from .agent_template_provider import AgentTemplateProvider
-from .agent_template_provider_item import AgentTemplateProviderItem
-from .agent_update_request_provider import AgentUpdateRequestProvider
-from .agent_update_request_provider_item import AgentUpdateRequestProviderItem
 from .chat_fhir_client_config import ChatFhirClientConfig
+from .chat_message_template import ChatMessageTemplate
+from .chat_session_template import ChatSessionTemplate
 from .json_patch import JsonPatch
 from .json_patch_operation import JsonPatchOperation
 from .json_patch_operation_op import JsonPatchOperationOp
 from .prompt_template import PromptTemplate
+from .provider_type import ProviderType
 from .success_response import SuccessResponse
 
 __all__ = [
     "AgentChatResponse",
-    "AgentCreateRequestProvider",
-    "AgentCreateRequestProviderItem",
     "AgentDeleteResponse",
     "AgentFhirConfig",
+    "AgentGetChatMessagesRequestOrder",
+    "AgentGetChatMessagesResponse",
     "AgentListResponse",
     "AgentPromptsResponse",
+    "AgentProvider",
     "AgentResponse",
     "AgentTemplate",
-    "AgentTemplateProvider",
-    "AgentTemplateProviderItem",
-    "AgentUpdateRequestProvider",
-    "AgentUpdateRequestProviderItem",
     "ChatFhirClientConfig",
+    "ChatMessageTemplate",
+    "ChatSessionTemplate",
     "JsonPatch",
     "JsonPatchOperation",
     "JsonPatchOperationOp",
     "PromptTemplate",
+    "ProviderType",
     "SuccessResponse",
 ]

phenoml/agent/types/agent_get_chat_messages_request_order.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+AgentGetChatMessagesRequestOrder = typing.Union[typing.Literal["asc", "desc"], typing.Any]

phenoml/agent/types/agent_get_chat_messages_response.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message_template import ChatMessageTemplate
+
+
+class AgentGetChatMessagesResponse(UniversalBaseModel):
+    messages: typing.Optional[typing.List[ChatMessageTemplate]] = None
+    total: typing.Optional[int] = None
+    session_id: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/agent/types/agent_provider.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .provider_type import ProviderType
+
+AgentProvider = typing.Union[ProviderType, typing.List[ProviderType]]

phenoml/agent/types/agent_template.py

@@ -5,7 +5,7 @@ import typing
 import pydantic
 from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .agent_fhir_config import AgentFhirConfig
-from .agent_template_provider import AgentTemplateProvider
+from .agent_provider import AgentProvider
 
 
 class AgentTemplate(UniversalBaseModel):
@@ -29,6 +29,11 @@ class AgentTemplate(UniversalBaseModel):
     Array of prompt IDs used by this agent
     """
 
+    tools: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Array of MCP server tool IDs used by this agent
+    """
+
     is_active: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Whether the agent is active
@@ -39,7 +44,7 @@ class AgentTemplate(UniversalBaseModel):
     Tags for categorizing the agent
     """
 
-    provider: typing.Optional[AgentTemplateProvider] = pydantic.Field(default=None)
+    provider: typing.Optional[AgentProvider] = pydantic.Field(default=None)
     """
     FHIR provider type - can be a single provider or array of providers
     """

phenoml/agent/types/chat_message_template.py

@@ -0,0 +1,72 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ChatMessageTemplate(UniversalBaseModel):
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat message ID
+    """
+
+    session_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session ID
+    """
+
+    role: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Message role
+    """
+
+    content: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Message content
+    """
+
+    created: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Message created time
+    """
+
+    updated: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Message updated time
+    """
+
+    user_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    User ID
+    """
+
+    function_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Function name
+    """
+
+    function_args: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Function arguments
+    """
+
+    function_result: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Function result
+    """
+
+    message_order: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Message order
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/agent/types/chat_session_template.py

@@ -0,0 +1,67 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ChatSessionTemplate(UniversalBaseModel):
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session ID
+    """
+
+    user_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    User ID
+    """
+
+    session_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session ID
+    """
+
+    status: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session status
+    """
+
+    system_prompt: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    System prompt
+    """
+
+    title: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session title
+    """
+
+    agent_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Agent ID
+    """
+
+    created: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session created time
+    """
+
+    last_message_at: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session last message time
+    """
+
+    updated: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Chat session updated time
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/agent/types/provider_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ProviderType = typing.Union[typing.Literal["medplum", "google_healthcare", "canvas", "hapi"], typing.Any]

phenoml/core/client_wrapper.py

@@ -22,10 +22,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "phenoml/0.0.1",
+            "User-Agent": "phenoml/0.0.2",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "phenoml",
-            "X-Fern-SDK-Version": "0.0.1",
+            "X-Fern-SDK-Version": "0.0.2",
             **(self.get_custom_headers() or {}),
         }
         headers["Authorization"] = f"Bearer {self._get_token()}"

phenoml/tools/__init__.py

@@ -11,9 +11,15 @@ from .types import (
     Lang2FhirAndCreateResponse,
     Lang2FhirAndSearchRequestProvider,
     Lang2FhirAndSearchResponse,
+    McpServerResponse,
+    McpServerResponseData,
+    McpServerToolCallResponse,
+    McpServerToolResponse,
+    McpServerToolResponseData,
     SearchConcept,
 )
 from .errors import BadRequestError, FailedDependencyError, ForbiddenError, InternalServerError, UnauthorizedError
+from . import mcp_server
 
 __all__ = [
     "BadRequestError",
@@ -28,6 +34,12 @@ __all__ = [
     "Lang2FhirAndCreateResponse",
     "Lang2FhirAndSearchRequestProvider",
     "Lang2FhirAndSearchResponse",
+    "McpServerResponse",
+    "McpServerResponseData",
+    "McpServerToolCallResponse",
+    "McpServerToolResponse",
+    "McpServerToolResponseData",
     "SearchConcept",
     "UnauthorizedError",
+    "mcp_server",
 ]