uncertainty-engine-types 0.0.13__py3-none-any.whl → 0.1.0__py3-none-any.whl

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

@@ -1 +1 @@
-__version__ = "0.0.13"
+__version__ = "0.1.0"

{uncertainty_engine_types-0.0.13.dist-info → uncertainty_engine_types-0.1.0.dist-info}/METADATA

@@ -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.dist-info → uncertainty_engine_types-0.1.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-uncertainty_engine_types/__init__.py,sha256=yUPp9RpvuwNXZUpmVwcqZ8Q51Oa8d9j0ika5sw0weB0,1866
+uncertainty_engine_types/__init__.py,sha256=QxuxItecFrZwOKNgPrdIO7i-YnEH3QAT3__86eTqSV8,1948
 uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
-uncertainty_engine_types/context.py,sha256=k9rlArn-L2x1P5vWy5khfuqJW5tJOqr3NlhubZCbyG4,209
+uncertainty_engine_types/context.py,sha256=1iDcfMLi_fKh_YVUAKif2NOdrA_vuXOQvBZKbF4uJZk,317
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
 uncertainty_engine_types/embeddings.py,sha256=sNDWjpuqmzOefKyQX--9EpdeTEyI4kQeSH4IELVUT7g,982
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
@@ -13,7 +13,7 @@ uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmn
 uncertainty_engine_types/message.py,sha256=5IY-S6Ipt6oYmnVSkRVv3QLSH-lKQvpxp_qF5YGSbok,209
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
 uncertainty_engine_types/model_config.py,sha256=AeN5oR06uZJMGoAJoaI_Ix4nCy5nNcU2Wh69ijP39rQ,286
-uncertainty_engine_types/node_info.py,sha256=2p87fCdX_CrdT2ZODtGJ-mR1Mvg46WhaE1GMddaEVak,800
+uncertainty_engine_types/node_info.py,sha256=XHONP6rNbAcLOHFxi-NPl1-UrKAH9P8C1yUr_rDiyCg,1362
 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
@@ -21,7 +21,7 @@ uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBX
 uncertainty_engine_types/uncertainty_plot.py,sha256=kJr0SuJ6JeTxaf2adpDNWqx7vVLZRpKG8tFbdx90pas,547
 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=SxvRes_80c-42zwvn5BXPvpGVs_EZ69FUEtUBF1k9Ts,23
-uncertainty_engine_types-0.0.13.dist-info/METADATA,sha256=lmGp8vQgY44E6PoGzr8CfFz0sjkw05L4ZQZ7yWmPSJw,2816
-uncertainty_engine_types-0.0.13.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.0.13.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+uncertainty_engine_types-0.1.0.dist-info/METADATA,sha256=jvQ6vZ_5TG74GjT0maweL1klrkSWR9iIYgUE8zP0qgs,2815
+uncertainty_engine_types-0.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.1.0.dist-info/RECORD,,