added env vars to launch response

Author: scarere

tests/vec_inf/cli/test_cli.py

@@ -40,6 +40,7 @@ def test_launch_command_success(runner):
             "model_weights_parent_dir": "/model-weights",
             "vocab_size": "128000",  # Changed to string
             "vllm_args": {"max_model_len": 8192},
+            "env": {"CACHE": "/cache"},
         }
         mock_client.launch_model.return_value = mock_response
 

tests/vec_inf/cli/test_helper.py

@@ -36,6 +36,7 @@ def test_format_table_output(self):
             "model_weights_parent_dir": "/model-weights",
             "log_dir": "/tmp/logs",
             "vllm_args": {"max_model_len": 8192, "enable_prefix_caching": True},
+            "env": {"CACHE": "/cache"},
         }
 
         formatter = LaunchResponseFormatter(model_name, params)
@@ -63,6 +64,7 @@ def test_format_table_output_with_minimal_params(self):
             "model_weights_parent_dir": "/weights",
             "log_dir": "/logs",
             "vllm_args": {},
+            "env": {},
         }
 
         formatter = LaunchResponseFormatter(model_name, params)

vec_inf/cli/_helper.py

@@ -79,6 +79,11 @@ def format_table_output(self) -> Table:
         for arg, value in self.params["vllm_args"].items():
             table.add_row(f"  {arg}:", str(value))
 
+        # Add Environment Variable Configuration Details
+        table.add_row("Environment Variables", style="magenta")
+        for arg, value in self.params["env"].items():
+            table.add_row(f"  {arg}:", str(value))
+
         return table