dagster-cloud 1.11.3__py3-none-any.whl → 1.11.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -308,6 +308,19 @@ class DagsterCloudAgent:
308
308
 
309
309
  self._check_initial_deployment_names()
310
310
 
311
+ serving = []
312
+ queues = list(filter(None, self._instance.agent_queues_config.queues))
313
+ if queues:
314
+ serving.append(f"queues{queues}")
315
+ if self._instance.deployment_names:
316
+ serving.append(f"deployments{self._instance.deployment_names}")
317
+ if self._instance.include_all_serverless_deployments:
318
+ serving.append("all serverless deployments")
319
+ if self._instance.includes_branch_deployments:
320
+ serving.append("branch deployments")
321
+
322
+ self._logger.info(f"Agent is serving: {', '.join(serving)}")
323
+
311
324
  self._check_update_workspace(
312
325
  user_code_launcher,
313
326
  upload_all=user_code_launcher.upload_snapshots_on_startup,
dagster_cloud/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "1.11.3"
1
+ __version__ = "1.11.5"
@@ -64,6 +64,7 @@ class Client:
64
64
  grace_period: int = DEFAULT_ECS_GRACE_PERIOD,
65
65
  launch_type: str = "FARGATE",
66
66
  show_debug_cluster_info: bool = True,
67
+ assign_public_ip: Optional[bool] = None,
67
68
  ):
68
69
  self.ecs = ecs_client if ecs_client else boto3.client("ecs", config=config)
69
70
  self.logs = boto3.client("logs", config=config)
@@ -85,6 +86,7 @@ class Client:
85
86
  self.grace_period = check.int_param(grace_period, "grace_period")
86
87
  self.launch_type = check.str_param(launch_type, "launch_type")
87
88
  self._namespace: Optional[str] = None
89
+ self._assign_public_ip_override = assign_public_ip
88
90
 
89
91
  @property
90
92
  def ec2(self):
@@ -115,14 +117,17 @@ class Client:
115
117
  @property
116
118
  @cached_method
117
119
  def network_configuration(self):
120
+ if self.launch_type != "FARGATE":
121
+ assign_public_ip = None
122
+ elif self._assign_public_ip_override is not None:
123
+ assign_public_ip = "ENABLED" if self._assign_public_ip_override else "DISABLED"
124
+ else:
125
+ assign_public_ip = self._infer_assign_public_ip()
126
+
118
127
  network_configuration = {
119
128
  "awsvpcConfiguration": {
120
129
  "subnets": self.subnet_ids,
121
- **(
122
- {"assignPublicIp": self._assign_public_ip()}
123
- if self.launch_type == "FARGATE"
124
- else {}
125
- ),
130
+ **({"assignPublicIp": assign_public_ip} if assign_public_ip else {}),
126
131
  },
127
132
  }
128
133
 
@@ -760,7 +765,7 @@ class Client:
760
765
  if service["Name"] == service_name:
761
766
  return service["Id"]
762
767
 
763
- def _assign_public_ip(self):
768
+ def _infer_assign_public_ip(self):
764
769
  # https://docs.aws.amazon.com/AmazonECS/latest/userguide/fargate-task-networking.html
765
770
  # Assign a public IP if any of the subnets are public
