atlan-application-sdk 0.1.1rc44__py3-none-any.whl → 0.1.1rc45__py3-none-any.whl

application_sdk/application/__init__.py

@@ -4,6 +4,7 @@ from typing import Any, Dict, List, Optional, Tuple, Type
 from application_sdk.activities import ActivitiesInterface
 from application_sdk.clients.base import BaseClient
 from application_sdk.clients.utils import get_workflow_client
+from application_sdk.constants import ENABLE_MCP
 from application_sdk.events.models import EventRegistration
 from application_sdk.handlers.base import BaseHandler
 from application_sdk.observability.logger_adaptor import get_logger
@@ -29,7 +30,7 @@ class BaseApplication:
         self,
         name: str,
         server: Optional[ServerInterface] = None,
-        application_manifest: Optional[dict] = None,
+        application_manifest: Optional[Dict[str, Any]] = None,
         client_class: Optional[Type[BaseClient]] = None,
         handler_class: Optional[Type[BaseHandler]] = None,
     ):
@@ -39,6 +40,9 @@ class BaseApplication:
         Args:
             name (str): The name of the application.
             server (ServerInterface): The server class for the application.
+            application_manifest (Optional[Dict[str, Any]]): Application manifest configuration.
+            client_class (Optional[Type[BaseClient]]): Client class for the application.
+            handler_class (Optional[Type[BaseHandler]]): Handler class for the application.
         """
         self.application_name = name
 
@@ -49,14 +53,21 @@ class BaseApplication:
 
         self.workflow_client = get_workflow_client(application_name=name)
 
-        self.application_manifest: Dict[str, Any] = application_manifest
+        self.application_manifest: Optional[Dict[str, Any]] = application_manifest
         self.bootstrap_event_registration()
 
         self.client_class = client_class or BaseClient
         self.handler_class = handler_class or BaseHandler
 
+        # MCP configuration
+        self.mcp_server: Optional["MCPServer"] = None
+        if ENABLE_MCP:
+            from application_sdk.server.mcp import MCPServer
+
+            self.mcp_server = MCPServer(application_name=name)
+
     def bootstrap_event_registration(self):
-        self.event_subscriptions = {}
+        self.event_subscriptions: Dict[str, EventWorkflowTrigger] = {}
         if self.application_manifest is None:
             logger.warning("No application manifest found, skipping event registration")
             return
@@ -122,8 +133,8 @@ class BaseApplication:
         ]
         workflow_activities = []
         for workflow_class, activities_class in workflow_and_activities_classes:
-            workflow_activities.extend(
-                workflow_class.get_activities(activities_class())
+            workflow_activities.extend(  # type: ignore
+                workflow_class.get_activities(activities_class())  # type: ignore
             )
 
         self.worker = Worker(
@@ -134,6 +145,13 @@ class BaseApplication:
             activity_executor=activity_executor,
         )
 
+        # Register MCP tools if ENABLED_MCP is True and an MCP server is initialized
+        if self.mcp_server:
+            logger.info("Registering MCP tools from workflow and activities classes")
+            await self.mcp_server.register_tools(  # type: ignore
+                workflow_and_activities_classes=workflow_and_activities_classes
+            )
+
     async def start_workflow(self, workflow_args, workflow_class) -> Any:
         """
         Start a new workflow execution.
@@ -147,7 +165,7 @@ class BaseApplication:
         """
         if self.workflow_client is None:
             raise ValueError("Workflow client not initialized")
-        return await self.workflow_client.start_workflow(workflow_args, workflow_class)
+        return await self.workflow_client.start_workflow(workflow_args, workflow_class)  # type: ignore
 
     async def start_worker(self, daemon: bool = True):
         """
