Add additional imports to __init__.py

restatedev · Jul 18, 2024 · 3bc51f9 · 3bc51f9
1 parent 0e90dd1
commit 3bc51f9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/examples/example.py b/examples/example.py
@@ -14,11 +14,9 @@
 
 
 from datetime import timedelta
-from restate.service import Service
-from restate.context import Context, ObjectContext, ObjectSharedContext
-from restate.context import WorkflowContext, WorkflowSharedContext
-from restate.object import VirtualObject
-from restate.workflow import Workflow
+from restate import Service, Context
+from restate import VirtualObject, ObjectContext, ObjectSharedContext
+from restate import Workflow, WorkflowContext, WorkflowSharedContext
 import restate
 
 greeter = Service("greeter")

diff --git a/python/restate/__init__.py b/python/restate/__init__.py
@@ -12,4 +12,13 @@
 Restate SDK for Python
 """
 
+from .service import Service
+from .object import VirtualObject
+from .workflow import Workflow
+
+# types
+from .context import Context, ObjectContext, ObjectSharedContext
+from .context import WorkflowContext, WorkflowSharedContext
+from .context import DurablePromise
+
 from .endpoint import app