atlan-application-sdk 2.0.0__py3-none-any.whl → 2.1.0__py3-none-any.whl

application_sdk/activities/common/sql_utils.py

@@ -131,7 +131,9 @@ async def finalize_multidb_results(
     dataframe_list: List[
         Union[AsyncIterator["pd.DataFrame"], Iterator["pd.DataFrame"]]
     ],
-    setup_parquet_output_func: Callable[[str, bool], Optional[ParquetFileWriter]],
+    setup_parquet_output_func: Callable[
+        [str, bool, Optional[str]], Optional[ParquetFileWriter]
+    ],
     output_path: str,
     typename: str,
 ) -> Optional[Union[ActivityStatistics, "pd.DataFrame"]]:
@@ -189,7 +191,9 @@ async def finalize_multidb_results(
                 return concatenated
 
             # Create new parquet output for concatenated data
-            concatenated_parquet_output = setup_parquet_output_func(output_path, True)
+            concatenated_parquet_output = setup_parquet_output_func(
+                output_path, True, typename
+            )
             if concatenated_parquet_output:
                 await concatenated_parquet_output.write(concatenated)  # type: ignore
                 return await concatenated_parquet_output.close()

application_sdk/io/__init__.py

@@ -49,9 +49,104 @@ if TYPE_CHECKING:
 
 
 class Reader(ABC):
+    """Abstract base class for reader data sources.
+
+    This class defines the interface for reader handlers that can read data
+    from various sources in different formats. Follows Python's file I/O
+    pattern with read/close semantics and supports context managers.
+
+    Attributes:
+        path (str): Path where the reader will read from.
+        _is_closed (bool): Whether the reader has been closed.
+        _downloaded_files (List[str]): List of downloaded temporary files to clean up.
+        cleanup_on_close (bool): Whether to clean up downloaded temp files on close.
+
+    Example:
+        Using close() explicitly::
+
+            reader = ParquetFileReader(path="/data/input")
+            df = await reader.read()
+            await reader.close()  # Cleans up any downloaded temp files
+
+        Using context manager (recommended)::
+
+            async with ParquetFileReader(path="/data/input") as reader:
+                df = await reader.read()
+            # close() called automatically
+
+        Reading in batches with context manager::
+
+            async with JsonFileReader(path="/data/input") as reader:
+                async for batch in reader.read_batches():
+                    process(batch)
+            # close() called automatically
     """
-    Abstract base class for reader data sources.
-    """
+
+    path: str
+    _is_closed: bool = False
+    _downloaded_files: List[str] = []
+    cleanup_on_close: bool = True
+
+    async def __aenter__(self) -> "Reader":
+        """Enter the async context manager.
+
+        Returns:
+            Reader: The reader instance.
+        """
+        return self
+
+    async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        """Exit the async context manager, closing the reader.
+
+        Args:
+            exc_type: Exception type if an exception was raised.
+            exc_val: Exception value if an exception was raised.
+            exc_tb: Exception traceback if an exception was raised.
+        """
+        await self.close()
+
+    async def close(self) -> None:
+        """Close the reader and clean up any downloaded temporary files.
+
+        This method cleans up any temporary files that were downloaded from
+        the object store during read operations. Calling close() multiple
+        times is safe (subsequent calls are no-ops).
+
+        Note:
+            Set ``cleanup_on_close=False`` during initialization to retain
+            downloaded files after closing.
+
+        Example::
+
+            reader = ParquetFileReader(path="/data/input")
+            df = await reader.read()
+            await reader.close()  # Cleans up temp files
+        """
+        if self._is_closed:
+            return
+
+        if self.cleanup_on_close and self._downloaded_files:
+            await self._cleanup_downloaded_files()
+
+        self._is_closed = True
+
+    async def _cleanup_downloaded_files(self) -> None:
+        """Clean up downloaded temporary files.
+
+        Override this method in subclasses for custom cleanup behavior.
+        """
+        import shutil
+
+        for file_path in self._downloaded_files:
+            try:
+                if os.path.isfile(file_path):
+                    os.remove(file_path)
+                elif os.path.isdir(file_path):
+                    shutil.rmtree(file_path, ignore_errors=True)
+            except Exception as e:
+                logger.warning(f"Failed to clean up temporary file {file_path}: {e}")
+
+        self._downloaded_files.clear()
 
     @abstractmethod
     def read_batches(
@@ -62,27 +157,27 @@ class Reader(ABC):
         Iterator["daft.DataFrame"],
         AsyncIterator["daft.DataFrame"],
     ]:
-        """
-        Get an iterator of batched pandas DataFrames.
+        """Get an iterator of batched pandas DataFrames.
 
         Returns:
             Iterator["pd.DataFrame"]: An iterator of batched pandas DataFrames.
 
         Raises:
             NotImplementedError: If the method is not implemented.
+            ValueError: If the reader has been closed.
         """
         raise NotImplementedError
 
     @abstractmethod
     async def read(self) -> Union["pd.DataFrame", "daft.DataFrame"]:
-        """
-        Get a single pandas or daft DataFrame.
+        """Get a single pandas or daft DataFrame.
 
         Returns:
             Union["pd.DataFrame", "daft.DataFrame"]: A pandas or daft DataFrame.
 
         Raises:
             NotImplementedError: If the method is not implemented.
+            ValueError: If the reader has been closed.
         """
         raise NotImplementedError
 

application_sdk/io/json.py

@@ -27,9 +27,36 @@ activity.logger = logger
 
 
 class JsonFileReader(Reader):
-    """
-    JSON File Reader class to read data from JSON files using daft and pandas.
+    """JSON File Reader class to read data from JSON files using daft and pandas.
+
     Supports reading both single files and directories containing multiple JSON files.
+    Follows Python's file I/O pattern with read/close semantics and supports context managers.
+
+    Attributes:
+        path (str): Path to JSON file or directory containing JSON files.
+        chunk_size (int): Number of rows per batch.
+        file_names (Optional[List[str]]): List of specific file names to read.
+        dataframe_type (DataframeType): Type of dataframe to return (pandas or daft).
+        cleanup_on_close (bool): Whether to clean up downloaded temp files on close.
+
+    Example:
+        Using context manager (recommended)::
+
+            async with JsonFileReader(path="/data/input") as reader:
+                df = await reader.read()
+            # close() called automatically, temp files cleaned up
+
+        Reading in batches::
+
+            async with JsonFileReader(path="/data/input", chunk_size=50000) as reader:
+                async for batch in reader.read_batches():
+                    process(batch)
+
+        Using close() explicitly::
+
+            reader = JsonFileReader(path="/data/input")
+            df = await reader.read()
+            await reader.close()  # Clean up downloaded temp files
     """
 
     def __init__(
@@ -38,6 +65,7 @@ class JsonFileReader(Reader):
         file_names: Optional[List[str]] = None,
         chunk_size: Optional[int] = 100000,
         dataframe_type: DataframeType = DataframeType.pandas,
+        cleanup_on_close: bool = True,
     ):
         """Initialize the JsonInput class.
 
@@ -48,6 +76,8 @@ class JsonFileReader(Reader):
                 Wildcards are not supported.
             file_names (Optional[List[str]]): List of specific file names to read. Defaults to None.
             chunk_size (int): Number of rows per batch. Defaults to 100000.
+            dataframe_type (DataframeType): Type of dataframe to read. Defaults to DataframeType.pandas.
+            cleanup_on_close (bool): Whether to clean up downloaded temp files on close. Defaults to True.
 
         Raises:
             ValueError: When path is not provided or when single file path is combined with file_names
@@ -65,12 +95,22 @@ class JsonFileReader(Reader):
         self.chunk_size = chunk_size
         self.file_names = file_names
         self.dataframe_type = dataframe_type
+        self.cleanup_on_close = cleanup_on_close
+        self._is_closed = False
+        self._downloaded_files: List[str] = []
 
     async def read(self) -> Union["pd.DataFrame", "daft.DataFrame"]:
+        """Read the data from the JSON files and return as a single DataFrame.
+
+        Returns:
+            Union[pd.DataFrame, daft.DataFrame]: Combined dataframe from JSON files.
+
+        Raises:
+            ValueError: If the reader has been closed or dataframe_type is unsupported.
         """
-        Method to read the data from the json files in the path
-        and return as a single combined pandas dataframe
-        """
+        if self._is_closed:
+            raise ValueError("Cannot read from a closed reader")
+
         if self.dataframe_type == DataframeType.pandas:
             return await self._get_dataframe()
         elif self.dataframe_type == DataframeType.daft:
@@ -84,10 +124,18 @@ class JsonFileReader(Reader):
         AsyncIterator["pd.DataFrame"],
         AsyncIterator["daft.DataFrame"],
     ]:
