Sdk core related updates

temporalio/bridge/client.py

@@ -11,6 +11,7 @@
 
 import google.protobuf.message
 
+import temporalio.bridge.runtime
 import temporalio.bridge.temporal_sdk_bridge
 from temporalio.bridge.temporal_sdk_bridge import RPCError
 
@@ -68,16 +69,30 @@ class Client:
     """RPC client using SDK Core."""
 
     @staticmethod
-    async def connect(config: ClientConfig) -> Client:
+    async def connect(
+        runtime: temporalio.bridge.runtime.Runtime, config: ClientConfig
+    ) -> Client:
         """Establish connection with server."""
         return Client(
-            await temporalio.bridge.temporal_sdk_bridge.connect_client(config)
+            runtime,
+            await temporalio.bridge.temporal_sdk_bridge.connect_client(
+                runtime._ref, config
+            ),
         )
 
-    def __init__(self, ref: temporalio.bridge.temporal_sdk_bridge.ClientRef):
+    def __init__(
+        self,
+        runtime: temporalio.bridge.runtime.Runtime,
+        ref: temporalio.bridge.temporal_sdk_bridge.ClientRef,
+    ):
         """Initialize client with underlying SDK Core reference."""
+        self._runtime = runtime
         self._ref = ref
 
+    def update_metadata(self, metadata: Mapping[str, str]) -> None:
+        """Update underlying metadata on Core client."""
+        self._ref.update_metadata(metadata)
+
     async def call(
         self,
         *,

temporalio/bridge/proto/workflow_commands/__init__.py

@@ -7,6 +7,7 @@
     CompleteWorkflowExecution,
     ContinueAsNewWorkflowExecution,
     FailWorkflowExecution,
+    ModifyWorkflowProperties,
     QueryResult,
     QuerySuccess,
     RequestCancelActivity,
@@ -31,6 +32,7 @@
     "CompleteWorkflowExecution",
     "ContinueAsNewWorkflowExecution",
     "FailWorkflowExecution",
+    "ModifyWorkflowProperties",
     "QueryResult",
     "QuerySuccess",
     "RequestCancelActivity",

temporalio/bridge/proto/workflow_commands/workflow_commands_pb2.pyi

@@ -90,6 +90,7 @@ class WorkflowCommand(google.protobuf.message.Message):
     SCHEDULE_LOCAL_ACTIVITY_FIELD_NUMBER: builtins.int
     REQUEST_CANCEL_LOCAL_ACTIVITY_FIELD_NUMBER: builtins.int
     UPSERT_WORKFLOW_SEARCH_ATTRIBUTES_FIELD_NUMBER: builtins.int
+    MODIFY_WORKFLOW_PROPERTIES_FIELD_NUMBER: builtins.int
     @property
     def start_timer(self) -> global___StartTimer: ...
     @property
@@ -138,6 +139,8 @@ class WorkflowCommand(google.protobuf.message.Message):
     def upsert_workflow_search_attributes(
         self,
     ) -> global___UpsertWorkflowSearchAttributes: ...
+    @property
+    def modify_workflow_properties(self) -> global___ModifyWorkflowProperties: ...
     def __init__(
         self,
         *,
@@ -165,6 +168,7 @@ class WorkflowCommand(google.protobuf.message.Message):
         request_cancel_local_activity: global___RequestCancelLocalActivity | None = ...,
         upsert_workflow_search_attributes: global___UpsertWorkflowSearchAttributes
         | None = ...,
+        modify_workflow_properties: global___ModifyWorkflowProperties | None = ...,
     ) -> None: ...
     def HasField(
         self,
@@ -183,6 +187,8 @@ class WorkflowCommand(google.protobuf.message.Message):
             b"continue_as_new_workflow_execution",
             "fail_workflow_execution",
             b"fail_workflow_execution",
+            "modify_workflow_properties",
+            b"modify_workflow_properties",
             "request_cancel_activity",
             b"request_cancel_activity",
             "request_cancel_external_workflow_execution",
@@ -226,6 +232,8 @@ class WorkflowCommand(google.protobuf.message.Message):
             b"continue_as_new_workflow_execution",
             "fail_workflow_execution",
             b"fail_workflow_execution",
+            "modify_workflow_properties",
+            b"modify_workflow_properties",
             "request_cancel_activity",
             b"request_cancel_activity",
             "request_cancel_external_workflow_execution",
@@ -273,6 +281,7 @@ class WorkflowCommand(google.protobuf.message.Message):
         "schedule_local_activity",
         "request_cancel_local_activity",
         "upsert_workflow_search_attributes",
+        "modify_workflow_properties",
     ] | None: ...
 
 global___WorkflowCommand = WorkflowCommand
@@ -352,7 +361,6 @@ class ScheduleActivity(google.protobuf.message.Message):
     SEQ_FIELD_NUMBER: builtins.int
     ACTIVITY_ID_FIELD_NUMBER: builtins.int
     ACTIVITY_TYPE_FIELD_NUMBER: builtins.int
-    NAMESPACE_FIELD_NUMBER: builtins.int
     TASK_QUEUE_FIELD_NUMBER: builtins.int
     HEADERS_FIELD_NUMBER: builtins.int
     ARGUMENTS_FIELD_NUMBER: builtins.int
@@ -367,7 +375,6 @@ class ScheduleActivity(google.protobuf.message.Message):
     """/ Lang's incremental sequence number, used as the operation identifier"""
     activity_id: builtins.str
     activity_type: builtins.str
-    namespace: builtins.str
     task_queue: builtins.str
     """The name of the task queue to place this activity request in"""
     @property
@@ -422,7 +429,6 @@ class ScheduleActivity(google.protobuf.message.Message):
         seq: builtins.int = ...,
         activity_id: builtins.str = ...,
         activity_type: builtins.str = ...,
-        namespace: builtins.str = ...,
         task_queue: builtins.str = ...,
         headers: collections.abc.Mapping[
             builtins.str, temporalio.api.common.v1.message_pb2.Payload
@@ -472,8 +478,6 @@ class ScheduleActivity(google.protobuf.message.Message):
             b"headers",
             "heartbeat_timeout",
             b"heartbeat_timeout",
-            "namespace",
-            b"namespace",
             "retry_policy",
             b"retry_policy",
             "schedule_to_close_timeout",
@@ -1531,3 +1535,27 @@ class UpsertWorkflowSearchAttributes(google.protobuf.message.Message):
     ) -> None: ...
 
 global___UpsertWorkflowSearchAttributes = UpsertWorkflowSearchAttributes
+
+class ModifyWorkflowProperties(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    UPSERTED_MEMO_FIELD_NUMBER: builtins.int
+    @property
+    def upserted_memo(self) -> temporalio.api.common.v1.message_pb2.Memo:
+        """If set, update the workflow memo with the provided values. The values will be merged with
+        the existing memo. If the user wants to delete values, a default/empty Payload should be
+        used as the value for the key being deleted.
+        """
+    def __init__(
+        self,
+        *,
+        upserted_memo: temporalio.api.common.v1.message_pb2.Memo | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing_extensions.Literal["upserted_memo", b"upserted_memo"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self, field_name: typing_extensions.Literal["upserted_memo", b"upserted_memo"]
+    ) -> None: ...
+
+global___ModifyWorkflowProperties = ModifyWorkflowProperties

temporalio/bridge/runtime.py

@@ -0,0 +1,147 @@
+"""Telemetry for SDK Core. (unstable)
+
+Nothing in this module should be considered stable. The API may change.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import ClassVar, Mapping, Optional
+
+import temporalio.bridge.temporal_sdk_bridge
+
+_default_runtime: Optional[Runtime] = None
+
+
+class Runtime:
+    """Runtime for SDK Core.
+
+    Users are encouraged to use :py:meth:`default`. It can be set with
+    :py:meth:`set_default`.
+    """
+
+    @staticmethod
+    def default() -> Runtime:
+        """Get the default runtime, creating if not already created.
+
+        If the default runtime needs to be different, it should be done with
+        :py:meth:`set_default` before this is called or ever used.
+
+        Returns:
+            The default runtime.
+        """
+        global _default_runtime
+        if not _default_runtime:
+            _default_runtime = Runtime(telemetry=TelemetryConfig())
+        return _default_runtime
+
+    @staticmethod
+    def set_default(runtime: Runtime, *, error_if_already_set: bool = True) -> None:
+        """Set the default runtime to the given runtime.
+
+        This should be called before any Temporal client is created, but can
+        change the existing one. Any clients and workers created with the
+        previous runtime will stay on that runtime.
+
+        Args:
+            runtime: The runtime to set.
+            error_if_already_set: If True and default is already set, this will
+                raise a RuntimeError.
+        """
+        global _default_runtime
+        if _default_runtime and error_if_already_set:
+            raise RuntimeError("Runtime default already set")
+        _default_runtime = runtime
+
+    def __init__(self, *, telemetry: TelemetryConfig) -> None:
+        """Create a default runtime with the given telemetry config."""
+        self._ref = temporalio.bridge.temporal_sdk_bridge.init_runtime(telemetry)
+
+
+def format_filter(core_level: str, other_level: str) -> str:
+    """Helper to build a filter from Core and other level.
+
+    Levels can be ``ERROR``, ``WARN``, ``INFO``, ``DEBUG``, or ``TRACE``.
+
+    Args:
+        core_level: Level for SDK Core.
+        other_level: Level for other things besides Core.
+
+    Returns:
+        Formatted string for use as a ``filter`` in telemetry configs.
+    """
+    return f"{other_level},temporal_sdk_core={core_level},temporal_client={core_level},temporal_sdk={core_level}"
+
+
+@dataclass(frozen=True)
+class TracingConfig:
+    """Configuration for Core tracing."""
+
+    filter: str
+    """Filter string for tracing. Use :py:func:`format_filter`."""
+
+    opentelemetry: OpenTelemetryConfig
+    """Configuration for OpenTelemetry tracing collector."""
+
+
+@dataclass(frozen=True)
+class LoggingConfig:
+    """Configuration for Core logging."""
+
+    filter: str
+    """Filter string for logging. Use :py:func:`format_filter`."""
+
+    forward: bool = False
+    """If true, logs are not on console but instead forwarded."""
+
+    default: ClassVar[LoggingConfig]
+    """Default logging configuration of Core WARN level and other ERROR
+    level.
+    """
+
+
+LoggingConfig.default = LoggingConfig(filter=format_filter("WARN", "ERROR"))
+
+
+@dataclass(frozen=True)
+class MetricsConfig:
+    """Configuration for Core metrics.
+
+    One and only one of :py:attr:`opentelemetry` or :py:attr:`prometheus` must
+    be set.
+    """
+
+    opentelemetry: Optional[OpenTelemetryConfig] = None
+    """Configuration for OpenTelemetry metrics collector."""
+
+    prometheus: Optional[PrometheusConfig] = None
+    """Configuration for Prometheus metrics endpoint."""
+
+
+@dataclass(frozen=True)
+class OpenTelemetryConfig:
+    """Configuration for OpenTelemetry collector."""
+
+    url: str
+    headers: Mapping[str, str]
+
+
+@dataclass(frozen=True)
+class PrometheusConfig:
+    """Configuration for Prometheus metrics endpoint."""
+
+    bind_address: str
+
+
+@dataclass(frozen=True)
+class TelemetryConfig:
+    """Configuration for Core telemetry."""
+
+    tracing: Optional[TracingConfig] = None
+    """Tracing configuration."""
+
+    logging: Optional[LoggingConfig] = LoggingConfig.default
+    """Logging configuration."""
+
+    metrics: Optional[PrometheusConfig] = None
+    """Metrics configuration."""