matrice-compute 0.1.21__py3-none-any.whl → 0.1.22__py3-none-any.whl

matrice_compute/scaling.py

@@ -677,7 +677,7 @@ class Scaling:
 
     @log_errors(log_error=True)
     def assign_jobs(self, is_gpu):
-        """Assign jobs to the instance using Kafka (with REST fallback).
+        """Assign jobs to the instance using REST API.
 
         Args:
             is_gpu: Boolean or any value indicating if this is a GPU instance.
@@ -690,30 +690,41 @@ class Scaling:
         is_gpu_bool = bool(is_gpu)
         logging.info("Assigning jobs for instance %s (GPU: %s)", self.instance_id, is_gpu_bool)
 
-        payload = {
-            "instanceID": self.instance_id,
-            "isGPUInstance": is_gpu_bool,
-        }
-
-        # Define REST fallback function
-        def rest_fallback():
-            is_gpu_str = str(is_gpu_bool).lower()
-            path = f"/v1/actions/assign_jobs/{is_gpu_str}/{self.instance_id}"
-            resp = self.rpc.get(path=path)
-            return self.handle_response(
-                resp,
-                "Pinged successfully",
-                "Could not ping the scaling jobs",
-            )
-
-        # Use hybrid approach: Kafka first, REST fallback, cache if both fail
-        return self._hybrid_request(
-            api="assign_jobs",
-            payload=payload,
-            request_topic=self.kafka_config["action_request_topic"] if self.enable_kafka else None,
-            response_topic=self.kafka_config["action_response_topic"] if self.enable_kafka else None,
-            rest_fallback_func=rest_fallback
+        # Use REST API directly
+        is_gpu_str = str(is_gpu_bool).lower()
+        path = f"/v1/actions/assign_jobs/{is_gpu_str}/{self.instance_id}"
+        resp = self.rpc.get(path=path)
+        return self.handle_response(
+            resp,
+            "Pinged successfully",
+            "Could not ping the scaling jobs",
         )
+
+        # # Kafka approach (commented out - using REST only)
+        # payload = {
+        #     "instanceID": self.instance_id,
+        #     "isGPUInstance": is_gpu_bool,
+        # }
+        # 
+        # # Define REST fallback function
+        # def rest_fallback():
+        #     is_gpu_str = str(is_gpu_bool).lower()
+        #     path = f"/v1/actions/assign_jobs/{is_gpu_str}/{self.instance_id}"
+        #     resp = self.rpc.get(path=path)
+        #     return self.handle_response(
+        #         resp,
+        #         "Pinged successfully",
+        #         "Could not ping the scaling jobs",
+        #     )
+        # 
+        # # Use hybrid approach: Kafka first, REST fallback, cache if both fail
+        # return self._hybrid_request(
+        #     api="assign_jobs",
+        #     payload=payload,
+        #     request_topic=self.kafka_config["action_request_topic"] if self.enable_kafka else None,
+        #     response_topic=self.kafka_config["action_response_topic"] if self.enable_kafka else None,
+        #     rest_fallback_func=rest_fallback
+        # )
           
 
     @log_errors(log_error=True)

{matrice_compute-0.1.21.dist-info → matrice_compute-0.1.22.dist-info}/METADATA

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

{matrice_compute-0.1.21.dist-info → matrice_compute-0.1.22.dist-info}/RECORD

@@ -7,11 +7,11 @@ matrice_compute/instance_utils.py,sha256=tCI_A3L5iohw62acmlXuOJns0DjIkvwN4znlUAI
 matrice_compute/prechecks.py,sha256=W9YmNF3RcLhOf4U8WBlExvFqDw1aGWSNTlJtA73lbDQ,17196
 matrice_compute/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 matrice_compute/resources_tracker.py,sha256=pkdt0aVKx_TpY_Sq---73w9INkDffZZe3mZGlp1EftE,22573
-matrice_compute/scaling.py,sha256=RBqfhButiocWueu-OeUbo8KquYs3BpYi2GiksFM9H10,45998
+matrice_compute/scaling.py,sha256=VMRhUYqPtYPWPd3Qt0PtHnYw04EIPokhYMEnQHzYbXE,46444
 matrice_compute/shutdown_manager.py,sha256=0MYV_AqygqR9NEntYf7atUC-PbWXyNkm1f-8c2aizgA,13234
 matrice_compute/task_utils.py,sha256=3qIutiQdYPyGRxH9ZwLbqdg8sZcnp6jp08pszWCRFl0,2820
-matrice_compute-0.1.21.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
-matrice_compute-0.1.21.dist-info/METADATA,sha256=_2gHnRw1cg9bwDjtSJ2_Dkb7XzTfrrQyvMWdFNB1RE8,1038
-matrice_compute-0.1.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice_compute-0.1.21.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
-matrice_compute-0.1.21.dist-info/RECORD,,
+matrice_compute-0.1.22.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_compute-0.1.22.dist-info/METADATA,sha256=A3T2jtGU5YaQaYqa-5d33c8vFDNoLNUsxMLftZ7qIT4,1038
+matrice_compute-0.1.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_compute-0.1.22.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
+matrice_compute-0.1.22.dist-info/RECORD,,