vellum-ai 0.9.11__py3-none-any.whl → 0.9.13__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- vellum/__init__.py +28 -22
- vellum/client.py +20 -20
- vellum/core/client_wrapper.py +1 -1
- vellum/resources/ad_hoc/client.py +39 -2
- vellum/resources/metric_definitions/client.py +11 -11
- vellum/types/__init__.py +28 -22
- vellum/types/{array_input_request.py → array_input.py} +5 -5
- vellum/types/chat_history_input.py +30 -0
- vellum/types/{code_execution_package_request.py → code_execution_package.py} +1 -1
- vellum/types/code_executor_input.py +24 -0
- vellum/types/{code_executor_secret_input_request.py → code_executor_secret_input.py} +1 -1
- vellum/types/document_processing_state.py +7 -0
- vellum/types/document_read.py +2 -11
- vellum/types/{error_input_request.py → error_input.py} +3 -3
- vellum/types/{function_call_input_request.py → function_call_input.py} +3 -3
- vellum/types/{function_definition_prompt_block.py → function_definition.py} +6 -7
- vellum/types/json_input.py +29 -0
- vellum/types/metric_definition_input.py +9 -0
- vellum/types/{number_input_request.py → number_input.py} +1 -1
- vellum/types/prompt_block.py +1 -4
- vellum/types/{search_results_input_request.py → search_results_input.py} +3 -3
- vellum/types/slim_document.py +2 -11
- vellum/types/string_input.py +29 -0
- vellum/types/vellum_error_code_enum.py +6 -1
- {vellum_ai-0.9.11.dist-info → vellum_ai-0.9.13.dist-info}/METADATA +1 -1
- {vellum_ai-0.9.11.dist-info → vellum_ai-0.9.13.dist-info}/RECORD +28 -25
- vellum/types/code_executor_input_request.py +0 -24
- vellum/types/metric_definition_input_request.py +0 -11
- vellum/types/processing_state_enum.py +0 -5
- {vellum_ai-0.9.11.dist-info → vellum_ai-0.9.13.dist-info}/LICENSE +0 -0
- {vellum_ai-0.9.11.dist-info → vellum_ai-0.9.13.dist-info}/WHEEL +0 -0
vellum/types/__init__.py
CHANGED
@@ -13,7 +13,7 @@ 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 .
|
16
|
+
from .array_input import ArrayInput
|
17
17
|
from .array_variable_value import ArrayVariableValue
|
18
18
|
from .array_variable_value_item import ArrayVariableValueItem
|
19
19
|
from .array_vellum_value import ArrayVellumValue
|
@@ -37,6 +37,7 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1 import
|
|
37
37
|
from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request import (
|
38
38
|
BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
|
39
39
|
)
|
40
|
+
from .chat_history_input import ChatHistoryInput
|
40
41
|
from .chat_history_input_request import ChatHistoryInputRequest
|
41
42
|
from .chat_history_variable_value import ChatHistoryVariableValue
|
42
43
|
from .chat_history_vellum_value import ChatHistoryVellumValue
|
@@ -58,11 +59,11 @@ from .code_execution_node_result_data import CodeExecutionNodeResultData
|
|
58
59
|
from .code_execution_node_result_output import CodeExecutionNodeResultOutput
|
59
60
|
from .code_execution_node_search_results_result import CodeExecutionNodeSearchResultsResult
|
60
61
|
from .code_execution_node_string_result import CodeExecutionNodeStringResult
|
61
|
-
from .
|
62
|
+
from .code_execution_package import CodeExecutionPackage
|
62
63
|
from .code_execution_runtime import CodeExecutionRuntime
|
63
|
-
from .
|
64
|
+
from .code_executor_input import CodeExecutorInput
|
64
65
|
from .code_executor_response import CodeExecutorResponse
|
65
|
-
from .
|
66
|
+
from .code_executor_secret_input import CodeExecutorSecretInput
|
66
67
|
from .compile_prompt_deployment_expand_meta_request import CompilePromptDeploymentExpandMetaRequest
|
67
68
|
from .compile_prompt_meta import CompilePromptMeta
|
68
69
|
from .components_schemas_pdf_search_result_meta_source import ComponentsSchemasPdfSearchResultMetaSource
|
@@ -84,6 +85,7 @@ from .document_index_chunking_request import DocumentIndexChunkingRequest
|
|
84
85
|
from .document_index_indexing_config import DocumentIndexIndexingConfig
|
85
86
|
from .document_index_indexing_config_request import DocumentIndexIndexingConfigRequest
|
86
87
|
from .document_index_read import DocumentIndexRead
|
88
|
+
from .document_processing_state import DocumentProcessingState
|
87
89
|
from .document_read import DocumentRead
|
88
90
|
from .document_status import DocumentStatus
|
89
91
|
from .enriched_normalized_completion import EnrichedNormalizedCompletion
|
@@ -92,7 +94,7 @@ from .entity_visibility import EntityVisibility
|
|
92
94
|
from .environment_enum import EnvironmentEnum
|
93
95
|
from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
|
94
96
|
from .ephemeral_prompt_cache_config_type_enum import EphemeralPromptCacheConfigTypeEnum
|
95
|
-
from .
|
97
|
+
from .error_input import ErrorInput
|
96
98
|
from .error_variable_value import ErrorVariableValue
|
97
99
|
from .error_vellum_value import ErrorVellumValue
|
98
100
|
from .error_vellum_value_request import ErrorVellumValueRequest
|
@@ -135,12 +137,12 @@ from .function_call_chat_message_content import FunctionCallChatMessageContent
|
|
135
137
|
from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
|
136
138
|
from .function_call_chat_message_content_value import FunctionCallChatMessageContentValue
|
137
139
|
from .function_call_chat_message_content_value_request import FunctionCallChatMessageContentValueRequest
|
138
|
-
from .
|
140
|
+
from .function_call_input import FunctionCallInput
|
139
141
|
from .function_call_request import FunctionCallRequest
|
140
142
|
from .function_call_variable_value import FunctionCallVariableValue
|
141
143
|
from .function_call_vellum_value import FunctionCallVellumValue
|
142
144
|
from .function_call_vellum_value_request import FunctionCallVellumValueRequest
|
143
|
-
from .
|
145
|
+
from .function_definition import FunctionDefinition
|
144
146
|
from .generate_options_request import GenerateOptionsRequest
|
145
147
|
from .generate_request import GenerateRequest
|
146
148
|
from .generate_response import GenerateResponse
|
@@ -178,6 +180,7 @@ from .instructor_vectorizer_config import InstructorVectorizerConfig
|
|
178
180
|
from .instructor_vectorizer_config_request import InstructorVectorizerConfigRequest
|
179
181
|
from .iteration_state_enum import IterationStateEnum
|
180
182
|
from .jinja_prompt_block import JinjaPromptBlock
|
183
|
+
from .json_input import JsonInput
|
181
184
|
from .json_input_request import JsonInputRequest
|
182
185
|
from .json_variable_value import JsonVariableValue
|
183
186
|
from .json_vellum_value import JsonVellumValue
|
@@ -193,7 +196,7 @@ from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
|
|
193
196
|
from .metadata_filter_rule_request import MetadataFilterRuleRequest
|
194
197
|
from .metadata_filters_request import MetadataFiltersRequest
|
195
198
|
from .metric_definition_execution import MetricDefinitionExecution
|
196
|
-
from .
|
199
|
+
from .metric_definition_input import MetricDefinitionInput
|
197
200
|
from .metric_node_result import MetricNodeResult
|
198
201
|
from .ml_model_usage import MlModelUsage
|
199
202
|
from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
|
@@ -238,7 +241,7 @@ from .node_output_compiled_string_value import NodeOutputCompiledStringValue
|
|
238
241
|
from .node_output_compiled_value import NodeOutputCompiledValue
|
239
242
|
from .normalized_log_probs import NormalizedLogProbs
|
240
243
|
from .normalized_token_log_probs import NormalizedTokenLogProbs
|
241
|
-
from .
|
244
|
+
from .number_input import NumberInput
|
242
245
|
from .number_variable_value import NumberVariableValue
|
243
246
|
from .number_vellum_value import NumberVellumValue
|
244
247
|
from .number_vellum_value_request import NumberVellumValueRequest
|
@@ -265,7 +268,6 @@ from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequ
|
|
265
268
|
from .plain_text_prompt_block import PlainTextPromptBlock
|
266
269
|
from .price import Price
|
267
270
|
from .processing_failure_reason_enum import ProcessingFailureReasonEnum
|
268
|
-
from .processing_state_enum import ProcessingStateEnum
|
269
271
|
from .prompt_block import PromptBlock
|
270
272
|
from .prompt_block_state import PromptBlockState
|
271
273
|
from .prompt_deployment_expand_meta_request import PromptDeploymentExpandMetaRequest
|
@@ -313,7 +315,7 @@ from .search_result_merging_request import SearchResultMergingRequest
|
|
313
315
|
from .search_result_meta import SearchResultMeta
|
314
316
|
from .search_result_meta_request import SearchResultMetaRequest
|
315
317
|
from .search_result_request import SearchResultRequest
|
316
|
-
from .
|
318
|
+
from .search_results_input import SearchResultsInput
|
317
319
|
from .search_results_variable_value import SearchResultsVariableValue
|
318
320
|
from .search_results_vellum_value import SearchResultsVellumValue
|
319
321
|
from .search_results_vellum_value_request import SearchResultsVellumValueRequest
|
@@ -332,6 +334,7 @@ from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
|
|
332
334
|
from .streaming_workflow_node_result_event import StreamingWorkflowNodeResultEvent
|
333
335
|
from .string_chat_message_content import StringChatMessageContent
|
334
336
|
from .string_chat_message_content_request import StringChatMessageContentRequest
|
337
|
+
from .string_input import StringInput
|
335
338
|
from .string_input_request import StringInputRequest
|
336
339
|
from .string_variable_value import StringVariableValue
|
337
340
|
from .string_vellum_value import StringVellumValue
|
@@ -507,7 +510,7 @@ __all__ = [
|
|
507
510
|
"ArrayChatMessageContentItem",
|
508
511
|
"ArrayChatMessageContentItemRequest",
|
509
512
|
"ArrayChatMessageContentRequest",
|
510
|
-
"
|
513
|
+
"ArrayInput",
|
511
514
|
"ArrayVariableValue",
|
512
515
|
"ArrayVariableValueItem",
|
513
516
|
"ArrayVellumValue",
|
@@ -523,6 +526,7 @@ __all__ = [
|
|
523
526
|
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
|
524
527
|
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
|
525
528
|
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
|
529
|
+
"ChatHistoryInput",
|
526
530
|
"ChatHistoryInputRequest",
|
527
531
|
"ChatHistoryVariableValue",
|
528
532
|
"ChatHistoryVellumValue",
|
@@ -544,11 +548,11 @@ __all__ = [
|
|
544
548
|
"CodeExecutionNodeResultOutput",
|
545
549
|
"CodeExecutionNodeSearchResultsResult",
|
546
550
|
"CodeExecutionNodeStringResult",
|
547
|
-
"
|
551
|
+
"CodeExecutionPackage",
|
548
552
|
"CodeExecutionRuntime",
|
549
|
-
"
|
553
|
+
"CodeExecutorInput",
|
550
554
|
"CodeExecutorResponse",
|
551
|
-
"
|
555
|
+
"CodeExecutorSecretInput",
|
552
556
|
"CompilePromptDeploymentExpandMetaRequest",
|
553
557
|
"CompilePromptMeta",
|
554
558
|
"ComponentsSchemasPdfSearchResultMetaSource",
|
@@ -570,6 +574,7 @@ __all__ = [
|
|
570
574
|
"DocumentIndexIndexingConfig",
|
571
575
|
"DocumentIndexIndexingConfigRequest",
|
572
576
|
"DocumentIndexRead",
|
577
|
+
"DocumentProcessingState",
|
573
578
|
"DocumentRead",
|
574
579
|
"DocumentStatus",
|
575
580
|
"EnrichedNormalizedCompletion",
|
@@ -578,7 +583,7 @@ __all__ = [
|
|
578
583
|
"EnvironmentEnum",
|
579
584
|
"EphemeralPromptCacheConfig",
|
580
585
|
"EphemeralPromptCacheConfigTypeEnum",
|
581
|
-
"
|
586
|
+
"ErrorInput",
|
582
587
|
"ErrorVariableValue",
|
583
588
|
"ErrorVellumValue",
|
584
589
|
"ErrorVellumValueRequest",
|
@@ -621,12 +626,12 @@ __all__ = [
|
|
621
626
|
"FunctionCallChatMessageContentRequest",
|
622
627
|
"FunctionCallChatMessageContentValue",
|
623
628
|
"FunctionCallChatMessageContentValueRequest",
|
624
|
-
"
|
629
|
+
"FunctionCallInput",
|
625
630
|
"FunctionCallRequest",
|
626
631
|
"FunctionCallVariableValue",
|
627
632
|
"FunctionCallVellumValue",
|
628
633
|
"FunctionCallVellumValueRequest",
|
629
|
-
"
|
634
|
+
"FunctionDefinition",
|
630
635
|
"GenerateOptionsRequest",
|
631
636
|
"GenerateRequest",
|
632
637
|
"GenerateResponse",
|
@@ -660,6 +665,7 @@ __all__ = [
|
|
660
665
|
"InstructorVectorizerConfigRequest",
|
661
666
|
"IterationStateEnum",
|
662
667
|
"JinjaPromptBlock",
|
668
|
+
"JsonInput",
|
663
669
|
"JsonInputRequest",
|
664
670
|
"JsonVariableValue",
|
665
671
|
"JsonVellumValue",
|
@@ -675,7 +681,7 @@ __all__ = [
|
|
675
681
|
"MetadataFilterRuleRequest",
|
676
682
|
"MetadataFiltersRequest",
|
677
683
|
"MetricDefinitionExecution",
|
678
|
-
"
|
684
|
+
"MetricDefinitionInput",
|
679
685
|
"MetricNodeResult",
|
680
686
|
"MlModelUsage",
|
681
687
|
"NamedScenarioInputChatHistoryVariableValueRequest",
|
@@ -720,7 +726,7 @@ __all__ = [
|
|
720
726
|
"NodeOutputCompiledValue",
|
721
727
|
"NormalizedLogProbs",
|
722
728
|
"NormalizedTokenLogProbs",
|
723
|
-
"
|
729
|
+
"NumberInput",
|
724
730
|
"NumberVariableValue",
|
725
731
|
"NumberVellumValue",
|
726
732
|
"NumberVellumValueRequest",
|
@@ -747,7 +753,6 @@ __all__ = [
|
|
747
753
|
"PlainTextPromptBlock",
|
748
754
|
"Price",
|
749
755
|
"ProcessingFailureReasonEnum",
|
750
|
-
"ProcessingStateEnum",
|
751
756
|
"PromptBlock",
|
752
757
|
"PromptBlockState",
|
753
758
|
"PromptDeploymentExpandMetaRequest",
|
@@ -795,7 +800,7 @@ __all__ = [
|
|
795
800
|
"SearchResultMeta",
|
796
801
|
"SearchResultMetaRequest",
|
797
802
|
"SearchResultRequest",
|
798
|
-
"
|
803
|
+
"SearchResultsInput",
|
799
804
|
"SearchResultsVariableValue",
|
800
805
|
"SearchResultsVellumValue",
|
801
806
|
"SearchResultsVellumValueRequest",
|
@@ -814,6 +819,7 @@ __all__ = [
|
|
814
819
|
"StreamingWorkflowNodeResultEvent",
|
815
820
|
"StringChatMessageContent",
|
816
821
|
"StringChatMessageContentRequest",
|
822
|
+
"StringInput",
|
817
823
|
"StringInputRequest",
|
818
824
|
"StringVariableValue",
|
819
825
|
"StringVellumValue",
|
@@ -2,15 +2,15 @@
|
|
2
2
|
|
3
3
|
from __future__ import annotations
|
4
4
|
from ..core.pydantic_utilities import UniversalBaseModel
|
5
|
-
from .
|
5
|
+
from .array_vellum_value import ArrayVellumValue
|
6
6
|
import pydantic
|
7
7
|
import typing
|
8
|
-
from .
|
8
|
+
from .vellum_value import VellumValue
|
9
9
|
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
10
10
|
from ..core.pydantic_utilities import update_forward_refs
|
11
11
|
|
12
12
|
|
13
|
-
class
|
13
|
+
class ArrayInput(UniversalBaseModel):
|
14
14
|
"""
|
15
15
|
A user input representing a Vellum Array value
|
16
16
|
"""
|
@@ -21,7 +21,7 @@ class ArrayInputRequest(UniversalBaseModel):
|
|
21
21
|
"""
|
22
22
|
|
23
23
|
type: typing.Literal["ARRAY"] = "ARRAY"
|
24
|
-
value: typing.List[
|
24
|
+
value: typing.List[VellumValue]
|
25
25
|
|
26
26
|
if IS_PYDANTIC_V2:
|
27
27
|
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
@@ -33,4 +33,4 @@ class ArrayInputRequest(UniversalBaseModel):
|
|
33
33
|
extra = pydantic.Extra.allow
|
34
34
|
|
35
35
|
|
36
|
-
update_forward_refs(
|
36
|
+
update_forward_refs(ArrayVellumValue, ArrayInput=ArrayInput)
|
@@ -0,0 +1,30 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
from ..core.pydantic_utilities import UniversalBaseModel
|
4
|
+
import pydantic
|
5
|
+
import typing
|
6
|
+
from .chat_message import ChatMessage
|
7
|
+
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
8
|
+
|
9
|
+
|
10
|
+
class ChatHistoryInput(UniversalBaseModel):
|
11
|
+
"""
|
12
|
+
A user input representing a list of chat messages
|
13
|
+
"""
|
14
|
+
|
15
|
+
name: str = pydantic.Field()
|
16
|
+
"""
|
17
|
+
The variable's name, as defined in the deployment.
|
18
|
+
"""
|
19
|
+
|
20
|
+
type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
|
21
|
+
value: typing.List[ChatMessage]
|
22
|
+
|
23
|
+
if IS_PYDANTIC_V2:
|
24
|
+
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
25
|
+
else:
|
26
|
+
|
27
|
+
class Config:
|
28
|
+
frozen = True
|
29
|
+
smart_union = True
|
30
|
+
extra = pydantic.Extra.allow
|
@@ -0,0 +1,24 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
import typing
|
4
|
+
from .string_input import StringInput
|
5
|
+
from .json_input import JsonInput
|
6
|
+
from .chat_history_input import ChatHistoryInput
|
7
|
+
from .number_input import NumberInput
|
8
|
+
from .search_results_input import SearchResultsInput
|
9
|
+
from .error_input import ErrorInput
|
10
|
+
from .array_input import ArrayInput
|
11
|
+
from .function_call_input import FunctionCallInput
|
12
|
+
from .code_executor_secret_input import CodeExecutorSecretInput
|
13
|
+
|
14
|
+
CodeExecutorInput = typing.Union[
|
15
|
+
StringInput,
|
16
|
+
JsonInput,
|
17
|
+
ChatHistoryInput,
|
18
|
+
NumberInput,
|
19
|
+
SearchResultsInput,
|
20
|
+
ErrorInput,
|
21
|
+
ArrayInput,
|
22
|
+
FunctionCallInput,
|
23
|
+
CodeExecutorSecretInput,
|
24
|
+
]
|
vellum/types/document_read.py
CHANGED
@@ -4,7 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
|
|
4
4
|
import typing
|
5
5
|
import pydantic
|
6
6
|
import datetime as dt
|
7
|
-
from .
|
7
|
+
from .document_processing_state import DocumentProcessingState
|
8
8
|
from .document_status import DocumentStatus
|
9
9
|
from .document_document_to_document_index import DocumentDocumentToDocumentIndex
|
10
10
|
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
@@ -23,16 +23,7 @@ class DocumentRead(UniversalBaseModel):
|
|
23
23
|
A human-readable label for the document. Defaults to the originally uploaded file's file name.
|
24
24
|
"""
|
25
25
|
|
26
|
-
processing_state:
|
27
|
-
"""
|
28
|
-
The current processing state of the document
|
29
|
-
|
30
|
-
- `QUEUED` - Queued
|
31
|
-
- `PROCESSING` - Processing
|
32
|
-
- `PROCESSED` - Processed
|
33
|
-
- `FAILED` - Failed
|
34
|
-
"""
|
35
|
-
|
26
|
+
processing_state: DocumentProcessingState
|
36
27
|
status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
|
37
28
|
"""
|
38
29
|
The current status of the document
|
@@ -3,11 +3,11 @@
|
|
3
3
|
from ..core.pydantic_utilities import UniversalBaseModel
|
4
4
|
import pydantic
|
5
5
|
import typing
|
6
|
-
from .
|
6
|
+
from .vellum_error import VellumError
|
7
7
|
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
8
8
|
|
9
9
|
|
10
|
-
class
|
10
|
+
class ErrorInput(UniversalBaseModel):
|
11
11
|
"""
|
12
12
|
A user input representing a Vellum Error value
|
13
13
|
"""
|
@@ -18,7 +18,7 @@ class ErrorInputRequest(UniversalBaseModel):
|
|
18
18
|
"""
|
19
19
|
|
20
20
|
type: typing.Literal["ERROR"] = "ERROR"
|
21
|
-
value:
|
21
|
+
value: VellumError
|
22
22
|
|
23
23
|
if IS_PYDANTIC_V2:
|
24
24
|
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
@@ -3,11 +3,11 @@
|
|
3
3
|
from ..core.pydantic_utilities import UniversalBaseModel
|
4
4
|
import pydantic
|
5
5
|
import typing
|
6
|
-
from .
|
6
|
+
from .function_call import FunctionCall
|
7
7
|
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
8
8
|
|
9
9
|
|
10
|
-
class
|
10
|
+
class FunctionCallInput(UniversalBaseModel):
|
11
11
|
"""
|
12
12
|
A user input representing a Vellum Function Call value
|
13
13
|
"""
|
@@ -18,7 +18,7 @@ class FunctionCallInputRequest(UniversalBaseModel):
|
|
18
18
|
"""
|
19
19
|
|
20
20
|
type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
|
21
|
-
value:
|
21
|
+
value: FunctionCall
|
22
22
|
|
23
23
|
if IS_PYDANTIC_V2:
|
24
24
|
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
@@ -8,19 +8,18 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
|
8
8
|
import pydantic
|
9
9
|
|
10
10
|
|
11
|
-
class
|
11
|
+
class FunctionDefinition(UniversalBaseModel):
|
12
12
|
"""
|
13
13
|
A block that represents a function definition in a prompt template.
|
14
14
|
"""
|
15
15
|
|
16
16
|
state: typing.Optional[PromptBlockState] = None
|
17
17
|
cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
function_strict: typing.Optional[bool] = None
|
18
|
+
name: typing.Optional[str] = None
|
19
|
+
description: typing.Optional[str] = None
|
20
|
+
parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
|
21
|
+
forced: typing.Optional[bool] = None
|
22
|
+
strict: typing.Optional[bool] = None
|
24
23
|
|
25
24
|
if IS_PYDANTIC_V2:
|
26
25
|
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
@@ -0,0 +1,29 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
from ..core.pydantic_utilities import UniversalBaseModel
|
4
|
+
import pydantic
|
5
|
+
import typing
|
6
|
+
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
7
|
+
|
8
|
+
|
9
|
+
class JsonInput(UniversalBaseModel):
|
10
|
+
"""
|
11
|
+
A user input representing a JSON object
|
12
|
+
"""
|
13
|
+
|
14
|
+
name: str = pydantic.Field()
|
15
|
+
"""
|
16
|
+
The variable's name
|
17
|
+
"""
|
18
|
+
|
19
|
+
type: typing.Literal["JSON"] = "JSON"
|
20
|
+
value: typing.Optional[typing.Any] = None
|
21
|
+
|
22
|
+
if IS_PYDANTIC_V2:
|
23
|
+
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
24
|
+
else:
|
25
|
+
|
26
|
+
class Config:
|
27
|
+
frozen = True
|
28
|
+
smart_union = True
|
29
|
+
extra = pydantic.Extra.allow
|
@@ -0,0 +1,9 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
import typing
|
4
|
+
from .string_input import StringInput
|
5
|
+
from .json_input import JsonInput
|
6
|
+
from .chat_history_input import ChatHistoryInput
|
7
|
+
from .number_input import NumberInput
|
8
|
+
|
9
|
+
MetricDefinitionInput = typing.Union[StringInput, JsonInput, ChatHistoryInput, NumberInput]
|
vellum/types/prompt_block.py
CHANGED
@@ -3,13 +3,10 @@
|
|
3
3
|
from __future__ import annotations
|
4
4
|
import typing
|
5
5
|
from .jinja_prompt_block import JinjaPromptBlock
|
6
|
-
from .function_definition_prompt_block import FunctionDefinitionPromptBlock
|
7
6
|
from .variable_prompt_block import VariablePromptBlock
|
8
7
|
from .rich_text_prompt_block import RichTextPromptBlock
|
9
8
|
import typing
|
10
9
|
|
11
10
|
if typing.TYPE_CHECKING:
|
12
11
|
from .chat_message_prompt_block import ChatMessagePromptBlock
|
13
|
-
PromptBlock = typing.Union[
|
14
|
-
JinjaPromptBlock, "ChatMessagePromptBlock", FunctionDefinitionPromptBlock, VariablePromptBlock, RichTextPromptBlock
|
15
|
-
]
|
12
|
+
PromptBlock = typing.Union[JinjaPromptBlock, "ChatMessagePromptBlock", VariablePromptBlock, RichTextPromptBlock]
|
@@ -3,11 +3,11 @@
|
|
3
3
|
from ..core.pydantic_utilities import UniversalBaseModel
|
4
4
|
import pydantic
|
5
5
|
import typing
|
6
|
-
from .
|
6
|
+
from .search_result import SearchResult
|
7
7
|
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
8
8
|
|
9
9
|
|
10
|
-
class
|
10
|
+
class SearchResultsInput(UniversalBaseModel):
|
11
11
|
"""
|
12
12
|
A user input representing a search results value
|
13
13
|
"""
|
@@ -18,7 +18,7 @@ class SearchResultsInputRequest(UniversalBaseModel):
|
|
18
18
|
"""
|
19
19
|
|
20
20
|
type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
|
21
|
-
value: typing.List[
|
21
|
+
value: typing.List[SearchResult]
|
22
22
|
|
23
23
|
if IS_PYDANTIC_V2:
|
24
24
|
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
vellum/types/slim_document.py
CHANGED
@@ -4,7 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
|
|
4
4
|
import pydantic
|
5
5
|
import typing
|
6
6
|
import datetime as dt
|
7
|
-
from .
|
7
|
+
from .document_processing_state import DocumentProcessingState
|
8
8
|
from .processing_failure_reason_enum import ProcessingFailureReasonEnum
|
9
9
|
from .document_status import DocumentStatus
|
10
10
|
from .document_document_to_document_index import DocumentDocumentToDocumentIndex
|
@@ -32,16 +32,7 @@ class SlimDocument(UniversalBaseModel):
|
|
32
32
|
Human-friendly name for this document.
|
33
33
|
"""
|
34
34
|
|
35
|
-
processing_state:
|
36
|
-
"""
|
37
|
-
An enum value representing where this document is along its processing lifecycle. Note that this is different than its indexing lifecycle.
|
38
|
-
|
39
|
-
- `QUEUED` - Queued
|
40
|
-
- `PROCESSING` - Processing
|
41
|
-
- `PROCESSED` - Processed
|
42
|
-
- `FAILED` - Failed
|
43
|
-
"""
|
44
|
-
|
35
|
+
processing_state: DocumentProcessingState
|
45
36
|
processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic.Field(default=None)
|
46
37
|
"""
|
47
38
|
An enum value representing why the document could not be processed. Is null unless processing_state is FAILED.
|
@@ -0,0 +1,29 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
from ..core.pydantic_utilities import UniversalBaseModel
|
4
|
+
import pydantic
|
5
|
+
import typing
|
6
|
+
from ..core.pydantic_utilities import IS_PYDANTIC_V2
|
7
|
+
|
8
|
+
|
9
|
+
class StringInput(UniversalBaseModel):
|
10
|
+
"""
|
11
|
+
A user input representing a string value
|
12
|
+
"""
|
13
|
+
|
14
|
+
name: str = pydantic.Field()
|
15
|
+
"""
|
16
|
+
The variable's name
|
17
|
+
"""
|
18
|
+
|
19
|
+
type: typing.Literal["STRING"] = "STRING"
|
20
|
+
value: str
|
21
|
+
|
22
|
+
if IS_PYDANTIC_V2:
|
23
|
+
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
|
24
|
+
else:
|
25
|
+
|
26
|
+
class Config:
|
27
|
+
frozen = True
|
28
|
+
smart_union = True
|
29
|
+
extra = pydantic.Extra.allow
|
@@ -4,7 +4,12 @@ import typing
|
|
4
4
|
|
5
5
|
VellumErrorCodeEnum = typing.Union[
|
6
6
|
typing.Literal[
|
7
|
-
"INVALID_REQUEST",
|
7
|
+
"INVALID_REQUEST",
|
8
|
+
"INVALID_INPUTS",
|
9
|
+
"PROVIDER_ERROR",
|
10
|
+
"REQUEST_TIMEOUT",
|
11
|
+
"INTERNAL_SERVER_ERROR",
|
12
|
+
"USER_DEFINED_ERROR",
|
8
13
|
],
|
9
14
|
typing.Any,
|
10
15
|
]
|