uncertainty-engine-types 0.10.0__py3-none-any.whl → 0.11.0__py3-none-any.whl

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/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
     """

{uncertainty_engine_types-0.10.0.dist-info → uncertainty_engine_types-0.11.0.dist-info}/METADATA

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

{uncertainty_engine_types-0.10.0.dist-info → uncertainty_engine_types-0.11.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-uncertainty_engine_types/__init__.py,sha256=QxuxItecFrZwOKNgPrdIO7i-YnEH3QAT3__86eTqSV8,1948
+uncertainty_engine_types/__init__.py,sha256=0FAX_KGWHoFlMgPmIUEqR5YfTSr7ffvdIt3I_O-1ETk,2005
 uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
 uncertainty_engine_types/context.py,sha256=gi1izptPe05ksem5TGZNIzgzO_QmomzXwTGqvB5I9Sc,914
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
@@ -13,7 +13,7 @@ uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmn
 uncertainty_engine_types/message.py,sha256=sX1tq_BbZC9r9-Vf0A3WHiOFRE2hAAlY-7rD52kbsoU,381
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
 uncertainty_engine_types/model_config.py,sha256=pCawKIo2l0O3eRLE5Ese7jaxomIyPXGW3XIcAx6TckY,558
-uncertainty_engine_types/node_info.py,sha256=ifSEkDVcS_B38jXx52nedU-wXe2GHD1Yt8Fqv0V3Ix0,1434
+uncertainty_engine_types/node_info.py,sha256=R1AZczUnYMhUkTgNMmW2d1YJ5JG7ai4Nfewgh0XuPMc,1880
 uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
 uncertainty_engine_types/sql.py,sha256=SBzmgMEZ-sa-OBRoZbmKiOqddop4zJixgxx-JCus9fY,298
@@ -22,6 +22,6 @@ uncertainty_engine_types/uncertainty_plot.py,sha256=kJr0SuJ6JeTxaf2adpDNWqx7vVLZ
 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=H9NWRZb7NbeRRPLP_V1fARmLNXranorVM-OOY-8_2ug,22
-uncertainty_engine_types-0.10.0.dist-info/METADATA,sha256=1GK1FkKrMzPgLQAI78_Oqbjxcb3yxNHH24BtGdEjatM,2816
-uncertainty_engine_types-0.10.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-uncertainty_engine_types-0.10.0.dist-info/RECORD,,
+uncertainty_engine_types-0.11.0.dist-info/METADATA,sha256=oRFqeZERfP-ejwdnKru7T9RpqLOtbJDht_ZItE6w72s,2816
+uncertainty_engine_types-0.11.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+uncertainty_engine_types-0.11.0.dist-info/RECORD,,