22
33from __future__ import annotations
44
5- from typing import Dict , Optional
5+ from typing import Dict , List , Optional
66
77import httpx
88
@@ -90,6 +90,7 @@ def create(
9090 environment_parameters : Optional [ScenarioEnvironmentParam ] | NotGiven = NOT_GIVEN ,
9191 metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
9292 reference_output : Optional [str ] | NotGiven = NOT_GIVEN ,
93+ required_environment_variables : Optional [List [str ]] | NotGiven = NOT_GIVEN ,
9394 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
9495 # The extra values given here take precedence over values defined on the client or passed to this method.
9596 extra_headers : Headers | None = None ,
@@ -117,6 +118,9 @@ def create(
117118 can be the result of a git diff or a sequence of command actions to apply to the
118119 environment.
119120
121+ required_environment_variables: Environment variables required to run the scenario. If these variables are not
122+ provided, the scenario will fail to start.
123+
120124 extra_headers: Send extra headers
121125
122126 extra_query: Add additional query parameters to the request
@@ -137,6 +141,7 @@ def create(
137141 "environment_parameters" : environment_parameters ,
138142 "metadata" : metadata ,
139143 "reference_output" : reference_output ,
144+ "required_environment_variables" : required_environment_variables ,
140145 },
141146 scenario_create_params .ScenarioCreateParams ,
142147 ),
@@ -192,6 +197,7 @@ def update(
192197 metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
193198 name : Optional [str ] | NotGiven = NOT_GIVEN ,
194199 reference_output : Optional [str ] | NotGiven = NOT_GIVEN ,
200+ required_env_vars : Optional [List [str ]] | NotGiven = NOT_GIVEN ,
195201 scoring_contract : Optional [ScoringContractUpdateParam ] | NotGiven = NOT_GIVEN ,
196202 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
197203 # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -219,6 +225,8 @@ def update(
219225 can be the result of a git diff or a sequence of command actions to apply to the
220226 environment.
221227
228+ required_env_vars: Environment variables required to run the benchmark.
229+
222230 scoring_contract: The scoring contract for the Scenario.
223231
224232 extra_headers: Send extra headers
@@ -242,6 +250,7 @@ def update(
242250 "metadata" : metadata ,
243251 "name" : name ,
244252 "reference_output" : reference_output ,
253+ "required_env_vars" : required_env_vars ,
245254 "scoring_contract" : scoring_contract ,
246255 },
247256 scenario_update_params .ScenarioUpdateParams ,
@@ -369,6 +378,7 @@ def start_run(
369378 benchmark_run_id : Optional [str ] | NotGiven = NOT_GIVEN ,
370379 metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
371380 run_name : Optional [str ] | NotGiven = NOT_GIVEN ,
381+ run_profile : Optional [scenario_start_run_params .RunProfile ] | NotGiven = NOT_GIVEN ,
372382 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
373383 # The extra values given here take precedence over values defined on the client or passed to this method.
374384 extra_headers : Headers | None = None ,
@@ -389,6 +399,8 @@ def start_run(
389399
390400 run_name: Display name of the run.
391401
402+ run_profile: Runtime configuration to use for this benchmark run
403+
392404 extra_headers: Send extra headers
393405
394406 extra_query: Add additional query parameters to the request
@@ -407,6 +419,7 @@ def start_run(
407419 "benchmark_run_id" : benchmark_run_id ,
408420 "metadata" : metadata ,
409421 "run_name" : run_name ,
422+ "run_profile" : run_profile ,
410423 },
411424 scenario_start_run_params .ScenarioStartRunParams ,
412425 ),
@@ -513,6 +526,7 @@ async def create(
513526 environment_parameters : Optional [ScenarioEnvironmentParam ] | NotGiven = NOT_GIVEN ,
514527 metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
515528 reference_output : Optional [str ] | NotGiven = NOT_GIVEN ,
529+ required_environment_variables : Optional [List [str ]] | NotGiven = NOT_GIVEN ,
516530 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
517531 # The extra values given here take precedence over values defined on the client or passed to this method.
518532 extra_headers : Headers | None = None ,
@@ -540,6 +554,9 @@ async def create(
540554 can be the result of a git diff or a sequence of command actions to apply to the
541555 environment.
542556
557+ required_environment_variables: Environment variables required to run the scenario. If these variables are not
558+ provided, the scenario will fail to start.
559+
543560 extra_headers: Send extra headers
544561
545562 extra_query: Add additional query parameters to the request
@@ -560,6 +577,7 @@ async def create(
560577 "environment_parameters" : environment_parameters ,
561578 "metadata" : metadata ,
562579 "reference_output" : reference_output ,
580+ "required_environment_variables" : required_environment_variables ,
563581 },
564582 scenario_create_params .ScenarioCreateParams ,
565583 ),
@@ -615,6 +633,7 @@ async def update(
615633 metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
616634 name : Optional [str ] | NotGiven = NOT_GIVEN ,
617635 reference_output : Optional [str ] | NotGiven = NOT_GIVEN ,
636+ required_env_vars : Optional [List [str ]] | NotGiven = NOT_GIVEN ,
618637 scoring_contract : Optional [ScoringContractUpdateParam ] | NotGiven = NOT_GIVEN ,
619638 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
620639 # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -642,6 +661,8 @@ async def update(
642661 can be the result of a git diff or a sequence of command actions to apply to the
643662 environment.
644663
664+ required_env_vars: Environment variables required to run the benchmark.
665+
645666 scoring_contract: The scoring contract for the Scenario.
646667
647668 extra_headers: Send extra headers
@@ -665,6 +686,7 @@ async def update(
665686 "metadata" : metadata ,
666687 "name" : name ,
667688 "reference_output" : reference_output ,
689+ "required_env_vars" : required_env_vars ,
668690 "scoring_contract" : scoring_contract ,
669691 },
670692 scenario_update_params .ScenarioUpdateParams ,
@@ -792,6 +814,7 @@ async def start_run(
792814 benchmark_run_id : Optional [str ] | NotGiven = NOT_GIVEN ,
793815 metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
794816 run_name : Optional [str ] | NotGiven = NOT_GIVEN ,
817+ run_profile : Optional [scenario_start_run_params .RunProfile ] | NotGiven = NOT_GIVEN ,
795818 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
796819 # The extra values given here take precedence over values defined on the client or passed to this method.
797820 extra_headers : Headers | None = None ,
@@ -812,6 +835,8 @@ async def start_run(
812835
813836 run_name: Display name of the run.
814837
838+ run_profile: Runtime configuration to use for this benchmark run
839+
815840 extra_headers: Send extra headers
816841
817842 extra_query: Add additional query parameters to the request
@@ -830,6 +855,7 @@ async def start_run(
830855 "benchmark_run_id" : benchmark_run_id ,
831856 "metadata" : metadata ,
832857 "run_name" : run_name ,
858+ "run_profile" : run_profile ,
833859 },
834860 scenario_start_run_params .ScenarioStartRunParams ,
835861 ),
0 commit comments