uncertainty-engine-types 0.14.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

uncertainty_engine_types/__init__.py

@@ -1,10 +1,13 @@
-from . import utils
-from .chat_history import ChatHistory
-from .context import Context, UserContext
-from .dataset import CSVDataset
-from .embeddings import TextEmbeddingsConfig, TextEmbeddingsProvider
-from .execution_error import ExecutionError
-from .file import (
+from uncertainty_engine_types import utils
+from uncertainty_engine_types.chat_history import ChatHistory
+from uncertainty_engine_types.context import Context, UserContext
+from uncertainty_engine_types.dataset import CSVDataset
+from uncertainty_engine_types.embeddings import (
+    TextEmbeddingsConfig,
+    TextEmbeddingsProvider,
+)
+from uncertainty_engine_types.execution_error import ExecutionError
+from uncertainty_engine_types.file import (
     PDF,
     Document,
     File,
@@ -17,28 +20,40 @@ from .file import (
     TabularData,
     WebPage,
 )
-from .graph import Graph, NodeElement, NodeId, SourceHandle, TargetHandle
-from .handle import Handle
-from .id import ResourceID
-from .job import JobInfo, JobStatus
-from .llm import LLMConfig, LLMProvider
-from .message import Message
-from .model import MachineLearningModel
-from .model_config import ModelConfig
-from .node_info import (
+from uncertainty_engine_types.graph import (
+    Graph,
+    NodeElement,
+    NodeId,
+    SourceHandle,
+    TargetHandle,
+)
+from uncertainty_engine_types.handle import Handle
+from uncertainty_engine_types.id import ResourceID
+from uncertainty_engine_types.job import JobInfo, JobStatus
+from uncertainty_engine_types.llm import LLMConfig, LLMProvider
+from uncertainty_engine_types.message import Message
+from uncertainty_engine_types.model import MachineLearningModel
+from uncertainty_engine_types.model_config import ModelConfig
+from uncertainty_engine_types.node_info import (
     NodeInfo,
     NodeInputInfo,
     NodeOutputInfo,
     NodeRequirementsInfo,
     ScalingInfo,
 )
-from .prompt import Prompt
-from .sensor_designer import SensorDesigner
-from .sql import SQLConfig, SQLKind
-from .token import Token
-from .uncertainty_plot import UncertaintyPlot
-from .vector_store import VectorStoreConfig, VectorStoreProvider
-from .version import __version__
+from uncertainty_engine_types.prompt import Prompt
+from uncertainty_engine_types.run_workflow import (
+    OverrideWorkflowInput,
+    OverrideWorkflowOutput,
+    RunWorkflowRequest,
+)
+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__
 
 __all__ = [
     "__version__",
@@ -88,4 +103,8 @@ __all__ = [
     "VectorStoreConfig",
     "VectorStoreProvider",
     "WebPage",
+    "OverrideWorkflowInput",
+    "OverrideWorkflowOutput",
+    "RunWorkflowRequest",
+    "ToolMetadata",
 ]

uncertainty_engine_types/run_workflow.py

@@ -0,0 +1,21 @@
+# Types for running a workflow via `run_workflow` or `queue_workflow` endpoints
+from typing import Any, List, Optional
+
+from pydantic import BaseModel
+
+
+class OverrideWorkflowInput(BaseModel):
+    node_label: str
+    input_handle: str
+    value: Any  # Required as any to allow for input to all nodes
+
+
+class OverrideWorkflowOutput(BaseModel):
+    node_label: str
+    output_handle: str
+    output_label: str
+
+
+class RunWorkflowRequest(BaseModel):
+    inputs: Optional[List[OverrideWorkflowInput]] = None
+    outputs: Optional[List[OverrideWorkflowOutput]] = None

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.14.0"
+__version__ = "0.16.0"

{uncertainty_engine_types-0.14.0.dist-info → uncertainty_engine_types-0.16.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.14.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.14.0.dist-info → uncertainty_engine_types-0.16.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-uncertainty_engine_types/__init__.py,sha256=0FAX_KGWHoFlMgPmIUEqR5YfTSr7ffvdIt3I_O-1ETk,2005
+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
@@ -15,13 +15,15 @@ uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk
 uncertainty_engine_types/model_config.py,sha256=pCawKIo2l0O3eRLE5Ese7jaxomIyPXGW3XIcAx6TckY,558
 uncertainty_engine_types/node_info.py,sha256=OEhn7vPjEgLtHMb8mqW40dP9ctFdWpXOSM9CTD9NiNk,2021
 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=EVkhkQ4Bbkx92jXuCmt2_JDjQExgnrq2d3eBQIP9UAo,23
-uncertainty_engine_types-0.14.0.dist-info/METADATA,sha256=m6eRQCLVYmasBRAX_iXdUxrFCXjxPZvupKyTCqsZv7U,2816
-uncertainty_engine_types-0.14.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-uncertainty_engine_types-0.14.0.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=3Msc5baw88UJubVj5AVFB8tExkT2OFIsNWe2leaoHhc,23
+uncertainty_engine_types-0.16.0.dist-info/METADATA,sha256=2Aglv8t0UPYKin33Bly5Z-4DRqQihPtCj9qwKm4v4Go,2816
+uncertainty_engine_types-0.16.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+uncertainty_engine_types-0.16.0.dist-info/RECORD,,