22
33from __future__ import annotations
44
5- from typing import Optional
5+ from typing import Dict , Optional
66
77import httpx
88
@@ -88,6 +88,8 @@ def create(
8888 name : str ,
8989 scoring_contract : ScoringContractParam ,
9090 environment_parameters : Optional [ScenarioEnvironmentParam ] | NotGiven = NOT_GIVEN ,
91+ metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
92+ reference_output : Optional [str ] | NotGiven = NOT_GIVEN ,
9193 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
9294 # The extra values given here take precedence over values defined on the client or passed to this method.
9395 extra_headers : Headers | None = None ,
@@ -109,6 +111,12 @@ def create(
109111
110112 environment_parameters: The Environment in which the Scenario will run.
111113
114+ metadata: User defined metadata to attach to the scenario for organization.
115+
116+ reference_output: A string representation of the reference output to solve the scenario. Commonly
117+ can be the result of a git diff or a sequence of command actions to apply to the
118+ environment.
119+
112120 extra_headers: Send extra headers
113121
114122 extra_query: Add additional query parameters to the request
@@ -127,6 +135,8 @@ def create(
127135 "name" : name ,
128136 "scoring_contract" : scoring_contract ,
129137 "environment_parameters" : environment_parameters ,
138+ "metadata" : metadata ,
139+ "reference_output" : reference_output ,
130140 },
131141 scenario_create_params .ScenarioCreateParams ,
132142 ),
@@ -281,6 +291,7 @@ def start_run(
281291 * ,
282292 scenario_id : str ,
283293 benchmark_run_id : Optional [str ] | NotGiven = NOT_GIVEN ,
294+ metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
284295 run_name : Optional [str ] | NotGiven = NOT_GIVEN ,
285296 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
286297 # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -298,6 +309,8 @@ def start_run(
298309
299310 benchmark_run_id: Benchmark to associate the run.
300311
312+ metadata: User defined metadata to attach to the run for organization.
313+
301314 run_name: Display name of the run.
302315
303316 extra_headers: Send extra headers
@@ -316,6 +329,7 @@ def start_run(
316329 {
317330 "scenario_id" : scenario_id ,
318331 "benchmark_run_id" : benchmark_run_id ,
332+ "metadata" : metadata ,
319333 "run_name" : run_name ,
320334 },
321335 scenario_start_run_params .ScenarioStartRunParams ,
@@ -423,6 +437,8 @@ async def create(
423437 name : str ,
424438 scoring_contract : ScoringContractParam ,
425439 environment_parameters : Optional [ScenarioEnvironmentParam ] | NotGiven = NOT_GIVEN ,
440+ metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
441+ reference_output : Optional [str ] | NotGiven = NOT_GIVEN ,
426442 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
427443 # The extra values given here take precedence over values defined on the client or passed to this method.
428444 extra_headers : Headers | None = None ,
@@ -444,6 +460,12 @@ async def create(
444460
445461 environment_parameters: The Environment in which the Scenario will run.
446462
463+ metadata: User defined metadata to attach to the scenario for organization.
464+
465+ reference_output: A string representation of the reference output to solve the scenario. Commonly
466+ can be the result of a git diff or a sequence of command actions to apply to the
467+ environment.
468+
447469 extra_headers: Send extra headers
448470
449471 extra_query: Add additional query parameters to the request
@@ -462,6 +484,8 @@ async def create(
462484 "name" : name ,
463485 "scoring_contract" : scoring_contract ,
464486 "environment_parameters" : environment_parameters ,
487+ "metadata" : metadata ,
488+ "reference_output" : reference_output ,
465489 },
466490 scenario_create_params .ScenarioCreateParams ,
467491 ),
@@ -616,6 +640,7 @@ async def start_run(
616640 * ,
617641 scenario_id : str ,
618642 benchmark_run_id : Optional [str ] | NotGiven = NOT_GIVEN ,
643+ metadata : Optional [Dict [str , str ]] | NotGiven = NOT_GIVEN ,
619644 run_name : Optional [str ] | NotGiven = NOT_GIVEN ,
620645 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
621646 # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -633,6 +658,8 @@ async def start_run(
633658
634659 benchmark_run_id: Benchmark to associate the run.
635660
661+ metadata: User defined metadata to attach to the run for organization.
662+
636663 run_name: Display name of the run.
637664
638665 extra_headers: Send extra headers
@@ -651,6 +678,7 @@ async def start_run(
651678 {
652679 "scenario_id" : scenario_id ,
653680 "benchmark_run_id" : benchmark_run_id ,
681+ "metadata" : metadata ,
654682 "run_name" : run_name ,
655683 },
656684 scenario_start_run_params .ScenarioStartRunParams ,
0 commit comments