dagster-cloud 1.11.10__py3-none-any.whl → 1.11.12__py3-none-any.whl

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/instance/__init__.py

@@ -34,7 +34,7 @@ from urllib3 import Retry
 from dagster_cloud.agent import AgentQueuesConfig
 from dagster_cloud.version import __version__
 
-from ..auth.constants import get_organization_name_from_agent_token
+from ..auth.constants import decode_agent_token
 from ..opentelemetry.config import opentelemetry_config_schema
 from ..opentelemetry.controller import OpenTelemetryController
 from ..storage.client import dagster_cloud_api_config
@@ -87,14 +87,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 +179,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 +242,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 +288,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 +318,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 +352,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 +426,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/storage/compute_logs/compute_log_manager.py

@@ -103,6 +103,7 @@ class CloudComputeLogManager(
             # for back-compat
             "run_id": log_key[0],
             "key": log_key[-1],
+            "method": "PUT",
         }
         if partial:
             params["partial"] = True
@@ -119,10 +120,9 @@ class CloudComputeLogManager(
         if resp_data.get("skip_upload"):
             return
 
-        self._upload_session.post(
+        self._upload_session.put(
             resp_data["url"],
-            data=resp_data["fields"],
-            files={"file": data},
+            data=data,
             timeout=self._instance.dagster_cloud_api_timeout,
         )
 

dagster_cloud/storage/defs_state/storage.py

@@ -1,18 +1,36 @@
+from pathlib import Path
 from typing import TYPE_CHECKING, Any, Optional
 
 import dagster._check as check
+from dagster._core.storage.defs_state.base import DefsStateStorage
 from dagster._serdes import ConfigurableClass, ConfigurableClassData
+from dagster_cloud_cli.core.artifacts import download_artifact, upload_artifact
+from dagster_cloud_cli.core.headers.auth import DagsterCloudInstanceScope
+from dagster_shared.serdes import deserialize_value
+from dagster_shared.serdes.objects.models.defs_state_info import DefsStateInfo
 from typing_extensions import Self
 
-from dagster_cloud.storage.defs_state.base_storage import BaseGraphQLDefsStateStorage
-
 if TYPE_CHECKING:
     from dagster_cloud.instance import DagsterCloudAgentInstance  # noqa: F401
 
+GET_LATEST_DEFS_STATE_INFO_QUERY = """
+    query getLatestDefsStateInfo {
+        latestDefsStateInfo
+    }
+"""
+
+SET_LATEST_VERSION_MUTATION = """
+    mutation setLatestDefsStateVersion($key: String!, $version: String!) {
+        defsState {
+            setLatestDefsStateVersion(key: $key, version: $version) {
+                ok
+            }
+        }
+    }
+"""
+
 
-class GraphQLDefsStateStorage(
-    BaseGraphQLDefsStateStorage["DagsterCloudAgentInstance"], ConfigurableClass
-):
+class GraphQLDefsStateStorage(DefsStateStorage["DagsterCloudAgentInstance"], ConfigurableClass):
     def __init__(
         self, inst_data: Optional[ConfigurableClassData] = None, override_graphql_client=None
     ):
@@ -53,3 +71,43 @@ class GraphQLDefsStateStorage(
             if self._override_graphql_client
             else self._instance.graphql_client  # pyright: ignore[reportAttributeAccessIssue]
         )
+
+    def _execute_query(self, query, variables=None, idempotent_mutation=False):
+        return self.graphql_client.execute(
+            query, variable_values=variables, idempotent_mutation=idempotent_mutation
+        )
+
+    def _get_artifact_key(self, key: str, version: str) -> str:
+        return f"__state__/{key}/{version}"
+
+    def download_state_to_path(self, key: str, version: str, path: Path) -> None:
+        download_artifact(
+            url=self.url,
+            scope=DagsterCloudInstanceScope.DEPLOYMENT,
+            api_token=self.api_token,
+            key=self._get_artifact_key(key, version),
+            path=path,
+            deployment=self.deployment,
+        )
+
+    def upload_state_from_path(self, key: str, version: str, path: Path) -> None:
+        upload_artifact(
+            url=self.url,
+            scope=DagsterCloudInstanceScope.DEPLOYMENT,
+            api_token=self.api_token,
+            key=self._get_artifact_key(key, version),
+            path=path,
+            deployment=self.deployment,
+        )
+        self.set_latest_version(key, version)
+
+    def get_latest_defs_state_info(self) -> Optional[DefsStateInfo]:
+        res = self._execute_query(GET_LATEST_DEFS_STATE_INFO_QUERY)
+        result = res["data"]["latestDefsStateInfo"]
+        if result is not None:
+            return deserialize_value(result, DefsStateInfo)
+        else:
+            return None
+
+    def set_latest_version(self, key: str, version: str) -> None:
+        self._execute_query(SET_LATEST_VERSION_MUTATION, variables={"key": key, "version": version})

dagster_cloud/version.py

@@ -1 +1 @@
-__version__ = "1.11.10"
+__version__ = "1.11.12"

{dagster_cloud-1.11.10.dist-info → dagster_cloud-1.11.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-cloud
-Version: 1.11.10
+Version: 1.11.12
 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.10
-Requires-Dist: dagster-shared==1.11.10
-Requires-Dist: dagster-cloud-cli==1.11.10
+Requires-Dist: dagster==1.11.12
+Requires-Dist: dagster-shared==1.11.12
+Requires-Dist: dagster-cloud-cli==1.11.12
 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.10; extra == "docker"
+Requires-Dist: dagster-docker==0.27.12; extra == "docker"
 Provides-Extra: kubernetes
 Requires-Dist: kubernetes; extra == "kubernetes"
-Requires-Dist: dagster-k8s==0.27.10; extra == "kubernetes"
+Requires-Dist: dagster-k8s==0.27.12; extra == "kubernetes"
 Provides-Extra: ecs
-Requires-Dist: dagster-aws==0.27.10; extra == "ecs"
+Requires-Dist: dagster-aws==0.27.12; extra == "ecs"
 Requires-Dist: boto3; extra == "ecs"
 Provides-Extra: sandbox
 Requires-Dist: supervisor; extra == "sandbox"

{dagster_cloud-1.11.10.dist-info → dagster_cloud-1.11.12.dist-info}/RECORD

@@ -1,7 +1,7 @@
 dagster_cloud/__init__.py,sha256=q3UY5eTfCSqZrr35mmn1b1B_AD1VbfR4Alj8cZ49Xg4,316
 dagster_cloud/constants.py,sha256=CPAqXJ99SWGMviksdIA2A9894FEvHChNk8UcP4TluYM,455
 dagster_cloud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dagster_cloud/version.py,sha256=-6noRcK_QpKJbYZJtuiOgAP4ZN45BdUcldnV7ljGK3A,24
+dagster_cloud/version.py,sha256=9Rb3yC-7xmQLAyvOVo4ZifFK2ok7cTSv1S1wYtretwE,24
 dagster_cloud/agent/__init__.py,sha256=_erVyIrxuHUiyNerwX8vNZcKZN8NAloTEkPq8vPZ3MI,811
 dagster_cloud/agent/dagster_cloud_agent.py,sha256=C72rumzRFFX41L2y7j-pJFrTAnvW60ZiS_AQRTUjExg,57013
 dagster_cloud/agent/queries.py,sha256=iI84GQ1Zxt5ryo6M1ELIaIae-gwUY14QPPMUeiFK97o,1837
@@ -19,7 +19,7 @@ dagster_cloud/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 dagster_cloud/api/dagster_cloud_api.py,sha256=6ge6WX8SCRKu9wTD0YgTBvCHgPN1_DFrIH_Y0Y9GFuA,14589
 dagster_cloud/artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dagster_cloud/auth/__init__.py,sha256=rUuAWah3YD-0tatQPFlmLNJpsGQ9ZRj7sUOSTrnf0zQ,248
-dagster_cloud/auth/constants.py,sha256=hl2aE8Wsg-Z6EDW9sDJAWMoEvTCPQgdlKOqoqnVP10Q,1242
+dagster_cloud/auth/constants.py,sha256=zFhKSUBkBKz5_eSTCp5FV8K_HFYJRGHDby-xC7KE4bo,1808
 dagster_cloud/batching/__init__.py,sha256=522PJZX1fmhpA91M32eDIwvYqn2PuqBodIOTcJmsXwc,40
 dagster_cloud/batching/batcher.py,sha256=kt9F-WdMnCJstCJFuHqagG_eHFTWjlfRQdhE-Cc9Hvw,7796
 dagster_cloud/dagster_insights/__init__.py,sha256=_qTVj3INjL-Y2YADy7uElk4HStZUBZoPD32jY5DbwF4,1864
@@ -46,7 +46,7 @@ dagster_cloud/execution/cloud_run_launcher/process.py,sha256=eYxvzcaJbqYKjHcXMUU
 dagster_cloud/execution/monitoring/__init__.py,sha256=2_o2UjeEPvfSpnOb0zky3ZQs-G9VHABAXzjvfMnAh8A,17782
 dagster_cloud/execution/utils/__init__.py,sha256=EfU-tK5-ScZYLF_zmWqUgIrmLsb9DDHWFN-CSg3BSXY,254
 dagster_cloud/execution/utils/process.py,sha256=Qz6Vp4d2siT7z_la7n1zybJLXfRG3hvTu9ePFJA4TA0,883
-dagster_cloud/instance/__init__.py,sha256=9BgD7WSrsx9HbSCLdQ-cwq-kZp7dLrRlGFTRbmhEfBw,26754
+dagster_cloud/instance/__init__.py,sha256=bZ4ZOUaa4Xg71DmYAHdxxTfSv9BEdjXB70esDWQyX-4,26193
 dagster_cloud/instrumentation/__init__.py,sha256=OKaT4qkwQNmCFdKveVrgB7JIo5sZD1F1rutQKdozK_4,886
 dagster_cloud/metadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dagster_cloud/metadata/source_code.py,sha256=yCUfD6V-wR2s4AUrmVpX_Dd7Gln7SYbtpuEpAH71J1I,5952
@@ -91,11 +91,10 @@ dagster_cloud/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 dagster_cloud/storage/client.py,sha256=jsqlQehNImOygYEAq0NoEVlbCn5EbIS0VZJabuVtxL8,2992
 dagster_cloud/storage/tags.py,sha256=YeWSWgpguqvpndFGMM0W7gORn0F8P3CDcanElE7n2Vo,2414
 dagster_cloud/storage/compute_logs/__init__.py,sha256=6O5IqKYrNZ6Yz0cataEa2_8u6kYAIP6PXOoeqh5jvjo,82
-dagster_cloud/storage/compute_logs/compute_log_manager.py,sha256=tGwHou-Qsa25XMwX6ULLE18PTjZiN9Pqx7bRs6Oym3Q,4878
+dagster_cloud/storage/compute_logs/compute_log_manager.py,sha256=OBNuGj0PECWmnAfbayFtRixztOcqaK-IGmgOD0AokfY,4857
 dagster_cloud/storage/defs_state/__init__.py,sha256=MCt72laA0K6t7dtZfqxlS9I7MYtFzI6lTR2NN40A9e8,72
-dagster_cloud/storage/defs_state/base_storage.py,sha256=_OtLfplIG5yVCZOjCZbdmEfCKm5yrgBdKXt6Wi8DMZ0,2976
 dagster_cloud/storage/defs_state/queries.py,sha256=UICxqqWH1gCpE_YR7TedxanES2gZ_7dR00y6KR8HWQA,365
-dagster_cloud/storage/defs_state/storage.py,sha256=OoikV-ptZtXr-t2izPkvrR-d94UdCV4lw4aXxAWZMkM,1792
+dagster_cloud/storage/defs_state/storage.py,sha256=V-YO63i8DBGCiCj5Ptu9U12jOMBCxQCvaF77YQWJrA8,4006
 dagster_cloud/storage/event_logs/__init__.py,sha256=N45uxmVOvMtOSYJFktLE32sJ6Ju9EFcwMH4b8L2DDFk,70
 dagster_cloud/storage/event_logs/queries.py,sha256=sgkHhHX-0Vr9WukcRjbuGSH8vjC9razSNnwqXBrihX8,19924
 dagster_cloud/storage/event_logs/storage.py,sha256=bu3GOrPu_RkCr29xic78-rRfI4h_ZTMOVZGWQj8jtSc,56438
@@ -129,7 +128,7 @@ dagster_cloud/workspace/user_code_launcher/__init__.py,sha256=6ib1U47aqB4R9BbzJV
 dagster_cloud/workspace/user_code_launcher/process.py,sha256=VZMfBf4OmXNwP0H0ckUMpqD-glqsBgACXS-xO6Pn6nM,14274
 dagster_cloud/workspace/user_code_launcher/user_code_launcher.py,sha256=DqWeZ_w-h4tP2ceLyvOHWeSPM6dUnr7gY_VmuSKsEYA,102911
 dagster_cloud/workspace/user_code_launcher/utils.py,sha256=t8Epee9MrXtRhWL-b_3avXxgMGrjLScUNWtBUUGpMCg,5285
-dagster_cloud-1.11.10.dist-info/METADATA,sha256=B1Akcb1W5inKpaInl8LMl6KhboqpA8i7kKa_9OL4Nr4,6584
-dagster_cloud-1.11.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dagster_cloud-1.11.10.dist-info/top_level.txt,sha256=2hMt-U33jyCgnywNrDB9Ih0EpaVmiO6dFkYcJ7Iwx4I,14
-dagster_cloud-1.11.10.dist-info/RECORD,,
+dagster_cloud-1.11.12.dist-info/METADATA,sha256=DP3Nwd6mkUdfyODjBTiBsHAN0oMSYwtRMubwl1IY4p4,6584
+dagster_cloud-1.11.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dagster_cloud-1.11.12.dist-info/top_level.txt,sha256=2hMt-U33jyCgnywNrDB9Ih0EpaVmiO6dFkYcJ7Iwx4I,14
+dagster_cloud-1.11.12.dist-info/RECORD,,

dagster_cloud/storage/defs_state/base_storage.py

@@ -1,90 +0,0 @@
-from abc import ABC, abstractmethod
-from pathlib import Path
-from typing import Any, Generic, Optional
-
-from dagster._core.instance.types import T_DagsterInstance
-from dagster._core.storage.defs_state.base import DefsStateStorage
-from dagster_cloud_cli.core.artifacts import download_artifact, upload_artifact
-from dagster_cloud_cli.core.headers.auth import DagsterCloudInstanceScope
-from dagster_shared.serdes import deserialize_value
-from dagster_shared.serdes.objects.models.defs_state_info import DefsStateInfo
-
-GET_LATEST_DEFS_STATE_INFO_QUERY = """
-    query getLatestDefsStateInfo {
-        latestDefsStateInfo
-    }
-"""
-
-SET_LATEST_VERSION_MUTATION = """
-    mutation setLatestDefsStateVersion($key: String!, $version: String!) {
-        defsState {
-            setLatestDefsStateVersion(key: $key, version: $version) {
-                ok
-            }
-        }
-    }
-"""
-
-
-class BaseGraphQLDefsStateStorage(
-    DefsStateStorage[T_DagsterInstance], ABC, Generic[T_DagsterInstance]
-):
-    """Base implementation of a DefsStateStorage that uses a GraphQL client to
-    interact with the Dagster+ API.
-    """
-
-    @property
-    @abstractmethod
-    def url(self) -> str: ...
-
-    @property
-    @abstractmethod
-    def api_token(self) -> str: ...
-
-    @property
-    @abstractmethod
-    def deployment(self) -> str: ...
-
-    @property
-    @abstractmethod
-    def graphql_client(self) -> Any: ...
-
-    def _execute_query(self, query, variables=None, idempotent_mutation=False):
-        return self.graphql_client.execute(
-            query, variable_values=variables, idempotent_mutation=idempotent_mutation
-        )
-
-    def _get_artifact_key(self, key: str, version: str) -> str:
-        return f"__state__/{key}/{version}"
-
-    def download_state_to_path(self, key: str, version: str, path: Path) -> None:
-        download_artifact(
-            url=self.url,
-            scope=DagsterCloudInstanceScope.DEPLOYMENT,
-            api_token=self.api_token,
-            key=self._get_artifact_key(key, version),
-            path=path,
-            deployment=self.deployment,
-        )
-
-    def upload_state_from_path(self, key: str, version: str, path: Path) -> None:
-        upload_artifact(
-            url=self.url,
-            scope=DagsterCloudInstanceScope.DEPLOYMENT,
-            api_token=self.api_token,
-            key=self._get_artifact_key(key, version),
-            path=path,
-            deployment=self.deployment,
-        )
-        self.set_latest_version(key, version)
-
-    def get_latest_defs_state_info(self) -> Optional[DefsStateInfo]:
-        res = self._execute_query(GET_LATEST_DEFS_STATE_INFO_QUERY)
-        result = res["data"]["latestDefsStateInfo"]
-        if result is not None:
-            return deserialize_value(result, DefsStateInfo)
-        else:
-            return None
-
-    def set_latest_version(self, key: str, version: str) -> None:
-        self._execute_query(SET_LATEST_VERSION_MUTATION, variables={"key": key, "version": version})