+        """Read the data from the JSON files and return as batched DataFrames.
+
+        Returns:
+            Union[AsyncIterator[pd.DataFrame], AsyncIterator[daft.DataFrame]]:
+                Async iterator of DataFrames.
+
+        Raises:
+            ValueError: If the reader has been closed or dataframe_type is unsupported.
         """
-        Method to read the data from the json files in the path
-        and return as a batched pandas dataframe
-        """
+        if self._is_closed:
+            raise ValueError("Cannot read from a closed reader")
+
         if self.dataframe_type == DataframeType.pandas:
             return self._get_batched_dataframe()
         elif self.dataframe_type == DataframeType.daft:
@@ -98,10 +146,7 @@ class JsonFileReader(Reader):
     async def _get_batched_dataframe(
         self,
     ) -> AsyncIterator["pd.DataFrame"]:
-        """
-        Method to read the data from the json files in the path
-        and return as a batched pandas dataframe
-        """
+        """Read the data from the JSON files and return as a batched pandas dataframe."""
         try:
             import pandas as pd
 
@@ -109,6 +154,8 @@ class JsonFileReader(Reader):
             json_files = await download_files(
                 self.path, self.extension, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(json_files)
             logger.info(f"Reading {len(json_files)} JSON files in batches")
 
             for json_file in json_files:
@@ -124,10 +171,7 @@ class JsonFileReader(Reader):
             raise
 
     async def _get_dataframe(self) -> "pd.DataFrame":
-        """
-        Method to read the data from the json files in the path
-        and return as a single combined pandas dataframe
-        """
+        """Read the data from the JSON files and return as a single pandas dataframe."""
         try:
             import pandas as pd
 
@@ -135,6 +179,8 @@ class JsonFileReader(Reader):
             json_files = await download_files(
                 self.path, self.extension, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(json_files)
             logger.info(f"Reading {len(json_files)} JSON files as pandas dataframe")
 
             return pd.concat(
@@ -149,10 +195,7 @@ class JsonFileReader(Reader):
     async def _get_batched_daft_dataframe(
         self,
     ) -> AsyncIterator["daft.DataFrame"]:  # noqa: F821
-        """
-        Method to read the data from the json files in the path
-        and return as a batched daft dataframe
-        """
+        """Read the data from the JSON files and return as a batched daft dataframe."""
         try:
             import daft
 
@@ -160,6 +203,8 @@ class JsonFileReader(Reader):
             json_files = await download_files(
                 self.path, self.extension, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(json_files)
             logger.info(f"Reading {len(json_files)} JSON files as daft batches")
 
             # Yield each discovered file as separate batch with chunking
@@ -170,10 +215,7 @@ class JsonFileReader(Reader):
             raise
 
     async def _get_daft_dataframe(self) -> "daft.DataFrame":  # noqa: F821
-        """
-        Method to read the data from the json files in the path
-        and return as a single combined daft dataframe
-        """
+        """Read the data from the JSON files and return as a single daft dataframe."""
         try:
             import daft
 
@@ -181,6 +223,8 @@ class JsonFileReader(Reader):
             json_files = await download_files(
                 self.path, self.extension, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(json_files)
             logger.info(f"Reading {len(json_files)} JSON files with daft")
 
             # Use the discovered/downloaded files directly

application_sdk/io/parquet.py

@@ -40,9 +40,37 @@ if TYPE_CHECKING:
 
 
 class ParquetFileReader(Reader):
-    """
-    Parquet File Reader class to read data from Parquet files using daft and pandas.
+    """Parquet File Reader class to read data from Parquet files using daft and pandas.
+
     Supports reading both single files and directories containing multiple parquet files.
+    Follows Python's file I/O pattern with read/close semantics and supports context managers.
+
+    Attributes:
+        path (str): Path to parquet file or directory containing parquet files.
+        chunk_size (int): Number of rows per batch.
+        buffer_size (int): Number of rows per batch for daft.
+        file_names (Optional[List[str]]): List of specific file names to read.
+        dataframe_type (DataframeType): Type of dataframe to return (pandas or daft).
+        cleanup_on_close (bool): Whether to clean up downloaded temp files on close.
+
+    Example:
+        Using context manager (recommended)::
+
+            async with ParquetFileReader(path="/data/input") as reader:
+                df = await reader.read()
+            # close() called automatically, temp files cleaned up
+
+        Reading in batches::
+
+            async with ParquetFileReader(path="/data/input", chunk_size=50000) as reader:
+                async for batch in reader.read_batches():
+                    process(batch)
+
+        Using close() explicitly::
+
+            reader = ParquetFileReader(path="/data/input")
+            df = await reader.read()
+            await reader.close()  # Clean up downloaded temp files
     """
 
     def __init__(
@@ -52,6 +80,7 @@ class ParquetFileReader(Reader):
         buffer_size: Optional[int] = 5000,
         file_names: Optional[List[str]] = None,
         dataframe_type: DataframeType = DataframeType.pandas,
+        cleanup_on_close: bool = True,
     ):
         """Initialize the Parquet input class.
 
@@ -64,6 +93,7 @@ class ParquetFileReader(Reader):
             buffer_size (int): Number of rows per batch. Defaults to 5000.
             file_names (Optional[List[str]]): List of file names to read. Defaults to None.
             dataframe_type (DataframeType): Type of dataframe to read. Defaults to DataframeType.pandas.
+            cleanup_on_close (bool): Whether to clean up downloaded temp files on close. Defaults to True.
 
         Raises:
             ValueError: When path is not provided or when single file path is combined with file_names
@@ -81,12 +111,22 @@ class ParquetFileReader(Reader):
         self.buffer_size = buffer_size
         self.file_names = file_names
         self.dataframe_type = dataframe_type
+        self.cleanup_on_close = cleanup_on_close
+        self._is_closed = False
+        self._downloaded_files: List[str] = []
 
     async def read(self) -> Union["pd.DataFrame", "daft.DataFrame"]:
+        """Read the data from the parquet files and return as a single DataFrame.
+
+        Returns:
+            Union[pd.DataFrame, daft.DataFrame]: Combined dataframe from parquet files.
+
+        Raises:
+            ValueError: If the reader has been closed or dataframe_type is unsupported.
         """
-        Method to read the data from the parquet files in the path
-        and return as a single combined pandas dataframe
-        """
+        if self._is_closed:
+            raise ValueError("Cannot read from a closed reader")
+
         if self.dataframe_type == DataframeType.pandas:
             return await self._get_dataframe()
         elif self.dataframe_type == DataframeType.daft:
