vellum-ai 0.7.8__py3-none-any.whl → 0.7.10__py3-none-any.whl

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