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

matrice_analytics/post_processing/face_reg/face_recognition_client.py

@@ -474,11 +474,11 @@ 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(self, deployment_id: str) -> Dict[str, Any]:
-        """Update deployment to notify facial recognition server
+    
+    async def update_deployment_action(self, deployment_id: str) -> Dict[str, Any]:
+        """Update deployment action in backend
         
-        API: PUT /v1/facial_recognition/update_deployment/:deployment_id
+        API: PUT /internal/v1/actions/update_facial_recognition_deployment/:server_id?app_deployment_id=:deployment_id
         
         Args:
             deployment_id: The deployment ID to update
@@ -487,34 +487,34 @@ class FacialRecognitionClient:
             Dict containing response data
         """
         if not deployment_id:
-            self.logger.warning("No deployment_id provided for update_deployment")
+            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 - deployment_id={deployment_id}")
+        self.logger.info(f"API REQUEST: Updating deployment action - deployment_id={deployment_id}")
 
-        # Use Matrice session for async RPC call
+        # 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/facial_recognition/update_deployment/{deployment_id}",
+                path=f"/v1/actions/update_facial_recognition_deployment/{self.server_id}?app_deployment_id={deployment_id}",
                 payload={},
-                base_url=self.server_base_url
+                base_url="https://prod.backend.app.matrice.ai"
             )
             
             if response.get('success', False):
-                self.logger.info(f"API RESPONSE: Deployment updated successfully - deployment_id={deployment_id}")
+                self.logger.info(f"API RESPONSE: Deployment action updated successfully - deployment_id={deployment_id}")
             else:
-                self.logger.warning(f"Failed to update deployment for deployment_id={deployment_id}: {response.get('error', 'Unknown error')}")
+                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 request failed - deployment_id={deployment_id} - {e}", exc_info=True)
+            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_action(self, deployment_id: str) -> Dict[str, Any]:
-        """Update deployment action in backend
+    async def update_deployment(self, deployment_id: str) -> Dict[str, Any]:
+        """Update deployment to notify facial recognition server
         
-        API: PUT /internal/v1/actions/update_facial_recognition_deployment/:server_id?app_deployment_id=:deployment_id
+        API: PUT /v1/facial_recognition/update_deployment/:deployment_id
         
         Args:
             deployment_id: The deployment ID to update
@@ -523,28 +523,28 @@ class FacialRecognitionClient:
             Dict containing response data
         """
         if not deployment_id:
-            self.logger.warning("No deployment_id provided for update_deployment_action")
+            self.logger.warning("No deployment_id provided for update_deployment")
             return {"success": False, "error": "deployment_id is required"}
 
-        self.logger.info(f"API REQUEST: Updating deployment action - deployment_id={deployment_id}")
+        self.logger.info(f"API REQUEST: Updating deployment - deployment_id={deployment_id}")
 
-        # Use Matrice session for async RPC call to backend (not facial recognition server)
+        # Use Matrice session for async RPC call
         try:
             response = await self.session.rpc.async_send_request(
                 method="PUT",
-                path=f"/internal/v1/actions/update_facial_recognition_deployment/{self.server_id}?app_deployment_id={deployment_id}",
+                path=f"/v1/facial_recognition/update_deployment/{deployment_id}",
                 payload={},
-                base_url="https://prod.backend.app.matrice.ai"
+                base_url=self.server_base_url
             )
             
             if response.get('success', False):
-                self.logger.info(f"API RESPONSE: Deployment action updated successfully - deployment_id={deployment_id}")
+                self.logger.info(f"API RESPONSE: Deployment 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')}")
+                self.logger.warning(f"Failed to update deployment 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)
+            self.logger.error(f"API ERROR: Update deployment request failed - deployment_id={deployment_id} - {e}", exc_info=True)
             return {"success": False, "error": str(e)}
     
     async def enroll_unknown_person(self, embedding: List[float], image_source: str = None, timestamp: str = None, location: str = None, employee_id: str = None) -> Dict[str, Any]:
@@ -602,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: