matrice-compute 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

matrice_compute/action_instance.py

@@ -75,7 +75,9 @@ class ActionInstance:
             "facial_recognition_setup": facial_recognition_setup_execute,
             "fe_fs_streaming": fe_fs_streaming_execute,
             "inference_ws_server": inference_ws_server_execute,
-            "lpr_setup": lpr_setup_execute
+            "fe_analytics_service": fe_analytics_service_execute,
+            "lpr_setup": lpr_setup_execute,
+            "inference_tracker_server": inference_tracker_setup_execute
         }
         if self.action_type not in self.actions_map:
             raise ValueError(f"Unknown action type: {self.action_type}")
@@ -471,11 +473,12 @@ class ActionInstance:
             # Try to get model codebase URLs from action_details first
             model_codebase_url = job_params.get("model_codebase_url")
             model_requirements_url = job_params.get("model_requirements_url")
+            dockerId = job_params.get("_idDocker")
 
             # Fallback to API calls if not provided in action_details
             if not model_codebase_url:
                 model_codebase_url, error, message = self.scaling.get_model_codebase(
-                    model_family
+                    dockerId
                 )
                 if error:
                     logging.warning(f"Failed to get model codebase URL: {message}")
@@ -486,7 +489,7 @@ class ActionInstance:
                 model_codebase_requirements_url = model_requirements_url
             else:
                 model_codebase_requirements_url, error, message = (
-                    self.scaling.get_model_codebase_requirements(model_family)
+                    self.scaling.get_model_codebase_requirements(dockerId)
                 )
                 if error:
                     logging.warning(
@@ -1173,7 +1176,7 @@ def inference_ws_server_execute(self: ActionInstance):
 def fe_fs_streaming_execute(self: ActionInstance):
     """
     Creates and setup the frontend for fs streaming.
-    Frontend streaming runs on port 3001 (localhost only with --net=host).
+    Frontend streaming runs on port 3000 (localhost only with --net=host).
     """
     action_details = self.get_action_details()
 
@@ -1183,7 +1186,7 @@ def fe_fs_streaming_execute(self: ActionInstance):
 
     self.setup_action_requirements(action_details)
 
-    # Frontend streaming with --net=host (Port: 3001)
+    # Frontend streaming with --net=host (Port: 3000)
     worker_cmd = (
         f"docker run -d --pull=always --net=host "
         f"--name fe_streaming "
@@ -1191,14 +1194,49 @@ def fe_fs_streaming_execute(self: ActionInstance):
         f'-e ENV="{os.environ.get("ENV", "prod")}" '
         f'-e MATRICE_SECRET_ACCESS_KEY="{self.matrice_secret_access_key}" '
         f'-e MATRICE_ACCESS_KEY_ID="{self.matrice_access_key_id}" '
+        f"-e PORT=3000 "
         f"{image}"
     )
-    logging.info("Starting frontend streaming (Port: 3001): %s", worker_cmd)
+    logging.info("Starting frontend streaming (Port: 3000): %s", worker_cmd)
 
     # Docker Command run
     self.start(worker_cmd, "fe_fs_streaming")
 
 
+@log_errors(raise_exception=False)
+def fe_analytics_service_execute(self: ActionInstance):
+    """
+    Creates and setup the frontend analytics service.
+    Frontend analytics service runs on port 3001 (localhost only with --net=host).
+    """
+    action_details = self.get_action_details()
+
+    if not action_details:
+        return
+    image = action_details["actionDetails"].get("docker")
+
+    self.setup_action_requirements(action_details)
+
+    project_id = action_details["_idProject"]
+    
+    # Frontend analytics service with --net=host (Port: 3001)
+    worker_cmd = (
+        f"docker run -d --pull=always --net=host "
+        f"--name fe-analytics "
+        f'-e NEXT_PUBLIC_DEPLOYMENT_ENV="{os.environ.get("ENV", "prod")}" '
+        f'-e MATRICE_SECRET_ACCESS_KEY="{self.matrice_secret_access_key}" '
+        f'-e MATRICE_ACCESS_KEY_ID="{self.matrice_access_key_id}" '
+        f'-e ACTION_ID="{self.action_record_id}" '
+        f"-e PORT=3001 "
+        f'-e PROJECT_ID="{project_id}" '
+        f"{image}"
+    )
+    logging.info("Starting frontend analytics service (Port: 3001): %s", worker_cmd)
+
+    # Docker Command run
+    self.start(worker_cmd, "fe_analytics_service")
+
+
 @log_errors(raise_exception=False)
 def synthetic_dataset_generation_execute(self: ActionInstance):
     """Execute synthetic dataset generation task."""
@@ -1571,3 +1609,34 @@ def kafka_setup_execute(self: ActionInstance):
 
     logging.info("Starting Kafka container (Ports: 9092, 9093): %s", cmd)
     self.start(cmd, "kafka_setup")
+
+
+@log_errors(raise_exception=False)
+def inference_tracker_setup_execute(self: ActionInstance):
+
+    """
+    Creates and start inference tracker.
+    Inference tracker runs on port 8110 (localhost only with --net=host).
+    """ 
+    
+    action_details = self.get_action_details()
+    if not action_details:
+        return
+    
+    image = self.docker_container
+    
+    self.setup_action_requirements(action_details)
+    
+    # This is the existing Docker run command
+    worker_cmd = (
+        f"docker run -d --pull=always --net=host "
+        f"--name inference-tracker-worker "
+        f"-v matrice_myvol:/matrice_data "
+        f'-e ENV="{os.environ.get("ENV", "prod")}" '
+        f'-e MATRICE_SECRET_ACCESS_KEY="{self.matrice_secret_access_key}" '
+        f'-e MATRICE_ACCESS_KEY_ID="{self.matrice_access_key_id}" '
+        f'-e ACTION_ID="{self.action_record_id}" '
+        f"{image}"
+    )
+    
+    self.start(worker_cmd, "inference_tracker_setup")

matrice_compute/scaling.py

@@ -674,16 +674,16 @@ class Scaling:
         )
 
     @log_errors(log_error=True)
