vellum-ai 0.7.8__py3-none-any.whl → 0.7.10__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 (40) hide show
  1. vellum/__init__.py +8 -0
  2. vellum/core/client_wrapper.py +1 -1
  3. vellum/resources/ml_models/client.py +56 -4
  4. vellum/terraform/__init__.py +2 -0
  5. vellum/terraform/_jsii/vellum-ai_vellum@0.0.0.jsii.tgz +0 -0
  6. vellum/terraform/data_vellum_document_index/__init__.py +10 -10
  7. vellum/terraform/data_vellum_ml_model/__init__.py +409 -0
  8. vellum/terraform/document_index/__init__.py +16 -16
  9. vellum/terraform/ml_model/__init__.py +806 -0
  10. vellum/terraform/provider/__init__.py +13 -13
  11. vellum/terraform/versions.json +1 -1
  12. vellum/types/__init__.py +8 -0
  13. vellum/types/ml_model_display_config_labelled.py +4 -0
  14. vellum/types/ml_model_display_config_request.py +4 -0
  15. vellum/types/ml_model_exec_config.py +5 -0
  16. vellum/types/ml_model_exec_config_request.py +5 -0
  17. vellum/types/open_api_array_property.py +1 -1
  18. vellum/types/open_api_array_property_request.py +1 -1
  19. vellum/types/open_api_boolean_property.py +1 -1
  20. vellum/types/open_api_boolean_property_request.py +1 -1
  21. vellum/types/open_api_const_property.py +1 -1
  22. vellum/types/open_api_const_property_request.py +1 -1
  23. vellum/types/open_api_integer_property.py +1 -1
  24. vellum/types/open_api_integer_property_request.py +1 -1
  25. vellum/types/open_api_number_property.py +1 -1
  26. vellum/types/open_api_number_property_request.py +1 -1
  27. vellum/types/open_api_object_property.py +1 -1
  28. vellum/types/open_api_object_property_request.py +1 -1
  29. vellum/types/open_api_one_of_property.py +1 -1
  30. vellum/types/open_api_one_of_property_request.py +1 -1
  31. vellum/types/open_api_property.py +26 -0
  32. vellum/types/open_api_property_request.py +26 -0
  33. vellum/types/open_api_ref_property.py +35 -0
  34. vellum/types/open_api_ref_property_request.py +35 -0
  35. vellum/types/open_api_string_property.py +1 -1
  36. vellum/types/open_api_string_property_request.py +1 -1
  37. {vellum_ai-0.7.8.dist-info → vellum_ai-0.7.10.dist-info}/METADATA +1 -1
  38. {vellum_ai-0.7.8.dist-info → vellum_ai-0.7.10.dist-info}/RECORD +40 -36
  39. {vellum_ai-0.7.8.dist-info → vellum_ai-0.7.10.dist-info}/LICENSE +0 -0
  40. {vellum_ai-0.7.8.dist-info → vellum_ai-0.7.10.dist-info}/WHEEL +0 -0
