atlan-application-sdk 0.1.1rc60__tar.gz → 0.1.1rc62__tar.gz

atlan_application_sdk-0.1.1rc62/.github/release_notes.md

@@ -0,0 +1,7 @@
+## v0.1.1rc62 (November 03, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc61...v0.1.1rc62
+
+### Features
+
+- Adding a new event whenever Token refreshes (#802) (by @Garavitey in [2e32ab4](https://github.com/atlanhq/application-sdk/commit/2e32ab4))

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## v0.1.1rc62 (November 03, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc61...v0.1.1rc62
+
+### Features
+
+- Adding a new event whenever Token refreshes (#802) (by @Garavitey in [2e32ab4](https://github.com/atlanhq/application-sdk/commit/2e32ab4))
+
+
+## v0.1.1rc61 (October 30, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc60...v0.1.1rc61
+
+
 ## v0.1.1rc60 (October 29, 2025)
 
 Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc59...v0.1.1rc60

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc60
+Version: 0.1.1rc62
 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
@@ -22,7 +22,7 @@ Requires-Python: >=3.11
 Requires-Dist: aiohttp>=3.10.0
 Requires-Dist: duckdb-engine>=0.17.0
 Requires-Dist: duckdb>=1.1.3
-Requires-Dist: fastapi[standard]>=0.115.0
+Requires-Dist: fastapi[standard]==0.120.2
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: opentelemetry-exporter-otlp>=1.27.0
 Requires-Dist: psutil>=7.0.0

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/application_sdk/clients/temporal.py

@@ -1,4 +1,5 @@
 import asyncio
+import time
 import uuid
 from concurrent.futures import ThreadPoolExecutor
 from typing import Any, Dict, Optional, Sequence, Type
@@ -26,10 +27,17 @@ from application_sdk.constants import (
     WORKFLOW_PORT,
     WORKFLOW_TLS_ENABLED_KEY,
 )
+from application_sdk.events.models import (
+    ApplicationEventNames,
+    Event,
+    EventTypes,
+    WorkerTokenRefreshEventData,
+)
 from application_sdk.interceptors.cleanup import CleanupInterceptor, cleanup
 from application_sdk.interceptors.events import EventInterceptor, publish_event
 from application_sdk.interceptors.lock import RedisLockInterceptor
 from application_sdk.observability.logger_adaptor import get_logger
+from application_sdk.services.eventstore import EventStore
 from application_sdk.services.secretstore import SecretStore
 from application_sdk.services.statestore import StateStore, StateType
 from application_sdk.workflows import WorkflowInterface
@@ -142,6 +150,39 @@ class TemporalWorkflowClient(WorkflowClient):
         """
         return self.namespace
 
+    async def _publish_token_refresh_event(self) -> None:
+        """Publish a token refresh event to the event store.
+
+        This method creates and publishes an event containing token refresh information,
+        including application name, deployment name, token expiry times, and refresh timestamp.
+        If event publishing fails, it logs a warning but does not raise an exception to avoid
+        disrupting the token refresh loop.
+
+        Note:
+            This method handles exceptions internally and will not propagate errors,
+            ensuring the token refresh loop continues even if event publishing fails.
+        """
+        try:
+            current_time = time.time()
+            worker_token_refresh_data = WorkerTokenRefreshEventData(
+                application_name=self.application_name,
+                deployment_name=DEPLOYMENT_NAME,
+                force_refresh=True,
+                token_expiry_time=self.auth_manager.get_token_expiry_time() or 0,
+                time_until_expiry=self.auth_manager.get_time_until_expiry() or 0,
+                refresh_timestamp=current_time,
+            )
+
+            event = Event(
+                event_type=EventTypes.APPLICATION_EVENT.value,
+                event_name=ApplicationEventNames.TOKEN_REFRESH.value,
+                data=worker_token_refresh_data.model_dump(),
+            )
+            await EventStore.publish_event(event)
+            logger.info("Published token refresh event")
+        except Exception as e:
+            logger.warning(f"Failed to publish token refresh event: {e}")
+
     async def _token_refresh_loop(self) -> None:
         """Background loop that refreshes the authentication token dynamically."""
         while True:
@@ -161,6 +202,9 @@ class TemporalWorkflowClient(WorkflowClient):
                 self._token_refresh_interval = (
                     self.auth_manager.calculate_refresh_interval()
                 )
+                # Publish token refresh event
+                await self._publish_token_refresh_event()
+
             except asyncio.CancelledError:
                 logger.info("Token refresh loop cancelled")
                 break

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/application_sdk/constants.py

@@ -51,7 +51,9 @@ SQL_SERVER_MIN_VERSION = os.getenv("ATLAN_SQL_SERVER_MIN_VERSION")
 SQL_QUERIES_PATH = os.getenv("ATLAN_SQL_QUERIES_PATH", "app/sql")
 
 # Output Path Constants
-#: Output path format for workflows (example: objectstore://bucket/artifacts/apps/{application_name}/workflows/{workflow_id}/{workflow_run_id})
+#: 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}"
 )
@@ -68,7 +70,9 @@ CLEANUP_BASE_PATHS = [
 ]
 
 # State Store Constants
-#: Path template for state store files (example: objectstore://bucket/persistent-artifacts/apps/{application_name}/{state_type}/{id}/config.json)
+#: 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"
 )
@@ -147,7 +151,9 @@ WORKER_START_EVENT_VERSION = "v1"
 #: 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: 100MB)
+#: Maximum gRPC message length in bytes for Dapr client.
+#:
+#: Default: 100MB
 DAPR_MAX_GRPC_MESSAGE_LENGTH = int(
     os.getenv("DAPR_MAX_GRPC_MESSAGE_LENGTH", "104857600")
 )
@@ -250,9 +256,9 @@ REDIS_HOST = os.getenv("REDIS_HOST", "")
 REDIS_PORT = os.getenv("REDIS_PORT", "")
 #: Redis password (required for authenticated Redis instances)
 REDIS_PASSWORD = os.getenv("REDIS_PASSWORD")
-#: Redis Sentinel service name (default: mymaster)
+#: Redis Sentinel service name. Default: mymaster
 REDIS_SENTINEL_SERVICE_NAME = os.getenv("REDIS_SENTINEL_SERVICE_NAME", "mymaster")
-#: Redis Sentinel hosts (comma-separated host:port pairs)
+#: Redis Sentinel hosts as comma-separated host:port pairs
 REDIS_SENTINEL_HOSTS = os.getenv("REDIS_SENTINEL_HOSTS", "")
 #: Whether to enable strict locking
 IS_LOCKING_DISABLED = os.getenv("IS_LOCKING_DISABLED", "true").lower() == "true"

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/application_sdk/events/models.py

@@ -29,6 +29,7 @@ class ApplicationEventNames(Enum):
     WORKER_END = "worker_end"
     APPLICATION_START = "application_start"
     APPLICATION_END = "application_end"
+    TOKEN_REFRESH = "token_refresh"
 
 
 class WorkflowStates(Enum):
@@ -174,3 +175,26 @@ class WorkerStartEventData(BaseModel):
     max_concurrent_activities: Optional[int]
     workflow_count: int
     activity_count: int
+
+
+class WorkerTokenRefreshEventData(BaseModel):
+    """Model for token refresh event data.
+
+    This model represents the data structure used when publishing token refresh events.
+    It contains information about the token refresh operation and agent status.
+
+    Attributes:
+        application_name: Name of the application the token belongs to.
+        deployment_name: Name of the deployment (e.g., dev, staging, prod).
+        force_refresh: Whether this was a forced refresh or automatic.
+        token_expiry_time: Unix timestamp when the new token expires.
+        time_until_expiry: Seconds until token expires.
+        refresh_timestamp: Unix timestamp when the refresh occurred.
+    """
+
+    application_name: str
+    deployment_name: str
+    force_refresh: bool
+    token_expiry_time: float
+    time_until_expiry: float
+    refresh_timestamp: float

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/application_sdk/version.py

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

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "atlan-application-sdk"
-version = "0.1.1rc60"
+version = "0.1.1rc62"
 description = "Atlan Application SDK is a Python library for developing applications on the Atlan Platform"
 license = "Apache-2.0"
 authors = [
@@ -23,7 +23,7 @@ dependencies = [
     "aiohttp>=3.10.0",
     "opentelemetry-exporter-otlp>=1.27.0",
     "psutil>=7.0.0",
-    "fastapi[standard]>=0.115.0",
+    "fastapi[standard]==0.120.2",
     "pyatlan>=8.0.2",
     "pydantic>=2.10.6",
     "loguru>=0.7.3",

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/tests/unit/clients/test_temporal_client.py

@@ -337,3 +337,110 @@ async def test_get_workflow_run_status_client_not_loaded(
     """Test get_workflow_run_status when client is not loaded."""
     with pytest.raises(ValueError, match="Client is not loaded"):
         await temporal_client.get_workflow_run_status("test_workflow_id", "test_run_id")
+
+
+@patch("application_sdk.clients.temporal.logger")
+@patch("application_sdk.clients.temporal.EventStore.publish_event")
+@patch("application_sdk.clients.temporal.time.time")
+@patch("application_sdk.clients.temporal.DEPLOYMENT_NAME", "test_deployment")
+async def test_publish_token_refresh_event_success(
+    mock_time: Mock,
+    mock_publish_event: AsyncMock,
+    mock_logger: MagicMock,
+    temporal_client: TemporalWorkflowClient,
+):
+    """Test successful token refresh event publishing."""
+    # Arrange
+    mock_time.return_value = 1234567890.0
+    mock_auth_manager = MagicMock()
+    mock_auth_manager.get_token_expiry_time.return_value = 1234567895.0
+    mock_auth_manager.get_time_until_expiry.return_value = 300.0
+    temporal_client.auth_manager = mock_auth_manager
+
+    # Act
+    await temporal_client._publish_token_refresh_event()
+
+    # Assert
+    mock_time.assert_called_once()
+    mock_auth_manager.get_token_expiry_time.assert_called_once()
+    mock_auth_manager.get_time_until_expiry.assert_called_once()
+    mock_publish_event.assert_called_once()
+
+    # Verify the event structure
+    call_args = mock_publish_event.call_args
+    event = call_args[0][0]  # First positional argument
+
+    assert event.event_type == "application_events"
+    assert event.event_name == "token_refresh"
+    assert event.data["application_name"] == "test_app"
+    assert event.data["deployment_name"] == "test_deployment"
+    assert event.data["force_refresh"] is True
+    assert event.data["token_expiry_time"] == 1234567895.0
+    assert event.data["time_until_expiry"] == 300.0
+    assert event.data["refresh_timestamp"] == 1234567890.0
+
+    mock_logger.info.assert_called_once_with("Published token refresh event")
+    mock_logger.warning.assert_not_called()
+
+
+@patch("application_sdk.clients.temporal.logger")
+@patch("application_sdk.clients.temporal.EventStore.publish_event")
+@patch("application_sdk.clients.temporal.time.time")
+@patch("application_sdk.clients.temporal.DEPLOYMENT_NAME", "test_deployment")
+async def test_publish_token_refresh_event_with_none_expiry_times(
+    mock_time: Mock,
+    mock_publish_event: AsyncMock,
+    mock_logger: MagicMock,
+    temporal_client: TemporalWorkflowClient,
+):
+    """Test token refresh event publishing with None expiry times."""
+    # Arrange
+    mock_time.return_value = 1234567890.0
+    mock_auth_manager = MagicMock()
+    mock_auth_manager.get_token_expiry_time.return_value = None
+    mock_auth_manager.get_time_until_expiry.return_value = None
+    temporal_client.auth_manager = mock_auth_manager
+
+    # Act
+    await temporal_client._publish_token_refresh_event()
+
+    # Assert
+    call_args = mock_publish_event.call_args
+    event = call_args[0][0]  # First positional argument
+
+    assert event.data["token_expiry_time"] == 0
+    assert event.data["time_until_expiry"] == 0
+
+    mock_logger.info.assert_called_once_with("Published token refresh event")
+    mock_logger.warning.assert_not_called()
+
+
+@patch("application_sdk.clients.temporal.logger")
+@patch("application_sdk.clients.temporal.EventStore.publish_event")
+@patch("application_sdk.clients.temporal.time.time")
+async def test_publish_token_refresh_event_exception_handling(
+    mock_time: Mock,
+    mock_publish_event: AsyncMock,
+    mock_logger: MagicMock,
+    temporal_client: TemporalWorkflowClient,
+):
+    """Test token refresh event publishing handles exceptions gracefully."""
+    # Arrange
+    mock_time.return_value = 1234567890.0
+    mock_auth_manager = MagicMock()
+    mock_auth_manager.get_token_expiry_time.return_value = 1234567895.0
+    mock_auth_manager.get_time_until_expiry.return_value = 300.0
+    temporal_client.auth_manager = mock_auth_manager
+
+    # Simulate event publishing failure
+    mock_publish_event.side_effect = Exception("Event store connection failed")
+
+    # Act - should not raise exception
+    await temporal_client._publish_token_refresh_event()
+
+    # Assert
+    mock_publish_event.assert_called_once()
+    mock_logger.info.assert_not_called()
+    mock_logger.warning.assert_called_once_with(
+        "Failed to publish token refresh event: Event store connection failed"
+    )

{atlan_application_sdk-0.1.1rc60 → atlan_application_sdk-0.1.1rc62}/tests/unit/services/test_eventstore.py

@@ -209,6 +209,7 @@ class TestApplicationEventNames:
             "worker_end",
             "application_start",
             "application_end",
+            "token_refresh",
         ]
         actual_names = [name.value for name in ApplicationEventNames]
         assert actual_names == expected_names