uncertainty-engine-types 0.8.0__py3-none-any.whl → 0.10.0__py3-none-any.whl

uncertainty_engine_types/context.py

@@ -1,6 +1,6 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
-from .node_info import NodeInfo
+from uncertainty_engine_types.node_info import NodeInfo
 
 
 class UserContext(BaseModel):
@@ -11,10 +11,30 @@ class UserContext(BaseModel):
 
 
 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/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/version.py

@@ -1 +1 @@
-__version__ = "0.8.0"
+__version__ = "0.9.0"

{uncertainty_engine_types-0.8.0.dist-info → uncertainty_engine_types-0.10.0.dist-info}/METADATA

@@ -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.dist-info → uncertainty_engine_types-0.10.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 uncertainty_engine_types/__init__.py,sha256=QxuxItecFrZwOKNgPrdIO7i-YnEH3QAT3__86eTqSV8,1948
 uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
-uncertainty_engine_types/context.py,sha256=73svFiBdxFgoDb2XMIl1u0qL2bYTZXBfV3ec7Swdj7k,348
+uncertainty_engine_types/context.py,sha256=gi1izptPe05ksem5TGZNIzgzO_QmomzXwTGqvB5I9Sc,914
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
 uncertainty_engine_types/embeddings.py,sha256=sNDWjpuqmzOefKyQX--9EpdeTEyI4kQeSH4IELVUT7g,982
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
@@ -13,7 +13,7 @@ uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmn
 uncertainty_engine_types/message.py,sha256=sX1tq_BbZC9r9-Vf0A3WHiOFRE2hAAlY-7rD52kbsoU,381
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
 uncertainty_engine_types/model_config.py,sha256=pCawKIo2l0O3eRLE5Ese7jaxomIyPXGW3XIcAx6TckY,558
-uncertainty_engine_types/node_info.py,sha256=XHONP6rNbAcLOHFxi-NPl1-UrKAH9P8C1yUr_rDiyCg,1362
+uncertainty_engine_types/node_info.py,sha256=ifSEkDVcS_B38jXx52nedU-wXe2GHD1Yt8Fqv0V3Ix0,1434
 uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
 uncertainty_engine_types/sql.py,sha256=SBzmgMEZ-sa-OBRoZbmKiOqddop4zJixgxx-JCus9fY,298
@@ -21,7 +21,7 @@ uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBX
 uncertainty_engine_types/uncertainty_plot.py,sha256=kJr0SuJ6JeTxaf2adpDNWqx7vVLZRpKG8tFbdx90pas,547
 uncertainty_engine_types/utils.py,sha256=72QVig8Kb5uIR-e1nofm-3x9CouebdQJIruDbq-aIn0,271
 uncertainty_engine_types/vector_store.py,sha256=9fYPJ04jWcy2DruyUSjiKQAgmqq-wgeAi5dBIrAOm30,392
-uncertainty_engine_types/version.py,sha256=iPlYCcIzuzW7T2HKDkmYlMkRI51dBLfNRxPPiWrfw9U,22
-uncertainty_engine_types-0.8.0.dist-info/METADATA,sha256=Js05bSFMWLe9r92idYdPQrodn1bPKPQbq53-4nOyREQ,2815
-uncertainty_engine_types-0.8.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-uncertainty_engine_types-0.8.0.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=H9NWRZb7NbeRRPLP_V1fARmLNXranorVM-OOY-8_2ug,22
+uncertainty_engine_types-0.10.0.dist-info/METADATA,sha256=1GK1FkKrMzPgLQAI78_Oqbjxcb3yxNHH24BtGdEjatM,2816
+uncertainty_engine_types-0.10.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+uncertainty_engine_types-0.10.0.dist-info/RECORD,,