great-expectations-cloud 20250902.0.dev1__py3-none-any.whl → 20260120.0.dev0__py3-none-any.whl

great_expectations_cloud/agent/message_service/asyncio_rabbit_mq_client.py

@@ -27,6 +27,7 @@ class OnMessagePayload:
     correlation_id: str
     delivery_tag: int
     body: bytes
+    redelivered: bool = False  # Set by RabbitMQ when message is redelivered
 
 
 class OnMessageFn(Protocol):
@@ -174,8 +175,12 @@ class AsyncRabbitMQClient:
         # param on_message is provided by the caller as an argument to AsyncRabbitMQClient.run
         correlation_id = header_frame.correlation_id
         delivery_tag = method_frame.delivery_tag
+        redelivered = method_frame.redelivered  # RabbitMQ sets this flag on redelivery
         payload = OnMessagePayload(
-            correlation_id=correlation_id, delivery_tag=delivery_tag, body=body
+            correlation_id=correlation_id,
+            delivery_tag=delivery_tag,
+            body=body,
+            redelivered=redelivered,
         )
         return on_message(payload)
 
@@ -190,10 +195,13 @@ class AsyncRabbitMQClient:
     def _on_consumer_canceled(self, method_frame: Basic.Cancel) -> None:
         """Callback invoked when the broker cancels the client's connection."""
         if self._channel is not None:
