diff --git a/genai-perf/README.md b/genai-perf/README.md index 6e7d6f21..0f2609aa 100644 --- a/genai-perf/README.md +++ b/genai-perf/README.md @@ -346,7 +346,7 @@ key authentication. To do so, you must add your API key directly in the command. Add the following flag to your command. ```bash --h "Authorization: Bearer ${API_KEY}" -H "Accept: text/event-stream" +-H "Authorization: Bearer ${API_KEY}" -H "Accept: text/event-stream" ```
@@ -467,7 +467,7 @@ Alternatively, a string representing a json formatted dict can be provided. (default: `None`) ##### `--header ` -##### `--h ` +##### `--H ` Add a custom header to the requests. Headers must be specified as 'Header:Value'. You can repeat this flag for multiple headers. (default: `None`) diff --git a/genai-perf/genai_perf/parser.py b/genai-perf/genai_perf/parser.py index 35ad00fb..d147eee1 100644 --- a/genai-perf/genai_perf/parser.py +++ b/genai-perf/genai_perf/parser.py @@ -753,7 +753,7 @@ def _add_input_args(parser): input_group.add_argument( "--header", - "-h", + "-H", action="append", help="Add a custom header to the requests. " "Headers must be specified as 'Header:Value'. " diff --git a/genai-perf/tests/test_cli.py b/genai-perf/tests/test_cli.py index 12751f4f..8655435d 100644 --- a/genai-perf/tests/test_cli.py +++ b/genai-perf/tests/test_cli.py @@ -186,7 +186,7 @@ def test_help_version_arguments_output_and_exit( ] }, ), - (["-h", "header_name:value"], {"header": ["header_name:value"]}), + (["-H", "header_name:value"], {"header": ["header_name:value"]}), (["--header", "header_name:value"], {"header": ["header_name:value"]}), ( ["--header", "header_name:value", "--header", "header_name_2:value_2"],