atlan-application-sdk 0.1.1rc45__tar.gz → 0.1.1rc46__tar.gz

atlan_application_sdk-0.1.1rc46/.github/release_notes.md

@@ -0,0 +1,7 @@
+## v0.1.1rc46 (September 18, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc45...v0.1.1rc46
+
+### Bug Fixes
+
+- Fix semaphore implementation to allow multiple batching without getting into max_concurrent_activities issue (#725) (by @Mustafa in [2a66f13](https://github.com/atlanhq/application-sdk/commit/2a66f13))

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+
+## v0.1.1rc46 (September 18, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc45...v0.1.1rc46
+
+### Bug Fixes
+
+- Fix semaphore implementation to allow multiple batching without getting into max_concurrent_activities issue (#725) (by @Mustafa in [2a66f13](https://github.com/atlanhq/application-sdk/commit/2a66f13))
+
 ## v0.1.1rc45 (September 18, 2025)
 
 Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc44...v0.1.1rc45

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc45
+Version: 0.1.1rc46
 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.1rc45 → atlan_application_sdk-0.1.1rc46}/application_sdk/activities/lock_management.py

@@ -5,15 +5,15 @@ allowing the workflow to orchestrate locking without hitting Temporal's
 deadlock timeout.
 """
 
-import asyncio
 import random
 from typing import Any, Dict
 
 from temporalio import activity
+from temporalio.exceptions import ApplicationError
 
 from application_sdk.clients.redis import RedisClientAsync
 from application_sdk.common.error_codes import ActivityError
-from application_sdk.constants import APPLICATION_NAME, LOCK_RETRY_INTERVAL
+from application_sdk.constants import APPLICATION_NAME
 from application_sdk.observability.logger_adaptor import get_logger
 
 logger = get_logger(__name__)
@@ -45,40 +45,41 @@ async def acquire_distributed_lock(
     """
     # Input validation
     if max_locks <= 0:
-        raise ActivityError(
-            f"{ActivityError.LOCK_ACQUISITION_ERROR}: max_locks must be greater than 0, got {max_locks}"
+        raise ApplicationError(
+            f"{ActivityError.LOCK_ACQUISITION_ERROR}: max_locks must be greater than 0, got {max_locks}",
+            non_retryable=True,
+        )
+    slot = random.randint(0, max_locks - 1)
+    resource_id = f"{APPLICATION_NAME}:{lock_name}:{slot}"
+
+    try:
+        async with RedisClientAsync() as redis_client:
+            # Acquire lock - connection will stay open until context exits
+            acquired = await redis_client._acquire_lock(
+                resource_id, owner_id, ttl_seconds
+            )
+            if acquired:
+                logger.info(f"Lock acquired for slot {slot}, resource: {resource_id}")
+                return {
+                    "status": True,
+                    "slot_id": slot,
+                    "resource_id": resource_id,
+                    "owner_id": owner_id,
+                }
+
+            raise ActivityError(
+                f"{ActivityError.LOCK_ACQUISITION_ERROR}: Lock not acquired for {resource_id}, will retry after some time"
+            )
+    except Exception as e:
+        # Redis connection or operation failed - propagate as activity error
+        if isinstance(e, (ActivityError)):
+            raise e
+        logger.error(f"Redis error during lock acquisition: {e}")
+        raise ApplicationError(
+            f"Redis error during lock acquisition for {resource_id}, error: {e}",
+            non_retryable=True,
+            type=type(e).__name__,
         )
-
-    async with RedisClientAsync() as redis_client:
-        while True:
-            slot = random.randint(0, max_locks - 1)
-            resource_id = f"{APPLICATION_NAME}:{lock_name}:{slot}"
-
-            try:
-                # Acquire lock - connection will stay open until context exits
-                acquired = await redis_client._acquire_lock(
-                    resource_id, owner_id, ttl_seconds
-                )
-                if acquired:
-                    logger.info(
-                        f"Lock acquired for slot {slot}, resource: {resource_id}"
-                    )
-                    return {
-                        "slot_id": slot,
-                        "resource_id": resource_id,
-                        "owner_id": owner_id,
-                    }
-                # If not acquired, continue retrying (lock held by another owner)
-
-            except Exception as e:
-                # Redis connection or operation failed - propagate as activity error
-                logger.error(f"Redis error during lock acquisition: {e}")
-                raise ActivityError(
-                    f"{ActivityError.LOCK_ACQUISITION_ERROR}: Redis error during lock acquisition for {resource_id}"
-                )
-
-            # Wait before retrying
-            await asyncio.sleep(LOCK_RETRY_INTERVAL)
 
 
 @activity.defn
@@ -94,8 +95,8 @@ async def release_distributed_lock(
     Returns:
         True if lock was released successfully, False otherwise
     """
-    async with RedisClientAsync() as redis_client:
-        try:
+    try:
+        async with RedisClientAsync() as redis_client:
             released, result = await redis_client._release_lock(resource_id, owner_id)
             if released:
                 logger.info(
@@ -103,8 +104,8 @@ async def release_distributed_lock(
                 )
             return released
 
-        except Exception as e:
-            logger.error(f"Redis error during lock release for {resource_id}: {e}")
-            # Don't raise exception for lock release failures - log and return False
-            # Lock release is best-effort and shouldn't fail the workflow
-            return False
+    except Exception as e:
+        logger.error(f"Redis error during lock release for {resource_id}: {e}")
+        # Don't raise exception for lock release failures - log and return False
+        # Lock release is best-effort and shouldn't fail the workflow
+        return False

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/application_sdk/interceptors/lock.py

@@ -22,6 +22,7 @@ from application_sdk.constants import (
     APPLICATION_NAME,
     IS_LOCKING_DISABLED,
     LOCK_METADATA_KEY,
+    LOCK_RETRY_INTERVAL,
 )
 from application_sdk.observability.logger_adaptor import get_logger
 
@@ -106,13 +107,17 @@ class RedisLockOutboundInterceptor(WorkflowOutboundInterceptor):
         lock_result = None
 
         try:
-            # Step 1: Acquire lock via dedicated activity (can take >2s safely)
-            start_to_close_timeout = workflow.info().execution_timeout
-            lock_result = await workflow.execute_activity(
+            # Step 1: Acquire lock via dedicated activity with Temporal retry policy
+            schedule_to_close_timeout = workflow.info().execution_timeout
+            lock_result = await workflow.execute_local_activity(
                 "acquire_distributed_lock",
                 args=[lock_name, max_locks, ttl_seconds, owner_id],
-                start_to_close_timeout=start_to_close_timeout,
-                retry_policy=RetryPolicy(maximum_attempts=1),
+                start_to_close_timeout=timedelta(seconds=30),
+                retry_policy=RetryPolicy(
+                    initial_interval=timedelta(seconds=int(LOCK_RETRY_INTERVAL)),
+                    backoff_coefficient=1.0,
+                ),
+                schedule_to_close_timeout=schedule_to_close_timeout,
             )
 
             logger.debug(f"Lock acquired: {lock_result}, executing {input.activity}")

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/application_sdk/version.py

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

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "atlan-application-sdk"
-version = "0.1.1rc45"
+version = "0.1.1rc46"
 description = "Atlan Application SDK is a Python library for developing applications on the Atlan Platform"
 license = "Apache-2.0"
 authors = [

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/tests/unit/activities/test_lock_management.py

@@ -3,6 +3,7 @@
 from unittest.mock import AsyncMock, patch
 
 import pytest
+from temporalio.exceptions import ApplicationError
 
 from application_sdk.activities.lock_management import (
     acquire_distributed_lock,
@@ -27,6 +28,7 @@ class TestAcquireDistributedLock:
         result = await acquire_distributed_lock("test_resource", 5, 100, "owner1")
 
         # Verify
+        assert result["status"] is True
         assert "slot_id" in result
         assert "resource_id" in result
         assert "owner_id" in result
@@ -36,7 +38,7 @@ class TestAcquireDistributedLock:
     @patch("application_sdk.activities.lock_management.RedisClientAsync")
     async def test_acquire_lock_invalid_max_locks(self, mock_redis_client_class):
         """Test lock acquisition with invalid max_locks."""
-        with pytest.raises(ActivityError) as exc_info:
+        with pytest.raises(ApplicationError) as exc_info:
             await acquire_distributed_lock("test_resource", 0, 100, "owner1")
 
         assert "ATLAN-ACTIVITY-503-01" in str(exc_info.value)
@@ -50,12 +52,27 @@ class TestAcquireDistributedLock:
         mock_client._acquire_lock.side_effect = Exception("Redis connection failed")
         mock_redis_client_class.return_value.__aenter__.return_value = mock_client
 
+        # Execute and verify
+        with pytest.raises(ApplicationError) as exc_info:
+            await acquire_distributed_lock("test_resource", 5, 100, "owner1")
+
+        assert "Redis error during lock acquisition" in str(exc_info.value)
+        assert "Redis connection failed" in str(exc_info.value)
+
+    @patch("application_sdk.activities.lock_management.RedisClientAsync")
+    async def test_acquire_lock_not_available(self, mock_redis_client_class):
+        """Test lock acquisition when lock is not available."""
+        # Setup mock to return False (lock not acquired)
+        mock_client = AsyncMock()
+        mock_client._acquire_lock.return_value = False
+        mock_redis_client_class.return_value.__aenter__.return_value = mock_client
+
         # Execute and verify
         with pytest.raises(ActivityError) as exc_info:
             await acquire_distributed_lock("test_resource", 5, 100, "owner1")
 
         assert "ATLAN-ACTIVITY-503-01" in str(exc_info.value)
-        assert "Redis error during lock acquisition" in str(exc_info.value)
+        assert "Lock not acquired" in str(exc_info.value)
 
 
 class TestReleaseDistributedLock:

{atlan_application_sdk-0.1.1rc45 → atlan_application_sdk-0.1.1rc46}/uv.lock

@@ -140,7 +140,7 @@ wheels = [
 
 [[package]]
 name = "atlan-application-sdk"
-version = "0.1.1rc45"
+version = "0.1.1rc46"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },

atlan_application_sdk-0.1.1rc45/.github/release_notes.md

@@ -1,9 +0,0 @@
-## v0.1.1rc45 (September 18, 2025)
-
-Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc44...v0.1.1rc45
-
-### Features
-
-- add MCP (Model Context Protocol) integration to application-sdk (#698) (by @AdvitXAtlan in [a5d496e](https://github.com/atlanhq/application-sdk/commit/a5d496e))
-- fix credentials and extra field information in docs (#724) (by @Nishchith Shetty in [a0808db](https://github.com/atlanhq/application-sdk/commit/a0808db))
-- add minimal API docs for endpoints exposed (#721) (by @Nishchith Shetty in [3301b61](https://github.com/atlanhq/application-sdk/commit/3301b61))