matrice-compute 0.1.16__tar.gz → 0.1.18__tar.gz

{matrice_compute-0.1.16 → matrice_compute-0.1.18}/PKG-INFO

@@ -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 → matrice_compute-0.1.18}/matrice_compute.egg-info/PKG-INFO

@@ -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 → matrice_compute-0.1.18}/src/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-0.1.16 → matrice_compute-0.1.18}/src/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,