dagster-cloud 1.11.11__tar.gz → 1.11.13__tar.gz

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-cloud
-Version: 1.11.11
+Version: 1.11.13
 Author-email: Elementl <support@elementl.com>
 License: Apache-2.0
 Project-URL: Homepage, https://dagster.io/cloud
@@ -28,9 +28,9 @@ Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
 Classifier: Operating System :: OS Independent
 Requires-Python: <3.14,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: dagster==1.11.11
-Requires-Dist: dagster-shared==1.11.11
-Requires-Dist: dagster-cloud-cli==1.11.11
+Requires-Dist: dagster==1.11.13
+Requires-Dist: dagster-shared==1.11.13
+Requires-Dist: dagster-cloud-cli==1.11.13
 Requires-Dist: opentelemetry-api<2,>=1.27.0
 Requires-Dist: opentelemetry-sdk<2,>=1.27.0
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc<2,>=1.27.0
@@ -65,12 +65,12 @@ Provides-Extra: insights
 Requires-Dist: pyarrow; extra == "insights"
 Provides-Extra: docker
 Requires-Dist: docker; extra == "docker"
-Requires-Dist: dagster-docker==0.27.11; extra == "docker"
+Requires-Dist: dagster-docker==0.27.13; extra == "docker"
 Provides-Extra: kubernetes
 Requires-Dist: kubernetes; extra == "kubernetes"
-Requires-Dist: dagster-k8s==0.27.11; extra == "kubernetes"
+Requires-Dist: dagster-k8s==0.27.13; extra == "kubernetes"
 Provides-Extra: ecs
-Requires-Dist: dagster-aws==0.27.11; extra == "ecs"
+Requires-Dist: dagster-aws==0.27.13; extra == "ecs"
 Requires-Dist: boto3; extra == "ecs"
 Provides-Extra: sandbox
 Requires-Dist: supervisor; extra == "sandbox"

dagster_cloud-1.11.13/dagster_cloud/__init__.py

@@ -0,0 +1,7 @@
+from dagster._core.libraries import DagsterLibraryRegistry
+
+from dagster_cloud.instance import DagsterCloudAgentInstance as DagsterCloudAgentInstance
+from dagster_cloud.storage.compute_logs import CloudComputeLogManager as CloudComputeLogManager
+from dagster_cloud.version import __version__
+
+DagsterLibraryRegistry.register("dagster-cloud", __version__)

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/agent/dagster_cloud_agent.py

@@ -31,6 +31,16 @@ from dagster._utils.typed_dict import init_optional_typeddict
 from dagster_cloud_cli.core.errors import DagsterCloudHTTPError, raise_http_error
 from dagster_cloud_cli.core.workspace import CodeLocationDeployData
 
+from dagster_cloud.agent.instrumentation.constants import DAGSTER_CLOUD_AGENT_METRIC_PREFIX
+from dagster_cloud.agent.instrumentation.run_launch import extract_run_attributes
+from dagster_cloud.agent.instrumentation.schedule import inspect_schedule_result
+from dagster_cloud.agent.instrumentation.sensor import inspect_sensor_result
+from dagster_cloud.agent.queries import (
+    ADD_AGENT_HEARTBEATS_MUTATION,
+    DEPLOYMENTS_QUERY,
+    GET_USER_CLOUD_REQUESTS_QUERY,
+    WORKSPACE_ENTRIES_QUERY,
+)
 from dagster_cloud.api.dagster_cloud_api import (
     AgentHeartbeat,
     AgentUtilizationMetrics,
@@ -48,26 +58,15 @@ from dagster_cloud.api.dagster_cloud_api import (
 )
 from dagster_cloud.batching import Batcher
 from dagster_cloud.instance import DagsterCloudAgentInstance
+from dagster_cloud.opentelemetry.observers.execution_observer import observe_execution
+from dagster_cloud.util import SERVER_HANDLE_TAG, compressed_namedtuple_upload_file, is_isolated_run
+from dagster_cloud.version import __version__
 from dagster_cloud.workspace.user_code_launcher import (
     DagsterCloudUserCodeLauncher,
     UserCodeLauncherEntry,
 )
 from dagster_cloud.workspace.user_code_launcher.utils import get_instance_ref_for_user_code
 
-from ..opentelemetry.observers.execution_observer import observe_execution
-from ..util import SERVER_HANDLE_TAG, compressed_namedtuple_upload_file, is_isolated_run
-from ..version import __version__
-from .instrumentation.constants import DAGSTER_CLOUD_AGENT_METRIC_PREFIX
-from .instrumentation.run_launch import extract_run_attributes
-from .instrumentation.schedule import inspect_schedule_result
-from .instrumentation.sensor import inspect_sensor_result
-from .queries import (
-    ADD_AGENT_HEARTBEATS_MUTATION,
-    DEPLOYMENTS_QUERY,
-    GET_USER_CLOUD_REQUESTS_QUERY,
-    WORKSPACE_ENTRIES_QUERY,
-)
-
 if TYPE_CHECKING:
     import datetime
 

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/anomaly_detection/__init__.py

@@ -1,7 +1,7 @@
-from .defs import (
+from dagster_cloud.anomaly_detection.defs import (
     build_anomaly_detection_freshness_checks as build_anomaly_detection_freshness_checks,
 )
-from .types import (
+from dagster_cloud.anomaly_detection.types import (
     AnomalyDetectionModelParams as AnomalyDetectionModelParams,
     BetaFreshnessAnomalyDetectionParams as BetaFreshnessAnomalyDetectionParams,
 )

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/anomaly_detection/defs.py

@@ -39,8 +39,8 @@ from dagster_cloud_cli.core.graphql_client import (
     create_cloud_webserver_client,
 )
 
-from .mutation import ANOMALY_DETECTION_INFERENCE_MUTATION
-from .types import (
+from dagster_cloud.anomaly_detection.mutation import ANOMALY_DETECTION_INFERENCE_MUTATION
+from dagster_cloud.anomaly_detection.types import (
     AnomalyDetectionModelParams,
     BetaFreshnessAnomalyDetectionParams,
     FreshnessAnomalyDetectionResult,

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/auth/constants.py

@@ -1,3 +1,4 @@
+import uuid
 from typing import Optional
 
 from dagster._core.errors import DagsterInvariantViolationError
@@ -19,12 +20,27 @@ def get_organization_public_id_from_api_token(api_token: str) -> Optional[str]:
     return split_token[2]
 
 
-def get_organization_name_from_agent_token(agent_token: str) -> Optional[str]:
+def decode_region_from_uuid(regional_token: str) -> Optional[str]:
+    try:
+        regional_uuid = uuid.UUID(regional_token)
+    except ValueError:
+        # if it's not an actual uuid, we can't decode region
+        return None
+
+    # custom uuids contain region subdomains in the first 2 bytes
+    if regional_uuid.version != 8 or regional_uuid.variant != uuid.RFC_4122:
+        return None
+
+    uuid_bytes = regional_uuid.bytes
+    return uuid_bytes[:2].decode("ascii")
+
+
+def decode_agent_token(agent_token: str) -> tuple[Optional[str], Optional[str]]:
     split_token = agent_token.split(":")
 
     # Legacy agent token format - organization must be specified in dagster.yaml
     if len(split_token) == 1:
-        return None
+        return None, None
 
     token_type, *token = split_token
 
@@ -35,6 +51,6 @@ def get_organization_name_from_agent_token(agent_token: str) -> Optional[str]:
             "Generate a new agent token in Dagster Cloud."
         )
 
-    organization, _identifier = token
-
-    return organization
+    # token format: agent:<org>:<uuid>
+    organization, uuid_str = token
+    return organization, decode_region_from_uuid(uuid_str)

dagster_cloud-1.11.13/dagster_cloud/batching/__init__.py

@@ -0,0 +1 @@
+from dagster_cloud.batching.batcher import Batcher as Batcher

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/__init__.py

@@ -1,15 +1,19 @@
 import sys
 from typing import Any
 
-from .snowflake.dbt_wrapper import dbt_with_snowflake_insights as dbt_with_snowflake_insights
-from .snowflake.definitions import (
+from dagster_cloud.dagster_insights.snowflake.dbt_wrapper import (
+    dbt_with_snowflake_insights as dbt_with_snowflake_insights,
+)
+from dagster_cloud.dagster_insights.snowflake.definitions import (
     create_snowflake_insights_asset_and_schedule as create_snowflake_insights_asset_and_schedule,
 )
-from .snowflake.snowflake_utils import meter_snowflake_query as meter_snowflake_query
+from dagster_cloud.dagster_insights.snowflake.snowflake_utils import (
+    meter_snowflake_query as meter_snowflake_query,
+)
 
 dagster_snowflake_req_imports = {"InsightsSnowflakeResource"}
 try:
-    from .snowflake.insights_snowflake_resource import (
+    from dagster_cloud.dagster_insights.snowflake.insights_snowflake_resource import (
         InsightsSnowflakeResource as InsightsSnowflakeResource,
     )
 except ImportError:
@@ -17,8 +21,10 @@ except ImportError:
 
 dagster_bigquery_req_imports = {"InsightsBigQueryResource", "dbt_with_bigquery_insights"}
 try:
-    from .bigquery.dbt_wrapper import dbt_with_bigquery_insights as dbt_with_bigquery_insights
-    from .bigquery.insights_bigquery_resource import (
+    from dagster_cloud.dagster_insights.bigquery.dbt_wrapper import (
+        dbt_with_bigquery_insights as dbt_with_bigquery_insights,
+    )
+    from dagster_cloud.dagster_insights.bigquery.insights_bigquery_resource import (
         InsightsBigQueryResource as InsightsBigQueryResource,
     )
 except ImportError:

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/bigquery/dbt_wrapper.py

@@ -18,8 +18,14 @@ from dagster_dbt import DbtCliInvocation
 from dagster_dbt.version import __version__ as dagster_dbt_version
 from packaging import version
 
-from ..insights_utils import extract_asset_info_from_event, handle_raise_on_error
-from .bigquery_utils import build_bigquery_cost_metadata, marker_asset_key_for_job
+from dagster_cloud.dagster_insights.bigquery.bigquery_utils import (
+    build_bigquery_cost_metadata,
+    marker_asset_key_for_job,
+)
+from dagster_cloud.dagster_insights.insights_utils import (
+    extract_asset_info_from_event,
+    handle_raise_on_error,
+)
 
 if TYPE_CHECKING:
     from dbt.adapters.base.impl import BaseAdapter

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/bigquery/insights_bigquery_resource.py

@@ -7,10 +7,12 @@ from dagster_gcp import BigQueryResource
 from dagster_gcp.bigquery.utils import setup_gcp_creds
 from google.cloud import bigquery
 
+from dagster_cloud.dagster_insights.bigquery.bigquery_utils import (
+    build_bigquery_cost_metadata,
+    marker_asset_key_for_job,
+)
 from dagster_cloud.dagster_insights.insights_utils import get_current_context_and_asset_key
 
-from .bigquery_utils import build_bigquery_cost_metadata, marker_asset_key_for_job
-
 OUTPUT_NON_ASSET_SIGIL = "__bigquery_query_metadata_"
 
 

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/snowflake/dagster_snowflake_insights.py

@@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Optional
 
 from dagster import AssetKey, AssetsDefinition, ScheduleDefinition
 
-from .snowflake_utils import OPAQUE_ID_SQL_SIGIL
+from dagster_cloud.dagster_insights.snowflake.snowflake_utils import OPAQUE_ID_SQL_SIGIL
 
 if TYPE_CHECKING:
     from dagster_snowflake import SnowflakeConnection
@@ -78,10 +78,6 @@ HAVING ARRAY_SIZE(opaque_ids) > 0
 
     costs: list[tuple[str, float, str]] = []
 
-    print(
-        f"{len(results) if results else 0} annotated queries returned from snowflake query_history"
-    )
-
     if not results:
         return []
 
@@ -94,8 +90,4 @@ HAVING ARRAY_SIZE(opaque_ids) > 0
         for opaque_id in opaque_ids:
             costs.append((opaque_id, float(cost), query_id))
 
-    print(
-        f"Reported costs for {len(costs)} of {total} asset materializations found in the"
-        " query_history."
-    )
     return costs

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/snowflake/dbt_wrapper.py

@@ -13,8 +13,15 @@ from dagster import (
     Output,
 )
 
-from ..insights_utils import extract_asset_info_from_event, handle_raise_on_error
-from .snowflake_utils import OPAQUE_ID_SQL_SIGIL, build_opaque_id_metadata, marker_asset_key_for_job
+from dagster_cloud.dagster_insights.insights_utils import (
+    extract_asset_info_from_event,
+    handle_raise_on_error,
+)
+from dagster_cloud.dagster_insights.snowflake.snowflake_utils import (
+    OPAQUE_ID_SQL_SIGIL,
+    build_opaque_id_metadata,
+    marker_asset_key_for_job,
+)
 
 if TYPE_CHECKING:
     from dagster_dbt import DbtCliInvocation

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/snowflake/definitions.py

@@ -2,7 +2,6 @@ import warnings
 from collections.abc import Sequence
 from dataclasses import dataclass
 from datetime import date, datetime, timedelta, timezone
-from pprint import pprint
 from typing import TYPE_CHECKING, Optional, Union
 
 from dagster import (
@@ -27,8 +26,10 @@ from dagster._core.storage.tags import (
     ASSET_PARTITION_RANGE_START_TAG,
 )
 
-from ..metrics_utils import put_cost_information
-from .dagster_snowflake_insights import get_cost_data_for_hour
+from dagster_cloud.dagster_insights.metrics_utils import put_cost_information
+from dagster_cloud.dagster_insights.snowflake.dagster_snowflake_insights import (
+    get_cost_data_for_hour,
+)
 
 if TYPE_CHECKING:
     from dagster_snowflake import SnowflakeConnection
@@ -154,7 +155,7 @@ def create_snowflake_insights_asset_and_schedule(
         )
 
         if dry_run:
-            pprint(costs)
+            pass
         else:
             context.log.info(
                 f"Submitting cost information for {len(costs)} queries to Dagster Insights"

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/dagster_insights/snowflake/insights_snowflake_resource.py

@@ -20,8 +20,7 @@ from dagster_snowflake import SnowflakeConnection, SnowflakeResource
 from snowflake.connector.cursor import SnowflakeCursor
 
 from dagster_cloud.dagster_insights.insights_utils import get_current_context_and_asset_key
-
-from .snowflake_utils import meter_snowflake_query
+from dagster_cloud.dagster_insights.snowflake.snowflake_utils import meter_snowflake_query
 
 
 def get_current_context_and_asset_key_or_warn() -> tuple[

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/execution/utils/process.py

@@ -4,7 +4,7 @@ from collections.abc import Sequence
 
 from dagster_shared.ipc import interrupt_ipc_subprocess_pid, open_ipc_subprocess
 
-from . import TaskStatus
+from dagster_cloud.execution.utils import TaskStatus
 
 
 def launch_process(args: Sequence[str]) -> int:

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/instance/__init__.py

@@ -32,14 +32,13 @@ from dagster_cloud_cli.core.headers.auth import DagsterCloudInstanceScope
 from urllib3 import Retry
 
 from dagster_cloud.agent import AgentQueuesConfig
+from dagster_cloud.auth.constants import decode_agent_token
+from dagster_cloud.opentelemetry.config import opentelemetry_config_schema
+from dagster_cloud.opentelemetry.controller import OpenTelemetryController
+from dagster_cloud.storage.client import dagster_cloud_api_config
+from dagster_cloud.util import get_env_names_from_config, is_isolated_run
 from dagster_cloud.version import __version__
 
-from ..auth.constants import get_organization_name_from_agent_token
-from ..opentelemetry.config import opentelemetry_config_schema
-from ..opentelemetry.controller import OpenTelemetryController
-from ..storage.client import dagster_cloud_api_config
-from ..util import get_env_names_from_config, is_isolated_run
-
 if TYPE_CHECKING:
     from requests import Session
 
@@ -87,14 +86,18 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
         super().__init__(*args, **kwargs)
 
         self._unprocessed_dagster_cloud_api_config = dagster_cloud_api
-        self._dagster_cloud_api_config = self._get_processed_config(
-            "dagster_cloud_api", dagster_cloud_api, dagster_cloud_api_config()
+        self._dagster_cloud_api_config = check.not_none(
+            self._get_processed_config(
+                "dagster_cloud_api",
+                dagster_cloud_api,
+                dagster_cloud_api_config(),
+            )
         )
 
         check.invariant(
             not (
-                self._dagster_cloud_api_config.get("deployment")  # pyright: ignore[reportOptionalMemberAccess]
-                and self._dagster_cloud_api_config.get("deployments")  # pyright: ignore[reportOptionalMemberAccess]
+                self._dagster_cloud_api_config.get("deployment")
+                and self._dagster_cloud_api_config.get("deployments")
             ),
             "Cannot set both deployment and deployments in `dagster_cloud_api`",
         )
@@ -175,16 +178,16 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
         return processed_config.value
 
     def _dagster_cloud_api_config_for_deployment(self, deployment_name: Optional[str]):
-        new_api_config = dict(copy.deepcopy(self._dagster_cloud_api_config))  # pyright: ignore[reportArgumentType,reportCallIssue]
+        new_api_config = dict(copy.deepcopy(self._dagster_cloud_api_config))
         if deployment_name:
-            new_api_config["deployment"] = deployment_name  # pyright: ignore[reportArgumentType]
+            new_api_config["deployment"] = deployment_name
             if self.includes_branch_deployments:
-                del new_api_config["branch_deployments"]  # pyright: ignore
-            new_api_config.pop("deployments", None)  # pyright: ignore
-            new_api_config.pop("all_serverless_deployments", None)  # pyright: ignore
+                del new_api_config["branch_deployments"]
+            new_api_config.pop("deployments", None)
+            new_api_config.pop("all_serverless_deployments", None)
         else:
-            new_api_config.pop("deployment", None)  # pyright: ignore
-            new_api_config.pop("deployments", None)  # pyright: ignore
+            new_api_config.pop("deployment", None)
+            new_api_config.pop("deployments", None)
 
         return new_api_config
 
@@ -238,11 +241,11 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
             return socket_option
 
     def _socket_options(self):
-        if self._dagster_cloud_api_config.get("socket_options") is None:  # pyright: ignore[reportOptionalMemberAccess]
+        if self._dagster_cloud_api_config.get("socket_options") is None:
             return None
 
         translated_socket_options = []
-        for socket_option in self._dagster_cloud_api_config["socket_options"]:  # pyright: ignore[reportOptionalSubscript]
+        for socket_option in self._dagster_cloud_api_config["socket_options"]:
             check.invariant(
                 len(socket_option) == 3, "Each socket option must be a list of three values"
             )
@@ -284,7 +287,7 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
                     adapter_kwargs=dict(
                         max_retries=Retry(
                             total=self.dagster_cloud_api_retries,
-                            backoff_factor=self._dagster_cloud_api_config["backoff_factor"],  # pyright: ignore[reportOptionalSubscript]
+                            backoff_factor=self._dagster_cloud_api_config["backoff_factor"],
                         ),
                         socket_options=self._socket_options(),
                     )
@@ -314,21 +317,26 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
 
     @property
     def dagster_cloud_url(self):
-        if "url" in self._dagster_cloud_api_config:  # pyright: ignore[reportOperatorIssue]
-            return self._dagster_cloud_api_config["url"]  # pyright: ignore[reportOptionalSubscript]
+        if "url" in self._dagster_cloud_api_config:
+            return self._dagster_cloud_api_config["url"]
 
-        organization = get_organization_name_from_agent_token(self.dagster_cloud_agent_token)  # pyright: ignore[reportArgumentType]
+        organization, region = decode_agent_token(self.dagster_cloud_agent_token)
         if not organization:
             raise DagsterInvariantViolationError(
                 "Could not derive Dagster Cloud URL from agent token. Create a new agent token or"
                 " set the `url` field under `dagster_cloud_api` in your `dagster.yaml`."
             )
 
-        return f"https://{organization}.agent.dagster.cloud"
+        return (
+            f"https://{organization}.agent.{region}.dagster.cloud"
+            if region
+            else f"https://{organization}.agent.dagster.cloud"
+        )
 
     @property
     def organization_name(self) -> Optional[str]:
-        return get_organization_name_from_agent_token(self.dagster_cloud_agent_token)  # pyright: ignore[reportArgumentType]
+        organization, _ = decode_agent_token(self.dagster_cloud_agent_token)
+        return organization
 
     @property
     def deployment_name(self) -> Optional[str]:
@@ -343,26 +351,31 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
 
     @property
     def deployment_names(self) -> list[str]:
-        if self._dagster_cloud_api_config.get("deployment"):  # pyright: ignore[reportOptionalMemberAccess]
-            return [self._dagster_cloud_api_config["deployment"]]  # pyright: ignore[reportOptionalSubscript]
+        if self._dagster_cloud_api_config.get("deployment"):
+            return [self._dagster_cloud_api_config["deployment"]]
 
-        return self._dagster_cloud_api_config.get("deployments", [])  # pyright: ignore[reportOptionalMemberAccess]
+        return self._dagster_cloud_api_config.get("deployments", [])
 
     @property
     def include_all_serverless_deployments(self) -> bool:
-        return self._dagster_cloud_api_config.get("all_serverless_deployments")  # pyright: ignore[reportOptionalMemberAccess,reportReturnType]
+        return self._dagster_cloud_api_config.get("all_serverless_deployments") or False
 
     @property
     def dagit_url(self):
-        organization = get_organization_name_from_agent_token(self.dagster_cloud_agent_token)  # pyright: ignore[reportArgumentType]
+        organization, region = decode_agent_token(self.dagster_cloud_agent_token)
         if not organization:
             raise Exception(
                 "Could not derive Dagster Cloud URL from agent token to generate a Dagit URL."
                 " Generate a new agent token in the Dagit UI."
             )
 
-        deployment = self._dagster_cloud_api_config.get("deployment")  # pyright: ignore[reportOptionalMemberAccess]
-        return f"https://{organization}.dagster.cloud/" + (f"{deployment}/" if deployment else "")
+        deployment = self._dagster_cloud_api_config.get("deployment")
+        base_url = (
+            f"https://{organization}.{region}.dagster.cloud/"
+            if region
+            else f"https://{organization}.dagster.cloud/"
+        )
+        return base_url + (f"{deployment}/" if deployment else "")
 
     @property
     def dagster_cloud_graphql_url(self):
@@ -412,33 +425,38 @@ class DagsterCloudAgentInstance(DagsterCloudInstance):
         return get_agent_headers(self._dagster_cloud_api_config, scope=scope)  # pyright: ignore[reportArgumentType]
 
     @property
-    def dagster_cloud_agent_token(self):
-        return self._dagster_cloud_api_config.get("agent_token")  # pyright: ignore[reportOptionalMemberAccess]
+    def dagster_cloud_agent_token(self) -> str:
+        check.invariant(
+            self._dagster_cloud_api_config.get("agent_token") is not None,
+            "No agent token found in dagster_cloud_api configuration. An agent token is required"
+            " for Dagster Cloud authentication.",
+        )
+        return self._dagster_cloud_api_config["agent_token"]
 
     @property
     def dagster_cloud_api_retries(self) -> int:
-        return self._dagster_cloud_api_config["retries"]  # pyright: ignore[reportOptionalSubscript]
+        return self._dagster_cloud_api_config["retries"]
 
     @property
     def dagster_cloud_api_timeout(self) -> int:
-        return self._dagster_cloud_api_config["timeout"]  # pyright: ignore[reportOptionalSubscript]
+        return self._dagster_cloud_api_config["timeout"]
 
     @property
     def dagster_cloud_api_proxies(self) -> Optional[dict[str, str]]:
         # Requests library modifies the proxies key so create a copy
         return (
             self._dagster_cloud_api_config.get("proxies").copy()  # pyright: ignore[reportOptionalMemberAccess]
-            if self._dagster_cloud_api_config.get("proxies")  # pyright: ignore[reportOptionalMemberAccess]
+            if self._dagster_cloud_api_config.get("proxies")
             else {}
         )
 
     @property
     def dagster_cloud_api_agent_label(self) -> Optional[str]:
-        return self._dagster_cloud_api_config.get("agent_label")  # pyright: ignore[reportOptionalMemberAccess]
+        return self._dagster_cloud_api_config.get("agent_label")
 
     @property
     def includes_branch_deployments(self) -> bool:
-        return self._dagster_cloud_api_config.get("branch_deployments", False)  # pyright: ignore[reportOptionalMemberAccess]
+        return self._dagster_cloud_api_config.get("branch_deployments", False)
 
     @property
     def instance_uuid(self) -> str:

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/opentelemetry/config/exporter.py

@@ -1,6 +1,6 @@
 from dagster import BoolSource, Enum, Field, IntSource, Map, Permissive, Shape, StringSource
 
-from ..enum import (
+from dagster_cloud.opentelemetry.enum import (
     AggregationTemporalityEnum,
     CompressionEnum,
     LoggingExporterEnum,

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/opentelemetry/observers/execution_observer.py

@@ -5,8 +5,10 @@ from typing import Callable, Optional
 from dagster import DagsterError
 from dagster._time import get_current_timestamp
 
-from ..controller import OpenTelemetryController
-from .dagster_exception_handler import extract_dagster_error_attributes
+from dagster_cloud.opentelemetry.controller import OpenTelemetryController
+from dagster_cloud.opentelemetry.observers.dagster_exception_handler import (
+    extract_dagster_error_attributes,
+)
 
 ResultEvaluatorCallback = Callable[..., str]
 

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/pex/grpc/__init__.py

@@ -1,8 +1,8 @@
-from .client import (
+from dagster_cloud.pex.grpc.client import (
     MultiPexGrpcClient as MultiPexGrpcClient,
     wait_for_grpc_server as wait_for_grpc_server,
 )
-from .types import (
+from dagster_cloud.pex.grpc.types import (
     CreatePexServerArgs as CreatePexServerArgs,
     GetPexServersArgs as GetPexServersArgs,
     PexServerHandle as PexServerHandle,

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/pex/grpc/client.py

@@ -12,8 +12,8 @@ from dagster._serdes import serialize_value
 from dagster._utils.error import SerializableErrorInfo, serializable_error_info_from_exc_info
 from dagster_shared.serdes.serdes import deserialize_value
 
-from .__generated__ import MultiPexApiStub, multi_pex_api_pb2
-from .types import (
+from dagster_cloud.pex.grpc.__generated__ import MultiPexApiStub, multi_pex_api_pb2
+from dagster_cloud.pex.grpc.types import (
     CreatePexServerArgs,
     CreatePexServerResponse,
     GetCrashedPexServersArgs,

dagster_cloud-1.11.13/dagster_cloud/pex/grpc/server/__init__.py

@@ -0,0 +1,2 @@
+from dagster_cloud.pex.grpc.server.manager import MultiPexManager as MultiPexManager
+from dagster_cloud.pex.grpc.server.server import run_multipex_server as run_multipex_server

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/pex/grpc/server/cli/__init__.py

@@ -9,8 +9,8 @@ from dagster._utils.interrupts import setup_interrupt_handlers
 from dagster_cloud_cli.core.workspace import PexMetadata
 from typer import Option, Typer
 
-from ..registry import PexS3Registry
-from ..server import run_multipex_server
+from dagster_cloud.pex.grpc.server.registry import PexS3Registry
+from dagster_cloud.pex.grpc.server.server import run_multipex_server
 
 app = Typer(hidden=True)
 

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/pex/grpc/server/manager.py

@@ -18,11 +18,10 @@ from dagster_shared import seven
 from dagster_shared.ipc import open_ipc_subprocess
 from pydantic import BaseModel, Extra
 
+from dagster_cloud.pex.grpc.server.registry import PexS3Registry
+from dagster_cloud.pex.grpc.types import PexServerHandle
 from dagster_cloud.workspace.user_code_launcher.utils import get_grpc_server_env
 
-from ..types import PexServerHandle
-from .registry import PexS3Registry
-
 logger = logging.getLogger("dagster.multipex")
 
 

{dagster_cloud-1.11.11 → dagster_cloud-1.11.13}/dagster_cloud/pex/grpc/server/registry.py

@@ -28,7 +28,11 @@ def _download_from_s3(filename: str, local_filepath: str):
 
     config = Config(retries={"max_attempts": 3, "mode": "standard"})
 
-    s3 = boto3.client("s3", region_name="us-west-2", config=config)
+    s3 = boto3.client(
+        "s3",
+        region_name=os.getenv("DAGSTER_CLOUD_SERVERLESS_REGION", "us-west-2"),
+        config=config,
+    )
 
     # TODO: move the bucket and prefix to pex_metdata
     s3_bucket_name = os.environ["DAGSTER_CLOUD_SERVERLESS_STORAGE_S3_BUCKET"]