atlan-application-sdk 0.1.1rc21__tar.gz → 0.1.1rc22__tar.gz

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/.github/actions/e2e-examples/action.yaml

@@ -125,6 +125,7 @@ runs:
         SNOWFLAKE_ACCOUNT_ID: ${{ inputs.snowflake-account-id }}
         SNOWFLAKE_USER: ${{ inputs.snowflake-user }}
         SNOWFLAKE_PASSWORD: ${{ inputs.snowflake-password }}
+        ATLAN_LOCAL_DEVELOPMENT: true
       run: |
         cd examples
         uv run run_examples.py

atlan_application_sdk-0.1.1rc22/.github/release_notes.md

@@ -0,0 +1,11 @@
+## v0.1.1rc22 (July 21, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc21...v0.1.1rc22
+
+### Features
+
+- use object store as state store, path update and fixes (#618) (by @Nishchith Shetty in [d439cb4](https://github.com/atlanhq/application-sdk/commit/d439cb4))
+
+### Bug Fixes
+
+- suppress daft dependency loggers (#611) (by @Nishchith Shetty in [3e379d4](https://github.com/atlanhq/application-sdk/commit/3e379d4))

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/CHANGELOG.md

@@ -1,6 +1,19 @@
 # Changelog
 
 
+## v0.1.1rc22 (July 21, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc21...v0.1.1rc22
+
+### Features
+
+- use object store as state store, path update and fixes (#618) (by @Nishchith Shetty in [d439cb4](https://github.com/atlanhq/application-sdk/commit/d439cb4))
+
+### Bug Fixes
+
+- suppress daft dependency loggers (#611) (by @Nishchith Shetty in [3e379d4](https://github.com/atlanhq/application-sdk/commit/3e379d4))
+
+
 ## v0.1.1rc21 (July 10, 2025)
 
 Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc20...v0.1.1rc21

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc21
+Version: 0.1.1rc22
 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
@@ -25,7 +25,7 @@ Requires-Dist: fastapi[standard]>=0.115.0
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: opentelemetry-exporter-otlp>=1.27.0
 Requires-Dist: psutil>=7.0.0
-Requires-Dist: pyatlan==7.1.1
+Requires-Dist: pyatlan>=7.1.1
 Requires-Dist: pydantic>=2.10.6
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: uvloop>=0.21.0; sys_platform != 'win32'

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/activities/__init__.py

@@ -13,16 +13,22 @@ Example:
     ...         await state.handler.do_something()
 """
 
+import os
 from abc import ABC
 from typing import Any, Dict, Generic, Optional, TypeVar
 
 from pydantic import BaseModel
 from temporalio import activity
 
-from application_sdk.activities.common.utils import auto_heartbeater, get_workflow_id
+from application_sdk.activities.common.utils import (
+    auto_heartbeater,
+    build_output_path,
+    get_workflow_id,
+)
 from application_sdk.common.error_codes import OrchestratorError
+from application_sdk.constants import TEMPORARY_PATH
 from application_sdk.handlers import HandlerInterface
-from application_sdk.inputs.statestore import StateStoreInput
+from application_sdk.inputs.statestore import StateStoreInput, StateType
 from application_sdk.observability.logger_adaptor import get_logger
 
 logger = get_logger(__name__)
@@ -183,7 +189,15 @@ class ActivitiesInterface(ABC, Generic[ActivitiesStateType]):
 
         try:
             # This already handles the Dapr call internally
-            return StateStoreInput.extract_configuration(workflow_id)
+            workflow_args = StateStoreInput.get_state(workflow_id, StateType.WORKFLOWS)
+            workflow_args["output_prefix"] = workflow_args.get(
+                "output_prefix", TEMPORARY_PATH
+            )
+            workflow_args["output_path"] = os.path.join(
+                workflow_args["output_prefix"], build_output_path()
+            )
+            return workflow_args
+
         except Exception as e:
             logger.error(
                 f"Failed to retrieve workflow configuration for {workflow_id}: {str(e)}",

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/activities/common/utils.py

@@ -11,6 +11,7 @@ from typing import Any, Awaitable, Callable, Optional, TypeVar, cast
 
 from temporalio import activity
 
+from application_sdk.constants import APPLICATION_NAME, WORKFLOW_OUTPUT_PATH_TEMPLATE
 from application_sdk.observability.logger_adaptor import get_logger
 
 logger = get_logger(__name__)
@@ -43,6 +44,34 @@ def get_workflow_id() -> str:
         raise Exception("Failed to get workflow id")
 
 
+def get_workflow_run_id() -> str:
+    """Get the workflow run ID from the current activity."""
+    try:
+        return activity.info().workflow_run_id
+    except Exception as e:
+        logger.error("Failed to get workflow run id", exc_info=e)
+        raise Exception("Failed to get workflow run id")
+
+
+def build_output_path() -> str:
+    """Build a standardized output path for workflow artifacts.
+
+    This method creates a consistent output path format across all workflows using the WORKFLOW_OUTPUT_PATH_TEMPLATE constant.
+
+    Returns:
+        str: The standardized output path.
+
+    Example:
+        >>> build_output_path()
+        "artifacts/apps/appName/workflows/wf-123/run-456"
+    """
+    return WORKFLOW_OUTPUT_PATH_TEMPLATE.format(
+        application_name=APPLICATION_NAME,
+        workflow_id=get_workflow_id(),
+        run_id=get_workflow_run_id(),
+    )
+
+
 def auto_heartbeater(fn: F) -> F:
     """Decorator that automatically sends heartbeats during activity execution.
 

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/activities/metadata_extraction/sql.py

@@ -137,8 +137,8 @@ class BaseSQLMetadataExtractionActivities(ActivitiesInterface):
         self._state[workflow_id].handler = handler
 
         if "credential_guid" in workflow_args:
-            credentials = SecretStoreInput.extract_credentials(
-                workflow_args["credential_guid"]
+            credentials = await SecretStoreInput.fetch_secret(
+                secret_key=workflow_args["credential_guid"]
             )
             await sql_client.load(credentials)
 

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/activities/query_extraction/sql.py

@@ -128,8 +128,8 @@ class SQLQueryExtractionActivities(ActivitiesInterface):
         workflow_id = get_workflow_id()
         sql_client = self.sql_client_class()
         if "credential_guid" in workflow_args:
-            credentials = SecretStoreInput.extract_credentials(
-                workflow_args["credential_guid"]
+            credentials = await SecretStoreInput.fetch_secret(
+                secret_key=workflow_args["credential_guid"]
             )
             await sql_client.load(credentials)
 

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/clients/temporal.py

@@ -28,6 +28,7 @@ from application_sdk.constants import (
     WORKFLOW_NAMESPACE,
     WORKFLOW_PORT,
 )
+from application_sdk.inputs.statestore import StateType
 from application_sdk.observability.logger_adaptor import get_logger
 from application_sdk.outputs.eventstore import (
     ApplicationEventNames,
@@ -312,13 +313,12 @@ class TemporalWorkflowClient(WorkflowClient):
         """
         if "credentials" in workflow_args:
             # remove credentials from workflow_args and add reference to credentials
-            workflow_args["credential_guid"] = SecretStoreOutput.store_credentials(
+            workflow_args["credential_guid"] = await SecretStoreOutput.save_secret(
                 workflow_args["credentials"]
             )
             del workflow_args["credentials"]
 
         workflow_id = workflow_args.get("workflow_id")
-        output_prefix = workflow_args.get("output_prefix", "/tmp/output")
         if not workflow_id:
             # if workflow_id is not provided, create a new one
             workflow_id = workflow_args.get("argo_workflow_name", str(uuid.uuid4()))
@@ -327,11 +327,12 @@ class TemporalWorkflowClient(WorkflowClient):
                 {
                     "application_name": self.application_name,
                     "workflow_id": workflow_id,
-                    "output_prefix": output_prefix,
                 }
             )
 
-            StateStoreOutput.store_configuration(workflow_id, workflow_args)
+            await StateStoreOutput.save_state_object(
+                id=workflow_id, value=workflow_args, type=StateType.WORKFLOWS
+            )
 
             logger.info(f"Created workflow config with ID: {workflow_id}")
 

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/common/credential_utils.py

@@ -43,7 +43,9 @@ async def resolve_credentials(credentials: Dict[str, Any]) -> Dict[str, Any]:
             )
 
         # Fetch and apply secret using SecretStoreInput
-        secret_data = await SecretStoreInput.fetch_secret(credential_source, secret_key)
+        secret_data = await SecretStoreInput.fetch_secret(
+            secret_key=secret_key, component_name=credential_source
+        )
         return SecretStoreInput.apply_secret_values(credentials, secret_data)
 
     except Exception as e:

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/common/utils.py

@@ -18,9 +18,7 @@ from typing import (
 )
 
 from application_sdk.common.error_codes import CommonError
-from application_sdk.inputs.statestore import StateStoreInput
 from application_sdk.observability.logger_adaptor import get_logger
-from application_sdk.outputs.statestore import StateStoreOutput
 
 logger = get_logger(__name__)
 
@@ -268,38 +266,6 @@ def normalize_filters(
     return normalized_filter_list
 
 
-def get_workflow_config(config_id: str) -> Dict[str, Any]:
-    """Gets the workflow configuration from the state store using config id.
-
-    Args:
-        config_id: The configuration ID to retrieve.
-
-    Returns:
-        dict: The workflow configuration.
-    """
-    return StateStoreInput.extract_configuration(config_id)
-
-
-def update_workflow_config(config_id: str, config: Dict[str, Any]) -> Dict[str, Any]:
-    """Updates the workflow configuration.
-
-    Args:
-        config_id: The configuration ID to update.
-        config: The new configuration dictionary.
-
-    Returns:
-        dict: The updated workflow configuration.
-    """
-    extracted_config = get_workflow_config(config_id)
-
-    for key in extracted_config.keys():
-        if key in config and config[key] is not None:
-            extracted_config[key] = config[key]
-
-    StateStoreOutput.store_configuration(config_id, extracted_config)
-    return extracted_config
-
-
 def read_sql_files(
     queries_prefix: str = f"{os.path.dirname(os.path.abspath(__file__))}/queries",
 ) -> Dict[str, str]:

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/constants.py

@@ -44,6 +44,28 @@ APP_DASHBOARD_PORT = int(os.getenv("ATLAN_APP_DASHBOARD_PORT", "8000"))
 SQL_SERVER_MIN_VERSION = os.getenv("ATLAN_SQL_SERVER_MIN_VERSION")
 #: Path to the SQL queries directory
 SQL_QUERIES_PATH = os.getenv("ATLAN_SQL_QUERIES_PATH", "app/sql")
+#: Whether to use local development mode (used for instance to fetch secrets from the local state store)
+LOCAL_DEVELOPMENT = os.getenv("ATLAN_LOCAL_DEVELOPMENT", "false").lower() == "true"
+
+
+# Output Path Constants
+#: Output path format for workflows (example: objectstore://bucket/artifacts/apps/{application_name}/workflows/{workflow_id}/{workflow_run_id})
+WORKFLOW_OUTPUT_PATH_TEMPLATE = (
+    "artifacts/apps/{application_name}/workflows/{workflow_id}/{run_id}"
+)
+
+# Temporary Path (used to store intermediate files)
+TEMPORARY_PATH = os.getenv("ATLAN_TEMPORARY_PATH", "./local/tmp/")
+
+# State Store Constants
+#: Path template for state store files (example: objectstore://bucket/persistent-artifacts/apps/{application_name}/{state_type}/{id}/config.json)
+STATE_STORE_PATH_TEMPLATE = (
+    "persistent-artifacts/apps/{application_name}/{state_type}/{id}/config.json"
+)
+
+# Observability Constants
+#: Directory for storing observability data
+OBSERVABILITY_DIR = "artifacts/apps/{application_name}/observability"
 
 # Workflow Client Constants
 #: Host address for the Temporal server
@@ -87,6 +109,7 @@ OBJECT_STORE_NAME = os.getenv("OBJECT_STORE_NAME", "objectstore")
 #: Name of the pubsub component in DAPR
 EVENT_STORE_NAME = os.getenv("EVENT_STORE_NAME", "eventstore")
 
+
 # Logger Constants
 #: Log level for the application (DEBUG, INFO, WARNING, ERROR, CRITICAL)
 LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()
@@ -120,10 +143,6 @@ OTEL_QUEUE_SIZE = int(os.getenv("OTEL_QUEUE_SIZE", "2048"))
 #: AWS Session Name
 AWS_SESSION_NAME = os.getenv("AWS_SESSION_NAME", "temp-session")
 
-# Observability Constants
-#: Directory for storing observability data
-OBSERVABILITY_DIR = os.environ.get("ATLAN_OBSERVABILITY_DIR", "/tmp/observability")
-
 # Log batching configuration
 LOG_BATCH_SIZE = int(os.environ.get("ATLAN_LOG_BATCH_SIZE", 100))
 LOG_FLUSH_INTERVAL_SECONDS = int(os.environ.get("ATLAN_LOG_FLUSH_INTERVAL_SECONDS", 10))
@@ -133,7 +152,6 @@ LOG_RETENTION_DAYS = int(os.environ.get("ATLAN_LOG_RETENTION_DAYS", 30))
 LOG_CLEANUP_ENABLED = bool(os.environ.get("ATLAN_LOG_CLEANUP_ENABLED", False))
 
 # Log Location configuration
-LOG_DIR = os.environ.get("ATLAN_LOG_DIR", "/tmp/observability")
 LOG_FILE_NAME = os.environ.get("ATLAN_LOG_FILE_NAME", "log.parquet")
 # Hive Partitioning Configuration
 ENABLE_HIVE_PARTITIONING = (

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/inputs/objectstore.py

@@ -88,11 +88,16 @@ class ObjectStoreInput:
 
         Args:
             download_file_prefix (str): The base path to calculate relative paths from.
+                example: /tmp/output
             file_path (str): The full path to where the file should be downloaded.
+                example: /tmp/output/persistent-artifacts/apps/myapp/data/wf-123/state.json
 
         Raises:
             Exception: If there's an error downloading the file from the object store.
         """
+        if not os.path.exists(os.path.dirname(file_path)):
+            os.makedirs(os.path.dirname(file_path), exist_ok=True)
+
         with DaprClient() as client:
             relative_path = os.path.relpath(file_path, download_file_prefix)
             metadata = {"key": relative_path, "fileName": relative_path}

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/inputs/secretstore.py

@@ -7,7 +7,8 @@ from typing import Any, Dict
 
 from dapr.clients import DaprClient
 
-from application_sdk.inputs.statestore import StateStoreInput
+from application_sdk.constants import LOCAL_DEVELOPMENT, SECRET_STORE_NAME
+from application_sdk.inputs.statestore import StateStoreInput, StateType
 from application_sdk.observability.logger_adaptor import get_logger
 
 logger = get_logger(__name__)
@@ -15,7 +16,34 @@ logger = get_logger(__name__)
 
 class SecretStoreInput:
     @classmethod
-    async def fetch_secret(cls, component_name: str, secret_key: str) -> Dict[str, Any]:
+    def get_secret(
+        cls, secret_key: str, component_name: str = SECRET_STORE_NAME
+    ) -> Dict[str, Any]:
+        """Get secret from the Dapr component.
+
+        Args:
+            secret_key: Key of the secret to fetch
+            component_name: Name of the Dapr component to fetch from
+
+        Returns:
+            Dict with processed secret data
+        """
+        try:
+            with DaprClient() as client:
+                dapr_secret_object = client.get_secret(
+                    store_name=component_name, key=secret_key
+                )
+                return cls._process_secret_data(dapr_secret_object.secret)
+        except Exception as e:
+            logger.error(
+                f"Failed to fetch secret using component {component_name}: {str(e)}"
+            )
+            raise
+
+    @classmethod
+    async def fetch_secret(
+        cls, secret_key: str, component_name: str = SECRET_STORE_NAME
+    ) -> Dict[str, Any]:
         """Fetch secret using the Dapr component.
 
         Args:
@@ -29,9 +57,15 @@ class SecretStoreInput:
             Exception: If secret fetching fails
         """
         try:
-            with DaprClient() as client:
-                secret = client.get_secret(store_name=component_name, key=secret_key)
-                return cls._process_secret_data(secret.secret)
+            secret = {}
+            if not LOCAL_DEVELOPMENT:
+                secret = cls.get_secret(secret_key, component_name)
+
+            credential_config = StateStoreInput.get_state(
+                secret_key, StateType.CREDENTIALS
+            )
+            secret.update(credential_config)
+            return secret
         except Exception as e:
             logger.error(
                 f"Failed to fetch secret using component {component_name}: {str(e)}"
@@ -93,25 +127,3 @@ class SecretStoreInput:
                     result_data["extra"][key] = secret_data[value]
 
         return result_data
-
-    @classmethod
-    def extract_credentials(cls, credential_guid: str) -> Dict[str, Any]:
-        """Extract credentials from the state store using the credential GUID.
-
-        Args:
-            credential_guid: The unique identifier for the credentials.
-
-        Returns:
-            Dict[str, Any]: The credentials if found.
-
-        Raises:
-            ValueError: If the credential_guid is invalid or credentials are not found.
-            Exception: If there's an error with the Dapr client operations.
-
-        Examples:
-            >>> SecretStoreInput.extract_credentials("1234567890")
-            {"username": "admin", "password": "password"}
-        """
-        if not credential_guid:
-            raise ValueError("Invalid credential GUID provided.")
-        return StateStoreInput.get_state(f"credential_{credential_guid}")

atlan_application_sdk-0.1.1rc22/application_sdk/inputs/statestore.py

@@ -0,0 +1,98 @@
+"""State store for the application."""
+
+import json
+import os
+from enum import Enum
+from typing import Any, Dict
+
+from temporalio import activity
+
+from application_sdk.constants import (
+    APPLICATION_NAME,
+    STATE_STORE_PATH_TEMPLATE,
+    TEMPORARY_PATH,
+)
+from application_sdk.inputs.objectstore import ObjectStoreInput
+from application_sdk.observability.logger_adaptor import get_logger
+
+logger = get_logger(__name__)
+activity.logger = logger
+
+
+class StateType(Enum):
+    WORKFLOWS = "workflows"
+    CREDENTIALS = "credentials"
+
+    @classmethod
+    def is_member(cls, type: str) -> bool:
+        return type in cls._value2member_map_
+
+
+def build_state_store_path(id: str, state_type: StateType) -> str:
+    """Build the state file path for the given id and type.
+
+    Args:
+        id: The unique identifier for the state.
+        state_type: The type of state (workflows, credentials, etc.).
+
+    Returns:
+        str: The constructed state file path.
+
+    Example:
+        >>> build_state_store_path("wf-123", "workflows")
+        'persistent-artifacts/apps/my-app/workflows/wf-123/config.json'
+    """
+    return STATE_STORE_PATH_TEMPLATE.format(
+        application_name=APPLICATION_NAME, state_type=state_type.value, id=id
+    )
+
+
+class StateStoreInput:
+    @classmethod
+    def get_state(cls, id: str, type: StateType) -> Dict[str, Any]:
+        """Get state from the store.
+
+        Args:
+            id: The key to retrieve the state for.
+            type: The type of state to retrieve.
+
+        Returns:
+            Dict[str, Any]: The retrieved state data.
+
+        Raises:
+            ValueError: If no state is found for the given key.
+            IOError: If there's an error with the Dapr client operations.
+
+        Example:
+            >>> from application_sdk.inputs.statestore import StateStoreInput
+
+            >>> state = StateStoreInput.get_state("wf-123")
+            >>> print(state)
+            {'test': 'test'}
+        """
+
+        state_file_path = build_state_store_path(id, type)
+        state = {}
+
+        try:
+            local_state_file_path = os.path.join(TEMPORARY_PATH, state_file_path)
+            ObjectStoreInput.download_file_from_object_store(
+                download_file_prefix=TEMPORARY_PATH,
+                file_path=local_state_file_path,
+            )
+
+            with open(local_state_file_path, "r") as file:
+                state = json.load(file)
+
+        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(
+                    f"No state found for {type.value} with id '{id}', returning empty dict"
+                )
+                pass
+            else:
+                logger.error(f"Failed to extract state: {str(e)}")
+                raise
+
+        return state

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/observability/logger_adaptor.py

@@ -24,7 +24,6 @@ from application_sdk.constants import (
     LOG_FLUSH_INTERVAL_SECONDS,
     LOG_LEVEL,
     LOG_RETENTION_DAYS,
-    OBSERVABILITY_DIR,
     OTEL_BATCH_DELAY_MS,
     OTEL_BATCH_SIZE,
     OTEL_EXPORTER_OTLP_ENDPOINT,
@@ -36,7 +35,10 @@ from application_sdk.constants import (
     SERVICE_VERSION,
 )
 from application_sdk.observability.observability import AtlanObservability
-from application_sdk.observability.utils import get_workflow_context
+from application_sdk.observability.utils import (
+    get_observability_dir,
+    get_workflow_context,
+)
 
 
 class LogExtraModel(BaseModel):
@@ -200,6 +202,13 @@ logging.basicConfig(
     level=logging.getLevelNamesMapping()[LOG_LEVEL], handlers=[InterceptHandler()]
 )
 
+DEPENDENCY_LOGGERS = ["daft_io.stats", "tracing.span"]
+
+# Configure external dependency loggers to reduce noise
+for logger_name in DEPENDENCY_LOGGERS:
+    logging.getLogger(logger_name).setLevel(logging.WARNING)
+
+
 # Add these constants
 SEVERITY_MAPPING = {
     "DEBUG": SeverityNumber.DEBUG,
@@ -245,7 +254,7 @@ class AtlanLoggerAdapter(AtlanObservability[LogRecordModel]):
             flush_interval=LOG_FLUSH_INTERVAL_SECONDS,
             retention_days=LOG_RETENTION_DAYS,
             cleanup_enabled=LOG_CLEANUP_ENABLED,
-            data_dir=OBSERVABILITY_DIR,
+            data_dir=get_observability_dir(),
             file_name=LOG_FILE_NAME,
         )
         self.logger_name = logger_name

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/observability/metrics_adaptor.py

@@ -19,7 +19,6 @@ from application_sdk.constants import (
     METRICS_FILE_NAME,
     METRICS_FLUSH_INTERVAL_SECONDS,
     METRICS_RETENTION_DAYS,
-    OBSERVABILITY_DIR,
     OTEL_BATCH_DELAY_MS,
     OTEL_EXPORTER_OTLP_ENDPOINT,
     OTEL_EXPORTER_TIMEOUT_SECONDS,
@@ -30,7 +29,10 @@ from application_sdk.constants import (
 )
 from application_sdk.observability.logger_adaptor import get_logger
 from application_sdk.observability.observability import AtlanObservability
-from application_sdk.observability.utils import get_workflow_context
+from application_sdk.observability.utils import (
+    get_observability_dir,
+    get_workflow_context,
+)
 
 
 class MetricType(Enum):
@@ -165,7 +167,7 @@ class AtlanMetricsAdapter(AtlanObservability[MetricRecord]):
             flush_interval=METRICS_FLUSH_INTERVAL_SECONDS,
             retention_days=METRICS_RETENTION_DAYS,
             cleanup_enabled=METRICS_CLEANUP_ENABLED,
-            data_dir=OBSERVABILITY_DIR,
+            data_dir=get_observability_dir(),
             file_name=METRICS_FILE_NAME,
         )
 

{atlan_application_sdk-0.1.1rc21 → atlan_application_sdk-0.1.1rc22}/application_sdk/observability/observability.py

@@ -20,10 +20,11 @@ from application_sdk.constants import (
     LOG_FILE_NAME,
     METRICS_FILE_NAME,
     OBJECT_STORE_NAME,
-    OBSERVABILITY_DIR,
     STATE_STORE_NAME,
+    TEMPORARY_PATH,
     TRACES_FILE_NAME,
 )
+from application_sdk.observability.utils import get_observability_dir
 
 
 class LogRecord(BaseModel):
@@ -440,8 +441,8 @@ class AtlanObservability(Generic[T], ABC):
                 with open(parquet_path, "rb") as f:
                     file_content = f.read()
                     metadata = {
-                        "key": os.path.relpath(parquet_path, self.data_dir),
-                        "blobName": os.path.relpath(parquet_path, self.data_dir),
+                        "key": os.path.relpath(parquet_path, TEMPORARY_PATH),
+                        "blobName": os.path.relpath(parquet_path, TEMPORARY_PATH),
                         "fileName": os.path.basename(parquet_path),
                     }
                     with DaprClient() as client:
@@ -614,9 +615,10 @@ class AtlanObservability(Generic[T], ABC):
 class DuckDBUI:
     """Class to handle DuckDB UI functionality."""
 
-    def __init__(self, db_path="/tmp/observability/observability.db"):
+    def __init__(self):
         """Initialize the DuckDB UI handler."""
-        self.db_path = db_path
+        self.observability_dir = get_observability_dir()
+        self.db_path = self.observability_dir + "/observability.db"
         self._duckdb_ui_con = None
 
     def _is_duckdb_ui_running(self, host="0.0.0.0", port=4213):
@@ -628,10 +630,10 @@ class DuckDBUI:
             result = sock.connect_ex((host, port))
             return result == 0
 
-    def start_ui(self, db_path=OBSERVABILITY_DIR):
+    def start_ui(self):
         """Start DuckDB UI and create views for Hive partitioned parquet files."""
         if not self._is_duckdb_ui_running():
-            os.makedirs(OBSERVABILITY_DIR, exist_ok=True)
+            os.makedirs(self.observability_dir, exist_ok=True)
             con = duckdb.connect(self.db_path)
 
             def process_partitioned_files(directory, prefix=""):
@@ -660,7 +662,7 @@ class DuckDBUI:
 
             # Process each type of data
             for data_type in ["logs", "metrics", "traces"]:
-                data_dir = os.path.join(OBSERVABILITY_DIR, data_type)
+                data_dir = os.path.join(self.observability_dir, data_type)
                 if os.path.exists(data_dir):
                     process_partitioned_files(data_dir, data_type)