vellum-ai 0.8.13__py3-none-any.whl → 0.8.15__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (75) hide show
  1. vellum/__init__.py +18 -0
  2. vellum/client.py +103 -34
  3. vellum/core/client_wrapper.py +1 -1
  4. vellum/core/http_client.py +6 -2
  5. vellum/core/pydantic_utilities.py +42 -11
  6. vellum/core/serialization.py +18 -0
  7. vellum/resources/ad_hoc/client.py +31 -10
  8. vellum/resources/deployments/client.py +13 -4
  9. vellum/resources/document_indexes/client.py +7 -2
  10. vellum/resources/documents/client.py +8 -8
  11. vellum/resources/sandboxes/client.py +7 -2
  12. vellum/resources/test_suite_runs/client.py +7 -2
  13. vellum/resources/test_suites/client.py +27 -6
  14. vellum/types/__init__.py +18 -0
  15. vellum/types/api_node_result_data.py +6 -2
  16. vellum/types/array_variable_value.py +27 -0
  17. vellum/types/array_variable_value_item.py +9 -0
  18. vellum/types/array_vellum_value.py +31 -0
  19. vellum/types/array_vellum_value_item.py +15 -1
  20. vellum/types/array_vellum_value_item_request.py +9 -0
  21. vellum/types/array_vellum_value_request.py +31 -0
  22. vellum/types/chat_history_variable_value.py +21 -0
  23. vellum/types/chat_history_vellum_value.py +25 -0
  24. vellum/types/chat_history_vellum_value_request.py +25 -0
  25. vellum/types/code_execution_node_array_result.py +6 -0
  26. vellum/types/code_execution_node_result.py +6 -0
  27. vellum/types/code_execution_node_result_data.py +6 -0
  28. vellum/types/create_test_suite_test_case_request.py +6 -0
  29. vellum/types/execute_workflow_response.py +6 -0
  30. vellum/types/execution_array_vellum_value.py +6 -0
  31. vellum/types/external_test_case_execution.py +6 -0
  32. vellum/types/external_test_case_execution_request.py +6 -0
  33. vellum/types/fulfilled_execute_workflow_workflow_result_event.py +8 -0
  34. vellum/types/fulfilled_workflow_node_result_event.py +8 -0
  35. vellum/types/initiated_workflow_node_result_event.py +6 -0
  36. vellum/types/named_test_case_array_variable_value.py +6 -0
  37. vellum/types/named_test_case_array_variable_value_request.py +8 -0
  38. vellum/types/node_input_compiled_array_value.py +6 -0
  39. vellum/types/node_output_compiled_array_value.py +6 -0
  40. vellum/types/paginated_test_suite_run_execution_list.py +6 -0
  41. vellum/types/paginated_test_suite_test_case_list.py +6 -0
  42. vellum/types/rejected_workflow_node_result_event.py +6 -0
  43. vellum/types/replace_test_suite_test_case_request.py +6 -0
  44. vellum/types/search_results_variable_value.py +21 -0
  45. vellum/types/search_results_vellum_value.py +25 -0
  46. vellum/types/search_results_vellum_value_request.py +25 -0
  47. vellum/types/streaming_workflow_node_result_event.py +8 -0
  48. vellum/types/templating_node_array_result.py +6 -0
  49. vellum/types/templating_node_result.py +6 -0
  50. vellum/types/templating_node_result_data.py +6 -0
  51. vellum/types/terminal_node_array_result.py +6 -0
  52. vellum/types/terminal_node_result.py +6 -0
  53. vellum/types/terminal_node_result_data.py +6 -0
  54. vellum/types/test_case_array_variable_value.py +6 -0
  55. vellum/types/test_suite_run_execution.py +6 -0
  56. vellum/types/test_suite_run_execution_array_output.py +6 -0
  57. vellum/types/test_suite_run_external_exec_config.py +6 -0
  58. vellum/types/test_suite_run_external_exec_config_data.py +6 -0
  59. vellum/types/test_suite_run_external_exec_config_data_request.py +8 -0
  60. vellum/types/test_suite_run_external_exec_config_request.py +8 -0
  61. vellum/types/test_suite_run_read.py +6 -0
  62. vellum/types/test_suite_test_case.py +6 -0
  63. vellum/types/test_suite_test_case_create_bulk_operation_request.py +8 -0
  64. vellum/types/test_suite_test_case_replace_bulk_operation_request.py +8 -0
  65. vellum/types/test_suite_test_case_upsert_bulk_operation_request.py +8 -0
  66. vellum/types/upsert_test_suite_test_case_request.py +6 -0
  67. vellum/types/workflow_execution_node_result_event.py +8 -0
  68. vellum/types/workflow_execution_workflow_result_event.py +8 -0
  69. vellum/types/workflow_output_array.py +6 -0
  70. vellum/types/workflow_result_event.py +8 -0
  71. vellum/types/workflow_result_event_output_data_array.py +6 -0
  72. {vellum_ai-0.8.13.dist-info → vellum_ai-0.8.15.dist-info}/METADATA +1 -1
  73. {vellum_ai-0.8.13.dist-info → vellum_ai-0.8.15.dist-info}/RECORD +75 -66
  74. {vellum_ai-0.8.13.dist-info → vellum_ai-0.8.15.dist-info}/LICENSE +0 -0
  75. {vellum_ai-0.8.13.dist-info → vellum_ai-0.8.15.dist-info}/WHEEL +0 -0
