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

Sign up to get free protection for your applications and to get access to all the features.
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
  """