Param parsing issue fix (#1182)

Author: VipulMascarenhas

ads/aqua/common/utils.py

@@ -832,7 +832,9 @@ def get_params_dict(params: Union[str, List[str]]) -> dict:
     """
     params_list = get_params_list(params) if isinstance(params, str) else params
     return {
-        split_result[0]: split_result[1] if len(split_result) > 1 else UNKNOWN
+        split_result[0]: " ".join(split_result[1:])
+        if len(split_result) > 1
+        else UNKNOWN
         for split_result in (x.split() for x in params_list)
     }
 
@@ -881,7 +883,9 @@ def build_params_string(params: dict) -> str:
         A params string.
     """
     return (
-        " ".join(f"{name} {value}" for name, value in params.items()).strip()
+        " ".join(
+            f"{name} {value}" if value else f"{name}" for name, value in params.items()
+        ).strip()
         if params
         else UNKNOWN
     )

tests/unitary/with_extras/aqua/test_utils.py

@@ -171,3 +171,15 @@ def test_list_os_files_with_extension(self, oss_mock: MagicMock):
         oss_mock_client.filepath = prefix
         resp = utils.list_os_files_with_extension(prefix, ".gguf")
         self.assertIn(obj2_name, resp)
+
+    @parameterized.expand(
+        [
+            "--gpu-memory-utilization 0.98 --max-loras 2 --lora-modules speech=artifact/speech-lora vision=artifact/vision-lora --dtype auto --trust-remote-code",
+            "--gpu-memory-utilization 0.98 --dtype auto --trust-remote-code",
+            "--trust-remote-code --max-model-len 4096",
+            "",
+        ]
+    )
+    def test_parse_params(self, params):
+        params_dict = utils.get_params_dict(params)
+        self.assertEqual(params, utils.build_params_string(params_dict))