fix json formatting

Author: scarere

vec_inf/cli/_helper.py

@@ -4,6 +4,7 @@
 handling the presentation of model information, status updates, and metrics.
 """
 
+import json
 from pathlib import Path
 from typing import Any, Union
 
@@ -27,9 +28,7 @@ class LaunchResponseFormatter:
     Parameters
     ----------
     model_name : str
-        Name of the launched model
-    params : dict[str, Any]
-        Launch parameters and configuration
+        Name of the launched model    params : dict[str, Any] Launch parameters and configuration
     """
 
     def __init__(self, model_name: str, params: dict[str, Any]):
@@ -116,28 +115,14 @@ def format_table_output(self) -> Table:
         for model_name in self.params["models"]:
             table.add_row("Model Name", model_name, style="magenta")
             # Add resource allocation details
-            table.add_row(
-                "Partition", f"  {self.params['models'][model_name]['partition']}"
-            )
+            table.add_row("Partition", f"  {self.params['models'][model_name]['partition']}")
             table.add_row("QoS", f"  {self.params['models'][model_name]['qos']}")
-            table.add_row(
-                "Time Limit", f"  {self.params['models'][model_name]['time']}"
-            )
-            table.add_row(
-                "Num Nodes", f"  {self.params['models'][model_name]['num_nodes']}"
-            )
-            table.add_row(
-                "GPUs/Node", f"  {self.params['models'][model_name]['gpus_per_node']}"
-            )
-            table.add_row(
-                "CPUs/Task", f"  {self.params['models'][model_name]['cpus_per_task']}"
-            )
-            table.add_row(
-                "Memory/Node", f"  {self.params['models'][model_name]['mem_per_node']}"
-            )
-            table.add_row(
-                "Log Directory", f"  {self.params['models'][model_name]['log_dir']}"
-            )
+            table.add_row("Time Limit", f"  {self.params['models'][model_name]['time']}")
+            table.add_row("Num Nodes", f"  {self.params['models'][model_name]['num_nodes']}")
+            table.add_row("GPUs/Node", f"  {self.params['models'][model_name]['gpus_per_node']}")
+            table.add_row("CPUs/Task", f"  {self.params['models'][model_name]['cpus_per_task']}")
+            table.add_row("Memory/Node", f"  {self.params['models'][model_name]['mem_per_node']}")
+            table.add_row("Log Directory", f"  {self.params['models'][model_name]['log_dir']}")
 
         return table
 
@@ -176,7 +161,12 @@ def output_json(self) -> None:
             json_data["pending_reason"] = self.status_info.pending_reason
         if self.status_info.failed_reason:
             json_data["failed_reason"] = self.status_info.failed_reason
-        click.echo(json_data)
+
+        # Convert dict to json compliant string with double quotes
+        json_string = json.dumps(json_data)
+
+        # Output json to console
+        click.echo(json_string)
 
     def output_table(self) -> Table:
         """Create and display rich table.
@@ -352,9 +342,7 @@ def __init__(self, console: Console, json_mode: bool = False):
         self.model_config = None
         self.model_names: list[str] = []
 
-    def _format_single_model_output(
-        self, config: ModelConfig
-    ) -> Union[dict[str, Any], Table]:
+    def _format_single_model_output(self, config: ModelConfig) -> Union[dict[str, Any], Table]:
         """Format output table for a single model.
 
         Parameters
@@ -372,9 +360,7 @@ def _format_single_model_output(
             excluded = {"venv", "log_dir"}
             config_dict = config.model_dump(exclude=excluded)
             # Convert Path objects to strings
-            config_dict["model_weights_parent_dir"] = str(
-                config_dict["model_weights_parent_dir"]
-            )
+            config_dict["model_weights_parent_dir"] = str(config_dict["model_weights_parent_dir"])
             return config_dict
 
         table = create_table(key_title="Model Config", value_title="Value")
@@ -387,9 +373,7 @@ def _format_single_model_output(
                     table.add_row(f"  {vllm_arg}:", str(vllm_value))
         return table
 
-    def _format_all_models_output(
-        self, model_infos: list[ModelInfo]
-    ) -> Union[list[str], list[Panel]]:
+    def _format_all_models_output(self, model_infos: list[ModelInfo]) -> Union[list[str], list[Panel]]:
         """Format output table for all models.
 
         Parameters