uncertainty-engine-types 0.15.0__py3-none-any.whl → 0.17.0__py3-none-any.whl

uncertainty_engine_types/__init__.py

@@ -50,6 +50,7 @@ from uncertainty_engine_types.run_workflow import (
 from uncertainty_engine_types.sensor_designer import SensorDesigner
 from uncertainty_engine_types.sql import SQLConfig, SQLKind
 from uncertainty_engine_types.token import Token
+from uncertainty_engine_types.tool_metadata import ToolMetadata
 from uncertainty_engine_types.uncertainty_plot import UncertaintyPlot
 from uncertainty_engine_types.vector_store import VectorStoreConfig, VectorStoreProvider
 from uncertainty_engine_types.version import __version__
@@ -105,4 +106,5 @@ __all__ = [
     "OverrideWorkflowInput",
     "OverrideWorkflowOutput",
     "RunWorkflowRequest",
+    "ToolMetadata",
 ]

uncertainty_engine_types/node_info.py

@@ -21,9 +21,9 @@ class NodeOutputInfo(BaseModel):
 
 
 class NodeRequirementsInfo(BaseModel):
-    cpu: int
-    gpu: bool
-    memory: int
+    cpu: int | None = None
+    gpu: bool | None = None
+    memory: int | None = None
     timeout: int
 
 

uncertainty_engine_types/tool_metadata.py

@@ -0,0 +1,41 @@
+from pydantic import BaseModel, Field
+
+from uncertainty_engine_types import NodeInputInfo, NodeOutputInfo
+
+NodeId = str
+"""Unique ID of a Uncertainty Engine node. Eg `label` field in the SDK"""
+HandleLabel = str
+"""Unique ID of a handle to a given node. Eg `input_variance`"""
+
+
+class ToolMetadata(BaseModel):
+    """Tool metadata."""
+
+    inputs: dict[NodeId, dict[HandleLabel, NodeInputInfo]] = Field(default_factory=dict)
+    """Defines which inputs on a workflow can be used as Tool Inputs"""
+    outputs: dict[NodeId, dict[HandleLabel, NodeOutputInfo]] = Field(
+        default_factory=dict
+    )
+    """Defines which outputs on a workflow can be used as Tool outputs"""
+
+    def is_empty(self) -> bool:
+        """Check if the metadata is completely empty"""
+        return not self.inputs and not self.outputs
+
+    def has_partial_data(self) -> bool:
+        """Check if only inputs or only outputs are defined"""
+        return bool(self.inputs) != bool(self.outputs)
+
+    def validate_complete(self) -> None:
+        """
+        Validate that tool metadata is complete (has both inputs and outputs).
+
+        Raises:
+            ValueError: If metadata has only inputs or only outputs
+        """
+        if self.has_partial_data():
+            raise ValueError(
+                "Tool metadata must have both inputs AND outputs defined. "
+                f"Currently has: inputs={'yes' if self.inputs else 'no'}, "
+                f"outputs={'yes' if self.outputs else 'no'}"
+            )

uncertainty_engine_types/version.py

@@ -1 +1 @@
-__version__ = "0.15.0"
+__version__ = "0.17.0"

{uncertainty_engine_types-0.15.0.dist-info → uncertainty_engine_types-0.17.0.dist-info}/METADATA

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

{uncertainty_engine_types-0.15.0.dist-info → uncertainty_engine_types-0.17.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-uncertainty_engine_types/__init__.py,sha256=lUVOcy0DDYPJKaVz_iAqVCkywP372PZrQbhvBqoIfBk,2812
+uncertainty_engine_types/__init__.py,sha256=BFXURIPv_EoF-NMxVB_Pr0smVr4Z0U4_bYttp7owfLQ,2896
 uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
 uncertainty_engine_types/context.py,sha256=gi1izptPe05ksem5TGZNIzgzO_QmomzXwTGqvB5I9Sc,914
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
@@ -13,16 +13,17 @@ 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=OEhn7vPjEgLtHMb8mqW40dP9ctFdWpXOSM9CTD9NiNk,2021
+uncertainty_engine_types/node_info.py,sha256=LbNXMKmPCNQoDWVSUlC2mpu3KOsMeG6WUy8pBBWFtT8,2063
 uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
 uncertainty_engine_types/run_workflow.py,sha256=bVKuP17uR8cToWkakoGSEpyYpa_0v7s4FFiL2xLZawg,564
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
 uncertainty_engine_types/sql.py,sha256=SBzmgMEZ-sa-OBRoZbmKiOqddop4zJixgxx-JCus9fY,298
 uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBXEpBY,178
+uncertainty_engine_types/tool_metadata.py,sha256=2pBTbMKhizcLMH7CtSCnAcS50MT61iwEcSRqMSTBAVw,1499
 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=wGIgxINRfcIKyk0LjIbc9UF9UwuclyCQZv_axTUzwNw,23
-uncertainty_engine_types-0.15.0.dist-info/METADATA,sha256=jDJIIGuyIos_pJ19oEwITh7-_0ZaeDX8XYwE6nGmO7g,2816
-uncertainty_engine_types-0.15.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-uncertainty_engine_types-0.15.0.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=XpM3lncCzPBIwMSvDN7i10pke_c6KdJtVLZYbCiaTRw,23
+uncertainty_engine_types-0.17.0.dist-info/METADATA,sha256=N3e3oxAmKB6-Y4ASkGuxsJIUCUr_jLuFD0O5CYWlqEY,2816
+uncertainty_engine_types-0.17.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+uncertainty_engine_types-0.17.0.dist-info/RECORD,,