durabletask 1.1.0.dev1__tar.gz → 1.1.0.dev2__tar.gz

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 1.1.0.dev1
+Version: 1.1.0.dev2
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/durabletask/entities/entity_instance_id.py

@@ -1,6 +1,3 @@
-from typing import Optional
-
-
 class EntityInstanceId:
     def __init__(self, entity: str, key: str):
         self.entity = entity
@@ -20,7 +17,7 @@ class EntityInstanceId:
         return str(self) < str(other)
 
     @staticmethod
-    def parse(entity_id: str) -> Optional["EntityInstanceId"]:
+    def parse(entity_id: str) -> "EntityInstanceId":
         """Parse a string representation of an entity ID into an EntityInstanceId object.
 
         Parameters
@@ -30,8 +27,13 @@ class EntityInstanceId:
 
         Returns
         -------
-        Optional[EntityInstanceId]
-            The parsed EntityInstanceId object, or None if the input is None.
+        EntityInstanceId
+            The parsed EntityInstanceId object.
+
+        Raises
+        ------
+        ValueError
+            If the input string is not in the correct format.
         """
         try:
             _, entity, key = entity_id.split("@", 2)

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/durabletask/entities/entity_metadata.py

@@ -44,8 +44,9 @@ class EntityMetadata:
 
     @staticmethod
     def from_entity_response(entity_response: pb.GetEntityResponse, includes_state: bool):
-        entity_id = EntityInstanceId.parse(entity_response.entity.instanceId)
-        if not entity_id:
+        try:
+            entity_id = EntityInstanceId.parse(entity_response.entity.instanceId)
+        except ValueError:
             raise ValueError("Invalid entity instance ID in entity response.")
         entity_state = None
         if includes_state:

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/durabletask/task.py

@@ -139,7 +139,8 @@ class OrchestrationContext(ABC):
         pass
 
     @abstractmethod
-    def call_entity(self, entity: EntityInstanceId,
+    def call_entity(self,
+                    entity: EntityInstanceId,
                     operation: str,
                     input: Optional[TInput] = None) -> Task:
         """Schedule entity function for execution.

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/durabletask/worker.py

@@ -750,9 +750,10 @@ class TaskHubGrpcWorker:
         for operation in req.operations:
             start_time = datetime.now(timezone.utc)
             executor = _EntityExecutor(self._registry, self._logger)
-            entity_instance_id = EntityInstanceId.parse(instance_id)
-            if not entity_instance_id:
-                raise RuntimeError(f"Invalid entity instance ID '{operation.requestId}' in entity operation request.")
+            try:
+                entity_instance_id = EntityInstanceId.parse(instance_id)
+            except ValueError:
+                raise RuntimeError(f"Invalid entity instance ID '{instance_id}' in entity operation request.")
 
             operation_result = None
 
@@ -1038,14 +1039,14 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
 
     def call_entity(
             self,
-            entity_id: EntityInstanceId,
+            entity: EntityInstanceId,
             operation: str,
             input: Optional[TInput] = None,
     ) -> task.Task:
         id = self.next_sequence_number()
 
         self.call_entity_function_helper(
-            id, entity_id, operation, input=input
+            id, entity, operation, input=input
         )
 
         return self._pending_tasks.get(id, task.CompletableTask())
@@ -1053,13 +1054,13 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
     def signal_entity(
             self,
             entity_id: EntityInstanceId,
-            operation: str,
+            operation_name: str,
             input: Optional[TInput] = None
     ) -> None:
         id = self.next_sequence_number()
 
         self.signal_entity_function_helper(
-            id, entity_id, operation, input
+            id, entity_id, operation_name, input
         )
 
     def lock_entities(self, entities: list[EntityInstanceId]) -> task.Task[EntityLock]:
@@ -1656,8 +1657,9 @@ class _OrchestrationExecutor:
                     raise _get_wrong_action_type_error(
                         entity_call_id, expected_method_name, action
                     )
-                entity_id = EntityInstanceId.parse(event.entityOperationCalled.targetInstanceId.value)
-                if not entity_id:
+                try:
+                    entity_id = EntityInstanceId.parse(event.entityOperationCalled.targetInstanceId.value)
+                except ValueError:
                     raise RuntimeError(f"Could not parse entity ID from targetInstanceId '{event.entityOperationCalled.targetInstanceId.value}'")
                 ctx._entity_task_id_map[event.entityOperationCalled.requestId] = (entity_id, entity_call_id)
             elif event.HasField("entityOperationSignaled"):

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/durabletask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 1.1.0.dev1
+Version: 1.1.0.dev2
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/durabletask.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ README.md
 pyproject.toml
 durabletask/__init__.py
 durabletask/client.py
+durabletask/py.typed
 durabletask/task.py
 durabletask/worker.py
 durabletask.egg-info/PKG-INFO

{durabletask-1.1.0.dev1 → durabletask-1.1.0.dev2}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durabletask"
-version = "1.1.0.dev1"
+version = "1.1.0.dev2"
 description = "A Durable Task Client SDK for Python"
 keywords = [
     "durable",