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

vellum/types/deployment_read.py

@@ -3,14 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .deployment_status import DeploymentStatus
 from .environment_enum import EnvironmentEnum
-from .model_type_deprecated import ModelTypeDeprecated
 from .vellum_variable import VellumVariable
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class DeploymentRead(pydantic.BaseModel):
     id: str
@@ -35,7 +37,6 @@ class DeploymentRead(pydantic.BaseModel):
             "* `PRODUCTION` - Production\n"
         )
     )
-    model_type: ModelTypeDeprecated
     active_model_version_ids: typing.List[str]
     last_deployed_on: dt.datetime
     input_variables: typing.List[VellumVariable]

vellum/types/document_document_to_document_index.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .indexing_state_enum import IndexingStateEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class DocumentDocumentToDocumentIndex(pydantic.BaseModel):
     id: str = pydantic.Field(description="Vellum-generated ID that uniquely identifies this link.")

vellum/types/document_index_read.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .document_index_status import DocumentIndexStatus
 from .environment_enum import EnvironmentEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class DocumentIndexRead(pydantic.BaseModel):
     id: str

vellum/types/document_read.py

@@ -3,13 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .document_document_to_document_index import DocumentDocumentToDocumentIndex
 from .document_status import DocumentStatus
 from .processing_state_enum import ProcessingStateEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class DocumentRead(pydantic.BaseModel):
     id: str

vellum/types/enriched_normalized_completion.py

@@ -3,13 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .finish_reason_enum import FinishReasonEnum
 from .normalized_log_probs import NormalizedLogProbs
 from .vellum_variable_type import VellumVariableType
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class EnrichedNormalizedCompletion(pydantic.BaseModel):
     id: str = pydantic.Field(description="The Vellum-generated ID of the completion.")
@@ -30,6 +33,7 @@ class EnrichedNormalizedCompletion(pydantic.BaseModel):
         description="The logprobs of the completion. Only present if specified in the original request options."
     )
     model_version_id: str = pydantic.Field(description="The ID of the model version used to generate this completion.")
+    prompt_version_id: typing.Optional[str]
     type: typing.Optional[VellumVariableType]
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/evaluation_params.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 EvaluationParams(pydantic.BaseModel):
     target: typing.Optional[str] = pydantic.Field(

vellum/types/evaluation_params_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 EvaluationParamsRequest(pydantic.BaseModel):
     target: typing.Optional[str] = pydantic.Field(

vellum/types/execute_workflow_stream_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 ExecuteWorkflowStreamErrorResponse(pydantic.BaseModel):
     detail: str = pydantic.Field(description="Details about why the request failed.")

vellum/types/generate_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 GenerateErrorResponse(pydantic.BaseModel):
     detail: str = pydantic.Field(description="Details about why the request failed.")

vellum/types/generate_options_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .logprobs_enum import LogprobsEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class GenerateOptionsRequest(pydantic.BaseModel):
     logprobs: typing.Optional[LogprobsEnum] = pydantic.Field(

vellum/types/generate_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_request import ChatMessageRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class GenerateRequest(pydantic.BaseModel):
     input_values: typing.Dict[str, typing.Any] = pydantic.Field(

vellum/types/generate_result.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .generate_result_data import GenerateResultData
 from .generate_result_error import GenerateResultError
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class GenerateResult(pydantic.BaseModel):
     data: typing.Optional[GenerateResultData] = pydantic.Field(

vellum/types/generate_result_data.py

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

vellum/types/generate_result_error.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 GenerateResultError(pydantic.BaseModel):
     message: str = pydantic.Field(description="The error message returned by the LLM provider.")

vellum/types/generate_stream_response.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .generate_stream_result import GenerateStreamResult
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class GenerateStreamResponse(pydantic.BaseModel):
     delta: GenerateStreamResult

vellum/types/generate_stream_result.py

@@ -3,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .generate_result_error import GenerateResultError
 from .generate_stream_result_data import GenerateStreamResultData
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class GenerateStreamResult(pydantic.BaseModel):
     request_index: int

vellum/types/generate_stream_result_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .enriched_normalized_completion import EnrichedNormalizedCompletion
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class GenerateStreamResultData(pydantic.BaseModel):
     completion_index: int

vellum/types/metadata_filter_config_request.py

@@ -3,13 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .logical_operator import LogicalOperator
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 from .metadata_filter_rule_request import MetadataFilterRuleRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class MetadataFilterConfigRequest(pydantic.BaseModel):
     combinator: typing.Optional[MetadataFilterRuleCombinator]

vellum/types/metadata_filter_rule_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 .logical_operator import LogicalOperator
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class MetadataFilterRuleRequest(pydantic.BaseModel):
     combinator: typing.Optional[MetadataFilterRuleCombinator]

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(
@@ -16,6 +19,7 @@ class ModelVersionBuildConfig(pydantic.BaseModel):
     sandbox_snapshot: typing.Optional[ModelVersionSandboxSnapshot] = pydantic.Field(
         description="Information about the sandbox snapshot that was used to create this model version, if applicable."
     )
+    prompt_version_id: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

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,21 +3,22 @@
 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
 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.")
     created: dt.datetime = pydantic.Field(description="Timestamp of when this model version was created.")
     label: str = pydantic.Field(description="Human-friendly name for this model version.")
-    model_type: ModelTypeDeprecated
     provider: ProviderEnum = pydantic.Field(
         description=(
             "Which LLM provider this model version is associated with.\n"
@@ -27,8 +28,9 @@ class ModelVersionRead(pydantic.BaseModel):
             "* `GOOGLE` - Google\n"
             "* `HOSTED` - Hosted\n"
             "* `MOSAICML` - MosaicML\n"
-            "* `MYSTIC` - Mystic\n"
             "* `OPENAI` - OpenAI\n"
+            "* `HUGGINGFACE` - HuggingFace\n"
+            "* `MYSTIC` - Mystic\n"
             "* `PYQ` - Pyq\n"
         )
     )

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_error_value.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class NodeInputCompiledErrorValue(pydantic.BaseModel):
+    node_input_id: str
+    key: str
+    value: typing.Optional[VellumError]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

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

@@ -7,6 +7,7 @@ import typing
 import typing_extensions
 
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
+from .node_input_compiled_error_value import NodeInputCompiledErrorValue
 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
@@ -58,10 +59,20 @@ class NodeInputVariableCompiledValue_SearchResults(NodeInputCompiledSearchResult
         allow_population_by_field_name = True
 
 
+class NodeInputVariableCompiledValue_Error(NodeInputCompiledErrorValue):
+    type: typing_extensions.Literal["ERROR"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 NodeInputVariableCompiledValue = typing.Union[
     NodeInputVariableCompiledValue_String,
     NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_Json,
     NodeInputVariableCompiledValue_ChatHistory,
     NodeInputVariableCompiledValue_SearchResults,
+    NodeInputVariableCompiledValue_Error,
 ]

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]