uncertainty-engine-types 0.14.0__py3-none-any.whl → 0.15.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,39 @@ 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.uncertainty_plot import UncertaintyPlot
+from uncertainty_engine_types.vector_store import VectorStoreConfig, VectorStoreProvider
+from uncertainty_engine_types.version import __version__
 
 __all__ = [
     "__version__",
@@ -88,4 +102,7 @@ __all__ = [
     "VectorStoreConfig",
     "VectorStoreProvider",
     "WebPage",
+    "OverrideWorkflowInput",
+    "OverrideWorkflowOutput",
+    "RunWorkflowRequest",
 ]

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/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.14.0
+Version: 0.15.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.15.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=lUVOcy0DDYPJKaVz_iAqVCkywP372PZrQbhvBqoIfBk,2812
 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,14 @@ 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/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=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,,