atlan-application-sdk 0.1.1rc28__py3-none-any.whl → 0.1.1rc30__py3-none-any.whl

application_sdk/common/dapr_utils.py

@@ -0,0 +1,32 @@
+from dapr import clients
+
+from application_sdk.observability.logger_adaptor import get_logger
+
+logger = get_logger(__name__)
+
+
+def is_component_registered(component_name: str) -> bool:
+    """Check if a DAPR component with the given name is registered.
+
+    Args:
+        component_name: Name of the component to check.
+
+    Returns:
+        True if the component is present, False otherwise or on metadata errors.
+    """
+    try:
+        with clients.DaprClient() as client:
+            metadata = client.get_metadata()
+            # Each registered component has fields: name, type (e.g., "eventstore")
+            for component in getattr(metadata, "registered_components", []):
+                if component.name == component_name:
+                    return True
+            return False
+    except Exception:
+        # If we cannot read metadata, behave conservatively and report unavailable
+        logger.warning(
+            "Failed to read Dapr metadata for component availability check; treating as unavailable",
+            exc_info=True,
+            extra={"component_name": component_name},
+        )
+        return False

application_sdk/constants.py

@@ -131,9 +131,9 @@ EVENT_STORE_NAME = os.getenv("EVENT_STORE_NAME", "eventstore")
 #: Whether to enable Atlan storage upload
 ENABLE_ATLAN_UPLOAD = os.getenv("ENABLE_ATLAN_UPLOAD", "false").lower() == "true"
 # Dapr Client Configuration
