uncertainty-engine-types 0.15.0__tar.gz → 0.16.0__tar.gz

{uncertainty_engine_types-0.15.0 → uncertainty_engine_types-0.16.0}/PKG-INFO

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

{uncertainty_engine_types-0.15.0 → uncertainty_engine_types-0.16.0}/pyproject.toml

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

{uncertainty_engine_types-0.15.0 → uncertainty_engine_types-0.16.0}/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-0.16.0/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-0.16.0/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.15.0/uncertainty_engine_types/version.py

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