uncertainty-engine-types 0.0.6__tar.gz → 0.0.7__tar.gz

{uncertainty_engine_types-0.0.6 → uncertainty_engine_types-0.0.7}/PKG-INFO

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

{uncertainty_engine_types-0.0.6 → uncertainty_engine_types-0.0.7}/pyproject.toml

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

{uncertainty_engine_types-0.0.6 → uncertainty_engine_types-0.0.7}/uncertainty_engine_types/__init__.py

@@ -1,13 +1,27 @@
 from .context import Context
 from .conversation import Conversation
+from .document import Document, FileLocation, FileType
 from .execution_error import ExecutionError
+from .file import (
+    PDF,
+    Document,
+    File,
+    FileLocation,
+    Image,
+    LocalStorage,
+    Mesh,
+    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
@@ -15,33 +29,45 @@ from .token import Token
 from .vector_store import VectorStoreConfig, VectorStoreProvider
 from .version import __version__
 
-
 __all__ = [
     "__version__",
     "Context",
     "Conversation",
+    "Document",
+    "Document",
     "ExecutionError",
+    "File",
+    "FileLocation",
+    "FileLocation",
+    "FileType",
     "Graph",
     "Handle",
+    "Image",
     "JobInfo",
     "JobStatus",
     "LLMConfig",
     "LLMProvider",
+    "LocalStorage",
     "MachineLearningModel",
+    "Mesh",
     "Message",
     "NodeElement",
     "NodeId",
     "NodeInfo",
     "NodeInputInfo",
     "NodeOutputInfo",
+    "PDF",
+    "S3Storage",
     "SensorDesigner",
     "SourceHandle",
     "SQLConfig",
     "SQLKind",
+    "SQLTable",
+    "TabularData",
     "TabularData",
     "TargetHandle",
     "Token",
     "VectorStoreConfig",
     "VectorStoreProvider",
-    "Versions",
+    "WebPage",
 ]

uncertainty_engine_types-0.0.7/uncertainty_engine_types/document.py

@@ -0,0 +1,29 @@
+from enum import StrEnum
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class FileLocation(StrEnum):
+    LOCAL = "local"
+    S3 = "s3"
+    SQL = "sql"
+    WWW = "www"
+
+
+class FileType(StrEnum):
+    CSV = "csv"
+    DOCX = "docx"
+    PDF = "pdf"
+    TXT = "txt"
+
+
+class Document(BaseModel):
+    """
+    Document identification.
+    """
+
+    location: FileLocation
+    file_type: FileType
+    path: str
+    excerpt: Optional[str] = None

uncertainty_engine_types-0.0.7/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-0.0.7/uncertainty_engine_types/handle.py

@@ -0,0 +1,27 @@
+from pydantic import BaseModel, model_validator
+
+
+class Handle(BaseModel):
+    node_name: str
+    node_handle: str
+
+    @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
+
+    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-0.0.7/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.0.6/uncertainty_engine_types/handle.py

@@ -1,15 +0,0 @@
-from pydantic import BaseModel
-
-
-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"
-            )
-
-        node_name, node_handle = handle_str.split(".")
-        super().__init__(node_name=node_name, node_handle=node_handle)

uncertainty_engine_types-0.0.6/uncertainty_engine_types/version.py

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