vellum-ai 0.0.41__py3-none-any.whl → 0.0.43__py3-none-any.whl

vellum/types/node_input_compiled_string_value.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class NodeInputCompiledStringValue(pydantic.BaseModel):
     node_input_id: str

vellum/types/node_input_variable_compiled_value.py

@@ -8,6 +8,7 @@ import typing_extensions
 
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
 from .node_input_compiled_json_value import NodeInputCompiledJsonValue
+from .node_input_compiled_number_value import NodeInputCompiledNumberValue
 from .node_input_compiled_search_results_value import NodeInputCompiledSearchResultsValue
 from .node_input_compiled_string_value import NodeInputCompiledStringValue
 
@@ -21,6 +22,15 @@ class NodeInputVariableCompiledValue_String(NodeInputCompiledStringValue):
         allow_population_by_field_name = True
 
 
+class NodeInputVariableCompiledValue_Number(NodeInputCompiledNumberValue):
+    type: typing_extensions.Literal["NUMBER"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 class NodeInputVariableCompiledValue_Json(NodeInputCompiledJsonValue):
     type: typing_extensions.Literal["JSON"]
 
@@ -50,6 +60,7 @@ class NodeInputVariableCompiledValue_SearchResults(NodeInputCompiledSearchResult
 
 NodeInputVariableCompiledValue = typing.Union[
     NodeInputVariableCompiledValue_String,
+    NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_Json,
     NodeInputVariableCompiledValue_ChatHistory,
     NodeInputVariableCompiledValue_SearchResults,

vellum/types/normalized_log_probs.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .normalized_token_log_probs import NormalizedTokenLogProbs
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class NormalizedLogProbs(pydantic.BaseModel):
     tokens: typing.List[NormalizedTokenLogProbs]

vellum/types/normalized_token_log_probs.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class NormalizedTokenLogProbs(pydantic.BaseModel):
     token: str

vellum/types/paginated_slim_document_list.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .slim_document import SlimDocument
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PaginatedSlimDocumentList(pydantic.BaseModel):
     count: typing.Optional[int]

vellum/types/prompt_node_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .prompt_node_result_data import PromptNodeResultData
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PromptNodeResult(pydantic.BaseModel):
     data: PromptNodeResultData

vellum/types/prompt_node_result_data.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PromptNodeResultData(pydantic.BaseModel):
     output_id: str

vellum/types/prompt_template_block.py

@@ -5,11 +5,14 @@ from __future__ import annotations
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .block_type_enum import BlockTypeEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PromptTemplateBlock(pydantic.BaseModel):
     id: str

vellum/types/prompt_template_block_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .prompt_template_block import PromptTemplateBlock
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PromptTemplateBlockData(pydantic.BaseModel):
     version: int

vellum/types/prompt_template_block_data_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .prompt_template_block_request import PromptTemplateBlockRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PromptTemplateBlockDataRequest(pydantic.BaseModel):
     version: int

vellum/types/prompt_template_block_properties.py

@@ -5,18 +5,21 @@ from __future__ import annotations
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_role import ChatMessageRole
-from .content_type import ContentType
+from .vellum_variable_type import VellumVariableType
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
 
 
 class PromptTemplateBlockProperties(pydantic.BaseModel):
     chat_role: typing.Optional[ChatMessageRole]
     chat_message_unterminated: typing.Optional[bool]
     template: typing.Optional[str]
-    template_type: typing.Optional[ContentType]
+    template_type: typing.Optional[VellumVariableType]
     function_name: typing.Optional[str]
     function_description: typing.Optional[str]
     function_parameters: typing.Optional[typing.Dict[str, typing.Any]]

vellum/types/prompt_template_block_properties_request.py

@@ -5,18 +5,21 @@ from __future__ import annotations
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_role import ChatMessageRole
-from .content_type import ContentType
+from .vellum_variable_type import VellumVariableType
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
 
 
 class PromptTemplateBlockPropertiesRequest(pydantic.BaseModel):
     chat_role: typing.Optional[ChatMessageRole]
     chat_message_unterminated: typing.Optional[bool]
     template: typing.Optional[str]
-    template_type: typing.Optional[ContentType]
+    template_type: typing.Optional[VellumVariableType]
     function_name: typing.Optional[str]
     function_description: typing.Optional[str]
     function_parameters: typing.Optional[typing.Dict[str, typing.Any]]

vellum/types/prompt_template_block_request.py

@@ -5,11 +5,14 @@ from __future__ import annotations
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .block_type_enum import BlockTypeEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class PromptTemplateBlockRequest(pydantic.BaseModel):
     id: str

vellum/types/register_prompt_error_response.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisterPromptErrorResponse(pydantic.BaseModel):
     detail: str = pydantic.Field(description="Details about why the request failed.")

vellum/types/register_prompt_model_parameters_request.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisterPromptModelParametersRequest(pydantic.BaseModel):
     temperature: float

vellum/types/register_prompt_prompt.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisterPromptPrompt(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the generated prompt.")

vellum/types/register_prompt_prompt_info_request.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .prompt_template_block_data_request import PromptTemplateBlockDataRequest
 from .registered_prompt_input_variable_request import RegisteredPromptInputVariableRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisterPromptPromptInfoRequest(pydantic.BaseModel):
     prompt_syntax_version: typing.Optional[int]

vellum/types/register_prompt_response.py

@@ -3,8 +3,6 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .register_prompt_prompt import RegisterPromptPrompt
 from .registered_prompt_deployment import RegisteredPromptDeployment
@@ -12,6 +10,11 @@ from .registered_prompt_model_version import RegisteredPromptModelVersion
 from .registered_prompt_sandbox import RegisteredPromptSandbox
 from .registered_prompt_sandbox_snapshot import RegisteredPromptSandboxSnapshot
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisterPromptResponse(pydantic.BaseModel):
     prompt: RegisterPromptPrompt = pydantic.Field(description="Information about the generated prompt")

vellum/types/registered_prompt_deployment.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisteredPromptDeployment(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the generated deployment.")

vellum/types/registered_prompt_input_variable_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .vellum_variable_type import VellumVariableType
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisteredPromptInputVariableRequest(pydantic.BaseModel):
     key: str

vellum/types/registered_prompt_model_version.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisteredPromptModelVersion(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the generated model version.")

vellum/types/registered_prompt_sandbox.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisteredPromptSandbox(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the generated sandbox.")

vellum/types/registered_prompt_sandbox_snapshot.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class RegisteredPromptSandboxSnapshot(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the generated sandbox snapshot.")

vellum/types/sandbox_metric_input_params.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .evaluation_params import EvaluationParams
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SandboxMetricInputParams(pydantic.BaseModel):
     params: typing.Optional[EvaluationParams]

vellum/types/sandbox_metric_input_params_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .evaluation_params_request import EvaluationParamsRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SandboxMetricInputParamsRequest(pydantic.BaseModel):
     params: typing.Optional[EvaluationParamsRequest]

vellum/types/sandbox_scenario.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .sandbox_metric_input_params import SandboxMetricInputParams
 from .scenario_input import ScenarioInput
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SandboxScenario(pydantic.BaseModel):
     label: typing.Optional[str]

vellum/types/scenario_input.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message import ChatMessage
 from .scenario_input_type_enum import ScenarioInputTypeEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ScenarioInput(pydantic.BaseModel):
     key: str

vellum/types/scenario_input_request.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_request import ChatMessageRequest
 from .scenario_input_type_enum import ScenarioInputTypeEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ScenarioInputRequest(pydantic.BaseModel):
     key: str

vellum/types/search_error_response.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchErrorResponse(pydantic.BaseModel):
     detail: str = pydantic.Field(description="Details about why the request failed.")

vellum/types/search_filters_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .metadata_filter_config_request import MetadataFilterConfigRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchFiltersRequest(pydantic.BaseModel):
     external_ids: typing.Optional[typing.List[str]] = pydantic.Field(

vellum/types/search_node_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_node_result_data import SearchNodeResultData
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchNodeResult(pydantic.BaseModel):
     data: SearchNodeResultData

vellum/types/search_node_result_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_result import SearchResult
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchNodeResultData(pydantic.BaseModel):
     results_output_id: str

vellum/types/search_request_options_request.py

@@ -3,13 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_filters_request import SearchFiltersRequest
 from .search_result_merging_request import SearchResultMergingRequest
 from .search_weights_request import SearchWeightsRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchRequestOptionsRequest(pydantic.BaseModel):
     limit: typing.Optional[int] = pydantic.Field(description="The maximum number of results to return.")

vellum/types/search_response.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_result import SearchResult
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchResponse(pydantic.BaseModel):
     results: typing.List[SearchResult] = pydantic.Field(

vellum/types/search_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_result_document import SearchResultDocument
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchResult(pydantic.BaseModel):
     text: str = pydantic.Field(description="The text of the chunk that matched the search query.")

vellum/types/search_result_document.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchResultDocument(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the document.")