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

vellum/types/model_version_build_config.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .model_version_sandbox_snapshot import ModelVersionSandboxSnapshot
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ModelVersionBuildConfig(pydantic.BaseModel):
     base_model: str = pydantic.Field(

vellum/types/model_version_compile_prompt_response.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .model_version_compiled_prompt import ModelVersionCompiledPrompt
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ModelVersionCompilePromptResponse(pydantic.BaseModel):
     prompt: ModelVersionCompiledPrompt = pydantic.Field(description="Information about the compiled prompt.")

vellum/types/model_version_compiled_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 ModelVersionCompiledPrompt(pydantic.BaseModel):
     text: str = pydantic.Field(

vellum/types/model_version_exec_config.py

@@ -3,13 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .model_version_exec_config_parameters import ModelVersionExecConfigParameters
 from .prompt_template_block_data import PromptTemplateBlockData
 from .vellum_variable import VellumVariable
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ModelVersionExecConfig(pydantic.BaseModel):
     parameters: ModelVersionExecConfigParameters = pydantic.Field(

vellum/types/model_version_exec_config_parameters.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 ModelVersionExecConfigParameters(pydantic.BaseModel):
     temperature: typing.Optional[float]

vellum/types/model_version_read.py

@@ -3,8 +3,6 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .model_type_deprecated import ModelTypeDeprecated
 from .model_version_build_config import ModelVersionBuildConfig
@@ -12,6 +10,11 @@ from .model_version_exec_config import ModelVersionExecConfig
 from .model_version_read_status_enum import ModelVersionReadStatusEnum
 from .provider_enum import ProviderEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ModelVersionRead(pydantic.BaseModel):
     id: str = pydantic.Field(description="Vellum-generated ID that uniquely identifies this model version.")

vellum/types/model_version_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 ModelVersionSandboxSnapshot(pydantic.BaseModel):
     id: str = pydantic.Field(description="The ID of the sandbox snapshot.")

vellum/types/node_input_compiled_chat_history_value.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message import ChatMessage
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class NodeInputCompiledChatHistoryValue(pydantic.BaseModel):
     node_input_id: str

vellum/types/node_input_compiled_json_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 NodeInputCompiledJsonValue(pydantic.BaseModel):
     node_input_id: str

vellum/types/node_input_compiled_number_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 NodeInputCompiledNumberValue(pydantic.BaseModel):
     node_input_id: str

vellum/types/node_input_compiled_search_results_value.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 NodeInputCompiledSearchResultsValue(pydantic.BaseModel):
     node_input_id: str

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/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,12 +5,15 @@ 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 .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]

vellum/types/prompt_template_block_properties_request.py

@@ -5,12 +5,15 @@ 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 .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]

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