@@ -10,6 +10,7 @@ from json.decoder import JSONDecodeError
10
10
  from ...core.api_error import ApiError
11
11
  from ...types.named_scenario_input_request import NamedScenarioInputRequest
12
12
  from ...types.sandbox_scenario import SandboxScenario
13
+ from ...core.serialization import convert_and_respect_annotation_metadata
13
14
  from ...core.client_wrapper import AsyncClientWrapper
14
15
 
15
16
  # this is used as the default value for optional parameters
@@ -166,7 +167,9 @@ class SandboxesClient:
166
167
  method="POST",
167
168
  json={
168
169
  "label": label,
169
- "inputs": inputs,
170
+ "inputs": convert_and_respect_annotation_metadata(
171
+ object_=inputs, annotation=typing.Sequence[NamedScenarioInputRequest], direction="write"
172
+ ),
170
173
  "scenario_id": scenario_id,
171
174
  },
172
175
  request_options=request_options,
@@ -400,7 +403,9 @@ class AsyncSandboxesClient:
400
403
  method="POST",
401
404
  json={
402
405
  "label": label,
403
- "inputs": inputs,
406
+ "inputs": convert_and_respect_annotation_metadata(
407
+ object_=inputs, annotation=typing.Sequence[NamedScenarioInputRequest], direction="write"
408
+ ),
404
409
  "scenario_id": scenario_id,
405
410
  },
406
411
  request_options=request_options,
@@ -5,6 +5,7 @@ from ...core.client_wrapper import SyncClientWrapper
5
5
  from ...types.test_suite_run_exec_config_request import TestSuiteRunExecConfigRequest
6
6
  from ...core.request_options import RequestOptions
7
7
  from ...types.test_suite_run_read import TestSuiteRunRead
8
+ from ...core.serialization import convert_and_respect_annotation_metadata
8
9
  from ...core.pydantic_utilities import parse_obj_as
9
10
  from json.decoder import JSONDecodeError
10
11
  from ...core.api_error import ApiError
@@ -72,7 +73,9 @@ class TestSuiteRunsClient:
72
73
  method="POST",
73
74
  json={
74
75
  "test_suite_id": test_suite_id,
75
- "exec_config": exec_config,
76
+ "exec_config": convert_and_respect_annotation_metadata(
77
+ object_=exec_config, annotation=TestSuiteRunExecConfigRequest, direction="write"
78
+ ),
76
79
  },
