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

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(

vellum/types/slim_document.py

@@ -3,14 +3,17 @@
 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_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SlimDocument(pydantic.BaseModel):
     id: str = pydantic.Field(description="Vellum-generated ID that uniquely identifies this document.")

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

vellum/types/submit_completion_actuals_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 SubmitCompletionActualsErrorResponse(pydantic.BaseModel):
     detail: str

vellum/types/templating_node_chat_history_result.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 TemplatingNodeChatHistoryResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_json_result.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 TemplatingNodeJsonResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_number_result.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 TemplatingNodeNumberResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_result.py

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

vellum/types/templating_node_result_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .templating_node_result_output import TemplatingNodeResultOutput
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeResultData(pydantic.BaseModel):
     output: TemplatingNodeResultOutput

vellum/types/templating_node_search_results_result.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 TemplatingNodeSearchResultsResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_string_result.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 TemplatingNodeStringResult(pydantic.BaseModel):
     id: str

vellum/types/terminal_node_chat_history_result.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 TerminalNodeChatHistoryResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_json_result.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 TerminalNodeJsonResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_number_result.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 TerminalNodeNumberResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_result.py

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

vellum/types/terminal_node_result_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .terminal_node_result_output import TerminalNodeResultOutput
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeResultData(pydantic.BaseModel):
     output: TerminalNodeResultOutput

vellum/types/terminal_node_search_results_result.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 TerminalNodeSearchResultsResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_string_result.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 TerminalNodeStringResult(pydantic.BaseModel):
     id: typing.Optional[str]

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

vellum/types/upload_document_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 UploadDocumentErrorResponse(pydantic.BaseModel):
     detail: str

vellum/types/upload_document_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 UploadDocumentResponse(pydantic.BaseModel):
     document_id: str = pydantic.Field(description="The ID of the newly created document.")

vellum/types/vellum_variable.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 VellumVariable(pydantic.BaseModel):
     id: str

vellum/types/workflow_event_error.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .workflow_execution_event_error_code import WorkflowExecutionEventErrorCode
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowEventError(pydantic.BaseModel):
     message: str

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

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

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

vellum/types/workflow_execution_node_result_event.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .workflow_node_result_event import WorkflowNodeResultEvent
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionNodeResultEvent(pydantic.BaseModel):
     execution_id: str

vellum/types/workflow_execution_workflow_result_event.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .workflow_result_event import WorkflowResultEvent
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionWorkflowResultEvent(pydantic.BaseModel):
     execution_id: str

vellum/types/workflow_node_result_event.py

@@ -3,14 +3,17 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .node_input_variable_compiled_value import NodeInputVariableCompiledValue
 from .workflow_event_error import WorkflowEventError
 from .workflow_node_result_data import WorkflowNodeResultData
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowNodeResultEvent(pydantic.BaseModel):
     id: str