diff --git a/comfy_cli/config_manager.py b/comfy_cli/config_manager.py index 616bf09..2330df6 100644 --- a/comfy_cli/config_manager.py +++ b/comfy_cli/config_manager.py @@ -34,14 +34,20 @@ def set(self, key, value): """ Set a key-value pair in the config file. """ - self.config["DEFAULT"][key] = value + self.config["DEFAULT"][key] = str(value) self.write_config() # Write changes to file immediately - def get(self, key): + def get(self, key, type_cast=str): """ Get a value from the config file. Returns None if the key does not exist. """ - return self.config["DEFAULT"].get(key, None) # Returns None if the key does not exist + value = self.config["DEFAULT"].get(key, None) # Returns None if the key does not exist + + # Handle boolean conversion + if type_cast == bool: + return value.lower() in ("true", "yes", "1") + + return type_cast(value) def load(self): config_file_path = self.get_config_file_path() diff --git a/comfy_cli/tracking.py b/comfy_cli/tracking.py index 3666fdd..aa0304e 100644 --- a/comfy_cli/tracking.py +++ b/comfy_cli/tracking.py @@ -45,7 +45,7 @@ def track_event(event_name: str, properties: any = None): if properties is None: properties = {} logging.debug(f"tracking event called with event_name: {event_name} and properties: {properties}") - enable_tracking = config_manager.get(constants.CONFIG_KEY_ENABLE_TRACKING) + enable_tracking = config_manager.get(constants.CONFIG_KEY_ENABLE_TRACKING, type_cast=bool) if not enable_tracking: return