@@ -162,39 +180,61 @@ class BaseApplication:
 
     async def setup_server(
         self,
-        workflow_class,
+        workflow_class: Type[WorkflowInterface],
         ui_enabled: bool = True,
         has_configmap: bool = False,
     ):
         """
-        Optionally set up a server for the application. (No-op by default)
+        Set up FastAPI server and automatically mount MCP if enabled.
+
+        Args:
+            workflow_class (WorkflowInterface): The workflow class for the application.
+            ui_enabled (bool): Whether to enable the UI.
+            has_configmap (bool): Whether to enable the configmap.
         """
         if self.workflow_client is None:
             await self.workflow_client.load()
 
-        # Overrides the application server. serves the UI, and handles the various triggers
+        mcp_http_app: Optional[Any] = None
+        lifespan: Optional[Any] = None
+
+        if self.mcp_server:
+            try:
+                mcp_http_app = await self.mcp_server.get_http_app()
+                lifespan = mcp_http_app.lifespan
+            except Exception as e:
+                logger.warning(f"Failed to get MCP HTTP app: {e}")
+
         self.server = APIServer(
+            lifespan=lifespan,
             workflow_client=self.workflow_client,
             ui_enabled=ui_enabled,
             handler=self.handler_class(client=self.client_class()),
             has_configmap=has_configmap,
         )
 
+        # Mount MCP at root
+        if mcp_http_app:
+            try:
+                self.server.app.mount("", mcp_http_app)  # Mount at root
+            except Exception as e:
+                logger.warning(f"Failed to mount MCP HTTP app: {e}")
+
+        # Register event-based workflows if any
         if self.event_subscriptions:
             for event_trigger in self.event_subscriptions.values():
-                if event_trigger.workflow_class is None:
+                if event_trigger.workflow_class is None:  # type: ignore
                     raise ValueError(
                         f"Workflow class not set for event trigger {event_trigger.event_id}"
                     )
 
