wizata-dsapi 2.0.0.dev18__tar.gz → 2.0.0.dev20__tar.gz

{wizata_dsapi-2.0.0.dev18/wizata_dsapi.egg-info → wizata_dsapi-2.0.0.dev20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev18
+Version: 2.0.0.dev20
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.0.0.dev18 → wizata_dsapi-2.0.0.dev20}/wizata_dsapi/__init__.py

@@ -9,7 +9,7 @@ from .execution import Execution, ExecutionStatus, ExecutionStepLog, AbortedExce
 from .execution_log import ExecutionLog
 from .experiment import Experiment
 from .ds_dataframe import DSDataFrame
-from .script import Script, ScriptConfig, ScriptType, ScriptLibrary, sanitize_blob_name, discover_from_wheel
+from .script import Script, ScriptConfig, ScriptType, ScriptCategory, ScriptLibrary, sanitize_blob_name, discover_from_wheel
 from .template import Template, TemplateProperty
 from .solution_component import SolutionComponent, SolutionType
 from .business_label import BusinessLabel
@@ -51,7 +51,7 @@ from .streamlit_utils import (get_streamlit_token, get_streamlit_domain, get_str
 
 from .search import FilterOperator, SearchQuery, SortOption, GroupSearchQuery
 from .notification import NotificationScope, DeviceToken, AlertRule, AlertActive, MonitoredDatapoint, \
-    NotificationPreference, RecipientList, Recipient, RecipientType
+    MonitoredPipeline, NotificationPreference, RecipientList, Recipient, RecipientType
 from .edge_device import EdgeDevice, EdgeDeviceStatus, EdgeDeviceType
 from .edge_module import EdgeModule, EdgeModuleType, WIZATA_MODULES
 from .edge_config import EdgeConsumer, EdgeWriter, EdgeTrigger

{wizata_dsapi-2.0.0.dev18 → wizata_dsapi-2.0.0.dev20}/wizata_dsapi/notification.py

@@ -19,7 +19,7 @@ RecipientList — named list of internal users (follow preferences) and free ent
                 (explicit address + channel).
 """
 
-from datetime import datetime
+from datetime import datetime, timezone
 from enum import Enum
 from typing import Optional
 import uuid
@@ -338,6 +338,54 @@ class MonitoredDatapoint:
         return f"MonitoredDatapoint(hardware_id={self.hardware_id!r})"
 
 
+class MonitoredPipeline:
+    """
+    Registry entry for a pipeline flagged for monitoring.
+
+    Stored in PostgreSQL ``"MonitoredPipeline"`` table.
+    PK is ``key`` (pipeline key string, one row per pipeline).
+
+    The pipeline_monitor job reads this table and creates independent
+    AlertRule/AlertActive entries per event type (not_triggered, failed).
+    """
+
+    def __init__(self, key: str = None):
+        self.key: Optional[str] = key
+        self.created_at: Optional[datetime] = None
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.key is not None:
+            obj["key"] = self.key
+        if self.created_at is not None:
+            obj["createdAt"] = int(self.created_at.timestamp() * 1000)
+        return obj
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "MonitoredPipeline":
+        obj = cls()
+        obj.key = data.get("key")
+        raw = data.get("createdAt") or data.get("created_at")
+        if raw is not None:
+            if isinstance(raw, (int, float)):
+                obj.created_at = datetime.fromtimestamp(raw / 1000, tz=timezone.utc)
+            elif isinstance(raw, datetime):
+                obj.created_at = raw
+        return obj
+
+    def api_id(self) -> str:
+        return self.key
+
+    def api_key(self) -> str:
+        return self.key
+
+    def set_id(self, value):
+        self.key = str(value)
+
+    def __repr__(self) -> str:
+        return f"MonitoredPipeline(key={self.key!r})"
+
+
 class NotificationPreference:
     """
     Per-user notification channel preferences stored in PostgreSQL

{wizata_dsapi-2.0.0.dev18 → wizata_dsapi-2.0.0.dev20}/wizata_dsapi/script.py

@@ -25,6 +25,18 @@ class ScriptType(Enum):
     RAW = 'raw'
 
 
+class ScriptCategory(Enum):
+    """
+    Defines the category of a script, determining which pipeline step type it creates.
+        - "function" data transformation scripts (creates a SCRIPT step)
+        - "model" ML model training scripts (creates a MODEL step)
+        - "plot" visualization scripts (creates a PLOT step)
+    """
+    FUNCTION = 'function'
+    MODEL = 'model'
+    PLOT = 'plot'
+
+
 def sanitize_blob_name(name: str) -> str:
     """
     Sanitize a script name for use as a blob path segment.
@@ -175,7 +187,7 @@ class Script(ApiDto):
         return "json"
 
     def __init__(self, script_id=None, description=None, function=None,
-                 script_type=None, version=None, library=None):
+                 script_type=None, version=None, library=None, category=None):
 
         # Id
         if script_id is None:
@@ -189,6 +201,7 @@ class Script(ApiDto):
         self.version = version
         self.library = library
         self.module = None
+        self.category = category
 
         # Source code property (not serialized)
         self.source = None
@@ -256,6 +269,8 @@ class Script(ApiDto):
             obj["module"] = self.module
         if self.source is not None:
             obj["source"] = self.source
+        if self.category is not None:
+            obj["category"] = self.category.value
         return obj
 
     def to_json(self, target: str = None):
@@ -287,6 +302,8 @@ class Script(ApiDto):
             self.module = obj["module"]
         if "source" in obj and obj["source"] is not None:
             self.source = obj["source"]
+        if "category" in obj and obj["category"] is not None:
+            self.category = ScriptCategory(obj["category"])
 
     def copy(self, myfunction):
         """

wizata_dsapi-2.0.0.dev20/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.0.0.dev20"

{wizata_dsapi-2.0.0.dev18 → wizata_dsapi-2.0.0.dev20/wizata_dsapi.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev18
+Version: 2.0.0.dev20
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-2.0.0.dev18/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.0.0.dev18"