neuromeka-vfm 0.1.2__tar.gz → 0.1.3__tar.gz

{neuromeka_vfm-0.1.2/src/neuromeka_vfm.egg-info → neuromeka_vfm-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuromeka_vfm
-Version: 0.1.2
+Version: 0.1.3
 Summary: Client utilities for Neuromeka VFM FoundationPose RPC (upload meshes, call server)
 Author: Neuromeka
 License: MIT License

{neuromeka_vfm-0.1.2 → neuromeka_vfm-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "neuromeka_vfm"
-version = "0.1.2"
+version = "0.1.3"
 description = "Client utilities for Neuromeka VFM FoundationPose RPC (upload meshes, call server)"
 readme = "README.md"
 requires-python = ">=3.8"

{neuromeka_vfm-0.1.2 → neuromeka_vfm-0.1.3}/src/neuromeka_vfm/segmentation.py

@@ -26,6 +26,25 @@ class Segmentation:
             self.call_time = {"add_image_prompt": 0, "register_first_frame": 0, "get_next": 0}
             self.call_count = {"add_image_prompt": 0, "register_first_frame": 0, "get_next": 0}
 
+    def _is_success(self, response):
+        """
+        Normalize server success flag.
+        Some servers return {"result": "SUCCESS"}, others {"success": true}, and
+        the segmentation server returns {"status": "success"}.
+        """
+        # Try known keys in order of common usage.
+        for key in ("result", "success", "status"):
+            flag = response.get(key)
+            if flag is None:
+                continue
+            if isinstance(flag, str):
+                return flag.lower() == "success"
+            if isinstance(flag, bool):
+                return flag
+            return bool(flag)
+        # Fallback: anything truthy counts as success.
+        return bool(response)
+
     def switch_compression_strategy(self, compression_strategy):
         if compression_strategy in STRATEGIES:
             self.compression_strategy_name = compression_strategy
@@ -46,7 +65,7 @@ class Segmentation:
             start = time.time()
         data = {"operation": "add_image_prompt", "object_name": object_name, "object_image": object_image}
         response = self.client.send_data(data)
-        if response.get("result") == "SUCCESS":
+        if self._is_success(response):
             self.image_prompt_names.add(object_name)
         if self.benchmark:
             self.call_time["add_image_prompt"] += time.time() - start
@@ -72,7 +91,7 @@ class Segmentation:
             "compression_strategy": self.compression_strategy_name,
         }
         response = self.client.send_data(data)
-        if response.get("result") == "SUCCESS":
+        if self._is_success(response):
             self.first_frame_registered = True
             self.tracking_object_ids = response["data"]["obj_ids"]
             masks = {}
@@ -98,7 +117,7 @@ class Segmentation:
         if self.benchmark:
             start = time.time()
         response = self.client.send_data({"operation": "get_next", "frame": self.compression_strategy.encode(frame)})
-        if response.get("result") == "SUCCESS":
+        if self._is_success(response):
             masks = {}
             for i, obj_id in enumerate(self.tracking_object_ids):
                 mask = self.compression_strategy.decode(response["data"]["masks"][i])

{neuromeka_vfm-0.1.2 → neuromeka_vfm-0.1.3/src/neuromeka_vfm.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuromeka_vfm
-Version: 0.1.2
+Version: 0.1.3
 Summary: Client utilities for Neuromeka VFM FoundationPose RPC (upload meshes, call server)
 Author: Neuromeka
 License: MIT License