Wire update context

Author: dandavison

temporalio/client.py

@@ -4991,6 +4991,12 @@ def __init__(
         self._workflow_run_id = workflow_run_id
         self._result_type = result_type
         self._known_outcome = known_outcome
+        self._data_converter = self._client.data_converter._with_context(
+            WorkflowSerializationContext(
+                namespace=self._client.namespace,
+                workflow_id=self.workflow_id,
+            )
+        )
 
     @property
     def id(self) -> str:
@@ -5038,14 +5044,12 @@ async def result(
         assert self._known_outcome
         if self._known_outcome.HasField("failure"):
             raise WorkflowUpdateFailedError(
-                await self._client.data_converter.decode_failure(
-                    self._known_outcome.failure
-                ),
+                await self._data_converter.decode_failure(self._known_outcome.failure),
             )
         if not self._known_outcome.success.payloads:
             return None  # type: ignore
         type_hints = [self._result_type] if self._result_type else None
-        results = await self._client.data_converter.decode(
+        results = await self._data_converter.decode(
             self._known_outcome.success.payloads, type_hints
         )
         if not results:
@@ -6227,8 +6231,14 @@ async def _build_update_workflow_execution_request(
             ),
         )
         if input.args:
+            context = temporalio.converter.WorkflowSerializationContext(
+                namespace=self._client.namespace,
+                workflow_id=workflow_id,
+            )
             req.request.input.args.payloads.extend(
-                await self._client.data_converter.encode(input.args)
+                await self._client.data_converter._with_context(context).encode(
+                    input.args
+                )
             )
         if input.headers is not None:
             await self._apply_headers(input.headers, req.request.input.header.fields)

temporalio/worker/_workflow_instance.py

@@ -207,11 +207,23 @@ def __init__(self, det: WorkflowInstanceDetails) -> None:
         # No init for AbstractEventLoop
         WorkflowInstance.__init__(self)
         temporalio.workflow._Runtime.__init__(self)
-        self._payload_converter = det.payload_converter_class()
         self._failure_converter = det.failure_converter_class()
         self._defn = det.defn
         self._workflow_input: Optional[ExecuteWorkflowInput] = None
         self._info = det.info
+
+        self._payload_converter = det.payload_converter_class()
+        if isinstance(
+            self._payload_converter, temporalio.converter.WithSerializationContext
+        ):
+            serialization_context = temporalio.converter.WorkflowSerializationContext(
+                namespace=self._info.namespace,
+                workflow_id=self._info.workflow_id,
+            )
+            self._payload_converter = self._payload_converter.with_context(
+                serialization_context
+            )
+
         self._extern_functions = det.extern_functions
         self._disable_eager_activity_execution = det.disable_eager_activity_execution
         self._worker_level_failure_exception_types = (