77
80
  request_options=request_options,
78
81
  omit=OMIT,
@@ -276,7 +279,9 @@ class AsyncTestSuiteRunsClient:
276
279
  method="POST",
277
280
  json={
278
281
  "test_suite_id": test_suite_id,
279
- "exec_config": exec_config,
282
+ "exec_config": convert_and_respect_annotation_metadata(
283
+ object_=exec_config, annotation=TestSuiteRunExecConfigRequest, direction="write"
284
+ ),
280
285
  },
281
286
  request_options=request_options,
282
287
  omit=OMIT,
@@ -10,6 +10,7 @@ from json.decoder import JSONDecodeError
10
10
  from ...core.api_error import ApiError
11
11
  from ...types.named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
12
12
  from ...types.test_suite_test_case import TestSuiteTestCase
13
+ from ...core.serialization import convert_and_respect_annotation_metadata
13
14
  from ...types.test_suite_test_case_bulk_operation_request import TestSuiteTestCaseBulkOperationRequest
14
15
  from ...types.test_suite_test_case_bulk_result import TestSuiteTestCaseBulkResult
15
16
  import json
@@ -165,8 +166,16 @@ class TestSuitesClient:
165
166
  "id": id,
166
167
  "external_id": external_id,
167
168
  "label": label,
168
- "input_values": input_values,
169
- "evaluation_values": evaluation_values,
169
+ "input_values": convert_and_respect_annotation_metadata(
170
+ object_=input_values,
171
+ annotation=typing.Sequence[NamedTestCaseVariableValueRequest],
172
+ direction="write",
173
+ ),
174
+ "evaluation_values": convert_and_respect_annotation_metadata(
175
+ object_=evaluation_values,
176
+ annotation=typing.Sequence[NamedTestCaseVariableValueRequest],
177
+ direction="write",
178
+ ),
170
179
  },
171
180
  request_options=request_options,
172
181
  omit=OMIT,
@@ -253,7 +262,9 @@ class TestSuitesClient:
253
262
  f"v1/test-suites/{jsonable_encoder(id)}/test-cases-bulk",
254
263
  base_url=self._client_wrapper.get_environment().default,
255
264
  method="POST",
256
- json=request,
265
+ json=convert_and_respect_annotation_metadata(
266
+ object_=request, annotation=typing.Sequence[TestSuiteTestCaseBulkOperationRequest], direction="write"
267
+ ),
257
268
  request_options=request_options,
258
269
  omit=OMIT,
259
270
  ) as _response:
@@ -489,8 +500,16 @@ class AsyncTestSuitesClient:
489
500
  "id": id,
490
501
  "external_id": external_id,
491
502
  "label": label,
492
- "input_values": input_values,
493
- "evaluation_values": evaluation_values,
503
+ "input_values": convert_and_respect_annotation_metadata(
504
+ object_=input_values,
505
+ annotation=typing.Sequence[NamedTestCaseVariableValueRequest],
506
+ direction="write",
507
+ ),
508
+ "evaluation_values": convert_and_respect_annotation_metadata(
509
+ object_=evaluation_values,
510
+ annotation=typing.Sequence[NamedTestCaseVariableValueRequest],
511
+ direction="write",
512
+ ),
494
513
  },
495
514
  request_options=request_options,
496
515
  omit=OMIT,
@@ -585,7 +604,9 @@ class AsyncTestSuitesClient:
585
604
  f"v1/test-suites/{jsonable_encoder(id)}/test-cases-bulk",
586
605
  base_url=self._client_wrapper.get_environment().default,
587
606
  method="POST",
588
- json=request,
607
+ json=convert_and_respect_annotation_metadata(
608
+ object_=request, annotation=typing.Sequence[TestSuiteTestCaseBulkOperationRequest], direction="write"
609
+ ),
589
610
  request_options=request_options,
