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

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/provider_enum.py

@@ -13,8 +13,9 @@ class ProviderEnum(str, enum.Enum):
     * `GOOGLE` - Google
     * `HOSTED` - Hosted
     * `MOSAICML` - MosaicML
-    * `MYSTIC` - Mystic
     * `OPENAI` - OpenAI
+    * `HUGGINGFACE` - HuggingFace
+    * `MYSTIC` - Mystic
     * `PYQ` - Pyq
     """
 
@@ -23,8 +24,9 @@ class ProviderEnum(str, enum.Enum):
     GOOGLE = "GOOGLE"
     HOSTED = "HOSTED"
     MOSAICML = "MOSAICML"
-    MYSTIC = "MYSTIC"
     OPENAI = "OPENAI"
+    HUGGINGFACE = "HUGGINGFACE"
+    MYSTIC = "MYSTIC"
     PYQ = "PYQ"
 
     def visit(
@@ -34,8 +36,9 @@ class ProviderEnum(str, enum.Enum):
         google: typing.Callable[[], T_Result],
         hosted: typing.Callable[[], T_Result],
         mosaicml: typing.Callable[[], T_Result],
-        mystic: typing.Callable[[], T_Result],
         openai: typing.Callable[[], T_Result],
+        huggingface: typing.Callable[[], T_Result],
+        mystic: typing.Callable[[], T_Result],
         pyq: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ProviderEnum.ANTHROPIC:
@@ -48,9 +51,11 @@ class ProviderEnum(str, enum.Enum):
             return hosted()
         if self is ProviderEnum.MOSAICML:
             return mosaicml()
-        if self is ProviderEnum.MYSTIC:
-            return mystic()
         if self is ProviderEnum.OPENAI:
             return openai()
+        if self is ProviderEnum.HUGGINGFACE:
+            return huggingface()
+        if self is ProviderEnum.MYSTIC:
+            return mystic()
         if self is ProviderEnum.PYQ:
             return pyq()

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.")

vellum/types/search_result_merging_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 SearchResultMergingRequest(pydantic.BaseModel):
     enabled: typing.Optional[bool] = pydantic.Field(description="Whether to enable merging results")

vellum/types/search_weights_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 SearchWeightsRequest(pydantic.BaseModel):
     semantic_similarity: typing.Optional[float] = pydantic.Field(