dbos 0.25.0a8__tar.gz → 0.25.0a9__tar.gz

{dbos-0.25.0a8 → dbos-0.25.0a9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbos
-Version: 0.25.0a8
+Version: 0.25.0a9
 Summary: Ultra-lightweight durable execution in Python
 Author-Email: "DBOS, Inc." <contact@dbos.dev>
 License: MIT
@@ -23,6 +23,7 @@ Requires-Dist: cryptography>=43.0.3
 Requires-Dist: rich>=13.9.4
 Requires-Dist: pyjwt>=2.10.1
 Requires-Dist: websockets>=15.0
+Requires-Dist: pyright>=1.1.398
 Description-Content-Type: text/markdown
 
 

{dbos-0.25.0a8 → dbos-0.25.0a9}/dbos/_core.py

@@ -81,7 +81,6 @@ from ._sys_db import (
 if TYPE_CHECKING:
     from ._dbos import (
         DBOS,
-        Workflow,
         WorkflowHandle,
         WorkflowHandleAsync,
         DBOSRegistry,
@@ -304,7 +303,7 @@ def _get_wf_invoke_func(
 def _execute_workflow_wthread(
     dbos: "DBOS",
     status: WorkflowStatusInternal,
-    func: "Workflow[P, R]",
+    func: "Callable[P, R]",
     ctx: DBOSContext,
     *args: Any,
     **kwargs: Any,
@@ -335,7 +334,7 @@ def _execute_workflow_wthread(
 async def _execute_workflow_async(
     dbos: "DBOS",
     status: WorkflowStatusInternal,
-    func: "Workflow[P, Coroutine[Any, Any, R]]",
+    func: "Callable[P, Coroutine[Any, Any, R]]",
     ctx: DBOSContext,
     *args: Any,
     **kwargs: Any,
@@ -449,7 +448,7 @@ def _get_new_wf() -> tuple[str, DBOSContext]:
 
 def start_workflow(
     dbos: "DBOS",
-    func: "Workflow[P, Union[R, Coroutine[Any, Any, R]]]",
+    func: "Callable[P, Union[R, Coroutine[Any, Any, R]]]",
     queue_name: Optional[str],
     execute_workflow: bool,
     *args: P.args,
@@ -531,7 +530,7 @@ def start_workflow(
 
 async def start_workflow_async(
     dbos: "DBOS",
-    func: "Workflow[P, Coroutine[Any, Any, R]]",
+    func: "Callable[P, Coroutine[Any, Any, R]]",
     queue_name: Optional[str],
     execute_workflow: bool,
     *args: P.args,

{dbos-0.25.0a8 → dbos-0.25.0a9}/dbos/_dbos.py

@@ -126,17 +126,6 @@ R = TypeVar("R", covariant=True)  # A generic type for workflow return values
 
 T = TypeVar("T")
 
-
-class DBOSCallProtocol(Protocol[P, R]):
-    __name__: str
-    __qualname__: str
-
-    def __call__(*args: P.args, **kwargs: P.kwargs) -> R: ...
-
-
-Workflow: TypeAlias = DBOSCallProtocol[P, R]
-
-
 IsolationLevel = Literal[
     "SERIALIZABLE",
     "REPEATABLE READ",
@@ -169,7 +158,7 @@ RegisteredJob = Tuple[
 
 class DBOSRegistry:
     def __init__(self) -> None:
-        self.workflow_info_map: dict[str, Workflow[..., Any]] = {}
+        self.workflow_info_map: dict[str, Callable[..., Any]] = {}
         self.function_type_map: dict[str, str] = {}
         self.class_info_map: dict[str, type] = {}
         self.instance_info_map: dict[str, object] = {}
@@ -713,7 +702,7 @@ class DBOS:
     @classmethod
     def start_workflow(
         cls,
-        func: Workflow[P, R],
+        func: Callable[P, R],
         *args: P.args,
         **kwargs: P.kwargs,
     ) -> WorkflowHandle[R]:
@@ -723,7 +712,7 @@ class DBOS:
     @classmethod
     async def start_workflow_async(
         cls,
-        func: Workflow[P, Coroutine[Any, Any, R]],
+        func: Callable[P, Coroutine[Any, Any, R]],
         *args: P.args,
         **kwargs: P.kwargs,
     ) -> WorkflowHandleAsync[R]:

{dbos-0.25.0a8 → dbos-0.25.0a9}/dbos/_queue.py

@@ -1,6 +1,6 @@
 import threading
 import traceback
-from typing import TYPE_CHECKING, Any, Coroutine, Optional, TypedDict
+from typing import TYPE_CHECKING, Any, Callable, Coroutine, Optional, TypedDict
 
 from psycopg import errors
 from sqlalchemy.exc import OperationalError
@@ -10,7 +10,7 @@ from dbos._utils import GlobalParams
 from ._core import P, R, execute_workflow_by_id, start_workflow, start_workflow_async
 
 if TYPE_CHECKING:
-    from ._dbos import DBOS, Workflow, WorkflowHandle, WorkflowHandleAsync
+    from ._dbos import DBOS, WorkflowHandle, WorkflowHandleAsync
 
 
 class QueueRateLimit(TypedDict):
@@ -59,7 +59,7 @@ class Queue:
         registry.queue_info_map[self.name] = self
 
     def enqueue(
-        self, func: "Workflow[P, R]", *args: P.args, **kwargs: P.kwargs
+        self, func: "Callable[P, R]", *args: P.args, **kwargs: P.kwargs
     ) -> "WorkflowHandle[R]":
         from ._dbos import _get_dbos_instance
 
@@ -68,7 +68,7 @@ class Queue:
 
     async def enqueue_async(
         self,
-        func: "Workflow[P, Coroutine[Any, Any, R]]",
+        func: "Callable[P, Coroutine[Any, Any, R]]",
         *args: P.args,
         **kwargs: P.kwargs,
     ) -> "WorkflowHandleAsync[R]":

{dbos-0.25.0a8 → dbos-0.25.0a9}/pyproject.toml

@@ -24,10 +24,11 @@ dependencies = [
     "rich>=13.9.4",
     "pyjwt>=2.10.1",
     "websockets>=15.0",
+    "pyright>=1.1.398",
 ]
 requires-python = ">=3.9"
 readme = "README.md"
-version = "0.25.0a8"
+version = "0.25.0a9"
 
 [project.license]
 text = "MIT"
@@ -86,4 +87,5 @@ dev = [
     "pytest-order>=1.3.0",
     "pdm-backend>=2.4.2",
     "pytest-asyncio>=0.25.0",
+    "pyright>=1.1.398",
 ]

{dbos-0.25.0a8 → dbos-0.25.0a9}/tests/test_classdecorators.py

@@ -419,7 +419,7 @@ def test_class_recovery(dbos: DBOS) -> None:
 def test_inst_recovery(dbos: DBOS) -> None:
     wfid = str(uuid.uuid4())
     exc_cnt: int = 0
-    last_inst: Optional[TestClass] = None
+    last_inst: Optional["TestClass"] = None
 
     @DBOS.dbos_class()
     class TestClass(DBOSConfiguredInstance):

{dbos-0.25.0a8 → dbos-0.25.0a9}/tests/test_fastapi.py

@@ -172,7 +172,7 @@ async def test_custom_lifespan(
     resource = None
     port = 8000
 
-    @asynccontextmanager
+    @asynccontextmanager  # pyright: ignore
     async def lifespan(app: FastAPI) -> Any:
         nonlocal resource
         resource = 1

{dbos-0.25.0a8 → dbos-0.25.0a9}/tests/test_fastapi_roles.py

@@ -308,7 +308,9 @@ def test_role_tracing() -> None:
     provider.add_span_processor(span_processor)
 
     def function_to_trace() -> None:
-        with provider.get_tracer(__name__).start_as_current_span("test-span") as span:
+        with provider.get_tracer(__name__).start_as_current_span(  # pyright: ignore
+            "test-span"
+        ) as span:
             span.set_attribute("testattribute", "value")
 
     # Clear any existing spans

{dbos-0.25.0a8 → dbos-0.25.0a9}/tests/test_workflow_cancel.py

@@ -71,8 +71,8 @@ def test_two_steps_cancel(dbos: DBOS, config: ConfigFile) -> None:
         return
 
     # run the workflow
+    wfuuid = str(uuid.uuid4())
     try:
-        wfuuid = str(uuid.uuid4())
         with SetWorkflowID(wfuuid):
             simple_workflow()
 
@@ -116,8 +116,8 @@ def test_two_transactions_cancel(dbos: DBOS, config: ConfigFile) -> None:
         return
 
     # run the workflow
+    wfuuid = str(uuid.uuid4())
     try:
-        wfuuid = str(uuid.uuid4())
         with SetWorkflowID(wfuuid):
             simple_workflow()