@@ -100,10 +140,18 @@ class ParquetFileReader(Reader):
         AsyncIterator["pd.DataFrame"],
         AsyncIterator["daft.DataFrame"],
     ]:
+        """Read the data from the parquet files and return as batched DataFrames.
+
+        Returns:
+            Union[AsyncIterator[pd.DataFrame], AsyncIterator[daft.DataFrame]]:
+                Async iterator of DataFrames.
+
+        Raises:
+            ValueError: If the reader has been closed or dataframe_type is unsupported.
         """
-        Method to read the data from the parquet files in the path
-        and return as a batched pandas dataframe
-        """
+        if self._is_closed:
+            raise ValueError("Cannot read from a closed reader")
+
         if self.dataframe_type == DataframeType.pandas:
             return self._get_batched_dataframe()
         elif self.dataframe_type == DataframeType.daft:
@@ -149,6 +197,8 @@ class ParquetFileReader(Reader):
             parquet_files = await download_files(
                 self.path, PARQUET_FILE_EXTENSION, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(parquet_files)
             logger.info(f"Reading {len(parquet_files)} parquet files")
 
             return pd.concat(
@@ -208,6 +258,8 @@ class ParquetFileReader(Reader):
             parquet_files = await download_files(
                 self.path, PARQUET_FILE_EXTENSION, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(parquet_files)
             logger.info(f"Reading {len(parquet_files)} parquet files in batches")
 
             # Process each file individually to maintain memory efficiency
@@ -259,6 +311,8 @@ class ParquetFileReader(Reader):
             parquet_files = await download_files(
                 self.path, PARQUET_FILE_EXTENSION, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(parquet_files)
             logger.info(f"Reading {len(parquet_files)} parquet files with daft")
 
             # Use the discovered/downloaded files directly
@@ -317,6 +371,8 @@ class ParquetFileReader(Reader):
             parquet_files = await download_files(
                 self.path, PARQUET_FILE_EXTENSION, self.file_names
             )
+            # Track downloaded files for cleanup on close
+            self._downloaded_files.extend(parquet_files)
             logger.info(f"Reading {len(parquet_files)} parquet files as daft batches")
 
             # Create a lazy dataframe without loading data into memory

application_sdk/observability/observability.py

@@ -426,16 +426,9 @@ class AtlanObservability(Generic[T], ABC):
                         chunk_start=0,
                         chunk_part=int(time()),
                     )
-                    logging.info(
-                        f"Successfully instantiated ParquetFileWriter for partition: {partition_path}"
-                    )
 
                     await parquet_writer._write_dataframe(dataframe=df)
 
-                    logging.info(
-                        f"Successfully wrote {len(df)} records to partition: {partition_path}"
-                    )
-
                 except Exception as partition_error:
                     logging.error(
                         f"Error processing partition {partition_path}: {str(partition_error)}"

application_sdk/services/objectstore.py

@@ -459,9 +459,22 @@ class ObjectStore:
 
             logger.info(f"Found {len(file_list)} files to download from: {source}")
 
+            # Normalize source prefix to use forward slashes for comparison
+            normalized_source = cls._normalize_object_store_key(source)
+
             # Download each file
             for file_path in file_list:
-                local_file_path = os.path.join(destination, file_path)
+                normalized_file_path = cls._normalize_object_store_key(file_path)
+                if normalized_file_path.startswith(normalized_source):
+                    # Extract relative path after the prefix
+                    relative_path = normalized_file_path[
+                        len(normalized_source) :
+                    ].lstrip("/")
+                else:
+                    # Fallback to just the filename
+                    relative_path = os.path.basename(normalized_file_path)
+
+                local_file_path = os.path.join(destination, relative_path)
                 await cls.download_file(file_path, local_file_path, store_name)
 
             logger.info(f"Successfully downloaded all files from: {source}")

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "2.0.0"
+__version__ = "2.1.0"

{atlan_application_sdk-2.0.0.dist-info → atlan_application_sdk-2.1.0.dist-info}/METADATA

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

@@ -1,13 +1,13 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
 application_sdk/constants.py,sha256=TvdmKQShVWBNQZdVF2y-fxuE31FmeraTnqQ9jT_n5XY,11567
-application_sdk/version.py,sha256=TY0SZFUH9N2qGF6tlbmJIww-CY7x-myMGotNEhFJko4,84
+application_sdk/version.py,sha256=b8rehaDy9Q_sCU-sBPVls3ZFH8_ZNacFO5aQrIaO8ts,84
 application_sdk/worker.py,sha256=DLMocpHvvwpdAopyXhxwM7ftaNlKvZMQfkgy1MFyiik,7561
 application_sdk/activities/__init__.py,sha256=6SiefuOPUDGfN3z6oPY4RkQLiUmkHpoDy5xadzpDzAw,11588
 application_sdk/activities/lock_management.py,sha256=6Wdf3jMKitoarHQP91PIJOoGFz4aaOLS_40c7n1yAOA,3902
 application_sdk/activities/.cursor/BUGBOT.md,sha256=FNykX5aMkdOhzgpiGqstOnSp9JN63iR2XP3onU4AGh8,15843
 application_sdk/activities/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/activities/common/models.py,sha256=43MF_w0EzEQiJvGIqF_FNet4X6MEmwqYd3YAsHdQn08,1362
-application_sdk/activities/common/sql_utils.py,sha256=csLM8H9L5NY5_rhxHBFo-jkMoOKjxhB9xaFbnLbkBGg,10177
+application_sdk/activities/common/sql_utils.py,sha256=QD4qOGkgJmlAGZKaSxqfC0AkjZVdTqdr6Q_Tw2CjIsM,10246
 application_sdk/activities/common/utils.py,sha256=ngyFmiZnMCAQtyu6vGeAlkzwNkM29MD_gBU5pWqOxJ4,8392
 application_sdk/activities/metadata_extraction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/activities/metadata_extraction/base.py,sha256=ENFojpxqKdN_eVSL4iet3cGfylPOfcl1jnflfo4zhs8,3920
@@ -64,15 +64,15 @@ application_sdk/interceptors/events.py,sha256=e0O6uK9_aCTmOORaTGN9RbcTg9_KNaakq-
 application_sdk/interceptors/lock.py,sha256=5ETm20zrTaH2b9fepN4Ckp1tGJV-uINqDrno_5RW3aw,6169
 application_sdk/interceptors/models.py,sha256=kEzJKvb-G1M7aKrLPgAmsukJXLXeh8hIJKwEkOiaY28,6115
 application_sdk/interceptors/.cursor/BUGBOT.md,sha256=pxmUF2c7dtaXAX8yAa1-LBa6FCrj_uw7aQcHrppjf1A,14570
-application_sdk/io/__init__.py,sha256=i5EbFW0IInErum4DTRAhWBjwT2EwiUHTpdYBHNmI-I4,24317
-application_sdk/io/json.py,sha256=yjgCMzbz4Ltyt0PCylVRrUqGl2FhQL_wfWW1KtSi-II,17435
-application_sdk/io/parquet.py,sha256=1J_C5c_LwDfuugIpiQqYgmU2uVcGearF5Kkrx-_34_I,31740
+application_sdk/io/__init__.py,sha256=Fse-fEyrpMlLUxwyFkH8vWWSXz8rdWGlAjZy5ulAZCU,27767
+application_sdk/io/json.py,sha256=sNSyHZCM_ZeaiJHUrolYVHKreBQqSCBsfsjD3JSkoD0,19729
+application_sdk/io/parquet.py,sha256=zy9H_TvWI5CkktJ582NH7Ut_5rUH_S0Jy7ZbTD0JxeI,34227
 application_sdk/io/utils.py,sha256=sn_8Q6HgjeC8uyZp2XGMAfqdJ8XzkIllOEVYXIH54DY,10724
 application_sdk/observability/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/observability/context.py,sha256=lJjpfxEjMY_hrdSDqq519YaWcztgc_1nM4d-mGV5shs,634
 application_sdk/observability/logger_adaptor.py,sha256=Fq5OE579ozr0EzsNYEh2H0q3POVAxtlWfJ-PSwWDGLM,30194
 application_sdk/observability/metrics_adaptor.py,sha256=5Oz02lUED60duryoVDF9mbD11fpxhbXi7P1609n_15Y,16446
-application_sdk/observability/observability.py,sha256=m5OgD_akc3YrkF5mCme2HcRaY7ysjLbVSaEs2C-K2Fs,24062
+application_sdk/observability/observability.py,sha256=O2rBal_0pmFRen7Yx4c4dSH1NyiT937b4bY2w63q-4U,23751
 application_sdk/observability/traces_adaptor.py,sha256=0eQJPN-tYA_dV8D3uEa5ZiX9g12NDuLnPaFuQMVDdL0,18242
 application_sdk/observability/utils.py,sha256=-02GAFom8Bg4SNyCTNYySmen2dzvLfTu43bqsNq1AH0,3096
 application_sdk/observability/decorators/observability_decorator.py,sha256=yd6qfrg1MmH5KcZ5Ydzb0RaBzmxx5FrmiI9qwvZx3EU,8963
@@ -92,7 +92,7 @@ application_sdk/services/__init__.py,sha256=H-5HZEPdr53MUfAggyHqHhRXDRLZFZsxvJgW
 application_sdk/services/_utils.py,sha256=0yHqDP6qNb1OT-bX2XRYQPZ5xkGkV13nyRw6GkPlHs8,1136
 application_sdk/services/atlan_storage.py,sha256=TKzXxu0yXeUcmZehwp8PcnQTC4A9w9RlZ0Fl-Xp1bLE,8509
 application_sdk/services/eventstore.py,sha256=wCT921KRzUe3fAWKC-bbM6_OtIJTKpSQrOutPQzMEgs,6745
-application_sdk/services/objectstore.py,sha256=JgQrL9z_6aG5micVd7I2N6l3RA4EUJh4T2BCuC_ATwQ,20161
+application_sdk/services/objectstore.py,sha256=dLljCsPPSr24bPKh0l3-xRblofzKVQ4LDfqDrMp8JGc,20819
 application_sdk/services/secretstore.py,sha256=Pmn1WlmHmgaDhWz5OXBB5_rKXQQMyLMzadwZSNKwc6Q,19070
 application_sdk/services/statestore.py,sha256=3-afiM3Vsoe1XDYRokdGTB5I5CwOKyieuX5RwIZf77o,9413
 application_sdk/test_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -152,8 +152,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-2.0.0.dist-info/METADATA,sha256=B9-UVeLlDGDuxko6Nvb6Y8zrLNueaATYaJZMESikYZU,5806
-atlan_application_sdk-2.0.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-atlan_application_sdk-2.0.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-2.0.0.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-2.0.0.dist-info/RECORD,,
+atlan_application_sdk-2.1.0.dist-info/METADATA,sha256=XTa1eREHTJv_zkHQhKdCtxe-txAkf9pmIqSEch_zbUA,5806
+atlan_application_sdk-2.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+atlan_application_sdk-2.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-2.1.0.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-2.1.0.dist-info/RECORD,,