uncertainty-engine-types 0.7.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

uncertainty_engine_types/context.py

@@ -1,19 +1,40 @@
-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):
     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/version.py

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

{uncertainty_engine_types-0.7.0.dist-info → uncertainty_engine_types-0.9.0.dist-info}/METADATA

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

{uncertainty_engine_types-0.7.0.dist-info → uncertainty_engine_types-0.9.0.dist-info}/RECORD

@@ -1,8 +1,7 @@
 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=1iDcfMLi_fKh_YVUAKif2NOdrA_vuXOQvBZKbF4uJZk,317
+uncertainty_engine_types/context.py,sha256=gi1izptPe05ksem5TGZNIzgzO_QmomzXwTGqvB5I9Sc,914
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
-uncertainty_engine_types/distributions.py,sha256=k9T1PmLFYJUSEYeml8vrM48vdF3Yio4679_E9twdZS4,198
 uncertainty_engine_types/embeddings.py,sha256=sNDWjpuqmzOefKyQX--9EpdeTEyI4kQeSH4IELVUT7g,982
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
 uncertainty_engine_types/file.py,sha256=J202bhOIVnvvvAG54sXyYF8fwjZGTUP3097ARnRnb3Q,579
@@ -22,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=RaANGbRu5e-vehwXI1-Qe2ggPPfs1TQaZj072JdbLk4,22
-uncertainty_engine_types-0.7.0.dist-info/METADATA,sha256=0t50EYJpW4a2LeueibBBfQcX2hpJ0r4t2he5TMkVAFg,2815
-uncertainty_engine_types-0.7.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-uncertainty_engine_types-0.7.0.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=H9NWRZb7NbeRRPLP_V1fARmLNXranorVM-OOY-8_2ug,22
+uncertainty_engine_types-0.9.0.dist-info/METADATA,sha256=NQnQZh0dJ5ntXnaAO8azV2gg0yKsPHoLUVAOIbLPF3U,2815
+uncertainty_engine_types-0.9.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+uncertainty_engine_types-0.9.0.dist-info/RECORD,,

uncertainty_engine_types/distributions.py

@@ -1,11 +0,0 @@
-from pydantic import BaseModel
-
-
-class Distribution(BaseModel):
-    type: str
-    parameters: dict[str, float]
-
-
-class ParameterDistribution(BaseModel):
-    name: str
-    distribution: Distribution