vellum/__init__.py CHANGED
@@ -346,6 +346,7 @@ from .types import (
346
346
  OpenApiPropertyRequest_Number,
347
347
  OpenApiPropertyRequest_Object,
348
348
  OpenApiPropertyRequest_OneOf,
349
+ OpenApiPropertyRequest_Ref,
349
350
  OpenApiPropertyRequest_String,
350
351
  OpenApiProperty_Array,
351
352
  OpenApiProperty_Boolean,
@@ -354,7 +355,10 @@ from .types import (
354
355
  OpenApiProperty_Number,
355
356
  OpenApiProperty_Object,
356
357
  OpenApiProperty_OneOf,
358
+ OpenApiProperty_Ref,
357
359
  OpenApiProperty_String,
360
+ OpenApiRefProperty,
361
+ OpenApiRefPropertyRequest,
358
362
  OpenApiStringProperty,
359
363
  OpenApiStringPropertyRequest,
360
364
  PaginatedDocumentIndexReadList,
@@ -1037,6 +1041,7 @@ __all__ = [
1037
1041
  "OpenApiPropertyRequest_Number",
1038
1042
  "OpenApiPropertyRequest_Object",
1039
1043
  "OpenApiPropertyRequest_OneOf",
1044
+ "OpenApiPropertyRequest_Ref",
1040
1045
  "OpenApiPropertyRequest_String",
1041
1046
  "OpenApiProperty_Array",
1042
1047
  "OpenApiProperty_Boolean",
@@ -1045,7 +1050,10 @@ __all__ = [
1045
1050
  "OpenApiProperty_Number",
1046
1051
  "OpenApiProperty_Object",
1047
1052
  "OpenApiProperty_OneOf",
1053
+ "OpenApiProperty_Ref",
1048
1054
  "OpenApiProperty_String",
1055
+ "OpenApiRefProperty",
1056
+ "OpenApiRefPropertyRequest",
1049
1057
  "OpenApiStringProperty",
1050
1058
  "OpenApiStringPropertyRequest",
1051
1059
  "PaginatedDocumentIndexReadList",
@@ -18,7 +18,7 @@ class BaseClientWrapper:
18
18
  headers: typing.Dict[str, str] = {
19
19
  "X-Fern-Language": "Python",
20
20
  "X-Fern-SDK-Name": "vellum-ai",
21
- "X-Fern-SDK-Version": "0.7.8",
21
+ "X-Fern-SDK-Version": "0.7.10",
22
22
  }
23
23
  headers["X_API_KEY"] = self.api_key
24
24
  return headers
@@ -285,6 +285,8 @@ class MlModelsClient:
285
285
  self,
286
286
  id: str,
287
287
  *,
288
+ exec_config: typing.Optional[MlModelExecConfigRequest] = OMIT,
289
+ parameter_config: typing.Optional[MlModelParameterConfigRequest] = OMIT,
288
290
  display_config: typing.Optional[MlModelDisplayConfigRequest] = OMIT,
289
291
  visibility: typing.Optional[VisibilityEnum] = OMIT,
290
292
  request_options: typing.Optional[RequestOptions] = None,
@@ -297,6 +299,12 @@ class MlModelsClient:
297
299
  id : str
298
300
  Either the ML Model's ID or its unique name
299
301
 
302
+ exec_config : typing.Optional[MlModelExecConfigRequest]
303
+ Configuration for how to execute the ML Model.
304
+
305
+ parameter_config : typing.Optional[MlModelParameterConfigRequest]
306
+ Configuration for the ML Model's parameters.
307
+
300
308
  display_config : typing.Optional[MlModelDisplayConfigRequest]
301
309
  Configuration for how to display the ML Model.
302
310
 
@@ -331,7 +339,12 @@ class MlModelsClient:
331
339
  f"v1/ml-models/{jsonable_encoder(id)}",
332
340
  base_url=self._client_wrapper.get_environment().default,
333
341
  method="PUT",
334
- json={"display_config": display_config, "visibility": visibility},
342
+ json={
343
+ "exec_config": exec_config,
344
+ "parameter_config": parameter_config,
345
+ "display_config": display_config,
346
+ "visibility": visibility,
347
+ },
335
348
  request_options=request_options,
336
349
  omit=OMIT,
337
350
  )
@@ -347,6 +360,8 @@ class MlModelsClient:
347
360
  self,
348
361
  id: str,
349
362
  *,
363
+ exec_config: typing.Optional[MlModelExecConfigRequest] = OMIT,
364
+ parameter_config: typing.Optional[MlModelParameterConfigRequest] = OMIT,
350
365
  display_config: typing.Optional[MlModelDisplayConfigRequest] = OMIT,
351
366
  visibility: typing.Optional[VisibilityEnum] = OMIT,
352
367
  request_options: typing.Optional[RequestOptions] = None,
@@ -359,6 +374,12 @@ class MlModelsClient:
359
374
  id : str
360
375
  Either the ML Model's ID or its unique name
361
376
 
377
+ exec_config : typing.Optional[MlModelExecConfigRequest]
378
+ Configuration for how to execute the ML Model.
379
+
380
+ parameter_config : typing.Optional[MlModelParameterConfigRequest]
381
+ Configuration for the ML Model's parameters.
382
+
362
383
  display_config : typing.Optional[MlModelDisplayConfigRequest]
363
384
  Configuration for how to display the ML Model.
364
385
 
@@ -393,7 +414,12 @@ class MlModelsClient:
393
414
  f"v1/ml-models/{jsonable_encoder(id)}",
394
415
  base_url=self._client_wrapper.get_environment().default,
395
416
  method="PATCH",
396
- json={"display_config": display_config, "visibility": visibility},
417
+ json={
418
+ "exec_config": exec_config,
419
+ "parameter_config": parameter_config,
420
+ "display_config": display_config,
421
+ "visibility": visibility,
422
+ },
397
423
  request_options=request_options,
398
424
  omit=OMIT,
399
425
  )
@@ -693,6 +719,8 @@ class AsyncMlModelsClient:
693
719
  self,
694
720
  id: str,
695
721
  *,
722
+ exec_config: typing.Optional[MlModelExecConfigRequest] = OMIT,
723
+ parameter_config: typing.Optional[MlModelParameterConfigRequest] = OMIT,
696
724
  display_config: typing.Optional[MlModelDisplayConfigRequest] = OMIT,
697
725
  visibility: typing.Optional[VisibilityEnum] = OMIT,
698
726
  request_options: typing.Optional[RequestOptions] = None,
@@ -705,6 +733,12 @@ class AsyncMlModelsClient:
705
733
  id : str
706
734
  Either the ML Model's ID or its unique name
707
735
 
736
+ exec_config : typing.Optional[MlModelExecConfigRequest]
737
+ Configuration for how to execute the ML Model.
738
+
739
+ parameter_config : typing.Optional[MlModelParameterConfigRequest]
740
+ Configuration for the ML Model's parameters.
741
+
708
742
  display_config : typing.Optional[MlModelDisplayConfigRequest]
709
743
  Configuration for how to display the ML Model.
710
744
 
@@ -747,7 +781,12 @@ class AsyncMlModelsClient:
747
781
  f"v1/ml-models/{jsonable_encoder(id)}",
748
782
  base_url=self._client_wrapper.get_environment().default,
749
783
  method="PUT",
750
- json={"display_config": display_config, "visibility": visibility},
784
+ json={
785
+ "exec_config": exec_config,
786
+ "parameter_config": parameter_config,
787
+ "display_config": display_config,
788
+ "visibility": visibility,
789
+ },
751
790
  request_options=request_options,
752
791
  omit=OMIT,
753
792
  )
@@ -763,6 +802,8 @@ class AsyncMlModelsClient:
763
802
  self,
764
803
  id: str,
765
804
  *,
805
+ exec_config: typing.Optional[MlModelExecConfigRequest] = OMIT,
806
+ parameter_config: typing.Optional[MlModelParameterConfigRequest] = OMIT,
766
807
  display_config: typing.Optional[MlModelDisplayConfigRequest] = OMIT,
767
808
  visibility: typing.Optional[VisibilityEnum] = OMIT,
768
809
  request_options: typing.Optional[RequestOptions] = None,
@@ -775,6 +816,12 @@ class AsyncMlModelsClient:
775
816
  id : str
776
817
  Either the ML Model's ID or its unique name
777
818
 
819
+ exec_config : typing.Optional[MlModelExecConfigRequest]
820
+ Configuration for how to execute the ML Model.
821
+
822
+ parameter_config : typing.Optional[MlModelParameterConfigRequest]
823
+ Configuration for the ML Model's parameters.
824
+
778
825
  display_config : typing.Optional[MlModelDisplayConfigRequest]
779
826
  Configuration for how to display the ML Model.
780
827
 
@@ -817,7 +864,12 @@ class AsyncMlModelsClient:
817
864
  f"v1/ml-models/{jsonable_encoder(id)}",
818
865
  base_url=self._client_wrapper.get_environment().default,
819
866
  method="PATCH",
820
- json={"display_config": display_config, "visibility": visibility},
867
+ json={
868
+ "exec_config": exec_config,
869
+ "parameter_config": parameter_config,
870
+ "display_config": display_config,
871
+ "visibility": visibility,
872
+ },
821
873
  request_options=request_options,
822
874
  omit=OMIT,
823
875
  )
@@ -17,7 +17,9 @@ from ._jsii import *
17
17
 
18
18
  __all__ = [
19
19
  "data_vellum_document_index",
20
+ "data_vellum_ml_model",
20
21
  "document_index",
22
+ "ml_model",
21
23
  "provider",
22
24
  ]
23
25
 
@@ -1,7 +1,7 @@
1
1
  '''
2
2
  # `data_vellum_document_index`
3
3
 
4
- Refer to the Terraform Registry for docs: [`data_vellum_document_index`](https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index).
4
+ Refer to the Terraform Registry for docs: [`data_vellum_document_index`](https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index).
5
5
  '''
6
6
  from pkgutil import extend_path
7
7
  __path__ = extend_path(__path__, __name__)
@@ -29,7 +29,7 @@ class DataVellumDocumentIndex(
29
29
  metaclass=jsii.JSIIMeta,
30
30
  jsii_type="vellum-ai_vellum.dataVellumDocumentIndex.DataVellumDocumentIndex",
31
31
  ):
32
- '''Represents a {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index vellum_document_index}.'''
32
+ '''Represents a {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index vellum_document_index}.'''
33
33
 
34
34
  def __init__(
35
35
  self,
@@ -46,12 +46,12 @@ class DataVellumDocumentIndex(
46
46
  provider: typing.Optional[_cdktf_9a9027ec.TerraformProvider] = None,
47
47
  provisioners: typing.Optional[typing.Sequence[typing.Union[typing.Union[_cdktf_9a9027ec.FileProvisioner, typing.Dict[builtins.str, typing.Any]], typing.Union[_cdktf_9a9027ec.LocalExecProvisioner, typing.Dict[builtins.str, typing.Any]], typing.Union[_cdktf_9a9027ec.RemoteExecProvisioner, typing.Dict[builtins.str, typing.Any]]]]] = None,
48
48
  ) -> None:
49
- '''Create a new {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index vellum_document_index} Data Source.
49
+ '''Create a new {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index vellum_document_index} Data Source.
50
50
 
51
51
  :param scope: The scope in which to define this construct.
52
52
  :param id_: The scoped construct ID. Must be unique amongst siblings in the same scope
53
- :param id: The Document Index's ID. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#id DataVellumDocumentIndex#id} Please be aware that the id field is automatically added to all resources in Terraform providers using a Terraform provider SDK version below 2. If you experience problems setting this value it might not be settable. Please take a look at the provider documentation to ensure it should be settable.
54
- :param name: A name that uniquely identifies this index within its workspace. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#name DataVellumDocumentIndex#name}
53
+ :param id: The Document Index's ID. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#id DataVellumDocumentIndex#id} Please be aware that the id field is automatically added to all resources in Terraform providers using a Terraform provider SDK version below 2. If you experience problems setting this value it might not be settable. Please take a look at the provider documentation to ensure it should be settable.
54
+ :param name: A name that uniquely identifies this index within its workspace. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#name DataVellumDocumentIndex#name}
55
55
  :param connection:
56
56
  :param count:
57
57
  :param depends_on:
@@ -91,7 +91,7 @@ class DataVellumDocumentIndex(
91
91
 
92
92
  :param scope: The scope in which to define this construct.
93
93
  :param import_to_id: The construct id used in the generated config for the DataVellumDocumentIndex to import.
94
- :param import_from_id: The id of the existing DataVellumDocumentIndex that should be imported. Refer to the {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#import import section} in the documentation of this resource for the id to use
94
+ :param import_from_id: The id of the existing DataVellumDocumentIndex that should be imported. Refer to the {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#import import section} in the documentation of this resource for the id to use
95
95
  :param provider: ? Optional instance of the provider where the DataVellumDocumentIndex to import is found.
96
96
  '''
97
97
  if __debug__:
@@ -215,8 +215,8 @@ class DataVellumDocumentIndexConfig(_cdktf_9a9027ec.TerraformMetaArguments):
215
215
  :param lifecycle:
216
216
  :param provider:
217
217
  :param provisioners:
218
- :param id: The Document Index's ID. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#id DataVellumDocumentIndex#id} Please be aware that the id field is automatically added to all resources in Terraform providers using a Terraform provider SDK version below 2. If you experience problems setting this value it might not be settable. Please take a look at the provider documentation to ensure it should be settable.
219
- :param name: A name that uniquely identifies this index within its workspace. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#name DataVellumDocumentIndex#name}
218
+ :param id: The Document Index's ID. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#id DataVellumDocumentIndex#id} Please be aware that the id field is automatically added to all resources in Terraform providers using a Terraform provider SDK version below 2. If you experience problems setting this value it might not be settable. Please take a look at the provider documentation to ensure it should be settable.
219
+ :param name: A name that uniquely identifies this index within its workspace. Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#name DataVellumDocumentIndex#name}
220
220
  '''
221
221
  if isinstance(lifecycle, dict):
222
222
  lifecycle = _cdktf_9a9027ec.TerraformResourceLifecycle(**lifecycle)
@@ -319,7 +319,7 @@ class DataVellumDocumentIndexConfig(_cdktf_9a9027ec.TerraformMetaArguments):
319
319
  def id(self) -> typing.Optional[builtins.str]:
320
320
  '''The Document Index's ID.
321
321
 
322
- Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#id DataVellumDocumentIndex#id}
322
+ Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#id DataVellumDocumentIndex#id}
323
323
 
324
324
  Please be aware that the id field is automatically added to all resources in Terraform providers using a Terraform provider SDK version below 2.
325
325
  If you experience problems setting this value it might not be settable. Please take a look at the provider documentation to ensure it should be settable.
@@ -331,7 +331,7 @@ class DataVellumDocumentIndexConfig(_cdktf_9a9027ec.TerraformMetaArguments):
331
331
  def name(self) -> typing.Optional[builtins.str]:
332
332
  '''A name that uniquely identifies this index within its workspace.
333
333
 
334
- Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.6/docs/data-sources/document_index#name DataVellumDocumentIndex#name}
334
+ Docs at Terraform Registry: {@link https://registry.terraform.io/providers/vellum-ai/vellum/0.0.7/docs/data-sources/document_index#name DataVellumDocumentIndex#name}
335
335
  '''
336
336
  result = self._values.get("name")
337
337
  return typing.cast(typing.Optional[builtins.str], result)