uncertainty-engine-types 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

uncertainty_engine_types/__init__.py

@@ -1,16 +1,29 @@
+from . import utils
 from .context import Context
 from .conversation import Conversation
 from .execution_error import ExecutionError
+from .file import (
+    Document,
+    File,
+    FileLocation,
+    Image,
+    LocalStorage,
+    Mesh,
+    PDF,
+    S3Storage,
+    SQLTable,
+    TabularData,
+    WebPage,
+)
 from .graph import Graph, NodeElement, NodeId, SourceHandle, TargetHandle
 from .handle import Handle
 from .job import JobInfo, JobStatus
 from .llm import LLMConfig, LLMProvider
 from .message import Message
 from .model import MachineLearningModel
-from .node_info import NodeInfo, NodeInputInfo, NodeOutputInfo, Versions
+from .node_info import NodeInfo, NodeInputInfo, NodeOutputInfo
 from .sensor_designer import SensorDesigner
 from .sql import SQLConfig, SQLKind
-from .tabular_data import TabularData
 from .token import Token
 from .vector_store import VectorStoreConfig, VectorStoreProvider
 from .version import __version__
@@ -20,28 +33,38 @@ __all__ = [
     "__version__",
     "Context",
     "Conversation",
+    "Document",
     "ExecutionError",
+    "File",
+    "FileLocation",
     "Graph",
     "Handle",
+    "Image",
     "JobInfo",
     "JobStatus",
     "LLMConfig",
     "LLMProvider",
+    "LocalStorage",
     "MachineLearningModel",
+    "Mesh",
     "Message",
     "NodeElement",
     "NodeId",
     "NodeInfo",
     "NodeInputInfo",
     "NodeOutputInfo",
+    "PDF",
+    "S3Storage",
     "SensorDesigner",
     "SourceHandle",
     "SQLConfig",
     "SQLKind",
+    "SQLTable",
     "TabularData",
     "TargetHandle",
     "Token",
+    "utils",
     "VectorStoreConfig",
     "VectorStoreProvider",
-    "Versions",
+    "WebPage",
 ]

uncertainty_engine_types/file.py

@@ -0,0 +1,48 @@
+from typing import Union
+
+from pydantic import BaseModel
+
+
+class S3Storage(BaseModel):
+    bucket: str
+    key: str
+
+
+class LocalStorage(BaseModel):
+    path: str
+
+
+FileLocation = Union[S3Storage, LocalStorage]
+
+
+class File(BaseModel):
+    pass
+
+
+class Document(File):
+    location: FileLocation
+
+
+class Image(File):
+    location: FileLocation
+
+
+class Mesh(File):
+    location: FileLocation
+
+
+class PDF(File):
+    location: FileLocation
+
+
+class SQLTable(File):
+    url: str
+    query: str
+
+
+class TabularData(File):
+    location: FileLocation
+
+
+class WebPage(File):
+    url: str

uncertainty_engine_types/handle.py

@@ -1,15 +1,27 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, model_validator
 
 
 class Handle(BaseModel):
     node_name: str
     node_handle: str
 
-    def __init__(self, handle_str: str):
-        if handle_str.count(".") != 1:
-            raise ValueError(
-                "Handle string must contain exactly one dot ('.') separating node and handle"
-            )
+    @model_validator(mode="before")
+    @classmethod
+    def split_handle(cls, values):
+        if isinstance(values, str):
+            parts = values.split(".")
+            if len(parts) != 2:
+                raise ValueError(
+                    "Handle string must contain exactly one dot ('.') separating node and handle"
+                )
+            return {"node_name": parts[0], "node_handle": parts[1]}
+        return values
 
-        node_name, node_handle = handle_str.split(".")
-        super().__init__(node_name=node_name, node_handle=node_handle)
+    def __init__(self, *args, **kwargs):
+        if args:
+            if len(args) == 1 and isinstance(args[0], str):
+                # Convert the positional argument to a dict via model_validate
+                kwargs = self.__class__.model_validate(args[0]).model_dump()
+            else:
+                raise TypeError("Invalid positional arguments")
+        super().__init__(**kwargs)

