atlan-application-sdk 0.1.1rc50__py3-none-any.whl → 0.1.1rc52__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/outputs/parquet.py

@@ -98,8 +98,8 @@ class ParquetOutput(Output):
         self.current_buffer_size = 0
         self.current_buffer_size_bytes = 0  # Track estimated buffer size in bytes
         self.max_file_size_bytes = int(
-            DAPR_MAX_GRPC_MESSAGE_LENGTH * 0.9
-        )  # 90% of DAPR limit as safety buffer
+            DAPR_MAX_GRPC_MESSAGE_LENGTH * 0.75
+        )  # 75% of DAPR limit as safety buffer
         self.chunk_start = chunk_start
         self.chunk_part = 0
         self.start_marker = start_marker

application_sdk/version.py

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

{atlan_application_sdk-0.1.1rc50.dist-info → atlan_application_sdk-0.1.1rc52.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc50
+Version: 0.1.1rc52
 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.1rc50.dist-info → atlan_application_sdk-0.1.1rc52.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=IKMb1DkxhDi2K0XaJVCTEXZHR_Eu-uL5cgErb-4UD54,88
+application_sdk/version.py,sha256=7YdRscQjBpl3S8wjyjN9i4Q5vuBeksOil1F1NzGu0Q4,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
@@ -79,7 +79,7 @@ application_sdk/observability/decorators/observability_decorator.py,sha256=yd6qf
 application_sdk/outputs/__init__.py,sha256=hrOPw0xuG9xP720Bt309TfbY2Qq_i51R8Xt3ZjwWDUY,15906
 application_sdk/outputs/iceberg.py,sha256=TdppOMEMfojMhGyBmhWeu1AJQexRyHM-huAYeJmhjdY,5533
 application_sdk/outputs/json.py,sha256=gYDDNOVb8EFxxeOkb6zKWZWjTEVgZLoapFM97_roK4A,10883
-application_sdk/outputs/parquet.py,sha256=-_7C7D7vc53brFjLgqa25Tnee6txwpj6V2yRfeH5hP4,20329
+application_sdk/outputs/parquet.py,sha256=lEUosbdLfLhZZPS9lZJxnQHEbLc5e1O7JeiDX958frw,20330
 application_sdk/outputs/.cursor/BUGBOT.md,sha256=KxEC3CIyRSK1YftZou5BgKc6PRXT3qQmBNFJp-HSyYE,11496
 application_sdk/server/__init__.py,sha256=KTqE1YPw_3WDVMWatJUuf9OOiobLM2K5SMaBrI62sCo,1568
 application_sdk/server/.cursor/BUGBOT.md,sha256=p_MMoWUW5G1894WfOKYReZKWCuyJT_OJz3rL5g21NbI,16566
@@ -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.1rc50.dist-info/METADATA,sha256=CPDcgj8T2mHi6CcG2Yyl73JsMI5D3Ppg_XLtmmJ4XYE,5567
-atlan_application_sdk-0.1.1rc50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc50.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc50.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc50.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc52.dist-info/METADATA,sha256=wBTA9rjeCH6j44tPMQAGDKWTNsajpdXwH5jkbXdPcPQ,5567
+atlan_application_sdk-0.1.1rc52.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc52.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc52.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc52.dist-info/RECORD,,