Drop env handling if entrypoint has args

Author: sjmonson

build/entrypoint.sh

@@ -1,17 +1,11 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
-# Required environment variables
-TARGET=${TARGET:-"http://localhost:8000"}
-MODEL=${MODEL:-"neuralmagic/Meta-Llama-3.1-8B-Instruct-quantized.w4a16"}
-RATE_TYPE=${RATE_TYPE:-"sweep"}
-DATA=${DATA:-"prompt_tokens=256,output_tokens=128"}
-MAX_REQUESTS=${MAX_REQUESTS:-"100"}
-MAX_SECONDS=${MAX_SECONDS:-""}
-
-# Output configuration
-OUTPUT_PATH=${OUTPUT_PATH:-"/results/guidellm_benchmark_results"}
-OUTPUT_FORMAT=${OUTPUT_FORMAT:-"json"}  # Can be json, yaml, or yml
+# If we receive any arguments switch to guidellm command
+if [ $# -gt 0 ]; then
+    echo "Running command: guidellm $*"
+    exec guidellm "$@"
+fi
 
 # Build the command
 CMD="guidellm benchmark --target \"${TARGET}\" --model \"${MODEL}\" --rate-type \"${RATE_TYPE}\" --data \"${DATA}\""
@@ -27,7 +21,7 @@ fi
 
 # Add output path with appropriate extension
 if [ ! -z "${OUTPUT_PATH}" ]; then
-    CMD="${CMD} --output-path \"${OUTPUT_PATH}.${OUTPUT_FORMAT}\""
+    CMD="${CMD} --output-path \"${OUTPUT_PATH}\""
 fi
 
 # Execute the command