590
611
  omit=OMIT,
591
612
  ) as _response:
vellum/types/__init__.py CHANGED
@@ -13,9 +13,12 @@ from .array_chat_message_content import ArrayChatMessageContent
13
13
  from .array_chat_message_content_item import ArrayChatMessageContentItem
14
14
  from .array_chat_message_content_item_request import ArrayChatMessageContentItemRequest
15
15
  from .array_chat_message_content_request import ArrayChatMessageContentRequest
16
+ from .array_variable_value import ArrayVariableValue
16
17
  from .array_variable_value_item import ArrayVariableValueItem
18
+ from .array_vellum_value import ArrayVellumValue
17
19
  from .array_vellum_value_item import ArrayVellumValueItem
18
20
  from .array_vellum_value_item_request import ArrayVellumValueItemRequest
21
+ from .array_vellum_value_request import ArrayVellumValueRequest
19
22
  from .basic_vectorizer_intfloat_multilingual_e_5_large import BasicVectorizerIntfloatMultilingualE5Large
20
23
  from .basic_vectorizer_intfloat_multilingual_e_5_large_request import BasicVectorizerIntfloatMultilingualE5LargeRequest
21
24
  from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1 import (
@@ -31,6 +34,9 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request
31
34
  BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
32
35
  )
33
36
  from .chat_history_input_request import ChatHistoryInputRequest
37
+ from .chat_history_variable_value import ChatHistoryVariableValue
38
+ from .chat_history_vellum_value import ChatHistoryVellumValue
39
+ from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
34
40
  from .chat_message import ChatMessage
35
41
  from .chat_message_content import ChatMessageContent
36
42
  from .chat_message_content_request import ChatMessageContentRequest
@@ -265,6 +271,9 @@ from .search_result_merging_request import SearchResultMergingRequest
265
271
  from .search_result_meta import SearchResultMeta
266
272
  from .search_result_meta_request import SearchResultMetaRequest
267
273
  from .search_result_request import SearchResultRequest
274
+ from .search_results_variable_value import SearchResultsVariableValue
275
+ from .search_results_vellum_value import SearchResultsVellumValue
276
+ from .search_results_vellum_value_request import SearchResultsVellumValueRequest
268
277
  from .search_weights_request import SearchWeightsRequest
269
278
  from .sentence_chunker_config import SentenceChunkerConfig
270
279
  from .sentence_chunker_config_request import SentenceChunkerConfigRequest