-#: Maximum gRPC message length in bytes for Dapr client (default: 16MB)
+#: Maximum gRPC message length in bytes for Dapr client (default: 100MB)
 DAPR_MAX_GRPC_MESSAGE_LENGTH = int(
-    os.getenv("DAPR_MAX_GRPC_MESSAGE_LENGTH", "16777216")
+    os.getenv("DAPR_MAX_GRPC_MESSAGE_LENGTH", "104857600")
 )
 #: Name of the deployment secret store component in DAPR
 DEPLOYMENT_SECRET_STORE_NAME = os.getenv(

application_sdk/handlers/sql.py

@@ -116,7 +116,7 @@ class BaseSQLHandler(HandlerInterface):
     async def fetch_metadata(
         self,
         metadata_type: Optional[str] = None,
-        database: Optional[str] = None,
+        database: str = "",
     ) -> List[Dict[str, str]]:
         """
         Fetch metadata based on the requested type.

application_sdk/inputs/objectstore.py

@@ -79,25 +79,11 @@ class ObjectStoreInput:
         try:
             # List all files in the object store path
             relative_path = os.path.relpath(file_path, download_file_prefix)
-            metadata = {"fileName": relative_path}
+            file_list = cls.list_all_files(relative_path, object_store_name)
 
-            try:
-                # Assuming the object store binding supports a "list" operation
-                response_data = cls._invoke_dapr_binding(
-                    operation=cls.OBJECT_LIST_OPERATION, metadata=metadata
-                )
-                file_list = orjson.loads(response_data.decode("utf-8"))
-            except Exception as e:
-                logger.error(
-                    f"Error listing files in object store path {download_file_prefix}: {str(e)}"
-                )
-                raise e
-
-            if not file_list:
-                logger.info(
-                    f"No files found in object store path: {download_file_prefix}"
-                )
-                return
+            logger.info(
+                f"Found list of files: {file_list} from: {download_file_prefix}"
+            )
 
             # Download each file
             for relative_path in file_list:

application_sdk/inputs/statestore.py

@@ -76,6 +76,7 @@ class StateStoreInput:
         state = {}
 
         try:
+            logger.info(f"Trying to download state object for {id} with type {type}")
             local_state_file_path = os.path.join(TEMPORARY_PATH, state_file_path)
             ObjectStoreInput.download_file_from_object_store(
                 download_file_prefix=TEMPORARY_PATH,
@@ -86,10 +87,11 @@ class StateStoreInput:
             with open(local_state_file_path, "r") as file:
                 state = json.load(file)
 
+            logger.info(f"State object downloaded for {id} with type {type}")
         except Exception as e:
             # local error message is "file not found", while in object store it is "object not found"
             if "not found" in str(e).lower():
-                logger.debug(
+                logger.info(
                     f"No state found for {type.value} with id '{id}', returning empty dict"
                 )
             else:

application_sdk/outputs/eventstore.py

@@ -14,6 +14,7 @@ from dapr import clients
 from pydantic import BaseModel, Field
 from temporalio import activity, workflow
 
+from application_sdk.common.dapr_utils import is_component_registered
 from application_sdk.constants import APPLICATION_NAME, EVENT_STORE_NAME
 from application_sdk.observability.logger_adaptor import get_logger
 
@@ -155,6 +156,11 @@ class EventStore:
         Example:
             >>> EventStore.create_generic_event(Event(event_type="test", data={"test": "test"}))
         """
+        if not is_component_registered(EVENT_STORE_NAME):
+            logger.warning(
+                "Skipping event publish because event store component is not registered",
+            )
+            return
         try:
             if enrich_metadata:
                 event = cls.enrich_event_metadata(event)

application_sdk/outputs/statestore.py

@@ -84,8 +84,8 @@ class StateStoreOutput:
             >>> from application_sdk.outputs.statestore import StateStoreOutput
             >>> await StateStoreOutput.save_state_object("wf-123", {"test": "test"}, "workflow")
         """
-
         try:
+            logger.info(f"Saving state object for {id} with type {type}")
             # get the current state from object store
             current_state = StateStoreInput.get_state(id, type)
             state_file_path = build_state_store_path(id, type)
@@ -106,7 +106,7 @@ class StateStoreOutput:
                 file_path=local_state_file_path,
                 object_store_name=UPSTREAM_OBJECT_STORE_NAME,
             )
-
+            logger.info(f"State object saved for {id} with type {type}")
             return current_state
         except Exception as e:
             logger.error(f"Failed to store state: {str(e)}")

application_sdk/server/fastapi/__init__.py

@@ -493,7 +493,7 @@ class APIServer(ServerInterface):
         metrics = get_metrics()
 
         metadata_type = body.root.get("type", "all")
-        database = body.root.get("database", None)
+        database = body.root.get("database", "")
 
         try:
             if not self.handler:

application_sdk/version.py

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

{atlan_application_sdk-0.1.1rc28.dist-info → atlan_application_sdk-0.1.1rc30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc28
+Version: 0.1.1rc30
 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.1rc28.dist-info → atlan_application_sdk-0.1.1rc30.dist-info}/RECORD

@@ -1,6 +1,6 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
-application_sdk/constants.py,sha256=Npmw6ojj731faIE3BMeXE5bucxWFkAD13cYx5cQ9PZw,9302
-application_sdk/version.py,sha256=dAt_fXyCvt-5sLv5kCnHWbaTLVTYgdCQ1V7EoOwVJDU,88
+application_sdk/constants.py,sha256=JMJdTVjIJZX1XrvxISAcPh3YQ3aYKBTSQ2thqiF8BR4,9304
+application_sdk/version.py,sha256=JWzagPnLJJesFA2rkpWvghXzlW-fdaQlCWDoL0vAsJ4,88
 application_sdk/worker.py,sha256=2fLjuKNJafhaQXrHzmxXYO22F4ZSc0igMjoxXVNBFfk,6167
 application_sdk/activities/__init__.py,sha256=EH5VTHcfGykIX7V1HsG0J1Z-1FbJEXTQOET0HdzFDjU,9519
 application_sdk/activities/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -23,6 +23,7 @@ application_sdk/clients/workflow.py,sha256=6bSqmA3sNCk9oY68dOjBUDZ9DhNKQxPD75qqE
 application_sdk/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/common/aws_utils.py,sha256=aeL3BTMzv1UWJ4KxfwY5EsfYnxtS1FKNJ4xKdHeoTjc,3438
 application_sdk/common/credential_utils.py,sha256=M9oraG2uPeOSbxUAOJlP2IeClsDD79EhNkdow42dFsI,3025
+application_sdk/common/dapr_utils.py,sha256=0yHqDP6qNb1OT-bX2XRYQPZ5xkGkV13nyRw6GkPlHs8,1136
 application_sdk/common/dataframe_utils.py,sha256=PId9vT6AUoq3tesiTd4sSUvW7RUhPWdAAEBLuOprks4,1262
 application_sdk/common/error_codes.py,sha256=uROVfOOMrGPO8JroWB3vs5rIEhr0GfcPqXAK9wdhcVQ,13742
 application_sdk/common/utils.py,sha256=ktCZLp-AEiyd-IPOgbD83Dg9qa8Z0Sj_mJmmdSzpOak,14759
@@ -42,15 +43,15 @@ application_sdk/docgen/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 application_sdk/docgen/parsers/directory.py,sha256=8Kk2sjb-0l2wLO_njdlcuHjv5akoNgmf-FmaDSaE4WM,7751
 application_sdk/docgen/parsers/manifest.py,sha256=3NP-dBTpHAUQa477usMIDaKSb_9xfLE8G3RX0T1Bq2s,3318
 application_sdk/handlers/__init__.py,sha256=U7kKwVWK0FZz1uIJ2ANN0C5tD83k_9Nyz0ns6ttr92g,1152
-application_sdk/handlers/sql.py,sha256=qyLbTG9HZYX5_PAsvoSKANVcK53JS3sLUrqXMgXr1T8,16112
+application_sdk/handlers/sql.py,sha256=oeB-sgWwPYo31xaD87TyMc0h51Sary1F-CmhExt9_Pk,16100
 application_sdk/inputs/__init__.py,sha256=_d-cUhcDyoJTJR3PdQkC831go6VDw9AM6Bg7-qm3NHI,1900
 application_sdk/inputs/iceberg.py,sha256=xiv1kNtVx1k0h3ZJbJeXjZwdfBGSy9j9orYP_AyCYlI,2756
 application_sdk/inputs/json.py,sha256=J1CVz0YGQHDyq840TyoBHE7Baua2yIFHzsrybiZbeWk,6079
-application_sdk/inputs/objectstore.py,sha256=mOC5Bv10tZg9y4oikff_QViHSWZ_FHiotI9xdrlAIok,8781
+application_sdk/inputs/objectstore.py,sha256=uOJW0uB3FrDmnyHFhAd23QOq3MKrAzhYdKCszdpMeF8,8219
 application_sdk/inputs/parquet.py,sha256=Hsi6Nz_KwxFMB6DcHSQov5y_hRkoeN7e4xfpYwogveo,6346
 application_sdk/inputs/secretstore.py,sha256=WPGvsPMLUaARvXA6JSa4uHp7tQcCW4NMIIUDQyM3F-I,3946
 application_sdk/inputs/sql_query.py,sha256=1EREgea6kKNaMIyX2HLJgbJ07rtAgLasd9NyvDcdZok,10636
-application_sdk/inputs/statestore.py,sha256=Ib87toiYzSEg7_7C0o6CeS-c0vsjaJqiIEIKwiwXAcE,2954
+application_sdk/inputs/statestore.py,sha256=ORWnv8ZCqC1wT4vlW4v5EemJT4oQ3t_DlpjKDAgTRTs,3117
 application_sdk/observability/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/observability/logger_adaptor.py,sha256=WTqnNg78W2SRGOQVhELVLn6KMRsurkG1kc7essL08Lk,29529
 application_sdk/observability/metrics_adaptor.py,sha256=4TYPNn38zLeqxwf7cUbe8wh_zwQlr-nyiXjJsiEhTEM,16445
@@ -60,15 +61,15 @@ application_sdk/observability/utils.py,sha256=MKEpT0WYtpATUgLgJDkGQaAP_t-jpDYMUK
 application_sdk/observability/decorators/observability_decorator.py,sha256=JNrWNXT5W4klmlAc5b8C3_VBjDu0PI64W2ptr7LMzk4,8110
 application_sdk/outputs/__init__.py,sha256=HX8VcN22xyrkoRWdjQj8TrC5dEUG7cPzOcvJhlprqAs,8415
 application_sdk/outputs/atlan_storage.py,sha256=HQLbuyOZQC-GxYAiCVJakIJizTWy926tdMGOHvaBlD8,6029
-application_sdk/outputs/eventstore.py,sha256=fEW1oo9ncKylF5uLXTgjyRQZCrj72Gta3CaeduSX8nw,5548
+application_sdk/outputs/eventstore.py,sha256=MY9x_2A8Gempa1B3GQJ-C-B4dVrqO5u4Sfpla-2u91Q,5827
 application_sdk/outputs/iceberg.py,sha256=IGtj5WDgqLu6vzDEvw5DLsKsjm29Krto3AHvWpemr0A,5311
 application_sdk/outputs/json.py,sha256=xF-8mY3BZRRejip4s9npIUuFaAxgFmBQVaLMkrI_iCI,14117
 application_sdk/outputs/objectstore.py,sha256=TJvgfkJpGRK129ttxY7qRYJ7ASKZA4R6-0BUA3Lk7mc,4450
 application_sdk/outputs/parquet.py,sha256=A2EnEx1zWjaXk10u3eJusmWxGxt8WR7CHXDaJgsKpq0,11040
 application_sdk/outputs/secretstore.py,sha256=JS9vUzb11leDpcMQSCnLJuE9Ww-9G3wMvCdUKBPaw9I,1342
-application_sdk/outputs/statestore.py,sha256=erviUTNyK-iLPe03i4r-G-5D3uAWdRnkW83Hnu7wI7o,3854
+application_sdk/outputs/statestore.py,sha256=XiEag2e9WW3_D3xbWQGoNrHiFJz9916qcIvhrROX8_8,3999
 application_sdk/server/__init__.py,sha256=KTqE1YPw_3WDVMWatJUuf9OOiobLM2K5SMaBrI62sCo,1568
-application_sdk/server/fastapi/__init__.py,sha256=QiIaR43QhgJaLeKs5lTghfxCAGxVxFnYAudy47akbP4,27783
+application_sdk/server/fastapi/__init__.py,sha256=MnyLA75XP0_k3_yGp9nw9CTxw4Qa1DIZIs__BgzqJWI,27781
 application_sdk/server/fastapi/models.py,sha256=mFFxteDS3ZYXaq7Apor_Meo5WNxTCxqdrMkmTKQjvP0,6687
 application_sdk/server/fastapi/utils.py,sha256=2XI4DylhRQsukhX67lpAzRNCHeFCSpbuNd7TlE2IBJA,1164
 application_sdk/server/fastapi/middleware/logmiddleware.py,sha256=CxcPtDmCbSfSZ8RyI09nIshVIbCokyyA9bByQJ2G_ns,2545
@@ -130,8 +131,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=_NhszxIgmcQI6lVpjJoyJRFLwPYvJw1Dyqox_m9K2RA,11947
 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.1rc28.dist-info/METADATA,sha256=wAdmNUPLPieA06duXajVG_NdweyzxwZMA60mDYShTXc,5473
-atlan_application_sdk-0.1.1rc28.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc28.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc28.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc28.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc30.dist-info/METADATA,sha256=iT4sH5JQ7eYpgokSzDzPqZIuMVJwMZxmUPxNK7iSGWo,5473
+atlan_application_sdk-0.1.1rc30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc30.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc30.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc30.dist-info/RECORD,,