primitive 0.1.9__tar.gz → 0.1.11__tar.gz

{primitive-0.1.9 → primitive-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.9
+Version: 0.1.11
 Project-URL: Documentation, https://github.com//primitivecorp/primitive-cli#readme
 Project-URL: Issues, https://github.com//primitivecorp/primitive-cli/issues
 Project-URL: Source, https://github.com//primitivecorp/primitive-cli

{primitive-0.1.9 → primitive-0.1.11}/src/primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@steins.studio>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.9"
+__version__ = "0.1.11"

{primitive-0.1.9 → primitive-0.1.11}/src/primitive/agent/actions.py

@@ -14,7 +14,11 @@ class Agent(BaseAction):
         logger.info(f" [*] Version: {__version__}")
 
         # self.primitive.hardware.update_hardware_system_info()
-        self.primitive.hardware.check_in_http(is_available=True, is_online=True)
+        try:
+            self.primitive.hardware.check_in_http(is_available=True, is_online=True)
+        except Exception as ex:
+            logger.error(f"Error checking in hardware: {ex}")
+            sys.exit(1)
 
         try:
             while True:

{primitive-0.1.9 → primitive-0.1.11}/src/primitive/hardware/actions.py

@@ -171,12 +171,6 @@ class Hardware(BaseAction):
 
                 gpu_config.append(gpu_info)
 
-        # Raise an error if there is no valid gpu config
-        if not gpu_config:
-            message = "No valid gpu configuration"
-            logger.error(message)
-            raise NotImplementedError(message)
-
         return gpu_config
 
     def _get_windows_computer_service_product_values(self) -> Dict[str, str]:
@@ -417,29 +411,43 @@ class Hardware(BaseAction):
         input = new_state
         variables = {"input": input}
         try:
-            result = self.primitive.session.execute(mutation, variable_values=variables)
-            previous_state = self.previous_state
-            self.previous_state = new_state.copy()
-
-            message = " [*] Checked in successfully: "
-            for key, value in new_state.items():
-                if value != previous_state.get(key, None):
-                    if value is True:
-                        message = (
-                            message
-                            + click.style(f"{key}: ")
-                            + click.style("💤")
-                            + click.style(" ==> ✅ ", fg="green")
-                        )
+            result = self.primitive.session.execute(
+                mutation, variable_values=variables, get_execution_result=True
+            )
+            checkin_success = result.data.get("checkIn").get("lastCheckIn")
+            if messages := result.data.get("checkIn").get("messages"):
+                for message in messages:
+                    logger.enable("primitive")
+                    if message.get("kind") == "ERROR":
+                        logger.error(message.get("message"))
                     else:
-                        message = (
-                            message
-                            + click.style(f"{key}: ")
-                            + click.style("✅")
-                            + click.style(" ==> 💤 ", fg="yellow")
-                        )
-            logger.info(message)
-
+                        logger.debug(message.get("message"))
+
+            if checkin_success:
+                previous_state = self.previous_state
+                self.previous_state = new_state.copy()
+
+                message = " [*] Checked in successfully: "
+                for key, value in new_state.items():
+                    if value != previous_state.get(key, None):
+                        if value is True:
+                            message = (
+                                message
+                                + click.style(f"{key}: ")
+                                + click.style("💤")
+                                + click.style(" ==> ✅ ", fg="green")
+                            )
+                        else:
+                            message = (
+                                message
+                                + click.style(f"{key}: ")
+                                + click.style("✅")
+                                + click.style(" ==> 💤 ", fg="yellow")
+                            )
+                logger.info(message)
+            else:
+                message = "Failed to check in!"
+                raise Exception(message)
             return result
         except client_exceptions.ClientConnectorError as exception:
             message = " [*] Failed to check in! "