-            LOGGER.info(
-                "Consumer was cancelled remotely, shutting down",
+            LOGGER.warning(
+                "rabbitmq.consumer.cancelled",
                 extra={
-                    "method_frame": method_frame,
+                    "consumer_tag": method_frame.consumer_tag
+                    if hasattr(method_frame, "consumer_tag")
+                    else None,
+                    "was_consuming": self.was_consuming,
                 },
             )
             self._channel.close()
@@ -232,11 +240,31 @@ class AsyncRabbitMQClient:
         self._reconnect()
         self._log_pika_exception("Connection open failed", reason)
 
-    def _on_connection_closed(
-        self, connection: AsyncioConnection, _unused_reason: pika.Exception
-    ) -> None:
+    def _on_connection_closed(self, connection: AsyncioConnection, reason: pika.Exception) -> None:
         """Callback invoked after the broker closes the connection"""
-        LOGGER.debug("Connection to RabbitMQ has been closed")
+        # Use DEBUG for expected closes, WARNING for unexpected
+        log_level = LOGGER.debug if self._closing else LOGGER.warning
+        if isinstance(reason, (ConnectionClosed, ChannelClosed)):
+            log_level(
+                "rabbitmq.connection.closed",
+                extra={
+                    "reply_code": reason.reply_code,
+                    "reply_text": reason.reply_text,
+                    "was_consuming": self.was_consuming,
+                    "is_closing": self._closing,
+                },
+            )
+        else:
+            log_level(
+                "rabbitmq.connection.closed",
+                extra={
+                    "reply_code": 0,  # Unknown/non-AMQP error
+                    "reply_text": str(reason),
+                    "reason_type": type(reason).__name__,
+                    "was_consuming": self.was_consuming,
+                    "is_closing": self._closing,
+                },
+            )
         self._channel = None
         self._is_unrecoverable = True
         if self._closing:

great_expectations_cloud/agent/message_service/subscriber.py

@@ -37,6 +37,8 @@ class EventContext:
             can be removed from the queue.
         redeliver_message: async callable to signal that the broker should
             try to deliver this message again.
+        redelivered: True if RabbitMQ is redelivering this message (another
+            consumer failed to acknowledge it).
     """
 
     event: Event
@@ -44,6 +46,7 @@ class EventContext:
     processed_successfully: Callable[[], None]
     processed_with_failures: Callable[[], None]
     redeliver_message: Callable[[], Coroutine[OnMessageCallback, None, None]]
+    redelivered: bool = False
 
 
 class OnMessageCallback(Protocol):
@@ -142,6 +145,7 @@ class Subscriber:
             processed_successfully=ack_callback,
             processed_with_failures=nack_callback,
             redeliver_message=redeliver_message,
+            redelivered=payload.redelivered,
         )
 
         return on_message(event_context)

great_expectations_cloud/agent/models.py

@@ -24,6 +24,18 @@ def all_subclasses(cls: type) -> list[type]:
     return all_sub_cls
 
 
+class DomainContext(BaseModel):
+    """
+    Encapsulates domain-related context information.
+
+    This dataclass consolidates organization_id and workspace_id to reduce
+    parameter proliferation and improve code maintainability.
+    """
+
+    organization_id: UUID
+    workspace_id: UUID
+
+
 class AgentBaseExtraForbid(BaseModel):
     class Config:
         # 2024-03-04: ZEL-501 Strictly enforce models for handling outdated APIs
@@ -39,6 +51,7 @@ class AgentBaseExtraIgnore(BaseModel):
 class EventBase(AgentBaseExtraIgnore):
     type: str
     organization_id: Optional[UUID] = None  # noqa: UP045
+    workspace_id: UUID
 
 
 class ScheduledEventBase(EventBase):
@@ -276,15 +289,23 @@ class CreateScheduledJobAndSetJobStartedRequest(AgentBaseExtraForbid):
     data: CreateScheduledJobAndSetJobStarted
 
 
-def build_failed_job_completed_status(error: BaseException) -> JobCompleted:
+def build_failed_job_completed_status(
+    error: BaseException,
+    processed_by: Literal["agent", "runner"] | None = None,
+) -> JobCompleted:
     if isinstance(error, GXCoreError):
         status = JobCompleted(
             success=False,
             error_stack_trace=str(error),
             error_code=error.error_code,
             error_params=error.get_error_params(),
+            processed_by=processed_by,
         )
     else:
-        status = JobCompleted(success=False, error_stack_trace=str(error))
+        status = JobCompleted(
+            success=False,
+            error_stack_trace=str(error),
+            processed_by=processed_by,
+        )
 
     return status

{great_expectations_cloud-20250902.0.dev1.dist-info → great_expectations_cloud-20260120.0.dev0.dist-info}/METADATA

@@ -1,22 +1,22 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: great_expectations_cloud
-Version: 20250902.0.dev1
+Version: 20260120.0.dev0
 Summary: Great Expectations Cloud
 License: Proprietary
+License-File: LICENSE
 Author: The Great Expectations Team
 Author-email: team@greatexpectations.io
-Requires-Python: >=3.11,<3.12
+Requires-Python: >=3.11.4,<3.12
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Software Development :: Quality Assurance
 Classifier: Topic :: Software Development :: Testing
-Requires-Dist: great-expectations[databricks,gx-redshift,mssql,postgresql,snowflake] (==1.5.10)
+Requires-Dist: great-expectations[databricks,gx-redshift,mssql,postgresql,snowflake,trino] (==1.11.1)
 Requires-Dist: orjson (>=3.9.7,<4.0.0,!=3.9.10)
 Requires-Dist: packaging (>=21.3,<26.0)
 Requires-Dist: pika (>=1.3.1,<2.0.0)

{great_expectations_cloud-20250902.0.dev1.dist-info → great_expectations_cloud-20260120.0.dev0.dist-info}/RECORD

@@ -1,34 +1,34 @@
 great_expectations_cloud/__init__.py,sha256=1mr5RDyA2N38eynvEfVbuYIbjFadeJfqZ-X9CrqYiVo,150
 great_expectations_cloud/agent/__init__.py,sha256=FqDFYbGefmNhwlvJwJbNovkwzny6mwaYH5LtTi6VlSU,464
 great_expectations_cloud/agent/actions/__init__.py,sha256=TYPe2j8EgaziXXgSLEdgjnbHKL56O6cQL1kjPnGbRFI,949
-great_expectations_cloud/agent/actions/agent_action.py,sha256=mGAovvT4CDaQlZaeNcz5tqfMIt6cul2anzxPwFrSxeA,1186
-great_expectations_cloud/agent/actions/draft_datasource_config_action.py,sha256=_v_yEPXZ3k8-lvdMCEPtrf6kegVYxnBo9UC0WoLaLwM,5117
-great_expectations_cloud/agent/actions/generate_data_quality_check_expectations_action.py,sha256=Ky1fqCspzhRRbCSUgWqYiLnL6iRDMDfYvgn1ZzfqzM0,23196
-great_expectations_cloud/agent/actions/list_asset_names.py,sha256=bDAYQtDbgDAOg8iu7HyGSTmBduhup7lLN0vGxoJZqN4,2510
-great_expectations_cloud/agent/actions/run_checkpoint.py,sha256=N2d07JDCG06kMve7yjPZQFlaGKoJw5dCbpKuBWw6Ssg,3751
-great_expectations_cloud/agent/actions/run_metric_list_action.py,sha256=tW64pNYJElkUmcd9baKllTSyR-NYyU-40QvMwSwwLqQ,3234
-great_expectations_cloud/agent/actions/run_scheduled_checkpoint.py,sha256=XMvv59izhzjSuw2YF7mlyNkeUQ3hxXj6G-ldAeSfl-s,2403
-great_expectations_cloud/agent/actions/run_window_checkpoint.py,sha256=khFmTSZ6W-MrInswgEHGrQAzVt-aelycurAMPbcLcLU,2264
+great_expectations_cloud/agent/actions/agent_action.py,sha256=F9zOgVmNJ_V2RhRbDXNMjd46-QVNFN8Lp_bmPvh5cwU,1189
+great_expectations_cloud/agent/actions/draft_datasource_config_action.py,sha256=NVN2GBSty-XvCW7pHDkLImHd3V0iJzpUNIh8rNGDuzs,5241
+great_expectations_cloud/agent/actions/generate_data_quality_check_expectations_action.py,sha256=dhiy_lkcePDy45fEX0uhk-m89_aYKtV6P5jfd-Dcax8,23611
+great_expectations_cloud/agent/actions/list_asset_names.py,sha256=pOL5ip8ZZJbZhDNSp44rjYkx93rKdf3U6f4fY-JLhvg,2576
+great_expectations_cloud/agent/actions/run_checkpoint.py,sha256=xlDYX6BHvHr3loz74hu4ARq5VStJ7-QYf80GPi-PuTY,5697
+great_expectations_cloud/agent/actions/run_metric_list_action.py,sha256=69nyR0vXjz_lKAHYczuEMQtbNIv0lf-DMiOBXmkwpuQ,3237
+great_expectations_cloud/agent/actions/run_scheduled_checkpoint.py,sha256=33lp6F_J1asgBmaHoqLLDL8wRnBw9MEZlQ3bosGQkwk,3425
+great_expectations_cloud/agent/actions/run_window_checkpoint.py,sha256=MCMbgY3dNUx546sNpXg5p3rCWNzDa6EQOUt20Nr5udo,2317
 great_expectations_cloud/agent/actions/unknown.py,sha256=mtWw9tDZqGZSiUWj7PtIlLFJ1dM-7AHBX3SO16-u2EM,739
-great_expectations_cloud/agent/actions/utils.py,sha256=DuUy6Ntsin0RIdgG2n1K85WJUATKHsqcCgtXKjr7scQ,1269
-great_expectations_cloud/agent/agent.py,sha256=Gd8gaamnh7Gw4uiW2-sVxbMaWdeNibHQR6LDryI_uVY,29761
+great_expectations_cloud/agent/actions/utils.py,sha256=0lzeASN1TYWJ35-H-sCRkcPMHzWU05SzKIyTv0CzvA8,1665
+great_expectations_cloud/agent/agent.py,sha256=JdyWvLdkzMq_kJgLsuhCDZz7NIOjBQZxaUv_UiwiEwg,40826
 great_expectations_cloud/agent/agent_warnings.py,sha256=9-xl_AI2V9Py4o7KzFOQjG3lYx-vZ36fq4w2iiPNiUw,362
 great_expectations_cloud/agent/cli.py,sha256=a_HmPxBMlVD59BEmkZnlbOOAFlezVMx9djZ2XIW-3W0,2885
 great_expectations_cloud/agent/config.py,sha256=c1BOr-TrxZnciWNRuu4atGtfRh-XSmwIS0osDCzQa04,1348
 great_expectations_cloud/agent/constants.py,sha256=SAEtcOwI-SxZiZSVoCFfEC5oCtYdgmyUmK7for_zg_Q,246
-great_expectations_cloud/agent/event_handler.py,sha256=kePpZ4EMIY0YWKeYH4W7lKY8C8OeER_Dkhwwadf3nTc,6069
+great_expectations_cloud/agent/event_handler.py,sha256=wcvAM0i6pnIsWlYstnZ2ThpByJYCyKrRmh0solC02bk,6091
 great_expectations_cloud/agent/exceptions.py,sha256=XIDBVSmBfFpVQA5i9rZqEtORIdi5E4_lwup0jP90TUk,1506
 great_expectations_cloud/agent/message_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-great_expectations_cloud/agent/message_service/asyncio_rabbit_mq_client.py,sha256=B2EwgG_Qdm1s_xkxbGPQxdRBkDSuhkKyT6rGFpXuqQ0,12391
-great_expectations_cloud/agent/message_service/subscriber.py,sha256=K8szy9uM1MUBkMaMZt9o5ExnjqEn1hw6zYaH7pQkjoM,5975
-great_expectations_cloud/agent/models.py,sha256=yZEir-ZShLbyJ8DZxMYNAZmSzbpOSH5wGRKyPo1LcJU,9984
+great_expectations_cloud/agent/message_service/asyncio_rabbit_mq_client.py,sha256=z1iYArWz5qhtsEYSK6Qd2LbiwZsEJOzOvunvtHVRuBo,13630
+great_expectations_cloud/agent/message_service/subscriber.py,sha256=ZEwwLaQe-yypYJWMkpTvRqEgzH7FbYH0-ean8LK1Qt8,6174
+great_expectations_cloud/agent/models.py,sha256=PKPxOXApET47s5Q-c9OqolgZoYghoULuvMFcCsl1soI,10478
 great_expectations_cloud/agent/run.py,sha256=V33RLoB1PFmJ0h0RfHG4SB5lN_Za8tW2Dua6GUpN9yY,639
 great_expectations_cloud/agent/utils.py,sha256=3OvdcXeK1gk2oJgqG4jPvBRwlMCn8LioULW3YgRtj98,2950
 great_expectations_cloud/logging/README.md,sha256=vbwU689x8SkGjzoBYQzZOzAvh28fR0RCa1XY5WD-Dgs,1762
 great_expectations_cloud/logging/logging_cfg.py,sha256=W6mlm4_Z2bjzM5TuKmFg_WZor2XoJm4DAoLGaf2O__I,6579
 great_expectations_cloud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-great_expectations_cloud-20250902.0.dev1.dist-info/LICENSE,sha256=_JJnoX6N_OkrAwlCRizCwil0tIjDAy2TG3GiJ50sM6k,2084
-great_expectations_cloud-20250902.0.dev1.dist-info/METADATA,sha256=lfPWV6KQi7QxYYifj3n0BkFeEKSxrummmkMp0Uq48_Y,12352
-great_expectations_cloud-20250902.0.dev1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-great_expectations_cloud-20250902.0.dev1.dist-info/entry_points.txt,sha256=ofJgdeS2gSzxXLyCAjfNhIaN1wmSyR7EAMs5qhVaXE4,68
-great_expectations_cloud-20250902.0.dev1.dist-info/RECORD,,
+great_expectations_cloud-20260120.0.dev0.dist-info/METADATA,sha256=UzLgYArcgE5nmCdO27EcqNQVSf7hClmv5Ol0xanWx1c,12331
+great_expectations_cloud-20260120.0.dev0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+great_expectations_cloud-20260120.0.dev0.dist-info/entry_points.txt,sha256=ofJgdeS2gSzxXLyCAjfNhIaN1wmSyR7EAMs5qhVaXE4,68
+great_expectations_cloud-20260120.0.dev0.dist-info/licenses/LICENSE,sha256=_JJnoX6N_OkrAwlCRizCwil0tIjDAy2TG3GiJ50sM6k,2084
+great_expectations_cloud-20260120.0.dev0.dist-info/RECORD,,

{great_expectations_cloud-20250902.0.dev1.dist-info → great_expectations_cloud-20260120.0.dev0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.1
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any