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

vellum/types/api_node_result_data.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ApiNodeResultData(pydantic.BaseModel):
+    text_output_id: str
+    text: typing.Optional[str]
+    json_output_id: str
+    json_: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(alias="json")
+    status_code_output_id: str
+    status_code: int
+
+    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
+        allow_population_by_field_name = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/chat_message.py

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

vellum/types/chat_message_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_role import ChatMessageRole
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ChatMessageRequest(pydantic.BaseModel):
     text: str

vellum/types/conditional_node_result.py

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

vellum/types/conditional_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 ConditionalNodeResultData(pydantic.BaseModel):
     source_handle_id: typing.Optional[str]

vellum/types/deployment_read.py

@@ -3,14 +3,17 @@
 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

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,12 +3,15 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
-from .content_type import ContentType
 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):
@@ -30,7 +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.")
-    type: typing.Optional[ContentType]
+    type: typing.Optional[VellumVariableType]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

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(

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,14 +3,17 @@
 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: float
-    max_tokens: int
+    temperature: typing.Optional[float]
+    max_tokens: typing.Optional[int]
     top_p: float
     frequency_penalty: float
     presence_penalty: 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

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+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
+    key: str
+    value: typing.Optional[int]
+
+    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_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