Worker package refactor (#15)

Author: cretz

docs/conf.py

@@ -66,6 +66,8 @@
 
 autodoc_typehints_description_target = "documented"
 
+autodoc_typehints_format = "short"
+
 autodoc_preserve_defaults = True
 
 autodoc_member_order = "bysource"

poetry.lock

@@ -42,6 +42,7 @@ mypy-protobuf = "^3.2.0"
 pydocstyle = "^6.1.1"
 pytest = "^6.2.5"
 pytest-asyncio = "^0.18.0"
+pytest-timeout = "^2.1.0"
 setuptools = "^60.9.3"
 setuptools-rust = "^1.1.2"
 Sphinx = "^4.4.0"
@@ -84,6 +85,8 @@ asyncio_mode = "auto"
 log_cli = true
 log_cli_level = "INFO"
 log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
+timeout = 60
+timeout_func_only = true
 
 [tool.isort]
 profile = "black"

pyproject.toml

@@ -22,7 +22,7 @@
     overload,
 )
 
-import typing_extensions
+from typing_extensions import TypedDict
 
 import temporalio.api.common.v1
 import temporalio.api.enums.v1
@@ -359,7 +359,7 @@ def get_activity_completion_handle(
         raise ValueError("Task token or workflow/run/activity ID must be present")
 
 
-class ClientConfig(typing_extensions.TypedDict, total=False):
+class ClientConfig(TypedDict, total=False):
     """TypedDict of config originally passed to :py:meth:`Client`."""
 
     service: temporalio.workflow_service.WorkflowService

temporalio/client.py

@@ -0,0 +1,21 @@
+"""Worker for processing Temporal workflows and/or activities."""
+
+from .activity import SharedHeartbeatSender, SharedStateManager
+from .interceptor import (
+    ActivityInboundInterceptor,
+    ActivityOutboundInterceptor,
+    ExecuteActivityInput,
+    Interceptor,
+)
+from .worker import Worker, WorkerConfig
+
+__all__ = [
+    "Worker",
+    "WorkerConfig",
+    "ExecuteActivityInput",
+    "Interceptor",
+    "ActivityInboundInterceptor",
+    "ActivityOutboundInterceptor",
+    "SharedStateManager",
+    "SharedHeartbeatSender",
+]