uncertainty-engine-types 0.8.0__tar.gz → 0.10.0__tar.gz

{uncertainty_engine_types-0.8.0 → uncertainty_engine_types-0.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.8.0
+Version: 0.10.0
 Summary: Common type definitions for the Uncertainty Engine
 Author: Freddy Wordingham
 Author-email: freddy@digilab.ai

{uncertainty_engine_types-0.8.0 → uncertainty_engine_types-0.10.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uncertainty-engine-types"
-version = "0.8.0"
+version = "0.10.0"
 description = "Common type definitions for the Uncertainty Engine"
 authors = [
     { name = "Freddy Wordingham", email = "freddy@digilab.ai" },

uncertainty_engine_types-0.10.0/uncertainty_engine_types/context.py

@@ -0,0 +1,40 @@
+from pydantic import BaseModel, ConfigDict
+
+from uncertainty_engine_types.node_info import NodeInfo
+
+
+class UserContext(BaseModel):
+    email: str
+    project_id: str
+    cost_code: str
+    user_id: str | None = None
+
+
+class Context(BaseModel):
+    """The context around an Uncertainty Engine node execution."""
+
+    sync: bool
+    """Whether to run the node synchronously."""
+
+    job_id: str
+    """The node execution job ID."""
+
+    queue_url: str
+    """The node queue URL."""
+
+    cache_url: str
+    """The node cache URL."""
+
+    timeout: int
+    """The node timeout in seconds."""
+
+    nodes: dict[str, NodeInfo]
+    """Dictionary of nodes and their runtime details."""
+
+    user: UserContext
+    """The context around the user executing the node."""
+
+    is_root: bool = False
+    """Indicates whether a node is the root node of a workflow."""
+
+    model_config = ConfigDict(use_attribute_docstrings=True)

{uncertainty_engine_types-0.8.0 → uncertainty_engine_types-0.10.0}/uncertainty_engine_types/node_info.py

@@ -54,6 +54,11 @@ class NodeInfo(BaseModel, extra="allow"):
 
     load_balancer_url: Optional[str] = None
     queue_url: Optional[str] = None
+    service_arn: Optional[str] = None
+    """
+    Service ARN.
+    """
+
     cache_url: Optional[str] = None
     version_types_lib: str = __version__
     version_base_image: int

uncertainty_engine_types-0.10.0/uncertainty_engine_types/version.py

@@ -0,0 +1 @@
+__version__ = "0.9.0"

uncertainty_engine_types-0.8.0/uncertainty_engine_types/context.py

@@ -1,20 +0,0 @@
-from pydantic import BaseModel
-
-from .node_info import NodeInfo
-
-
-class UserContext(BaseModel):
-    email: str
-    project_id: str
-    cost_code: str
-    user_id: str | None = None
-
-
-class Context(BaseModel):
-    sync: bool
-    job_id: str
-    queue_url: str
-    cache_url: str
-    timeout: int
-    nodes: dict[str, NodeInfo]
-    user: UserContext

uncertainty_engine_types-0.8.0/uncertainty_engine_types/version.py

@@ -1 +0,0 @@
-__version__ = "0.8.0"