Implement it in the SDK

Author: dandavison

temporalio/converter.py

@@ -384,6 +384,15 @@ def from_payloads(
                 ) from err
         return values
 
+    def with_context(self, context: Optional[SerializationContext]) -> Self:
+        instance = type(self).__new__(type(self))
+        converters = [
+            c.with_context(context) if isinstance(c, WithSerializationContext) else c
+            for c in self.converters.values()
+        ]
+        CompositePayloadConverter.__init__(instance, *converters)
+        return instance
+
 
 class DefaultPayloadConverter(CompositePayloadConverter):
     """Default payload converter compatible with other Temporal SDKs.

tests/test_serialization_context.py

@@ -5,8 +5,6 @@
 from dataclasses import dataclass, field
 from typing import Any, Optional, Type
 
-from typing_extensions import Self
-
 from temporalio import workflow
 from temporalio.api.common.v1 import Payload
 from temporalio.client import Client
@@ -80,15 +78,6 @@ def __init__(self):
             *DefaultPayloadConverter.default_encoding_payload_converters,
         )
 
-    def with_context(self, context: Optional[SerializationContext]) -> Self:
-        instance = type(self).__new__(type(self))
-        converters = [
-            c.with_context(context) if isinstance(c, WithSerializationContext) else c
-            for c in self.converters.values()
-        ]
-        CompositePayloadConverter.__init__(instance, *converters)
-        return instance
-
 
 data_converter = dataclasses.replace(
     DataConverter.default,