766
771
  route_tables = self.ec2.route_tables.filter(
@@ -90,6 +90,7 @@ class EcsUserCodeLauncher(DagsterCloudUserCodeLauncher[EcsServerHandleType], Con
90
90
  enable_ecs_exec=False,
91
91
  server_task_definition_prefix: str = "server",
92
92
  run_task_definition_prefix: str = "run",
93
+ assign_public_ip: Optional[bool] = None,
93
94
  **kwargs,
94
95
  ):
95
96
  self.ecs = boto3.client("ecs")
@@ -182,6 +183,7 @@ class EcsUserCodeLauncher(DagsterCloudUserCodeLauncher[EcsServerHandleType], Con
182
183
  timeout=self._ecs_timeout,
183
184
  grace_period=self._ecs_grace_period,
184
185
  launch_type=self.launch_type,
186
+ assign_public_ip=assign_public_ip,
185
187
  )
186
188
  super().__init__(**kwargs)
187
189
 
@@ -297,6 +299,16 @@ class EcsUserCodeLauncher(DagsterCloudUserCodeLauncher[EcsServerHandleType], Con
297
299
  "run_task_definition_prefix": Field(
298
300
  str, is_required=False, default_value="dagsterrun"
299
301
  ),
302
+ "assign_public_ip": Field(
303
+ Noneable(bool),
304
+ is_required=False,
305
+ default_value=None,
306
+ description=(
307
+ "When using the FARGATE launch type, the launcher will attempt to automatically determine if it is "
308
+ "necessary to assign a public IP to the ECS task. In complex network topologies, this automatic "
309
+ "determination may not be accurate. In this case, you can explicitly set this value to True or False."
310
+ ),
311
+ ),
300
312
  },
301
313
  SHARED_ECS_CONFIG,
302
314
  SHARED_USER_CODE_LAUNCHER_CONFIG,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dagster-cloud
3
- Version: 1.11.3
3
+ Version: 1.11.5
4
4
  Author-email: Elementl <support@elementl.com>
5
5
  License: Apache-2.0
6
6
  Project-URL: Homepage, https://dagster.io/cloud
@@ -28,9 +28,9 @@ Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
28
28
  Classifier: Operating System :: OS Independent
29
29
  Requires-Python: <3.14,>=3.9
30
30
  Description-Content-Type: text/markdown
31
- Requires-Dist: dagster==1.11.3
32
- Requires-Dist: dagster-shared==1.11.3
33
- Requires-Dist: dagster-cloud-cli==1.11.3
31
+ Requires-Dist: dagster==1.11.5
32
+ Requires-Dist: dagster-shared==1.11.5
33
+ Requires-Dist: dagster-cloud-cli==1.11.5
34
34
  Requires-Dist: opentelemetry-api<2,>=1.27.0
35
35
  Requires-Dist: opentelemetry-sdk<2,>=1.27.0
36
36
  Requires-Dist: opentelemetry-exporter-otlp-proto-grpc<2,>=1.27.0
@@ -65,12 +65,12 @@ Provides-Extra: insights
65
65
  Requires-Dist: pyarrow; extra == "insights"
66
66
  Provides-Extra: docker
67
67
  Requires-Dist: docker; extra == "docker"
68
- Requires-Dist: dagster-docker==0.27.3; extra == "docker"
68
+ Requires-Dist: dagster-docker==0.27.5; extra == "docker"
69
69
  Provides-Extra: kubernetes
70
70
  Requires-Dist: kubernetes; extra == "kubernetes"
71
- Requires-Dist: dagster-k8s==0.27.3; extra == "kubernetes"
71
+ Requires-Dist: dagster-k8s==0.27.5; extra == "kubernetes"
72
72
  Provides-Extra: ecs
73
- Requires-Dist: dagster-aws==0.27.3; extra == "ecs"
73
+ Requires-Dist: dagster-aws==0.27.5; extra == "ecs"
74
74
  Requires-Dist: boto3; extra == "ecs"
75
75
  Provides-Extra: sandbox
76
76
  Requires-Dist: supervisor; extra == "sandbox"
@@ -1,9 +1,9 @@
1
1
  dagster_cloud/__init__.py,sha256=q3UY5eTfCSqZrr35mmn1b1B_AD1VbfR4Alj8cZ49Xg4,316
2
2
  dagster_cloud/constants.py,sha256=CPAqXJ99SWGMviksdIA2A9894FEvHChNk8UcP4TluYM,455
3
3
  dagster_cloud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
- dagster_cloud/version.py,sha256=QRmtbmlWa_y4yftU0t90UOFDX99h0KZNr2EwU5DavlE,23
4
+ dagster_cloud/version.py,sha256=BXE7d2BLjQ0cOw7nqWZFd01VxpEfF1Vkf8tN6hJrH5U,23
5
5
  dagster_cloud/agent/__init__.py,sha256=_erVyIrxuHUiyNerwX8vNZcKZN8NAloTEkPq8vPZ3MI,811
6
- dagster_cloud/agent/dagster_cloud_agent.py,sha256=iGX6Ib_WT427r8LjD1Za1aGloSsP1j-cCiaVaJ29nQs,56496
6
+ dagster_cloud/agent/dagster_cloud_agent.py,sha256=jT1PSysqZFI-SAiS43oYLd5aVIlyczH6r2eIRCphHlc,57075
7
7
  dagster_cloud/agent/queries.py,sha256=iI84GQ1Zxt5ryo6M1ELIaIae-gwUY14QPPMUeiFK97o,1837
8
8
  dagster_cloud/agent/cli/__init__.py,sha256=rGbeQJ2Ap95wPYQuk5XbyHAfP9cs-XPUSmuVM_k278k,9084
9
9
  dagster_cloud/agent/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -113,8 +113,8 @@ dagster_cloud/workspace/config_schema/kubernetes.py,sha256=JIdZ5hX06hAw1lGT-5Acw
113
113
  dagster_cloud/workspace/docker/__init__.py,sha256=ioVI3SHL4gnzPaG7z25BTiejVTDpBgoCW_FwbRHB8FM,15112
114
114
  dagster_cloud/workspace/docker/utils.py,sha256=80vo8QphKxgb47JqDBOMZxTHN_MxReCjYXpppmfiue4,352
115
115
  dagster_cloud/workspace/ecs/__init__.py,sha256=0GliAaE4NYzMD0ALQIG20cRM5ou77bj_FaiysgqoOKI,65
116
- dagster_cloud/workspace/ecs/client.py,sha256=gabMBJv6GcR3KmJqESfdRTvIAArydTrzkvdNQWzywZ8,31355
117
- dagster_cloud/workspace/ecs/launcher.py,sha256=Cp4avJDdCM7sMQtJKDFkuwJF-uPx5pRoJ1qBUiSMC1Q,30475
116
+ dagster_cloud/workspace/ecs/client.py,sha256=9c4juhbKgzj7z0MBtEiOSOt0cYABq-c6GLXbyWX4Wic,31675
117
+ dagster_cloud/workspace/ecs/launcher.py,sha256=zfCZBXQib8hvVxY2Jhf5x6uL-wFSSai3zzdnXVGHAlo,31179
118
118
  dagster_cloud/workspace/ecs/run_launcher.py,sha256=rra0PMk_hdp2EkXGUkNvrrQ3dvzTrtqvwlI9ja83q54,574
119
119
  dagster_cloud/workspace/ecs/service.py,sha256=v-puyDEg2BzHA3RJqEhH8V04bUAcrYIlPCH1SThvwWw,4126
120
120
  dagster_cloud/workspace/ecs/utils.py,sha256=r5A4-GYZhOg7MvWfRvM_uVO9hbbd0QRdT_w2AmDmuoE,2970
@@ -125,7 +125,7 @@ dagster_cloud/workspace/user_code_launcher/__init__.py,sha256=6ib1U47aqB4R9BbzJV
125
125
  dagster_cloud/workspace/user_code_launcher/process.py,sha256=VZMfBf4OmXNwP0H0ckUMpqD-glqsBgACXS-xO6Pn6nM,14274
126
126
  dagster_cloud/workspace/user_code_launcher/user_code_launcher.py,sha256=RHHaO2LkGsFCWzSnuOQhiw2lnC7sdtflTba42Dkmcd0,102815
127
127
  dagster_cloud/workspace/user_code_launcher/utils.py,sha256=t8Epee9MrXtRhWL-b_3avXxgMGrjLScUNWtBUUGpMCg,5285
128
- dagster_cloud-1.11.3.dist-info/METADATA,sha256=-swQnH59rCnaM9_Hp0HHJdxUQ2I6cEsBRQutxHfNfnQ,6577
129
- dagster_cloud-1.11.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
130
- dagster_cloud-1.11.3.dist-info/top_level.txt,sha256=2hMt-U33jyCgnywNrDB9Ih0EpaVmiO6dFkYcJ7Iwx4I,14
131
- dagster_cloud-1.11.3.dist-info/RECORD,,
128
+ dagster_cloud-1.11.5.dist-info/METADATA,sha256=MCk3hOhMpsoONHS5d1MQqRPX-IEmy5Us475z6UfnsX0,6577
129
+ dagster_cloud-1.11.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
130
+ dagster_cloud-1.11.5.dist-info/top_level.txt,sha256=2hMt-U33jyCgnywNrDB9Ih0EpaVmiO6dFkYcJ7Iwx4I,14
131
+ dagster_cloud-1.11.5.dist-info/RECORD,,