vellum-ai 0.3.21__py3-none-any.whl → 0.3.23__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (58) hide show
  1. vellum/__init__.py +48 -0
  2. vellum/core/client_wrapper.py +1 -1
  3. vellum/resources/test_suite_runs/client.py +14 -0
  4. vellum/terraform/__init__.py +26 -0
  5. vellum/terraform/_jsii/__init__.py +27 -0
  6. vellum/terraform/_jsii/vellum-ai_vellum@0.0.0.jsii.tgz +0 -0
  7. vellum/terraform/constraints.json +4 -0
  8. vellum/terraform/data_vellum_document_index/__init__.py +409 -0
  9. vellum/terraform/document_index/__init__.py +490 -0
  10. vellum/terraform/provider/__init__.py +237 -0
  11. vellum/terraform/py.typed +1 -0
  12. vellum/terraform/versions.json +3 -0
  13. vellum/types/__init__.py +50 -0
  14. vellum/types/external_test_case_execution.py +35 -0
  15. vellum/types/external_test_case_execution_request.py +35 -0
  16. vellum/types/fulfilled_workflow_node_result_event.py +1 -1
  17. vellum/types/named_test_case_chat_history_variable_value.py +35 -0
  18. vellum/types/named_test_case_chat_history_variable_value_request.py +5 -1
  19. vellum/types/named_test_case_error_variable_value.py +35 -0
  20. vellum/types/named_test_case_error_variable_value_request.py +5 -1
  21. vellum/types/named_test_case_json_variable_value.py +34 -0
  22. vellum/types/named_test_case_json_variable_value_request.py +5 -1
  23. vellum/types/named_test_case_number_variable_value.py +34 -0
  24. vellum/types/named_test_case_number_variable_value_request.py +5 -1
  25. vellum/types/named_test_case_search_results_variable_value.py +35 -0
  26. vellum/types/named_test_case_search_results_variable_value_request.py +5 -1
  27. vellum/types/named_test_case_string_variable_value.py +34 -0
  28. vellum/types/named_test_case_string_variable_value_request.py +5 -1
  29. vellum/types/named_test_case_variable_value.py +82 -0
  30. vellum/types/prompt_template_block.py +2 -0
  31. vellum/types/prompt_template_block_request.py +2 -0
  32. vellum/types/prompt_template_block_state.py +22 -0
  33. vellum/types/sandbox_scenario.py +4 -0
  34. vellum/types/test_case_chat_history_variable_value.py +5 -0
  35. vellum/types/test_case_error_variable_value.py +5 -0
  36. vellum/types/test_case_json_variable_value.py +5 -0
  37. vellum/types/test_case_number_variable_value.py +5 -0
  38. vellum/types/test_case_search_results_variable_value.py +5 -0
  39. vellum/types/test_case_string_variable_value.py +5 -0
  40. vellum/types/test_suite_run_exec_config.py +14 -1
  41. vellum/types/test_suite_run_exec_config_request.py +14 -1
  42. vellum/types/test_suite_run_execution_chat_history_output.py +6 -1
  43. vellum/types/test_suite_run_execution_error_output.py +6 -1
  44. vellum/types/test_suite_run_execution_json_output.py +6 -1
  45. vellum/types/test_suite_run_execution_metric_definition.py +31 -0
  46. vellum/types/test_suite_run_execution_metric_result.py +3 -0
  47. vellum/types/test_suite_run_execution_number_output.py +6 -1
  48. vellum/types/test_suite_run_execution_search_results_output.py +6 -1
  49. vellum/types/test_suite_run_execution_string_output.py +6 -1
  50. vellum/types/test_suite_run_external_exec_config.py +38 -0
  51. vellum/types/test_suite_run_external_exec_config_data.py +33 -0
  52. vellum/types/test_suite_run_external_exec_config_data_request.py +33 -0
  53. vellum/types/test_suite_run_external_exec_config_request.py +38 -0
  54. vellum/types/test_suite_run_external_exec_config_type_enum.py +5 -0
  55. {vellum_ai-0.3.21.dist-info → vellum_ai-0.3.23.dist-info}/METADATA +2 -1
  56. {vellum_ai-0.3.21.dist-info → vellum_ai-0.3.23.dist-info}/RECORD +58 -33
  57. {vellum_ai-0.3.21.dist-info → vellum_ai-0.3.23.dist-info}/LICENSE +0 -0
  58. {vellum_ai-0.3.21.dist-info → vellum_ai-0.3.23.dist-info}/WHEEL +0 -0