-    def get_model_codebase_requirements(self, model_family_id):
+    def get_model_codebase_requirements(self, dockerId):
         """Get model codebase requirements.
 
         Args:
-            model_family_id: ID of the model family
+            dockerId: ID of the docker
 
         Returns:
             Tuple of (data, error, message) from API response
         """
-        path = f"/v1/model_store/get_user_requirements_download_path/{model_family_id}"
+        path = f"/v1/model_store/get_user_requirements_download_path/{dockerId}"
         resp = self.rpc.get(path=path)
         return self.handle_response(
             resp,

{matrice_compute-0.1.16.dist-info → matrice_compute-0.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_compute
-Version: 0.1.16
+Version: 0.1.18
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_compute-0.1.16.dist-info → matrice_compute-0.1.18.dist-info}/RECORD

@@ -1,5 +1,5 @@
 matrice_compute/__init__.py,sha256=ZzQcFsT005VCgq9VZUh565f4upOooEb_FwZ6RgweNZs,597
-matrice_compute/action_instance.py,sha256=cilzBD3o6K5CpDZEJCGMrNg0bCoUyOW3aCLNrMGyS10,60554
+matrice_compute/action_instance.py,sha256=NYjcDLlJ2k47Cx7yhTkSY3U-Eog9dRx0unvv1ip5Pqo,62904
 matrice_compute/actions_manager.py,sha256=5U-xM6tl_Z6x96bi-c7AJM9ru80LqTN8f5Oce8dAu_A,7780
 matrice_compute/actions_scaledown_manager.py,sha256=pJ0nduNwHWZ10GnqJNx0Ok7cVWabQ_M8E2Vb9pH3A_k,2002
 matrice_compute/instance_manager.py,sha256=8USyX09ZxLvnVNIrjRogbyUeMCfgWnasuRqYkkVF4tQ,10146
@@ -7,11 +7,11 @@ matrice_compute/instance_utils.py,sha256=cANKRUlUzfecnzVEMC6Gkg9K7GZajH9ojNPiChd
 matrice_compute/prechecks.py,sha256=W9YmNF3RcLhOf4U8WBlExvFqDw1aGWSNTlJtA73lbDQ,17196
 matrice_compute/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 matrice_compute/resources_tracker.py,sha256=n57IJmT5GjNEX8yQL7nbKv57bjvESYM-vRQcQ0DgQXQ,19256
-matrice_compute/scaling.py,sha256=3F8SWvy9wWczpJ6dbY5RrXWw5ByZlIzAPJklir3KIFI,35359
+matrice_compute/scaling.py,sha256=fAoXsDmMgtNjX-5gXB3KvaWeAi78rDDyRsG1GUmwykg,35332
 matrice_compute/shutdown_manager.py,sha256=0MYV_AqygqR9NEntYf7atUC-PbWXyNkm1f-8c2aizgA,13234
 matrice_compute/task_utils.py,sha256=3qIutiQdYPyGRxH9ZwLbqdg8sZcnp6jp08pszWCRFl0,2820
-matrice_compute-0.1.16.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
-matrice_compute-0.1.16.dist-info/METADATA,sha256=gTIsLb7gHIZCl4rvaQ5tKQW8b2OW2jfvLqyYxn_BMFo,1038
-matrice_compute-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice_compute-0.1.16.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
-matrice_compute-0.1.16.dist-info/RECORD,,
+matrice_compute-0.1.18.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_compute-0.1.18.dist-info/METADATA,sha256=NoeXPLhIa4x2IAUXphbVZXUF6u0YVgb-WVF1093ouDc,1038
+matrice_compute-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_compute-0.1.18.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
+matrice_compute-0.1.18.dist-info/RECORD,,