Use Worker::validate_namespace from sdk-core to get namespace information

Author: jmaeagle99

temporalio/bridge/proto/__init__.py

@@ -3,6 +3,7 @@
     ActivitySlotInfo,
     ActivityTaskCompletion,
     LocalActivitySlotInfo,
+    NamespaceInfo,
     NexusSlotInfo,
     WorkflowSlotInfo,
 )
@@ -12,6 +13,7 @@
     "ActivitySlotInfo",
     "ActivityTaskCompletion",
     "LocalActivitySlotInfo",
+    "NamespaceInfo",
     "NexusSlotInfo",
     "WorkflowSlotInfo",
 ]

temporalio/bridge/proto/core_interface_pb2.py

@@ -165,3 +165,54 @@ class NexusSlotInfo(google.protobuf.message.Message):
     ) -> None: ...
 
 global___NexusSlotInfo = NexusSlotInfo
+
+class NamespaceInfo(google.protobuf.message.Message):
+    """Info about a namespace"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class Limits(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        BLOB_SIZE_LIMIT_ERROR_FIELD_NUMBER: builtins.int
+        MEMO_SIZE_LIMIT_ERROR_FIELD_NUMBER: builtins.int
+        blob_size_limit_error: builtins.int
+        """Maximum size in bytes for payload fields in workflow history events
+        (e.g., workflow/activity inputs and results, failure details, signal payloads).
+        When exceeded, the server will reject the operation with an error.
+        """
+        memo_size_limit_error: builtins.int
+        """Maximum total memo size in bytes per workflow execution."""
+        def __init__(
+            self,
+            *,
+            blob_size_limit_error: builtins.int = ...,
+            memo_size_limit_error: builtins.int = ...,
+        ) -> None: ...
+        def ClearField(
+            self,
+            field_name: typing_extensions.Literal[
+                "blob_size_limit_error",
+                b"blob_size_limit_error",
+                "memo_size_limit_error",
+                b"memo_size_limit_error",
+            ],
+        ) -> None: ...
+
+    LIMITS_FIELD_NUMBER: builtins.int
+    @property
+    def limits(self) -> global___NamespaceInfo.Limits:
+        """Namespace configured limits"""
+    def __init__(
+        self,
+        *,
+        limits: global___NamespaceInfo.Limits | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing_extensions.Literal["limits", b"limits"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self, field_name: typing_extensions.Literal["limits", b"limits"]
+    ) -> None: ...
+
+global___NamespaceInfo = NamespaceInfo

temporalio/bridge/proto/core_interface_pb2.pyi

@@ -543,7 +543,7 @@ pub fn new_replay_worker<'a>(
 
 #[pymethods]
 impl WorkerRef {
-    fn validate<'p>(&self, py: Python<'p>) -> PyResult<Bound<PyAny, 'p>> {
+    fn validate_namespace<'p>(&self, py: Python<'p>) -> PyResult<Bound<'p, PyAny>> {
         self.runtime.assert_same_process("use worker")?;
         let worker = self.worker.as_ref().unwrap().clone();
         // Set custom slot supplier task locals so they can run futures.
@@ -555,11 +555,15 @@ impl WorkerRef {
             .expect("must only be set once");
 
         self.runtime.future_into_py(py, async move {
-            worker
-                .validate()
-                .await
-                .context("Worker validation failed")
-                .map_err(Into::into)
+            let bytes = match worker.validate_namespace().await {
+                Ok(info) => info.encode_to_vec(),
+                Err(err) => {
+                    return Err(PyRuntimeError::new_err(format!(
+                        "Worker validation failed: {err}"
+                    )))
+                }
+            };
+            Ok(bytes)
         })
     }
 

temporalio/bridge/src/worker.rs

@@ -25,7 +25,9 @@
 from temporalio.bridge.temporal_sdk_bridge import (
     CustomSlotSupplier as BridgeCustomSlotSupplier,
 )
-from temporalio.bridge.temporal_sdk_bridge import PollShutdownError  # type: ignore
+from temporalio.bridge.temporal_sdk_bridge import (
+    PollShutdownError,  # type: ignore # noqa: F401
+)
 from temporalio.worker._command_aware_visitor import CommandAwarePayloadVisitor
 
 
@@ -194,9 +196,13 @@ def __init__(self, ref: temporalio.bridge.temporal_sdk_bridge.WorkerRef) -> None
         """Create SDK core worker from a bridge worker."""
         self._ref = ref
 
-    async def validate(self) -> None:
+    async def validate_namespace(
+        self,
+    ) -> temporalio.bridge.proto.NamespaceInfo:
         """Validate the bridge worker."""
-        await self._ref.validate()  # type: ignore[reportOptionalMemberAccess]
+        return temporalio.bridge.proto.NamespaceInfo.FromString(
+            await self._ref.validate_namespace()  # type: ignore[reportOptionalMemberAccess]
+        )
 
     async def poll_workflow_activation(
         self,

temporalio/bridge/worker.py

@@ -715,7 +715,7 @@ def make_lambda(plugin: Plugin, next: Callable[[Worker], Awaitable[None]]):
 
     async def _run(self):
         # Eagerly validate which will do a namespace check in Core
-        await self._bridge_worker.validate()
+        _ = await self._bridge_worker.validate_namespace()
 
         if self._started:
             raise RuntimeError("Already started")