vellum/__init__.py CHANGED
@@ -104,6 +104,8 @@ from .types import (
104
104
  ExecutionVellumValue_Number,
105
105
  ExecutionVellumValue_SearchResults,
106
106
  ExecutionVellumValue_String,
107
+ ExternalTestCaseExecution,
108
+ ExternalTestCaseExecutionRequest,
107
109
  FinishReasonEnum,
108
110
  FulfilledEnum,
109
111
  FulfilledExecutePromptEvent,
@@ -155,12 +157,19 @@ from .types import (
155
157
  ModelVersionRead,
156
158
  ModelVersionReadStatusEnum,
157
159
  ModelVersionSandboxSnapshot,
160
+ NamedTestCaseChatHistoryVariableValue,
158
161
  NamedTestCaseChatHistoryVariableValueRequest,
162
+ NamedTestCaseErrorVariableValue,
159
163
  NamedTestCaseErrorVariableValueRequest,
164
+ NamedTestCaseJsonVariableValue,
160
165
  NamedTestCaseJsonVariableValueRequest,
166
+ NamedTestCaseNumberVariableValue,
161
167
  NamedTestCaseNumberVariableValueRequest,
168
+ NamedTestCaseSearchResultsVariableValue,
162
169
  NamedTestCaseSearchResultsVariableValueRequest,
170
+ NamedTestCaseStringVariableValue,
163
171
  NamedTestCaseStringVariableValueRequest,
172
+ NamedTestCaseVariableValue,
164
173
  NamedTestCaseVariableValueRequest,
165
174
  NamedTestCaseVariableValueRequest_ChatHistory,
166
175
  NamedTestCaseVariableValueRequest_Error,
@@ -168,6 +177,12 @@ from .types import (
168
177
  NamedTestCaseVariableValueRequest_Number,
169
178
  NamedTestCaseVariableValueRequest_SearchResults,
170
179
  NamedTestCaseVariableValueRequest_String,
180
+ NamedTestCaseVariableValue_ChatHistory,
181
+ NamedTestCaseVariableValue_Error,
182
+ NamedTestCaseVariableValue_Json,
183
+ NamedTestCaseVariableValue_Number,
184
+ NamedTestCaseVariableValue_SearchResults,
185
+ NamedTestCaseVariableValue_String,
171
186
  NodeInputCompiledArrayValue,
172
187
  NodeInputCompiledChatHistoryValue,
173
188
  NodeInputCompiledErrorValue,
@@ -233,6 +248,7 @@ from .types import (
233
248
  PromptTemplateBlockProperties,
234
249
  PromptTemplateBlockPropertiesRequest,
235
250
  PromptTemplateBlockRequest,
251
+ PromptTemplateBlockState,
236
252
  ProviderEnum,
237
253
  RawPromptExecutionOverridesRequest,
238
254
  RegisterPromptErrorResponse,
@@ -349,13 +365,16 @@ from .types import (
349
365
  TestSuiteRunExecConfig,
350
366
  TestSuiteRunExecConfigRequest,
351
367
  TestSuiteRunExecConfigRequest_DeploymentReleaseTag,
368
+ TestSuiteRunExecConfigRequest_External,
352
369
  TestSuiteRunExecConfigRequest_WorkflowReleaseTag,
353
370
  TestSuiteRunExecConfig_DeploymentReleaseTag,
371
+ TestSuiteRunExecConfig_External,
354
372
  TestSuiteRunExecConfig_WorkflowReleaseTag,
355
373
  TestSuiteRunExecution,
356
374
  TestSuiteRunExecutionChatHistoryOutput,
357
375
  TestSuiteRunExecutionErrorOutput,
358
376
  TestSuiteRunExecutionJsonOutput,
377
+ TestSuiteRunExecutionMetricDefinition,
359
378
  TestSuiteRunExecutionMetricResult,
360
379
  TestSuiteRunExecutionNumberOutput,
361
380
  TestSuiteRunExecutionOutput,
@@ -367,6 +386,11 @@ from .types import (
367
386
  TestSuiteRunExecutionOutput_String,
368
387
  TestSuiteRunExecutionSearchResultsOutput,
369
388
  TestSuiteRunExecutionStringOutput,
389
+ TestSuiteRunExternalExecConfig,
390
+ TestSuiteRunExternalExecConfigData,
391
+ TestSuiteRunExternalExecConfigDataRequest,
392
+ TestSuiteRunExternalExecConfigRequest,
393
+ TestSuiteRunExternalExecConfigTypeEnum,
370
394
  TestSuiteRunMetricErrorOutput,
371
395
  TestSuiteRunMetricErrorOutputTypeEnum,
372
396
  TestSuiteRunMetricNumberOutput,
@@ -596,6 +620,8 @@ __all__ = [
596
620
  "ExecutionVellumValue_Number",
597
621
  "ExecutionVellumValue_SearchResults",
598
622
  "ExecutionVellumValue_String",
623
+ "ExternalTestCaseExecution",
624
+ "ExternalTestCaseExecutionRequest",
599
625
  "FinishReasonEnum",
600
626
  "ForbiddenError",
601
627
  "FulfilledEnum",
@@ -649,12 +675,19 @@ __all__ = [
649
675
  "ModelVersionRead",
650
676
  "ModelVersionReadStatusEnum",
651
677
  "ModelVersionSandboxSnapshot",
678
+ "NamedTestCaseChatHistoryVariableValue",
652
679
  "NamedTestCaseChatHistoryVariableValueRequest",
680
+ "NamedTestCaseErrorVariableValue",
653
681
  "NamedTestCaseErrorVariableValueRequest",
682
+ "NamedTestCaseJsonVariableValue",
654
683
  "NamedTestCaseJsonVariableValueRequest",
684
+ "NamedTestCaseNumberVariableValue",
655
685
  "NamedTestCaseNumberVariableValueRequest",
686
+ "NamedTestCaseSearchResultsVariableValue",
656
687
  "NamedTestCaseSearchResultsVariableValueRequest",
688
+ "NamedTestCaseStringVariableValue",
657
689
  "NamedTestCaseStringVariableValueRequest",
690
+ "NamedTestCaseVariableValue",
658
691
  "NamedTestCaseVariableValueRequest",
659
692
  "NamedTestCaseVariableValueRequest_ChatHistory",
660
693
  "NamedTestCaseVariableValueRequest_Error",
@@ -662,6 +695,12 @@ __all__ = [
662
695
  "NamedTestCaseVariableValueRequest_Number",
663
696
  "NamedTestCaseVariableValueRequest_SearchResults",
664
697
  "NamedTestCaseVariableValueRequest_String",
698
+ "NamedTestCaseVariableValue_ChatHistory",
699
+ "NamedTestCaseVariableValue_Error",
700
+ "NamedTestCaseVariableValue_Json",
701
+ "NamedTestCaseVariableValue_Number",
702
+ "NamedTestCaseVariableValue_SearchResults",
703
+ "NamedTestCaseVariableValue_String",
665
704
  "NodeInputCompiledArrayValue",
666
705
  "NodeInputCompiledChatHistoryValue",
667
706
  "NodeInputCompiledErrorValue",
@@ -728,6 +767,7 @@ __all__ = [
728
767
  "PromptTemplateBlockProperties",
729
768
  "PromptTemplateBlockPropertiesRequest",
730
769
  "PromptTemplateBlockRequest",
770
+ "PromptTemplateBlockState",
731
771
  "ProviderEnum",
732
772
  "RawPromptExecutionOverridesRequest",
733
773
  "RegisterPromptErrorResponse",
@@ -844,13 +884,16 @@ __all__ = [
844
884
  "TestSuiteRunExecConfig",
845
885
  "TestSuiteRunExecConfigRequest",
846
886
  "TestSuiteRunExecConfigRequest_DeploymentReleaseTag",
887
+ "TestSuiteRunExecConfigRequest_External",
847
888
  "TestSuiteRunExecConfigRequest_WorkflowReleaseTag",
848
889
  "TestSuiteRunExecConfig_DeploymentReleaseTag",
890
+ "TestSuiteRunExecConfig_External",
849
891
  "TestSuiteRunExecConfig_WorkflowReleaseTag",
850
892
  "TestSuiteRunExecution",
851
893
  "TestSuiteRunExecutionChatHistoryOutput",
852
894
  "TestSuiteRunExecutionErrorOutput",
853
895
  "TestSuiteRunExecutionJsonOutput",
896
+ "TestSuiteRunExecutionMetricDefinition",
854
897
  "TestSuiteRunExecutionMetricResult",
855
898
  "TestSuiteRunExecutionNumberOutput",
856
899
  "TestSuiteRunExecutionOutput",
@@ -862,6 +905,11 @@ __all__ = [
862
905
  "TestSuiteRunExecutionOutput_String",
863
906
  "TestSuiteRunExecutionSearchResultsOutput",
864
907
  "TestSuiteRunExecutionStringOutput",
908
+ "TestSuiteRunExternalExecConfig",
909
+ "TestSuiteRunExternalExecConfigData",
910
+ "TestSuiteRunExternalExecConfigDataRequest",
911
+ "TestSuiteRunExternalExecConfigRequest",
912
+ "TestSuiteRunExternalExecConfigTypeEnum",
865
913
  "TestSuiteRunMetricErrorOutput",
866
914
  "TestSuiteRunMetricErrorOutputTypeEnum",
867
915
  "TestSuiteRunMetricNumberOutput",
@@ -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.3.21",
21
+ "X-Fern-SDK-Version": "0.3.23",
22
22
  }
23
23
  headers["X_API_KEY"] = self.api_key
24
24
  return headers
@@ -149,6 +149,7 @@ class TestSuiteRunsClient:
149
149
  self,
150
150
  id: str,
151
151
  *,
152
+ expand: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
152
153
  limit: typing.Optional[int] = None,
153
154
  offset: typing.Optional[int] = None,
154
155
  request_options: typing.Optional[RequestOptions] = None,
@@ -157,6 +158,11 @@ class TestSuiteRunsClient:
157
158
  Parameters:
158
159
  - id: str. A UUID string identifying this test suite run.
159
160
 
161
+ - expand: typing.Optional[typing.Union[str, typing.Sequence[str]]]. The response fields to expand for more information.
162
+
163
+ - 'results.metric_results.metric_label' expands the metric label for each metric result.
164
+ - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
165
+ - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
160
166
  - limit: typing.Optional[int]. Number of results to return per page.
161
167
 
162
168
  - offset: typing.Optional[int]. The initial index from which to return the results.
@@ -181,6 +187,7 @@ class TestSuiteRunsClient:
181
187
  params=jsonable_encoder(
182
188
  remove_none_from_dict(
183
189
  {
190
+ "expand": expand,
184
191
  "limit": limit,
185
192
  "offset": offset,
186
193
  **(
@@ -341,6 +348,7 @@ class AsyncTestSuiteRunsClient:
341
348
  self,
342
349
  id: str,
343
350
  *,
351
+ expand: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
344
352
  limit: typing.Optional[int] = None,
345
353
  offset: typing.Optional[int] = None,
346
354
  request_options: typing.Optional[RequestOptions] = None,
@@ -349,6 +357,11 @@ class AsyncTestSuiteRunsClient:
349
357
  Parameters:
350
358
  - id: str. A UUID string identifying this test suite run.
351
359
 
360
+ - expand: typing.Optional[typing.Union[str, typing.Sequence[str]]]. The response fields to expand for more information.
361
+
362
+ - 'results.metric_results.metric_label' expands the metric label for each metric result.
363
+ - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
364
+ - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
352
365
  - limit: typing.Optional[int]. Number of results to return per page.
353
366
 
354
367
  - offset: typing.Optional[int]. The initial index from which to return the results.
@@ -373,6 +386,7 @@ class AsyncTestSuiteRunsClient:
373
386
  params=jsonable_encoder(
374
387
  remove_none_from_dict(
375
388
  {
389
+ "expand": expand,
376
390
  "limit": limit,
377
391
  "offset": offset,
378
392
  **(
@@ -0,0 +1,26 @@
1
+ from pkgutil import extend_path
2
+ __path__ = extend_path(__path__, __name__)
3
+
4
+ import abc
5
+ import builtins
6
+ import datetime
7
+ import enum
8
+ import typing
9
+
10
+ import jsii
11
+ import publication
12
+ import typing_extensions
13
+
14
+ from typeguard import check_type
15
+
16
+ from ._jsii import *
17
+
18
+ __all__ = [
19
+ "data_vellum_document_index",
20
+ "document_index",
21
+ "provider",
22
+ ]
23
+
24
+ publication.publish()
25
+
26
+ # Loading modules to ensure their types are registered with the jsii runtime library
@@ -0,0 +1,27 @@
1
+ from pkgutil import extend_path
2
+ __path__ = extend_path(__path__, __name__)
3
+
4
+ import abc
5
+ import builtins
6
+ import datetime
7
+ import enum
8
+ import typing
9
+
10
+ import jsii
11
+ import publication
12
+ import typing_extensions
13
+
14
+ from typeguard import check_type
15
+
16
+ import cdktf._jsii
17
+ import constructs._jsii
18
+
19
+ __jsii_assembly__ = jsii.JSIIAssembly.load(
20
+ "vellum-ai_vellum", "0.0.0", __name__[0:-6], "vellum-ai_vellum@0.0.0.jsii.tgz"
21
+ )
22
+
23
+ __all__ = [
24
+ "__jsii_assembly__",
25
+ ]
26
+
27
+ publication.publish()
@@ -0,0 +1,4 @@
1
+ {
2
+ "cdktf": "0.20.6",
3
+ "providers": {}
4
+ }