matrice-analytics 0.1.58__tar.gz → 0.1.59__tar.gz

{matrice_analytics-0.1.58 → matrice_analytics-0.1.59}/PKG-INFO

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

{matrice_analytics-0.1.58 → matrice_analytics-0.1.59}/matrice_analytics.egg-info/PKG-INFO

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

{matrice_analytics-0.1.58 → matrice_analytics-0.1.59}/src/matrice_analytics/post_processing/face_reg/face_recognition_client.py

@@ -448,14 +448,29 @@ class FacialRecognitionClient:
             )
             
             embeddings_count = 0
-            if response: #.get('success', False):
-                data = response.get('data', [])
-                embeddings_count = len(data) if isinstance(data, list) else 0
-                self.logger.info(f"API RESPONSE: Retrieved {embeddings_count} staff embeddings")
+            if response:
+                # Handle both list and dict responses
+                if isinstance(response, list):
+                    # API returned list directly
+                    data = response
+                    embeddings_count = len(data)
+                    self.logger.info(f"API RESPONSE: Retrieved {embeddings_count} staff embeddings (list format)")
+                    # Return in standard format for consistency
+                    return {"success": True, "data": data}
+                elif isinstance(response, dict):
+                    # API returned dict with 'data' key
+                    data = response.get('data', [])
+                    embeddings_count = len(data) if isinstance(data, list) else 0
+                    self.logger.info(f"API RESPONSE: Retrieved {embeddings_count} staff embeddings (dict format)")
+                    return self._handle_response(response)
+                else:
+                    self.logger.error(f"Unexpected response type: {type(response)}")
+                    return {"success": False, "error": f"Unexpected response type: {type(response)}"}
             else:
-                self.logger.warning(f"Failed to get staff embeddings: {response.get('error', 'Unknown error')}")
+                error_msg = response.get('error', 'Unknown error') if isinstance(response, dict) else 'Empty response'
+                self.logger.warning(f"Failed to get staff embeddings: {error_msg}")
+                return {"success": False, "error": error_msg}
             
-            return self._handle_response(response)
         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)}