@@ -443,9 +452,12 @@ __all__ = [
443
452
  "ArrayChatMessageContentItem",
444
453
  "ArrayChatMessageContentItemRequest",
445
454
  "ArrayChatMessageContentRequest",
455
+ "ArrayVariableValue",
446
456
  "ArrayVariableValueItem",
457
+ "ArrayVellumValue",
447
458
  "ArrayVellumValueItem",
448
459
  "ArrayVellumValueItemRequest",
460
+ "ArrayVellumValueRequest",
449
461
  "BasicVectorizerIntfloatMultilingualE5Large",
450
462
  "BasicVectorizerIntfloatMultilingualE5LargeRequest",
451
463
  "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1",
@@ -453,6 +465,9 @@ __all__ = [
453
465
  "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
454
466
  "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
455
467
  "ChatHistoryInputRequest",
468
+ "ChatHistoryVariableValue",
469
+ "ChatHistoryVellumValue",
470
+ "ChatHistoryVellumValueRequest",
456
471
  "ChatMessage",
457
472
  "ChatMessageContent",
458
473
  "ChatMessageContentRequest",
@@ -687,6 +702,9 @@ __all__ = [
687
702
  "SearchResultMeta",
688
703
  "SearchResultMetaRequest",
689
704
  "SearchResultRequest",
705
+ "SearchResultsVariableValue",
706
+ "SearchResultsVellumValue",
707
+ "SearchResultsVellumValueRequest",
690
708
  "SearchWeightsRequest",
691
709
  "SentenceChunkerConfig",
692
710
  "SentenceChunkerConfigRequest",
@@ -2,15 +2,19 @@
2
2
 
3
3
  from ..core.pydantic_utilities import UniversalBaseModel
4
4
  import typing
5
- import pydantic
5
+ import typing_extensions
6
+ from ..core.serialization import FieldMetadata
6
7
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
8
+ import pydantic
7
9
 
8
10
 
9
11
  class ApiNodeResultData(UniversalBaseModel):
10
12
  text_output_id: str
11
13
  text: typing.Optional[str] = None
12
14
  json_output_id: str
13
- json_: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(alias="json", default=None)
15
+ json_: typing_extensions.Annotated[
16
+ typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="json")
17
+ ] = None
14
18
  status_code_output_id: str
15
19
  status_code: int
16
20
 
@@ -0,0 +1,27 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from __future__ import annotations
4
+ from ..core.pydantic_utilities import UniversalBaseModel
5
+ import typing
6
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
+ import pydantic
8
+ from ..core.pydantic_utilities import update_forward_refs
9
+
10
+
11
+ class ArrayVariableValue(UniversalBaseModel):
12
+ type: typing.Literal["ARRAY"] = "ARRAY"
13
+ value: typing.Optional[typing.List["ArrayVariableValueItem"]] = None
14
+
15
+ if IS_PYDANTIC_V2:
16
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
17
+ else:
18
+
19
+ class Config:
20
+ frozen = True
21
+ smart_union = True
22
+ extra = pydantic.Extra.allow
23
+
24
+
25
+ from .array_variable_value_item import ArrayVariableValueItem # noqa: E402
26
+
27
+ update_forward_refs(ArrayVariableValue)
@@ -1,5 +1,6 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  import typing
4
5
  from .string_variable_value import StringVariableValue
5
6
  from .number_variable_value import NumberVariableValue
@@ -7,7 +8,12 @@ from .json_variable_value import JsonVariableValue
7
8
  from .error_variable_value import ErrorVariableValue
8
9
  from .function_call_variable_value import FunctionCallVariableValue
9
10
  from .image_variable_value import ImageVariableValue
11
+ from .chat_history_variable_value import ChatHistoryVariableValue
12
+ from .search_results_variable_value import SearchResultsVariableValue
13
+ import typing
10
14
 
15
+ if typing.TYPE_CHECKING:
16
+ from .array_variable_value import ArrayVariableValue
11
17
  ArrayVariableValueItem = typing.Union[
12
18
  StringVariableValue,
13
19
  NumberVariableValue,
@@ -15,4 +21,7 @@ ArrayVariableValueItem = typing.Union[
15
21
  ErrorVariableValue,
16
22
  FunctionCallVariableValue,
17
23
  ImageVariableValue,
24
+ ChatHistoryVariableValue,
25
+ SearchResultsVariableValue,
26
+ "ArrayVariableValue",
18
27
  ]
@@ -0,0 +1,31 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from __future__ import annotations
4
+ from ..core.pydantic_utilities import UniversalBaseModel
5
+ import typing
6
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
+ import pydantic
8
+ from ..core.pydantic_utilities import update_forward_refs
9
+
10
+
11
+ class ArrayVellumValue(UniversalBaseModel):
12
+ """
13
+ A value representing an array of Vellum variable values.
14
+ """
15
+
16
+ type: typing.Literal["ARRAY"] = "ARRAY"
17
+ value: typing.Optional[typing.List["ArrayVellumValueItem"]] = None
18
+
19
+ if IS_PYDANTIC_V2:
20
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
21
+ else:
22
+
23
+ class Config:
24
+ frozen = True
25
+ smart_union = True
26
+ extra = pydantic.Extra.allow
27
+
28
+
29
+ from .array_vellum_value_item import ArrayVellumValueItem # noqa: E402
30
+
31
+ update_forward_refs(ArrayVellumValue)
@@ -1,5 +1,6 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  import typing
4
5
  from .string_vellum_value import StringVellumValue
5
6
  from .number_vellum_value import NumberVellumValue
@@ -7,7 +8,20 @@ from .json_vellum_value import JsonVellumValue
7
8
  from .image_vellum_value import ImageVellumValue
8
9
  from .function_call_vellum_value import FunctionCallVellumValue
9
10
  from .error_vellum_value import ErrorVellumValue
11
+ from .chat_history_vellum_value import ChatHistoryVellumValue
12
+ from .search_results_vellum_value import SearchResultsVellumValue
13
+ import typing
10
14
 
15
+ if typing.TYPE_CHECKING:
16
+ from .array_vellum_value import ArrayVellumValue
11
17
  ArrayVellumValueItem = typing.Union[
12
- StringVellumValue, NumberVellumValue, JsonVellumValue, ImageVellumValue, FunctionCallVellumValue, ErrorVellumValue
18
+ StringVellumValue,
19
+ NumberVellumValue,
20
+ JsonVellumValue,
21
+ ImageVellumValue,
22
+ FunctionCallVellumValue,
23
+ ErrorVellumValue,
24
+ ChatHistoryVellumValue,
25
+ SearchResultsVellumValue,
26
+ "ArrayVellumValue",
13
27
  ]
@@ -1,5 +1,6 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  import typing
4
5
  from .string_vellum_value_request import StringVellumValueRequest
5
6
  from .number_vellum_value_request import NumberVellumValueRequest
@@ -7,7 +8,12 @@ from .json_vellum_value_request import JsonVellumValueRequest
7
8
  from .image_vellum_value_request import ImageVellumValueRequest
8
9
  from .function_call_vellum_value_request import FunctionCallVellumValueRequest
9
10
  from .error_vellum_value_request import ErrorVellumValueRequest
11
+ from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
12
+ from .search_results_vellum_value_request import SearchResultsVellumValueRequest
13
+ import typing
10
14
 
15
+ if typing.TYPE_CHECKING:
16
+ from .array_vellum_value_request import ArrayVellumValueRequest
11
17
  ArrayVellumValueItemRequest = typing.Union[
12
18
  StringVellumValueRequest,
13
19
  NumberVellumValueRequest,
@@ -15,4 +21,7 @@ ArrayVellumValueItemRequest = typing.Union[
15
21
  ImageVellumValueRequest,
16
22
  FunctionCallVellumValueRequest,
17
23
  ErrorVellumValueRequest,
24
+ ChatHistoryVellumValueRequest,
25
+ SearchResultsVellumValueRequest,
26
+ "ArrayVellumValueRequest",
18
27
  ]
@@ -0,0 +1,31 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from __future__ import annotations
4
+ from ..core.pydantic_utilities import UniversalBaseModel
5
+ import typing
6
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
+ import pydantic
8
+ from ..core.pydantic_utilities import update_forward_refs
9
+
10
+
11
+ class ArrayVellumValueRequest(UniversalBaseModel):
12
+ """
13
+ A value representing an array of Vellum variable values.
14
+ """
15
+
16
+ type: typing.Literal["ARRAY"] = "ARRAY"
17
+ value: typing.Optional[typing.List["ArrayVellumValueItemRequest"]] = None
18
+
19
+ if IS_PYDANTIC_V2:
20
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
21
+ else:
22
+
23
+ class Config:
24
+ frozen = True
25
+ smart_union = True
26
+ extra = pydantic.Extra.allow
27
+
28
+
29
+ from .array_vellum_value_item_request import ArrayVellumValueItemRequest # noqa: E402
30
+
31
+ update_forward_refs(ArrayVellumValueRequest)
@@ -0,0 +1,21 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from ..core.pydantic_utilities import UniversalBaseModel
4
+ import typing
5
+ from .chat_message import ChatMessage
6
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
+ import pydantic
8
+
9
+
10
+ class ChatHistoryVariableValue(UniversalBaseModel):
11
+ type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
12
+ value: typing.Optional[typing.List[ChatMessage]] = None
13
+
14
+ if IS_PYDANTIC_V2:
15
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
16
+ else:
17
+
18
+ class Config:
19
+ frozen = True
20
+ smart_union = True
21
+ extra = pydantic.Extra.allow
@@ -0,0 +1,25 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from ..core.pydantic_utilities import UniversalBaseModel
4
+ import typing
5
+ from .chat_message import ChatMessage
6
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
+ import pydantic
8
+
9
+
10
+ class ChatHistoryVellumValue(UniversalBaseModel):
11
+ """
12
+ A value representing Chat History.
13
+ """
14
+
15
+ type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
16
+ value: typing.Optional[typing.List[ChatMessage]] = None
17
+
18
+ if IS_PYDANTIC_V2:
19
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
20
+ else:
21
+
22
+ class Config:
23
+ frozen = True
24
+ smart_union = True
25
+ extra = pydantic.Extra.allow
@@ -0,0 +1,25 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from ..core.pydantic_utilities import UniversalBaseModel
4
+ import typing
5
+ from .chat_message_request import ChatMessageRequest
6
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
+ import pydantic
8
+
9
+
10
+ class ChatHistoryVellumValueRequest(UniversalBaseModel):
11
+ """
12
+ A value representing Chat History.
13
+ """
14
+
15
+ type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
16
+ value: typing.Optional[typing.List[ChatMessageRequest]] = None
17
+
18
+ if IS_PYDANTIC_V2:
19
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
20
+ else:
21
+
22
+ class Config:
23
+ frozen = True
24
+ smart_union = True
25
+ extra = pydantic.Extra.allow
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_variable_value import ArrayVariableValue
4
6
  import typing
5
7
  from .array_variable_value_item import ArrayVariableValueItem
6
8
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
9
  import pydantic
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class CodeExecutionNodeArrayResult(UniversalBaseModel):
@@ -20,3 +23,6 @@ class CodeExecutionNodeArrayResult(UniversalBaseModel):
20
23
  frozen = True
21
24
  smart_union = True
22
25
  extra = pydantic.Extra.allow
26
+
27
+
28
+ update_forward_refs(ArrayVariableValue, CodeExecutionNodeArrayResult=CodeExecutionNodeArrayResult)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_variable_value import ArrayVariableValue
4
6
  import typing
5
7
  from .code_execution_node_result_data import CodeExecutionNodeResultData
6
8
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
9
  import pydantic
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class CodeExecutionNodeResult(UniversalBaseModel):
@@ -23,3 +26,6 @@ class CodeExecutionNodeResult(UniversalBaseModel):
23
26
  frozen = True
24
27
  smart_union = True
25
28
  extra = pydantic.Extra.allow
29
+
30
+
31
+ update_forward_refs(ArrayVariableValue, CodeExecutionNodeResult=CodeExecutionNodeResult)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_variable_value import ArrayVariableValue
4
6
  from .code_execution_node_result_output import CodeExecutionNodeResultOutput
5
7
  import typing
6
8
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
9
  import pydantic
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class CodeExecutionNodeResultData(UniversalBaseModel):
@@ -19,3 +22,6 @@ class CodeExecutionNodeResultData(UniversalBaseModel):
19
22
  frozen = True
20
23
  smart_union = True
21
24
  extra = pydantic.Extra.allow
25
+
26
+
27
+ update_forward_refs(ArrayVariableValue, CodeExecutionNodeResultData=CodeExecutionNodeResultData)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_vellum_value_request import ArrayVellumValueRequest
4
6
  import typing
5
7
  import pydantic
6
8
  from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
7
9
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
@@ -40,3 +43,6 @@ class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
40
43
  frozen = True
41
44
  smart_union = True
42
45
  extra = pydantic.Extra.allow
46
+
47
+
48
+ update_forward_refs(ArrayVellumValueRequest, CreateTestSuiteTestCaseRequest=CreateTestSuiteTestCaseRequest)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_vellum_value import ArrayVellumValue
4
6
  import typing
5
7
  from .execute_workflow_workflow_result_event import ExecuteWorkflowWorkflowResultEvent
6
8
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
7
9
  import pydantic
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class ExecuteWorkflowResponse(UniversalBaseModel):
@@ -21,3 +24,6 @@ class ExecuteWorkflowResponse(UniversalBaseModel):
21
24
  frozen = True
22
25
  smart_union = True
23
26
  extra = pydantic.Extra.allow
27
+
28
+
29
+ update_forward_refs(ArrayVellumValue, ExecuteWorkflowResponse=ExecuteWorkflowResponse)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_vellum_value import ArrayVellumValue
4
6
  import pydantic
5
7
  import typing
6
8
  from .array_vellum_value_item import ArrayVellumValueItem
7
9
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class ExecutionArrayVellumValue(UniversalBaseModel):
@@ -29,3 +32,6 @@ class ExecutionArrayVellumValue(UniversalBaseModel):
29
32
  frozen = True
30
33
  smart_union = True
31
34
  extra = pydantic.Extra.allow
35
+
36
+
37
+ update_forward_refs(ArrayVellumValue, ExecutionArrayVellumValue=ExecutionArrayVellumValue)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_vellum_value import ArrayVellumValue
4
6
  import typing
5
7
  from .named_test_case_variable_value import NamedTestCaseVariableValue
6
8
  import pydantic
7
9
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class ExternalTestCaseExecution(UniversalBaseModel):
@@ -23,3 +26,6 @@ class ExternalTestCaseExecution(UniversalBaseModel):
23
26
  frozen = True
24
27
  smart_union = True
25
28
  extra = pydantic.Extra.allow
29
+
30
+
31
+ update_forward_refs(ArrayVellumValue, ExternalTestCaseExecution=ExternalTestCaseExecution)
@@ -1,10 +1,13 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_vellum_value_request import ArrayVellumValueRequest
4
6
  import typing
5
7
  from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
6
8
  import pydantic
7
9
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
10
+ from ..core.pydantic_utilities import update_forward_refs
8
11
 
9
12
 
10
13
  class ExternalTestCaseExecutionRequest(UniversalBaseModel):
@@ -23,3 +26,6 @@ class ExternalTestCaseExecutionRequest(UniversalBaseModel):
23
26
  frozen = True
24
27
  smart_union = True
25
28
  extra = pydantic.Extra.allow
29
+
30
+
31
+ update_forward_refs(ArrayVellumValueRequest, ExternalTestCaseExecutionRequest=ExternalTestCaseExecutionRequest)
@@ -1,11 +1,14 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ from __future__ import annotations
3
4
  from ..core.pydantic_utilities import UniversalBaseModel
5
+ from .array_vellum_value import ArrayVellumValue
4
6
  import typing
5
7
  import datetime as dt
6
8
  from .workflow_output import WorkflowOutput
7
9
  from ..core.pydantic_utilities import IS_PYDANTIC_V2
8
10
  import pydantic
11
+ from ..core.pydantic_utilities import update_forward_refs
9
12
 
10
13
 
11
14
  class FulfilledExecuteWorkflowWorkflowResultEvent(UniversalBaseModel):
@@ -26,3 +29,8 @@ class FulfilledExecuteWorkflowWorkflowResultEvent(UniversalBaseModel):
26
29
  frozen = True
27
30
  smart_union = True
28
31
  extra = pydantic.Extra.allow
32
+
33
+
34
+ update_forward_refs(
35
+ ArrayVellumValue, FulfilledExecuteWorkflowWorkflowResultEvent=FulfilledExecuteWorkflowWorkflowResultEvent
36
+ )