uncertainty-engine-types 0.0.13__tar.gz → 0.1.0__tar.gz

{uncertainty_engine_types-0.0.13 → uncertainty_engine_types-0.1.0}/PKG-INFO

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

{uncertainty_engine_types-0.0.13 → uncertainty_engine_types-0.1.0}/pyproject.toml

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

{uncertainty_engine_types-0.0.13 → uncertainty_engine_types-0.1.0}/uncertainty_engine_types/__init__.py

@@ -1,6 +1,6 @@
 from . import utils
 from .chat_history import ChatHistory
-from .context import Context
+from .context import Context, UserContext
 from .dataset import CSVDataset
 from .embeddings import TextEmbeddingsConfig, TextEmbeddingsProvider
 from .execution_error import ExecutionError
@@ -25,7 +25,7 @@ from .llm import LLMConfig, LLMProvider
 from .message import Message
 from .model import MachineLearningModel
 from .model_config import ModelConfig
-from .node_info import NodeInfo, NodeInputInfo, NodeOutputInfo
+from .node_info import NodeInfo, NodeInputInfo, NodeOutputInfo, NodeRequirementsInfo
 from .prompt import Prompt
 from .sensor_designer import SensorDesigner
 from .sql import SQLConfig, SQLKind
@@ -60,6 +60,7 @@ __all__ = [
     "NodeInfo",
     "NodeInputInfo",
     "NodeOutputInfo",
+    "NodeRequirementsInfo",
     "PDF",
     "Prompt",
     "ResourceID",
@@ -74,6 +75,7 @@ __all__ = [
     "TextEmbeddingsConfig",
     "TextEmbeddingsProvider",
     "Token",
+    "UserContext",
     "UncertaintyPlot",
     "utils",
     "VectorStoreConfig",

{uncertainty_engine_types-0.0.13 → uncertainty_engine_types-0.1.0}/uncertainty_engine_types/context.py

@@ -3,6 +3,12 @@ from pydantic import BaseModel
 from .node_info import NodeInfo
 
 
+class UserContext(BaseModel):
+    email: str
+    project_id: str
+    cost_code: str
+
+
 class Context(BaseModel):
     sync: bool
     job_id: str
@@ -10,3 +16,4 @@ class Context(BaseModel):
     cache_url: str
     timeout: int
     nodes: dict[str, NodeInfo]
+    user: UserContext

{uncertainty_engine_types-0.0.13 → uncertainty_engine_types-0.1.0}/uncertainty_engine_types/node_info.py

@@ -20,7 +20,24 @@ class NodeOutputInfo(BaseModel):
     description: str
 
 
-class NodeInfo(BaseModel):
+class NodeRequirementsInfo(BaseModel):
+    cpu: int
+    gpu: bool
+    memory: int
+    timeout: int
+
+
+class NodeInfo(BaseModel, extra="allow"):
+    """
+    Node information.
+    """
+
+    # New properties must be added as optional. The Node Registry uses this
+    # model and must support Nodes that don't provide a full set of details.
+    #
+    # Likewise, the `extra="allow"` argument allows the Node Registry to
+    # deserialise `NodeInfo` models with properties added post-release.
+
     id: str
     label: str
     category: str
@@ -30,6 +47,11 @@ class NodeInfo(BaseModel):
     cost: int
     inputs: dict[str, NodeInputInfo]
     outputs: dict[str, NodeOutputInfo] = {}
+    requirements: Optional[NodeRequirementsInfo] = None
+    """
+    Deployment requirements.
+    """
+
     load_balancer_url: Optional[str] = None
     queue_url: Optional[str] = None
     cache_url: Optional[str] = None

uncertainty_engine_types-0.1.0/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.0.13/uncertainty_engine_types/version.py

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