-                self.server.register_workflow(
-                    workflow_class=event_trigger.workflow_class,
+                self.server.register_workflow(  # type: ignore
+                    workflow_class=event_trigger.workflow_class,  # type: ignore
                     triggers=[event_trigger],
                 )
 
-        # register the workflow on the application server
-        # the workflow is by default triggered by an HTTP POST request to the /start endpoint
-        self.server.register_workflow(
+        # Register the main workflow (HTTP POST /start endpoint)
+        self.server.register_workflow(  # type: ignore
             workflow_class=workflow_class,
             triggers=[HttpWorkflowTrigger()],
         )

application_sdk/constants.py

@@ -255,3 +255,9 @@ REDIS_SENTINEL_HOSTS = os.getenv("REDIS_SENTINEL_HOSTS", "")
 IS_LOCKING_DISABLED = os.getenv("IS_LOCKING_DISABLED", "true").lower() == "true"
 #: Retry interval for lock acquisition
 LOCK_RETRY_INTERVAL = int(os.getenv("LOCK_RETRY_INTERVAL", "5"))
+
+# MCP Configuration
+#: Flag to indicate if MCP should be enabled or not. Turning this to true will setup an MCP server along
+#: with the application.
+ENABLE_MCP = os.getenv("ENABLE_MCP", "false").lower() == "true"
+MCP_METADATA_KEY = "__atlan_application_sdk_mcp_metadata"

application_sdk/decorators/mcp_tool.py

@@ -0,0 +1,63 @@
+"""
+MCP tool decorator for marking activities as MCP tools.
+
+This module provides the @mcp_tool decorator that developers use to mark
+activities for automatic exposure via Model Context Protocol.
+"""
+
+from typing import Any, Callable, Optional
+
+from application_sdk.constants import MCP_METADATA_KEY
+from application_sdk.server.mcp import MCPMetadata
+
+
+def mcp_tool(
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+    visible: bool = True,
+    *args,
+    **kwargs,
+):
+    """
+    Decorator to mark functions as MCP tools.
+
+    Use this decorator to mark any function as an MCP tool. You can additionally use the `visible`
+    parameter to control whether the tool is visible at runtime or not.
+
+    Function parameters that are Pydantic models will be automatically converted into correct JSON schema
+    for the tool specification. This is handled by the underlying FastMCP server implementation.
+
+    Args:
+        name(Optional[str]): The name of the tool. Defaults to the function name.
+        description(Optional[str]): The description of the tool. Defaults to the function docstring.
+        visible(bool): Whether the MCP tool is visible at runtime or not. Defaults to True.
+        *args: Additional arguments to pass to the tool.
+        **kwargs: Additional keyword arguments to pass to the tool.
+
+    Examples:
+        >>> @mcp_tool(name="add_numbers", description="Add two numbers", visible=True)
+        >>> def add_numbers(self, a: int, b: int) -> int:
+        >>>     return a + b
+
+
+        >>> # Use with Temporal activity decorator
+        >>> @activity.defn
+        >>> @mcp_tool(name="get_weather", description="Get the weather for a given city")
+        >>> async def get_weather(self, city: str) -> str:
+        >>>     # ... activity implementation unchanged ...
+    """
+
+    def decorator(f: Callable[..., Any]) -> Callable[..., Any]:
+        mcp_metadata = MCPMetadata(
+            name=name if name else f.__name__,
+            description=description if description else f.__doc__,
+            visible=visible,
+            args=args,
+            kwargs=kwargs,
+        )
+
+        setattr(f, MCP_METADATA_KEY, mcp_metadata)
+
+        return f
+
+    return decorator

application_sdk/server/mcp/__init__.py

@@ -0,0 +1,4 @@
+from application_sdk.server.mcp.models import MCPMetadata
+from application_sdk.server.mcp.server import MCPServer
+
+__all__ = ["MCPServer", "MCPMetadata"]

application_sdk/server/mcp/models.py

@@ -0,0 +1,11 @@
+from typing import Any, Dict, Optional, Tuple
+
+from pydantic import BaseModel
+
+
+class MCPMetadata(BaseModel):
+    name: str
+    description: Optional[str]
+    visible: bool
+    args: Tuple[Any, ...] = ()
+    kwargs: Dict[str, Any] = {}

application_sdk/server/mcp/server.py

@@ -0,0 +1,96 @@
+"""
+MCP Server implementation using FastMCP for Atlan Application SDK.
+
+This module provides the MCPServer class that automatically discovers
+activities marked with @mcp_tool decorators and mounts them on FastAPI
+using streamable HTTP transport.
+"""
+
+from typing import Any, Callable, List, Optional, Tuple, Type
+
+from fastmcp import FastMCP
+from fastmcp.server.http import StarletteWithLifespan
+
+from application_sdk.activities import ActivitiesInterface
+from application_sdk.constants import MCP_METADATA_KEY
+from application_sdk.observability.logger_adaptor import get_logger
+from application_sdk.server.mcp.models import MCPMetadata
+from application_sdk.workflows import WorkflowInterface
+
+
+class MCPServer:
+    """
+    MCP Server using FastMCP 2.0 with FastAPI mounting capability.
+
+    This server automatically discovers activities marked with @mcp_tool
+    and creates a FastMCP server that can be mounted on FastAPI.
+    """
+
+    def __init__(self, application_name: str, instructions: Optional[str] = None):
+        """
+        Initialize the MCP server.
+
+        Args:
+            application_name (str): Name of the application
+            instructions (Optional[str]): Description for the MCP server
+        """
+        self.application_name = application_name
+
+        self.logger = get_logger(__name__)
+
+        # FastMCP Server
+        self.server = FastMCP(
+            name=f"{application_name} MCP",
+            instructions=instructions,
+            on_duplicate_tools="error",
+        )
+
+    async def register_tools(
+        self,
+        workflow_and_activities_classes: List[
+            Tuple[Type[WorkflowInterface], Type[ActivitiesInterface]]
+        ],
+    ) -> None:
+        """
+        Discover activities marked with @mcp_tool and register them.
+
+        Args:
+            workflow_and_activities_classes: List of (workflow_class, activities_class) tuples
+        """
+        activity_methods: List[Callable[..., Any]] = []
+        for workflow_class, activities_class in workflow_and_activities_classes:
+            activities_instance = activities_class()
+            activity_methods.extend(workflow_class.get_activities(activities_instance))  # type: ignore
+
+        for f in activity_methods:
+            mcp_metadata: Optional[MCPMetadata] = getattr(f, MCP_METADATA_KEY, None)
+            if not mcp_metadata:
+                self.logger.info(
+                    f"No MCP metadata found on activity method {f.__name__}. Skipping tool registration"
+                )
+                continue
+
+            if mcp_metadata.visible:
+                self.logger.info(
+                    f"Registering tool {mcp_metadata.name} with description: {mcp_metadata.description}"
+                )
+                self.server.tool(
+                    f,
+                    name=mcp_metadata.name,
+                    description=mcp_metadata.description,
+                    *mcp_metadata.args,
+                    **mcp_metadata.kwargs,
+                )
+            else:
+                self.logger.info(
+                    f"Tool {mcp_metadata.name} is marked as not visible. Skipping tool registration"
+                )
+
+        tools = await self.server.get_tools()
+        self.logger.info(f"Registered {len(tools)} tools: {list(tools.keys())}")
+
+    async def get_http_app(self) -> StarletteWithLifespan:
+        """
+        Get the HTTP app for the MCP server.
+        """
+        return self.server.http_app()

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "0.1.1rc44"
+__version__ = "0.1.1rc45"

{atlan_application_sdk-0.1.1rc44.dist-info → atlan_application_sdk-0.1.1rc45.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc44
+Version: 0.1.1rc45
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md

{atlan_application_sdk-0.1.1rc44.dist-info → atlan_application_sdk-0.1.1rc45.dist-info}/RECORD

@@ -1,6 +1,6 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
-application_sdk/constants.py,sha256=1THiejjOEgm4kHFN-PrwrUkfRk7q1pjOLWLm-t2ph1Q,10674
-application_sdk/version.py,sha256=PUC8knGCYDRf-xg5lgaXJ_F5evubFFRTEkWX9EULiq0,88
+application_sdk/constants.py,sha256=eLHmH9GukXCKK-u5a4bAqz8BeCOCusCM0eW0Q0Bwjns,10947
+application_sdk/version.py,sha256=p3JLlDmx39Ch7lMX6DS5ZKpVpATBikUiznrMlVxdpTA,88
 application_sdk/worker.py,sha256=i5f0AeKI39IfsLO05QkwC6uMz0zDPSJqP7B2byri1VI,7489
 application_sdk/activities/__init__.py,sha256=QaXLOBYbb0zPOY5kfDQh56qbXQFaYNXOjJ5PCvatiZ4,9530
 application_sdk/activities/lock_management.py,sha256=L__GZ9BsArwU1ntYwAgCKsSjCqN6QBeOfT-OT4WyD4Y,3983
@@ -14,7 +14,7 @@ application_sdk/activities/metadata_extraction/rest.py,sha256=47DEQpj8HBSa-_TImW
 application_sdk/activities/metadata_extraction/sql.py,sha256=ivIbTrkKAonijQQPfiOigoiXLWtA_-nLUn9lz09lpaU,34725
 application_sdk/activities/query_extraction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/activities/query_extraction/sql.py,sha256=l64cGyTmbtaGcg3qj1YXKyNWiWeRsWPEuQyqW06rxxQ,21165
-application_sdk/application/__init__.py,sha256=PbSImXYaQQ2IIee2SvI8AjDiSo2QcCFrM1PX3x-_RQs,8035
+application_sdk/application/__init__.py,sha256=hb5zBc4zi-10av8Ivbovhb0CEAwNgr3eFlfpRaMKVmI,9861
 application_sdk/application/metadata_extraction/sql.py,sha256=rOd06Wodr4GyzupCYxVSCsNcuNar1rJM66ej9vocNHw,8138
 application_sdk/clients/__init__.py,sha256=C9T84J7V6ZumcoWJPAxdd3tqSmbyciaGBJn-CaCCny0,1341
 application_sdk/clients/atlan.py,sha256=l6yV39fr1006SJFwkOTNDQlbSFlHCZQaUPfdUlzdVEg,5053
@@ -36,6 +36,7 @@ application_sdk/common/utils.py,sha256=ImCrlyCj5Mj571CVWfqy5MynVVju9xhn1ItSlJoae
 application_sdk/common/.cursor/BUGBOT.md,sha256=OkB5TMAEJFzaBfbNb3g9ZDPW2r1krQE_KEuJbytMPuI,12176
 application_sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/decorators/locks.py,sha256=-cdbICCMns3lkqZ4CCQabW1du8cEu9XSWlwzWTTbIPk,1411
+application_sdk/decorators/mcp_tool.py,sha256=uxuc0Qk1A_MMvnwWe19kYGn4cfipvwwkmITR5nBdHP8,2215
 application_sdk/decorators/.cursor/BUGBOT.md,sha256=iiS_41FKaJ4-L2jm9ziEqQhQNGYGNZzHVVk09c2cgN8,11250
 application_sdk/docgen/__init__.py,sha256=Gr_3uVEnSspKd_-R1YRsDABI-iP4170Dvg5jM2oD76A,7352
 application_sdk/docgen/exporters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -89,6 +90,9 @@ application_sdk/server/fastapi/middleware/logmiddleware.py,sha256=CxcPtDmCbSfSZ8
 application_sdk/server/fastapi/middleware/metrics.py,sha256=5ddHAIg5sT-u9tB_HHMGL3Cfu2g1rm9z7ksienIr9ks,1563
 application_sdk/server/fastapi/routers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/server/fastapi/routers/server.py,sha256=vfHQwZCysThzfeVFNVW1IjuAdL0c1Cs4fULKTBK2eNo,4209
+application_sdk/server/mcp/__init__.py,sha256=29HI-GkNn9wOPsb86Litmv0MhXWFBNVBOivJcNFWZas,154
+application_sdk/server/mcp/models.py,sha256=3jX_3wXBuFs6XX9OG-xoag8VkKYWO_Y5r5-CHcLU5vg,236
+application_sdk/server/mcp/server.py,sha256=HG8tFmcc-f9Wj3vZzs2oRoNJzN1s5hwjnKykSdTXgCQ,3450
 application_sdk/services/__init__.py,sha256=H-5HZEPdr53MUfAggyHqHhRXDRLZFZsxvJgWbr257Ds,465
 application_sdk/services/atlan_storage.py,sha256=TKzXxu0yXeUcmZehwp8PcnQTC4A9w9RlZ0Fl-Xp1bLE,8509
 application_sdk/services/eventstore.py,sha256=X03JzodKByXh8w8nOl658rnnZfMFTj0IkmiLVbd6IN8,6729
@@ -152,8 +156,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=6ZaVt84n-8U2ZvR9GR7uIJKv5v8CuyQjhlnoRJvDszc,12435
 application_sdk/workflows/query_extraction/__init__.py,sha256=n066_CX5RpJz6DIxGMkKS3eGSRg03ilaCtsqfJWQb7Q,117
 application_sdk/workflows/query_extraction/sql.py,sha256=kT_JQkLCRZ44ZpaC4QvPL6DxnRIIVh8gYHLqRbMI-hA,4826
-atlan_application_sdk-0.1.1rc44.dist-info/METADATA,sha256=FoTta0zU5XJfLfr0hmUeEyEXFOqZ0-Plb_CerNBVfcM,5567
-atlan_application_sdk-0.1.1rc44.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc44.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc44.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc44.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc45.dist-info/METADATA,sha256=nuX1LdMTNoLQW3eKYU9KdNp2vXPiLG3RnKxZeu5ZcpM,5567
+atlan_application_sdk-0.1.1rc45.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc45.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc45.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc45.dist-info/RECORD,,