matrice-analytics 0.1.60__py3-none-any.whl → 0.1.89__py3-none-any.whl

matrice_analytics/post_processing/face_reg/face_recognition_client.py

@@ -474,6 +474,42 @@ class FacialRecognitionClient:
         except Exception as e:
             self.logger.error(f"API ERROR: Get all staff embeddings request failed - {e}", exc_info=True)
             return {"success": False, "error": str(e)}
+    
+    async def update_deployment_action(self, deployment_id: str) -> Dict[str, Any]:
+        """Update deployment action in backend
+        
+        API: PUT /internal/v1/actions/update_facial_recognition_deployment/:server_id?app_deployment_id=:deployment_id
+        
+        Args:
+            deployment_id: The deployment ID to update
+            
+        Returns:
+            Dict containing response data
+        """
+        if not deployment_id:
+            self.logger.warning("No deployment_id provided for update_deployment_action")
+            return {"success": False, "error": "deployment_id is required"}
+
+        self.logger.info(f"API REQUEST: Updating deployment action - deployment_id={deployment_id}")
+
+        # Use Matrice session for async RPC call to backend (not facial recognition server).
+        try:
+            response = await self.session.rpc.async_send_request(
+                method="PUT",
+                path=f"/v1/actions/update_facial_recognition_deployment/{self.server_id}?app_deployment_id={deployment_id}",
+                payload={},
+                base_url="https://prod.backend.app.matrice.ai"
+            )
+            
+            if response.get('success', False):
+                self.logger.info(f"API RESPONSE: Deployment action updated successfully - deployment_id={deployment_id}")
+            else:
+                self.logger.warning(f"Failed to update deployment action for deployment_id={deployment_id}: {response.get('error', 'Unknown error')}")
+            
+            return self._handle_response(response)
+        except Exception as e:
+            self.logger.error(f"API ERROR: Update deployment action request failed - deployment_id={deployment_id} - {e}", exc_info=True)
+            return {"success": False, "error": str(e)}
 
     async def update_deployment(self, deployment_id: str) -> Dict[str, Any]:
         """Update deployment to notify facial recognition server
@@ -566,17 +602,47 @@ class FacialRecognitionClient:
                 payload={},
                 base_url=self.server_base_url
             )
-            
+
             if response.get('success', False):
                 self.logger.info(f"API RESPONSE: Service is healthy")
             else:
                 self.logger.warning(f"Health check failed: {response.get('error', 'Unknown error')}")
-            
+
             return self._handle_response(response)
         except Exception as e:
             self.logger.error(f"API ERROR: Health check request failed - {e}", exc_info=True)
             return {"success": False, "error": str(e)}
 
+    async def get_redis_details(self) -> Dict[str, Any]:
+        """Get Redis connection details from facial recognition server
+
+        API: GET /v1/facial_recognition/get_redis_details
+
+        Returns:
+            Dict containing Redis connection details (REDIS_IP, REDIS_PORT, REDIS_PASSWORD)
+        """
+
+        self.logger.info(f"API REQUEST: Getting Redis connection details")
+
+        # Use Matrice session for async RPC call
+        try:
+            response = await self.session.rpc.async_send_request(
+                method="GET",
+                path=f"/v1/facial_recognition/get_redis_details",
+                payload={},
+                base_url=self.server_base_url
+            )
+
+            if response.get('success', False):
+                self.logger.info(f"API RESPONSE: Redis details retrieved successfully")
+            else:
+                self.logger.warning(f"Failed to get Redis details: {response.get('error', 'Unknown error')}")
+
+            return self._handle_response(response)
+        except Exception as e:
+            self.logger.error(f"API ERROR: Get Redis details request failed - {e}", exc_info=True)
+            return {"success": False, "error": str(e)}
+
     def _handle_response(self, response: Dict[str, Any]) -> Dict[str, Any]:
         """Handle RPC response and errors"""
         try: