vellum-ai 0.7.5__py3-none-any.whl → 0.7.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (35) hide show
  1. vellum/__init__.py +76 -72
  2. vellum/core/client_wrapper.py +1 -1
  3. vellum/resources/ml_models/client.py +6 -17
  4. vellum/types/__init__.py +80 -77
  5. vellum/types/ml_model_display_config_labelled.py +0 -1
  6. vellum/types/ml_model_display_config_request.py +0 -1
  7. vellum/types/ml_model_parameter_config.py +14 -14
  8. vellum/types/ml_model_parameter_config_request.py +14 -14
  9. vellum/types/ml_model_read.py +0 -6
  10. vellum/types/{ml_model_build_config.py → named_scenario_input_json_variable_value_request.py} +7 -2
  11. vellum/types/named_scenario_input_request.py +27 -1
  12. vellum/types/{array_parameter_config.py → open_api_array_property.py} +6 -6
  13. vellum/types/{array_parameter_config_request.py → open_api_array_property_request.py} +6 -6
  14. vellum/types/{boolean_parameter_config.py → open_api_boolean_property.py} +1 -1
  15. vellum/types/{boolean_parameter_config_request.py → open_api_boolean_property_request.py} +1 -1
  16. vellum/types/{const_parameter_config.py → open_api_const_property.py} +1 -1
  17. vellum/types/{const_parameter_config_request.py → open_api_const_property_request.py} +1 -1
  18. vellum/types/{integer_parameter_config.py → open_api_integer_property.py} +1 -1
  19. vellum/types/{integer_parameter_config_request.py → open_api_integer_property_request.py} +1 -1
  20. vellum/types/{number_parameter_config.py → open_api_number_property.py} +1 -1
  21. vellum/types/{number_parameter_config_request.py → open_api_number_property_request.py} +1 -1
  22. vellum/types/{object_parameter_config.py → open_api_object_property.py} +7 -7
  23. vellum/types/{object_parameter_config_request.py → open_api_object_property_request.py} +7 -7
  24. vellum/types/{one_of_parameter_config.py → open_api_one_of_property.py} +4 -4
  25. vellum/types/{one_of_parameter_config_request.py → open_api_one_of_property_request.py} +4 -4
  26. vellum/types/{parameter_config.py → open_api_property.py} +25 -25
  27. vellum/types/{parameter_config_request.py → open_api_property_request.py} +25 -25
  28. vellum/types/{string_parameter_config.py → open_api_string_property.py} +1 -1
  29. vellum/types/{string_parameter_config_request.py → open_api_string_property_request.py} +1 -1
  30. vellum/types/scenario_input.py +25 -1
  31. vellum/types/{ml_model_build_config_request.py → scenario_input_json_variable_value.py} +7 -2
  32. {vellum_ai-0.7.5.dist-info → vellum_ai-0.7.7.dist-info}/METADATA +1 -1
  33. {vellum_ai-0.7.5.dist-info → vellum_ai-0.7.7.dist-info}/RECORD +35 -35
  34. {vellum_ai-0.7.5.dist-info → vellum_ai-0.7.7.dist-info}/LICENSE +0 -0
  35. {vellum_ai-0.7.5.dist-info → vellum_ai-0.7.7.dist-info}/WHEEL +0 -0
vellum/types/__init__.py CHANGED
@@ -17,8 +17,6 @@ from .array_chat_message_content_item_request import (
17
17
  ArrayChatMessageContentItemRequest_String,
18
18
  )
19
19
  from .array_chat_message_content_request import ArrayChatMessageContentRequest
20
- from .array_parameter_config import ArrayParameterConfig
21
- from .array_parameter_config_request import ArrayParameterConfigRequest
22
20
  from .array_variable_value_item import (
23
21
  ArrayVariableValueItem,
24
22
  ArrayVariableValueItem_Error,
@@ -60,8 +58,6 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1 import
60
58
  from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request import (
61
59
  BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
62
60
  )
63
- from .boolean_parameter_config import BooleanParameterConfig
64
- from .boolean_parameter_config_request import BooleanParameterConfigRequest
65
61
  from .chat_history_input_request import ChatHistoryInputRequest
66
62
  from .chat_message import ChatMessage
67
63
  from .chat_message_content import (
@@ -105,8 +101,6 @@ from .compile_prompt_deployment_expand_meta_request import CompilePromptDeployme
105
101
  from .compile_prompt_meta import CompilePromptMeta
106
102
  from .conditional_node_result import ConditionalNodeResult
107
103
  from .conditional_node_result_data import ConditionalNodeResultData
108
- from .const_parameter_config import ConstParameterConfig
109
- from .const_parameter_config_request import ConstParameterConfigRequest
110
104
  from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
111
105
  from .deployment_provider_payload_response import DeploymentProviderPayloadResponse
112
106
  from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
@@ -237,8 +231,6 @@ from .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
237
231
  from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
238
232
  from .instructor_vectorizer_config import InstructorVectorizerConfig
239
233
  from .instructor_vectorizer_config_request import InstructorVectorizerConfigRequest
240
- from .integer_parameter_config import IntegerParameterConfig
241
- from .integer_parameter_config_request import IntegerParameterConfigRequest
242
234
  from .iteration_state_enum import IterationStateEnum
243
235
  from .json_input_request import JsonInputRequest
244
236
  from .json_variable_value import JsonVariableValue
@@ -254,8 +246,6 @@ from .metadata_filter_config_request import MetadataFilterConfigRequest
254
246
  from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
255
247
  from .metadata_filter_rule_request import MetadataFilterRuleRequest
256
248
  from .metric_node_result import MetricNodeResult
257
- from .ml_model_build_config import MlModelBuildConfig
258
- from .ml_model_build_config_request import MlModelBuildConfigRequest
259
249
  from .ml_model_developer import MlModelDeveloper
260
250
  from .ml_model_developer_enum_value_label import MlModelDeveloperEnumValueLabel
261
251
  from .ml_model_display_config_labelled import MlModelDisplayConfigLabelled
@@ -289,9 +279,11 @@ from .ml_model_tokenizer_config_request import (
289
279
  )
290
280
  from .ml_model_usage import MlModelUsage
291
281
  from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
282
+ from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
292
283
  from .named_scenario_input_request import (
293
284
  NamedScenarioInputRequest,
294
285
  NamedScenarioInputRequest_ChatHistory,
286
+ NamedScenarioInputRequest_Json,
295
287
  NamedScenarioInputRequest_String,
296
288
  )
297
289
  from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
@@ -373,15 +365,9 @@ from .node_output_compiled_value import (
373
365
  )
374
366
  from .normalized_log_probs import NormalizedLogProbs
375
367
  from .normalized_token_log_probs import NormalizedTokenLogProbs
376
- from .number_parameter_config import NumberParameterConfig
377
- from .number_parameter_config_request import NumberParameterConfigRequest
378
368
  from .number_variable_value import NumberVariableValue
379
369
  from .number_vellum_value import NumberVellumValue
380
370
  from .number_vellum_value_request import NumberVellumValueRequest
381
- from .object_parameter_config import ObjectParameterConfig
382
- from .object_parameter_config_request import ObjectParameterConfigRequest
383
- from .one_of_parameter_config import OneOfParameterConfig
384
- from .one_of_parameter_config_request import OneOfParameterConfigRequest
385
371
  from .open_ai_vectorizer_config import OpenAiVectorizerConfig
386
372
  from .open_ai_vectorizer_config_request import OpenAiVectorizerConfigRequest
387
373
  from .open_ai_vectorizer_text_embedding_3_large import OpenAiVectorizerTextEmbedding3Large
@@ -390,6 +376,44 @@ from .open_ai_vectorizer_text_embedding_3_small import OpenAiVectorizerTextEmbed
390
376
  from .open_ai_vectorizer_text_embedding_3_small_request import OpenAiVectorizerTextEmbedding3SmallRequest
391
377
  from .open_ai_vectorizer_text_embedding_ada_002 import OpenAiVectorizerTextEmbeddingAda002
392
378
  from .open_ai_vectorizer_text_embedding_ada_002_request import OpenAiVectorizerTextEmbeddingAda002Request
379
+ from .open_api_array_property import OpenApiArrayProperty
380
+ from .open_api_array_property_request import OpenApiArrayPropertyRequest
381
+ from .open_api_boolean_property import OpenApiBooleanProperty
382
+ from .open_api_boolean_property_request import OpenApiBooleanPropertyRequest
383
+ from .open_api_const_property import OpenApiConstProperty
384
+ from .open_api_const_property_request import OpenApiConstPropertyRequest
385
+ from .open_api_integer_property import OpenApiIntegerProperty
386
+ from .open_api_integer_property_request import OpenApiIntegerPropertyRequest
387
+ from .open_api_number_property import OpenApiNumberProperty
388
+ from .open_api_number_property_request import OpenApiNumberPropertyRequest
389
+ from .open_api_object_property import OpenApiObjectProperty
390
+ from .open_api_object_property_request import OpenApiObjectPropertyRequest
391
+ from .open_api_one_of_property import OpenApiOneOfProperty
392
+ from .open_api_one_of_property_request import OpenApiOneOfPropertyRequest
393
+ from .open_api_property import (
394
+ OpenApiProperty,
395
+ OpenApiProperty_Array,
396
+ OpenApiProperty_Boolean,
397
+ OpenApiProperty_Const,
398
+ OpenApiProperty_Integer,
399
+ OpenApiProperty_Number,
400
+ OpenApiProperty_Object,
401
+ OpenApiProperty_OneOf,
402
+ OpenApiProperty_String,
403
+ )
404
+ from .open_api_property_request import (
405
+ OpenApiPropertyRequest,
406
+ OpenApiPropertyRequest_Array,
407
+ OpenApiPropertyRequest_Boolean,
408
+ OpenApiPropertyRequest_Const,
409
+ OpenApiPropertyRequest_Integer,
410
+ OpenApiPropertyRequest_Number,
411
+ OpenApiPropertyRequest_Object,
412
+ OpenApiPropertyRequest_OneOf,
413
+ OpenApiPropertyRequest_String,
414
+ )
415
+ from .open_api_string_property import OpenApiStringProperty
416
+ from .open_api_string_property_request import OpenApiStringPropertyRequest
393
417
  from .paginated_document_index_read_list import PaginatedDocumentIndexReadList
394
418
  from .paginated_ml_model_read_list import PaginatedMlModelReadList
395
419
  from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
@@ -397,28 +421,6 @@ from .paginated_slim_document_list import PaginatedSlimDocumentList
397
421
  from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
398
422
  from .paginated_test_suite_run_execution_list import PaginatedTestSuiteRunExecutionList
399
423
  from .paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
400
- from .parameter_config import (
401
- ParameterConfig,
402
- ParameterConfig_Array,
403
- ParameterConfig_Boolean,
404
- ParameterConfig_Const,
405
- ParameterConfig_Integer,
406
- ParameterConfig_Number,
407
- ParameterConfig_Object,
408
- ParameterConfig_OneOf,
409
- ParameterConfig_String,
410
- )
411
- from .parameter_config_request import (
412
- ParameterConfigRequest,
413
- ParameterConfigRequest_Array,
414
- ParameterConfigRequest_Boolean,
415
- ParameterConfigRequest_Const,
416
- ParameterConfigRequest_Integer,
417
- ParameterConfigRequest_Number,
418
- ParameterConfigRequest_Object,
419
- ParameterConfigRequest_OneOf,
420
- ParameterConfigRequest_String,
421
- )
422
424
  from .pdf_search_result_meta_source import PdfSearchResultMetaSource
423
425
  from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequest
424
426
  from .processing_failure_reason_enum import ProcessingFailureReasonEnum
@@ -454,8 +456,9 @@ from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
454
456
  from .release_tag_source import ReleaseTagSource
455
457
  from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
456
458
  from .sandbox_scenario import SandboxScenario
457
- from .scenario_input import ScenarioInput, ScenarioInput_ChatHistory, ScenarioInput_String
459
+ from .scenario_input import ScenarioInput, ScenarioInput_ChatHistory, ScenarioInput_Json, ScenarioInput_String
458
460
  from .scenario_input_chat_history_variable_value import ScenarioInputChatHistoryVariableValue
461
+ from .scenario_input_json_variable_value import ScenarioInputJsonVariableValue
459
462
  from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
460
463
  from .search_filters_request import SearchFiltersRequest
461
464
  from .search_node_result import SearchNodeResult
@@ -485,8 +488,6 @@ from .streaming_workflow_node_result_event import StreamingWorkflowNodeResultEve
485
488
  from .string_chat_message_content import StringChatMessageContent
486
489
  from .string_chat_message_content_request import StringChatMessageContentRequest
487
490
  from .string_input_request import StringInputRequest
488
- from .string_parameter_config import StringParameterConfig
489
- from .string_parameter_config_request import StringParameterConfigRequest
490
491
  from .string_variable_value import StringVariableValue
491
492
  from .string_vellum_value import StringVellumValue
492
493
  from .string_vellum_value_request import StringVellumValueRequest
@@ -767,8 +768,6 @@ __all__ = [
767
768
  "ArrayChatMessageContentItem_Image",
768
769
  "ArrayChatMessageContentItem_String",
769
770
  "ArrayChatMessageContentRequest",
770
- "ArrayParameterConfig",
771
- "ArrayParameterConfigRequest",
772
771
  "ArrayVariableValueItem",
773
772
  "ArrayVariableValueItem_Error",
774
773
  "ArrayVariableValueItem_FunctionCall",
@@ -796,8 +795,6 @@ __all__ = [
796
795
  "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
797
796
  "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
798
797
  "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
799
- "BooleanParameterConfig",
800
- "BooleanParameterConfigRequest",
801
798
  "ChatHistoryInputRequest",
802
799
  "ChatMessage",
803
800
  "ChatMessageContent",
@@ -835,8 +832,6 @@ __all__ = [
835
832
  "CompilePromptMeta",
836
833
  "ConditionalNodeResult",
837
834
  "ConditionalNodeResultData",
838
- "ConstParameterConfig",
839
- "ConstParameterConfigRequest",
840
835
  "CreateTestSuiteTestCaseRequest",
841
836
  "DeploymentProviderPayloadResponse",
842
837
  "DeploymentProviderPayloadResponsePayload",
@@ -951,8 +946,6 @@ __all__ = [
951
946
  "InitiatedWorkflowNodeResultEvent",
952
947
  "InstructorVectorizerConfig",
953
948
  "InstructorVectorizerConfigRequest",
954
- "IntegerParameterConfig",
955
- "IntegerParameterConfigRequest",
956
949
  "IterationStateEnum",
957
950
  "JsonInputRequest",
958
951
  "JsonVariableValue",
@@ -968,8 +961,6 @@ __all__ = [
968
961
  "MetadataFilterRuleCombinator",
969
962
  "MetadataFilterRuleRequest",
970
963
  "MetricNodeResult",
971
- "MlModelBuildConfig",
972
- "MlModelBuildConfigRequest",
973
964
  "MlModelDeveloper",
974
965
  "MlModelDeveloperEnumValueLabel",
975
966
  "MlModelDisplayConfigLabelled",
@@ -999,8 +990,10 @@ __all__ = [
999
990
  "MlModelTokenizerConfig_Tiktoken",
1000
991
  "MlModelUsage",
1001
992
  "NamedScenarioInputChatHistoryVariableValueRequest",
993
+ "NamedScenarioInputJsonVariableValueRequest",
1002
994
  "NamedScenarioInputRequest",
1003
995
  "NamedScenarioInputRequest_ChatHistory",
996
+ "NamedScenarioInputRequest_Json",
1004
997
  "NamedScenarioInputRequest_String",
1005
998
  "NamedScenarioInputStringVariableValueRequest",
1006
999
  "NamedTestCaseArrayVariableValue",
@@ -1073,15 +1066,9 @@ __all__ = [
1073
1066
  "NodeOutputCompiledValue_String",
1074
1067
  "NormalizedLogProbs",
1075
1068
  "NormalizedTokenLogProbs",
1076
- "NumberParameterConfig",
1077
- "NumberParameterConfigRequest",
1078
1069
  "NumberVariableValue",
1079
1070
  "NumberVellumValue",
1080
1071
  "NumberVellumValueRequest",
1081
- "ObjectParameterConfig",
1082
- "ObjectParameterConfigRequest",
1083
- "OneOfParameterConfig",
1084
- "OneOfParameterConfigRequest",
1085
1072
  "OpenAiVectorizerConfig",
1086
1073
  "OpenAiVectorizerConfigRequest",
1087
1074
  "OpenAiVectorizerTextEmbedding3Large",
@@ -1090,6 +1077,40 @@ __all__ = [
1090
1077
  "OpenAiVectorizerTextEmbedding3SmallRequest",
1091
1078
  "OpenAiVectorizerTextEmbeddingAda002",
1092
1079
  "OpenAiVectorizerTextEmbeddingAda002Request",
1080
+ "OpenApiArrayProperty",
1081
+ "OpenApiArrayPropertyRequest",
1082
+ "OpenApiBooleanProperty",
1083
+ "OpenApiBooleanPropertyRequest",
1084
+ "OpenApiConstProperty",
1085
+ "OpenApiConstPropertyRequest",
1086
+ "OpenApiIntegerProperty",
1087
+ "OpenApiIntegerPropertyRequest",
1088
+ "OpenApiNumberProperty",
1089
+ "OpenApiNumberPropertyRequest",
1090
+ "OpenApiObjectProperty",
1091
+ "OpenApiObjectPropertyRequest",
1092
+ "OpenApiOneOfProperty",
1093
+ "OpenApiOneOfPropertyRequest",
1094
+ "OpenApiProperty",
1095
+ "OpenApiPropertyRequest",
1096
+ "OpenApiPropertyRequest_Array",
1097
+ "OpenApiPropertyRequest_Boolean",
1098
+ "OpenApiPropertyRequest_Const",
1099
+ "OpenApiPropertyRequest_Integer",
1100
+ "OpenApiPropertyRequest_Number",
1101
+ "OpenApiPropertyRequest_Object",
1102
+ "OpenApiPropertyRequest_OneOf",
1103
+ "OpenApiPropertyRequest_String",
1104
+ "OpenApiProperty_Array",
1105
+ "OpenApiProperty_Boolean",
1106
+ "OpenApiProperty_Const",
1107
+ "OpenApiProperty_Integer",
1108
+ "OpenApiProperty_Number",
1109
+ "OpenApiProperty_Object",
1110
+ "OpenApiProperty_OneOf",
1111
+ "OpenApiProperty_String",
1112
+ "OpenApiStringProperty",
1113
+ "OpenApiStringPropertyRequest",
1093
1114
  "PaginatedDocumentIndexReadList",
1094
1115
  "PaginatedMlModelReadList",
1095
1116
  "PaginatedSlimDeploymentReadList",
@@ -1097,24 +1118,6 @@ __all__ = [
1097
1118
  "PaginatedSlimWorkflowDeploymentList",
1098
1119
  "PaginatedTestSuiteRunExecutionList",
1099
1120
  "PaginatedTestSuiteTestCaseList",
1100
- "ParameterConfig",
1101
- "ParameterConfigRequest",
1102
- "ParameterConfigRequest_Array",
1103
- "ParameterConfigRequest_Boolean",
1104
- "ParameterConfigRequest_Const",
1105
- "ParameterConfigRequest_Integer",
1106
- "ParameterConfigRequest_Number",
1107
- "ParameterConfigRequest_Object",
1108
- "ParameterConfigRequest_OneOf",
1109
- "ParameterConfigRequest_String",
1110
- "ParameterConfig_Array",
1111
- "ParameterConfig_Boolean",
1112
- "ParameterConfig_Const",
1113
- "ParameterConfig_Integer",
1114
- "ParameterConfig_Number",
1115
- "ParameterConfig_Object",
1116
- "ParameterConfig_OneOf",
1117
- "ParameterConfig_String",
1118
1121
  "PdfSearchResultMetaSource",
1119
1122
  "PdfSearchResultMetaSourceRequest",
1120
1123
  "ProcessingFailureReasonEnum",
@@ -1148,8 +1151,10 @@ __all__ = [
1148
1151
  "SandboxScenario",
1149
1152
  "ScenarioInput",
1150
1153
  "ScenarioInputChatHistoryVariableValue",
1154
+ "ScenarioInputJsonVariableValue",
1151
1155
  "ScenarioInputStringVariableValue",
1152
1156
  "ScenarioInput_ChatHistory",
1157
+ "ScenarioInput_Json",
1153
1158
  "ScenarioInput_String",
1154
1159
  "SearchFiltersRequest",
1155
1160
  "SearchNodeResult",
@@ -1181,8 +1186,6 @@ __all__ = [
1181
1186
  "StringChatMessageContent",
1182
1187
  "StringChatMessageContentRequest",
1183
1188
  "StringInputRequest",
1184
- "StringParameterConfig",
1185
- "StringParameterConfigRequest",
1186
1189
  "StringVariableValue",
1187
1190
  "StringVellumValue",
1188
1191
  "StringVellumValueRequest",
@@ -12,7 +12,6 @@ class MlModelDisplayConfigLabelled(pydantic_v1.BaseModel):
12
12
  label: str
13
13
  description: str
14
14
  tags: typing.List[MlModelDisplayTagEnumValueLabel]
15
- default_display_priority: float
16
15
 
17
16
  def json(self, **kwargs: typing.Any) -> str:
18
17
  kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -12,7 +12,6 @@ class MlModelDisplayConfigRequest(pydantic_v1.BaseModel):
12
12
  label: str
13
13
  description: str
14
14
  tags: typing.List[MlModelDisplayTag]
15
- default_display_priority: float
16
15
 
17
16
  def json(self, **kwargs: typing.Any) -> str:
18
17
  kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -5,23 +5,23 @@ import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
- from .array_parameter_config import ArrayParameterConfig
9
- from .integer_parameter_config import IntegerParameterConfig
10
- from .number_parameter_config import NumberParameterConfig
11
- from .object_parameter_config import ObjectParameterConfig
12
- from .parameter_config import ParameterConfig
8
+ from .open_api_array_property import OpenApiArrayProperty
9
+ from .open_api_integer_property import OpenApiIntegerProperty
10
+ from .open_api_number_property import OpenApiNumberProperty
11
+ from .open_api_object_property import OpenApiObjectProperty
12
+ from .open_api_property import OpenApiProperty
13
13
 
14
14
 
15
15
  class MlModelParameterConfig(pydantic_v1.BaseModel):
16
- temperature: typing.Optional[NumberParameterConfig] = None
17
- max_tokens: typing.Optional[IntegerParameterConfig] = None
18
- stop: typing.Optional[ArrayParameterConfig] = None
19
- top_p: typing.Optional[NumberParameterConfig] = None
20
- top_k: typing.Optional[IntegerParameterConfig] = None
21
- frequency_penalty: typing.Optional[NumberParameterConfig] = None
22
- presence_penalty: typing.Optional[NumberParameterConfig] = None
23
- logit_bias: typing.Optional[ObjectParameterConfig] = None
24
- custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[ParameterConfig]]] = None
16
+ temperature: typing.Optional[OpenApiNumberProperty] = None
17
+ max_tokens: typing.Optional[OpenApiIntegerProperty] = None
18
+ stop: typing.Optional[OpenApiArrayProperty] = None
19
+ top_p: typing.Optional[OpenApiNumberProperty] = None
20
+ top_k: typing.Optional[OpenApiIntegerProperty] = None
21
+ frequency_penalty: typing.Optional[OpenApiNumberProperty] = None
22
+ presence_penalty: typing.Optional[OpenApiNumberProperty] = None
23
+ logit_bias: typing.Optional[OpenApiObjectProperty] = None
24
+ custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[OpenApiProperty]]] = None
25
25
 
26
26
  def json(self, **kwargs: typing.Any) -> str:
27
27
  kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -5,23 +5,23 @@ import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
- from .array_parameter_config_request import ArrayParameterConfigRequest
9
- from .integer_parameter_config_request import IntegerParameterConfigRequest
10
- from .number_parameter_config_request import NumberParameterConfigRequest
11
- from .object_parameter_config_request import ObjectParameterConfigRequest
12
- from .parameter_config_request import ParameterConfigRequest
8
+ from .open_api_array_property_request import OpenApiArrayPropertyRequest
9
+ from .open_api_integer_property_request import OpenApiIntegerPropertyRequest
10
+ from .open_api_number_property_request import OpenApiNumberPropertyRequest
11
+ from .open_api_object_property_request import OpenApiObjectPropertyRequest
12
+ from .open_api_property_request import OpenApiPropertyRequest
13
13
 
14
14
 
15
15
  class MlModelParameterConfigRequest(pydantic_v1.BaseModel):
16
- temperature: typing.Optional[NumberParameterConfigRequest] = None
17
- max_tokens: typing.Optional[IntegerParameterConfigRequest] = None
18
- stop: typing.Optional[ArrayParameterConfigRequest] = None
19
- top_p: typing.Optional[NumberParameterConfigRequest] = None
20
- top_k: typing.Optional[IntegerParameterConfigRequest] = None
21
- frequency_penalty: typing.Optional[NumberParameterConfigRequest] = None
22
- presence_penalty: typing.Optional[NumberParameterConfigRequest] = None
23
- logit_bias: typing.Optional[ObjectParameterConfigRequest] = None
24
- custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[ParameterConfigRequest]]] = None
16
+ temperature: typing.Optional[OpenApiNumberPropertyRequest] = None
17
+ max_tokens: typing.Optional[OpenApiIntegerPropertyRequest] = None
18
+ stop: typing.Optional[OpenApiArrayPropertyRequest] = None
19
+ top_p: typing.Optional[OpenApiNumberPropertyRequest] = None
20
+ top_k: typing.Optional[OpenApiIntegerPropertyRequest] = None
21
+ frequency_penalty: typing.Optional[OpenApiNumberPropertyRequest] = None
22
+ presence_penalty: typing.Optional[OpenApiNumberPropertyRequest] = None
23
+ logit_bias: typing.Optional[OpenApiObjectPropertyRequest] = None
24
+ custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[OpenApiPropertyRequest]]] = None
25
25
 
26
26
  def json(self, **kwargs: typing.Any) -> str:
27
27
  kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -6,7 +6,6 @@ import typing
6
6
  from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .hosted_by_enum import HostedByEnum
9
- from .ml_model_build_config import MlModelBuildConfig
10
9
  from .ml_model_developer_enum_value_label import MlModelDeveloperEnumValueLabel
11
10
  from .ml_model_display_config_labelled import MlModelDisplayConfigLabelled
12
11
  from .ml_model_exec_config import MlModelExecConfig
@@ -69,11 +68,6 @@ class MlModelRead(pydantic_v1.BaseModel):
69
68
  - `DISABLED` - Disabled
70
69
  """
71
70
 
72
- build_config: MlModelBuildConfig = pydantic_v1.Field()
73
- """
74
- Configuration for how the ML Model was built.
75
- """
76
-
77
71
  exec_config: MlModelExecConfig = pydantic_v1.Field()
78
72
  """
79
73
  Configuration for how to execute the ML Model.
@@ -7,8 +7,13 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class MlModelBuildConfig(pydantic_v1.BaseModel):
11
- finetune_data: typing.Dict[str, typing.Any]
10
+ class NamedScenarioInputJsonVariableValueRequest(pydantic_v1.BaseModel):
11
+ """
12
+ Named Prompt Sandbox Scenario input value that is of type JSON
13
+ """
14
+
15
+ value: typing.Any
16
+ name: str
12
17
 
13
18
  def json(self, **kwargs: typing.Any) -> str:
14
19
  kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -34,6 +34,30 @@ class NamedScenarioInputRequest_String(pydantic_v1.BaseModel):
34
34
  json_encoders = {dt.datetime: serialize_datetime}
35
35
 
36
36
 
37
+ class NamedScenarioInputRequest_Json(pydantic_v1.BaseModel):
38
+ value: typing.Any
39
+ name: str
40
+ type: typing.Literal["JSON"] = "JSON"
41
+
42
+ def json(self, **kwargs: typing.Any) -> str:
43
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
44
+ return super().json(**kwargs_with_defaults)
45
+
46
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
47
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
48
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
49
+
50
+ return deep_union_pydantic_dicts(
51
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
52
+ )
53
+
54
+ class Config:
55
+ frozen = True
56
+ smart_union = True
57
+ extra = pydantic_v1.Extra.allow
58
+ json_encoders = {dt.datetime: serialize_datetime}
59
+
60
+
37
61
  class NamedScenarioInputRequest_ChatHistory(pydantic_v1.BaseModel):
38
62
  value: typing.Optional[typing.List[ChatMessageRequest]] = None
39
63
  name: str
@@ -58,4 +82,6 @@ class NamedScenarioInputRequest_ChatHistory(pydantic_v1.BaseModel):
58
82
  json_encoders = {dt.datetime: serialize_datetime}
59
83
 
60
84
 
61
- NamedScenarioInputRequest = typing.Union[NamedScenarioInputRequest_String, NamedScenarioInputRequest_ChatHistory]
85
+ NamedScenarioInputRequest = typing.Union[
86
+ NamedScenarioInputRequest_String, NamedScenarioInputRequest_Json, NamedScenarioInputRequest_ChatHistory
87
+ ]
@@ -9,7 +9,7 @@ from ..core.datetime_utils import serialize_datetime
9
9
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
10
10
 
11
11
 
12
- class ArrayParameterConfig(pydantic_v1.BaseModel):
12
+ class OpenApiArrayProperty(pydantic_v1.BaseModel):
13
13
  """
14
14
  An OpenAPI specification of a parameter with type 'array'
15
15
  """
@@ -17,9 +17,9 @@ class ArrayParameterConfig(pydantic_v1.BaseModel):
17
17
  min_items: typing.Optional[int] = None
18
18
  max_items: typing.Optional[int] = None
19
19
  unique_items: typing.Optional[bool] = None
20
- items: ParameterConfig
21
- prefix_items: typing.Optional[typing.List[ParameterConfig]] = None
22
- contains: typing.Optional[ParameterConfig] = None
20
+ items: OpenApiProperty
21
+ prefix_items: typing.Optional[typing.List[OpenApiProperty]] = None
22
+ contains: typing.Optional[OpenApiProperty] = None
23
23
  min_contains: typing.Optional[int] = None
24
24
  max_contains: typing.Optional[int] = None
25
25
  default: typing.Optional[typing.List[typing.Any]] = None
@@ -45,6 +45,6 @@ class ArrayParameterConfig(pydantic_v1.BaseModel):
45
45
  json_encoders = {dt.datetime: serialize_datetime}
46
46
 
47
47
 
48
- from .parameter_config import ParameterConfig # noqa: E402
48
+ from .open_api_property import OpenApiProperty # noqa: E402
49
49
 
50
- ArrayParameterConfig.update_forward_refs()
50
+ OpenApiArrayProperty.update_forward_refs()
@@ -9,7 +9,7 @@ from ..core.datetime_utils import serialize_datetime
9
9
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
10
10
 
11
11
 
12
- class ArrayParameterConfigRequest(pydantic_v1.BaseModel):
12
+ class OpenApiArrayPropertyRequest(pydantic_v1.BaseModel):
13
13
  """
14
14
  An OpenAPI specification of a parameter with type 'array'
15
15
  """
@@ -17,9 +17,9 @@ class ArrayParameterConfigRequest(pydantic_v1.BaseModel):
17
17
  min_items: typing.Optional[int] = None
18
18
  max_items: typing.Optional[int] = None
19
19
  unique_items: typing.Optional[bool] = None
20
- items: ParameterConfigRequest
21
- prefix_items: typing.Optional[typing.List[ParameterConfigRequest]] = None
22
- contains: typing.Optional[ParameterConfigRequest] = None
20
+ items: OpenApiPropertyRequest
21
+ prefix_items: typing.Optional[typing.List[OpenApiPropertyRequest]] = None
22
+ contains: typing.Optional[OpenApiPropertyRequest] = None
23
23
  min_contains: typing.Optional[int] = None
24
24
  max_contains: typing.Optional[int] = None
25
25
  default: typing.Optional[typing.List[typing.Any]] = None
@@ -45,6 +45,6 @@ class ArrayParameterConfigRequest(pydantic_v1.BaseModel):
45
45
  json_encoders = {dt.datetime: serialize_datetime}
46
46
 
47
47
 
48
- from .parameter_config_request import ParameterConfigRequest # noqa: E402
48
+ from .open_api_property_request import OpenApiPropertyRequest # noqa: E402
49
49
 
50
- ArrayParameterConfigRequest.update_forward_refs()
50
+ OpenApiArrayPropertyRequest.update_forward_refs()
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class BooleanParameterConfig(pydantic_v1.BaseModel):
10
+ class OpenApiBooleanProperty(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'boolean'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class BooleanParameterConfigRequest(pydantic_v1.BaseModel):
10
+ class OpenApiBooleanPropertyRequest(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'boolean'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class ConstParameterConfig(pydantic_v1.BaseModel):
10
+ class OpenApiConstProperty(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'const'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class ConstParameterConfigRequest(pydantic_v1.BaseModel):
10
+ class OpenApiConstPropertyRequest(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'const'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class IntegerParameterConfig(pydantic_v1.BaseModel):
10
+ class OpenApiIntegerProperty(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'integer'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class IntegerParameterConfigRequest(pydantic_v1.BaseModel):
10
+ class OpenApiIntegerPropertyRequest(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'integer'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class NumberParameterConfig(pydantic_v1.BaseModel):
10
+ class OpenApiNumberProperty(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'number'
13
13
  """
@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
7
7
  from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
- class NumberParameterConfigRequest(pydantic_v1.BaseModel):
10
+ class OpenApiNumberPropertyRequest(pydantic_v1.BaseModel):
11
11
  """
12
12
  An OpenAPI specification of a parameter with type 'number'
13
13
  """