Fix check required field for batch launch

Author: XkunW

vec_inf/client/_helper.py

@@ -554,11 +554,17 @@ def _get_launch_params(
                     raise ValueError(
                         f"Mismatch found for {arg}: {params[arg]} != {params['models'][model_name][arg]}, check your configuration"
                     )
-            # Check for required fields, will raise an error if missing any
-            utils.check_required_fields(
+            # Check for required fields and return environment variable overrides
+            env_overrides = utils.check_required_fields(
                 {**params["models"][model_name], **common_params}
             )
 
+            for arg, value in env_overrides.items():
+                if arg in common_params:
+                    params[arg] = value
+                else:
+                    params["models"][model_name][arg] = value
+
         return params
 
     def _build_launch_command(self) -> str:

vec_inf/client/_utils.py

@@ -436,20 +436,23 @@ def find_matching_dirs(
     return matched
 
 
-def check_required_fields(params: dict[str, Any]) -> None:
+def check_required_fields(params: dict[str, Any]) -> dict[str, Any]:
     """Check for required fields without default vals and their corresponding env vars.
 
     Parameters
     ----------
     params : dict[str, Any]
         Dictionary of parameters to check.
     """
+    env_overrides = {}
     for arg in REQUIRED_ARGS:
         if not params.get(arg):
             default_value = os.getenv(REQUIRED_ARGS[arg])
             if default_value:
                 params[arg] = default_value
+                env_overrides[arg] = default_value
             else:
                 raise MissingRequiredFieldsError(
                     f"{arg} is required, please set it in the command arguments or environment variables"
                 )
+    return env_overrides