Added post-prom. pipelines to ml_service; [skip ci]

Author: Sebastijan-Dominis

ml_service/backend/pipelines/models/pipelines_cli_args.py

@@ -143,3 +143,17 @@ class RunAllWorkflowsInput(BaseModel):
     logging_level: LOGGING_LEVEL = "INFO"
     owner: str | None = "Sebastijan"
     skip_if_existing: bool | None = True
+
+class InferInput(BaseModel):
+    """Model for the input of the inference pipeline."""
+    problem: str
+    segment: str
+    snapshot_bindings_id: str
+    logging_level: LOGGING_LEVEL = "INFO"
+
+class MonitorInput(BaseModel):
+    """Model for the input of the monitoring pipeline."""
+    problem: str
+    segment: str
+    inference_run_id: str | None = None
+    logging_level: LOGGING_LEVEL = "INFO"

ml_service/backend/routers/pipelines.py

@@ -15,6 +15,8 @@
     ExplainInput,
     FreezeAllFeatureSetsInput,
     FreezeFeaturesInput,
+    InferInput,
+    MonitorInput,
     PromoteInput,
     RegisterRawSnapshotInput,
     RunAllWorkflowsInput,
@@ -163,3 +165,23 @@ def run_all_workflows(payload: Annotated[RunAllWorkflowsInput, Body(...)], reque
         payload=payload,
         boolean_args=[],
     )
+
+@router.post("/infer", status_code=200)
+@limiter.limit("10/minute")
+def infer(payload: Annotated[InferInput, Body(...)], request: Request): # type: ignore
+    """Executes the inference pipeline."""
+    return execute_pipeline(
+        module_path="pipelines.post_promotion.infer",
+        payload=payload,
+        boolean_args=[],
+    )
+
+@router.post("/monitor", status_code=200)
+@limiter.limit("10/minute")
+def monitor(payload: Annotated[MonitorInput, Body(...)], request: Request): # type: ignore
+    """Executes the monitoring pipeline."""
+    return execute_pipeline(
+        module_path="pipelines.post_promotion.monitor",
+        payload=payload,
+        boolean_args=[],
+    )

ml_service/frontend/pipelines/pipelines_registry.py

@@ -11,6 +11,8 @@
     ExplainInput,
     FreezeAllFeatureSetsInput,
     FreezeFeaturesInput,
+    InferInput,
+    MonitorInput,
     PromoteInput,
     RegisterRawSnapshotInput,
     RunAllWorkflowsInput,
@@ -367,6 +369,61 @@
             }
         }
     },
+    {
+        "name": "Infer",
+        "endpoint": "pipelines/infer",
+        "args_schema": InferInput,
+        "field_metadata": {
+            "problem": {
+                "placeholder": "Model problem, e.g., 'no_show'",
+                "optional": False
+            },
+            "segment": {
+                "placeholder": "Model segment name, e.g., 'city_hotel_online_ta'",
+                "optional": False
+            },
+            "version": {
+                "placeholder": "Model version, e.g., 'v1'",
+                "optional": False
+            },
+            "snapshot_bindings_id": {
+                "placeholder": "A snapshot binding ID to define which snapshot to load for each dataset",
+                "optional": False
+            },
+            "logging_level": {
+                "optional": True,
+                "value": "INFO"
+            }
+        }
+    },
+    {
+        "name": "Monitor",
+        "endpoint": "pipelines/monitor",
+        "args_schema": MonitorInput,
+        "field_metadata": {
+            "problem": {
+                "placeholder": "Model problem, e.g., 'no_show'",
+                "optional": False
+            },
+            "segment": {
+                "placeholder": "Model segment name, e.g., 'city_hotel_online_ta'",
+                "optional": False
+            },
+            "version": {
+                "placeholder":"Model version, e.g., 'v1'",
+                "optional": False
+            },
+           "inference_run_id": {
+                "placeholder":"Inference run id dir name under predictions/{problem}/{segment} (default = latest)",
+                "optional": True,
+            },
+           "logging_level": {
+                "optional": True,
+                "value":"INFO"
+            }
+        }
+
+    },
     {
         "name": "Execute All Data Preprocessing",
         "endpoint": "pipelines/execute_all_data_preprocessing",