uncertainty_engine_types/utils.py

@@ -0,0 +1,9 @@
+from pydantic import ValidationError
+
+
+def format_pydantic_errors(e: ValidationError) -> str:
+    msgs = []
+    for error in e.errors():
+        loc = " -> ".join(map(str, error["loc"]))
+        msgs.append(f"Error at '{loc}': {error['msg']}")
+    return "\n".join(msgs)

uncertainty_engine_types/version.py

@@ -1 +1 @@
-__version__ = "0.0.6"
+__version__ = "0.0.8"

{uncertainty_engine_types-0.0.6.dist-info → uncertainty_engine_types-0.0.8.dist-info}/METADATA

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

{uncertainty_engine_types-0.0.6.dist-info → uncertainty_engine_types-0.0.8.dist-info}/RECORD

@@ -1,9 +1,10 @@
-uncertainty_engine_types/__init__.py,sha256=6pB7JPLiwVbNJJ65-SxaK_-Uj8Kc5kKlj_VTwYt6weM,1188
+uncertainty_engine_types/__init__.py,sha256=D9bTgnPGoBMJqAod_s7kyzjRO00p5E90iC6TThLB8PI,1480
 uncertainty_engine_types/context.py,sha256=k9rlArn-L2x1P5vWy5khfuqJW5tJOqr3NlhubZCbyG4,209
 uncertainty_engine_types/conversation.py,sha256=uDHTr4uPzjkcflhyFAanfiQ1KbEIs4C9jogrwzOEt_s,122
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
+uncertainty_engine_types/file.py,sha256=J202bhOIVnvvvAG54sXyYF8fwjZGTUP3097ARnRnb3Q,579
 uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUjL2Qg,265
-uncertainty_engine_types/handle.py,sha256=DEu3aCwJsOuinQ2Gxdy6dZbwnCHAXkplIT4WNL6o7oQ,444
+uncertainty_engine_types/handle.py,sha256=buX_xEdBwT7vZRs5uCEPB6VCbSWIFbCE-NYIL4fZSjg,942
 uncertainty_engine_types/job.py,sha256=eYDe-MW4s-kXGL1Ke1VxdwoduW66orJAa0L1XwGdYi0,347
 uncertainty_engine_types/llm.py,sha256=oivoY3aDv5Rsk8FWA-vOPEFfK9a6-SqSyu01AlEuO7o,406
 uncertainty_engine_types/message.py,sha256=KncFJQHL_ko8nfOuso_eh7i2gB0y4nXI6gBW9U8lAEY,971
@@ -11,10 +12,10 @@ uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk
 uncertainty_engine_types/node_info.py,sha256=2p87fCdX_CrdT2ZODtGJ-mR1Mvg46WhaE1GMddaEVak,800
 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/tabular_data.py,sha256=i_1LfgQqxDanPRWuDkP3Hd8-9eVet6FbPIlPZlAi1n0,76
 uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBXEpBY,178
+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=QiiYsv0kcJaB8wCWyT-FnI2b6be87HA-CrrIUn8LQhg,22
-uncertainty_engine_types-0.0.6.dist-info/METADATA,sha256=PQkB_d60eZxSWxPEyxRBq6bPpPvYsAdb4iuxEySMia0,2815
-uncertainty_engine_types-0.0.6.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.0.6.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=wOJN3HxAgnSon5vWYU3Txm2UZ_7tBHDKXUKZIH-mXX8,22
+uncertainty_engine_types-0.0.8.dist-info/METADATA,sha256=XkrqvGnn2EGgHII5lf4oWhbswMBH2SWZdW6QaTNoJfM,2815
+uncertainty_engine_types-0.0.8.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.0.8.dist-info/RECORD,,

uncertainty_engine_types/tabular_data.py

@@ -1,5 +0,0 @@
-from pydantic import BaseModel
-
-
-class TabularData(BaseModel):
-    csv: str