atlan-application-sdk 0.1.1rc51__py3-none-any.whl → 0.1.1rc53__py3-none-any.whl

application_sdk/activities/__init__.py

@@ -241,6 +241,16 @@ class ActivitiesInterface(ABC, Generic[ActivitiesStateType]):
             if not handler:
                 raise ValueError("Preflight check handler not found")
 
+            # Verify handler and client are properly initialized
+            if hasattr(handler, "sql_client") and handler.sql_client:
+                if not handler.sql_client.engine:
+                    logger.error("SQL client engine is not initialized")
+                    raise ValueError(
+                        "Preflight check failed: SQL client engine not initialized. "
+                        "This may indicate credentials were not loaded properly."
+                    )
+                logger.info("SQL client engine verified as initialized")
+
             result = await handler.preflight_check(
                 {"metadata": workflow_args["metadata"]}
             )

application_sdk/activities/metadata_extraction/sql.py

@@ -166,16 +166,17 @@ class BaseSQLMetadataExtractionActivities(ActivitiesInterface):
 
         sql_client = self.sql_client_class()
 
-        handler = self.handler_class(sql_client)
-        self._state[workflow_id].handler = handler
-
+        # Load credentials BEFORE creating handler to avoid race condition
         if "credential_guid" in workflow_args:
             credentials = await SecretStore.get_credentials(
                 workflow_args["credential_guid"]
             )
             await sql_client.load(credentials)
 
+        # Assign sql_client and handler to state AFTER credentials are loaded
         self._state[workflow_id].sql_client = sql_client
+        handler = self.handler_class(sql_client)
+        self._state[workflow_id].handler = handler
 
         # Create transformer with required parameters from ApplicationConstants
         transformer_params = {

application_sdk/clients/temporal.py

@@ -441,9 +441,7 @@ class TemporalWorkflowClient(WorkflowClient):
                 "execution_duration_seconds": execution_info.execution_duration.ToSeconds(),
             }
             if include_last_executed_run_id:
-                workflow_info["last_executed_run_id"] = (
-                    execution_info.root_execution.run_id
-                )
+                workflow_info["last_executed_run_id"] = execution_info.execution.run_id
             return workflow_info
         except Exception as e:
             if (

application_sdk/version.py

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

{atlan_application_sdk-0.1.1rc51.dist-info → atlan_application_sdk-0.1.1rc53.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc51
+Version: 0.1.1rc53
 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
@@ -38,6 +38,8 @@ Provides-Extra: iam-auth
 Requires-Dist: boto3>=1.38.6; extra == 'iam-auth'
 Provides-Extra: iceberg
 Requires-Dist: pyiceberg>=0.8.1; extra == 'iceberg'
+Provides-Extra: mcp
+Requires-Dist: fastmcp>=2.12.3; extra == 'mcp'
 Provides-Extra: pandas
 Requires-Dist: pandas>=2.2.3; extra == 'pandas'
 Provides-Extra: scale-data-generator

{atlan_application_sdk-0.1.1rc51.dist-info → atlan_application_sdk-0.1.1rc53.dist-info}/RECORD

@@ -1,8 +1,8 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
 application_sdk/constants.py,sha256=EDGR-3SuCxNV-3x0D4wA9is9vBbVWa3nHvJ8r2w4lYY,10977
-application_sdk/version.py,sha256=y3YjdWXoWFHawTe7w3qur8cONKs4SWKqLlcBNZ3J2jw,88
+application_sdk/version.py,sha256=0GDfah3Is6w2M-p8lHG9431TleRLSCtZ79vP9LRgIfY,88
 application_sdk/worker.py,sha256=i5f0AeKI39IfsLO05QkwC6uMz0zDPSJqP7B2byri1VI,7489
-application_sdk/activities/__init__.py,sha256=QaXLOBYbb0zPOY5kfDQh56qbXQFaYNXOjJ5PCvatiZ4,9530
+application_sdk/activities/__init__.py,sha256=OhbOrz8ildUCFe4xrjYvceQM36dlDpKTPtw0OnTaPwU,10090
 application_sdk/activities/lock_management.py,sha256=oX2qPpfEu_xP0MiaCakVGk9ivZDvG4EddVZag1DuHSE,3976
 application_sdk/activities/.cursor/BUGBOT.md,sha256=FNykX5aMkdOhzgpiGqstOnSp9JN63iR2XP3onU4AGh8,15843
 application_sdk/activities/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -11,7 +11,7 @@ application_sdk/activities/common/utils.py,sha256=nSNGkY5eS5pPc8etdPWkXBFTSaConG
 application_sdk/activities/metadata_extraction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/activities/metadata_extraction/base.py,sha256=ENFojpxqKdN_eVSL4iet3cGfylPOfcl1jnflfo4zhs8,3920
 application_sdk/activities/metadata_extraction/rest.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-application_sdk/activities/metadata_extraction/sql.py,sha256=9VdQ9Kv3vCuK6b1US9akvdv9A8NvFVVOvK3Svxkj0ws,34284
+application_sdk/activities/metadata_extraction/sql.py,sha256=0nma1lQned2K7VUmBK_GIPDjt3iU7mRGFvdxxmm80ec,34436
 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=hb5zBc4zi-10av8Ivbovhb0CEAwNgr3eFlfpRaMKVmI,9861
@@ -23,7 +23,7 @@ application_sdk/clients/base.py,sha256=TIn3pG89eXUc1XSYf4jk66m1vajWp0WxcCQOOltda
 application_sdk/clients/models.py,sha256=yPgmiqt3I7am2NdFF10BnKiwfgjLceEXrpie0rvGCg8,1509
 application_sdk/clients/redis.py,sha256=IfAD32vLp88BCvsDTaQtxFHxzHlEx4V7TK7h1HwDDBg,15917
 application_sdk/clients/sql.py,sha256=r-8rghnATLRxxISchLZaNnMCAZMRLKyPAwPZengzMMY,19846
-application_sdk/clients/temporal.py,sha256=g7DXlnMLlWOHAdyXB0TCDx0aWdAwjazDnIbMgNNgKgw,18359
+application_sdk/clients/temporal.py,sha256=jC3U8LmW8G6gg-Qmxk0rcAifIGF0KekwP1UkMGXN7RA,18314
 application_sdk/clients/utils.py,sha256=zLFOJbTr_6TOqnjfVFGY85OtIXZ4FQy_rquzjaydkbY,779
 application_sdk/clients/workflow.py,sha256=6bSqmA3sNCk9oY68dOjBUDZ9DhNKQxPD75qqE0cfldc,6104
 application_sdk/clients/.cursor/BUGBOT.md,sha256=7nEDUqWBEMI_uU6eK1jCSZGeXoQtLQcKwOrDn8AIDWo,10595
@@ -156,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.1rc51.dist-info/METADATA,sha256=p8FBqKsRYzX0DN9zf36MgkQOjqs8ADYtmYx93wIkRR4,5567
-atlan_application_sdk-0.1.1rc51.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc51.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc51.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc51.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc53.dist-info/METADATA,sha256=IIkNqjucZ5QB-EUMFTFaucgdxl-U6ouulLKzH3atL_I,5634
+atlan_application_sdk-0.1.1rc53.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc53.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc53.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc53.dist-info/RECORD,,