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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.7
+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.7 → uncertainty_engine_types-0.0.8}/pyproject.toml

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

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

@@ -1,15 +1,15 @@
+from . import utils
 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,
+    PDF,
     S3Storage,
     SQLTable,
     TabularData,
@@ -24,22 +24,19 @@ from .model import MachineLearningModel
 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__
 
+
 __all__ = [
     "__version__",
     "Context",
     "Conversation",
     "Document",
-    "Document",
     "ExecutionError",
     "File",
     "FileLocation",
-    "FileLocation",
-    "FileType",
     "Graph",
     "Handle",
     "Image",
@@ -64,9 +61,9 @@ __all__ = [
     "SQLKind",
     "SQLTable",
     "TabularData",
-    "TabularData",
     "TargetHandle",
     "Token",
+    "utils",
     "VectorStoreConfig",
     "VectorStoreProvider",
     "WebPage",

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

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

uncertainty_engine_types-0.0.7/uncertainty_engine_types/document.py

@@ -1,29 +0,0 @@
-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/tabular_data.py

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

uncertainty_engine_types-0.0.7/uncertainty_engine_types/version.py

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