uncertainty-engine-types 0.10.0__tar.gz → 0.12.0__tar.gz

{uncertainty_engine_types-0.10.0 → uncertainty_engine_types-0.12.0}/PKG-INFO

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

{uncertainty_engine_types-0.10.0 → uncertainty_engine_types-0.12.0}/pyproject.toml

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

{uncertainty_engine_types-0.10.0 → uncertainty_engine_types-0.12.0}/uncertainty_engine_types/__init__.py

@@ -25,7 +25,13 @@ 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, NodeRequirementsInfo
+from .node_info import (
+    NodeInfo,
+    NodeInputInfo,
+    NodeOutputInfo,
+    NodeRequirementsInfo,
+    ScalingInfo,
+)
 from .prompt import Prompt
 from .sensor_designer import SensorDesigner
 from .sql import SQLConfig, SQLKind
@@ -65,6 +71,7 @@ __all__ = [
     "Prompt",
     "ResourceID",
     "S3Storage",
+    "ScalingInfo",
     "SensorDesigner",
     "SourceHandle",
     "SQLConfig",

{uncertainty_engine_types-0.10.0 → uncertainty_engine_types-0.12.0}/uncertainty_engine_types/node_info.py

@@ -1,6 +1,6 @@
 from typing import Any, Optional
 
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from .version import __version__
 
@@ -27,6 +27,18 @@ class NodeRequirementsInfo(BaseModel):
     timeout: int
 
 
+class ScalingInfo(BaseModel):
+    """Scaling configuration."""
+
+    model_config = ConfigDict(use_attribute_docstrings=True)
+
+    max: int = 1
+    """Maximum number of service tasks to scale out to."""
+
+    min: int = 0
+    """Minimum number of service tasks to scale in to."""
+
+
 class NodeInfo(BaseModel, extra="allow"):
     """
     Node information.
@@ -52,7 +64,14 @@ class NodeInfo(BaseModel, extra="allow"):
     Deployment requirements.
     """
 
+    scaling: ScalingInfo = ScalingInfo()
+    """Scaling configuration."""
+
     load_balancer_url: Optional[str] = None
+
+    queue_name: Optional[str] = None
+    """Name of the node's job queue."""
+
     queue_url: Optional[str] = None
     service_arn: Optional[str] = None
     """
@@ -63,3 +82,6 @@ class NodeInfo(BaseModel, extra="allow"):
     version_types_lib: str = __version__
     version_base_image: int
     version_node: int
+
+    tags: list[str] = []
+    """Tags associated with the node."""

uncertainty_engine_types-0.12.0/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.10.0/uncertainty_engine_types/version.py

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