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

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_error_result.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
+from .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TemplatingNodeErrorResult(pydantic.BaseModel):
+    id: 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/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_result_output.py

@@ -7,6 +7,7 @@ import typing
 import typing_extensions
 
 from .templating_node_chat_history_result import TemplatingNodeChatHistoryResult
+from .templating_node_error_result import TemplatingNodeErrorResult
 from .templating_node_json_result import TemplatingNodeJsonResult
 from .templating_node_number_result import TemplatingNodeNumberResult
 from .templating_node_search_results_result import TemplatingNodeSearchResultsResult
@@ -58,10 +59,20 @@ class TemplatingNodeResultOutput_SearchResults(TemplatingNodeSearchResultsResult
         allow_population_by_field_name = True
 
 
+class TemplatingNodeResultOutput_Error(TemplatingNodeErrorResult):
+    type: typing_extensions.Literal["ERROR"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 TemplatingNodeResultOutput = typing.Union[
     TemplatingNodeResultOutput_String,
     TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_Json,
     TemplatingNodeResultOutput_ChatHistory,
     TemplatingNodeResultOutput_SearchResults,
+    TemplatingNodeResultOutput_Error,
 ]

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_error_result.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 TerminalNodeErrorResult(pydantic.BaseModel):
+    id: typing.Optional[str]
+    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
+    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/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_result_output.py

@@ -7,6 +7,7 @@ import typing
 import typing_extensions
 
 from .terminal_node_chat_history_result import TerminalNodeChatHistoryResult
+from .terminal_node_error_result import TerminalNodeErrorResult
 from .terminal_node_json_result import TerminalNodeJsonResult
 from .terminal_node_number_result import TerminalNodeNumberResult
 from .terminal_node_search_results_result import TerminalNodeSearchResultsResult
@@ -58,10 +59,20 @@ class TerminalNodeResultOutput_SearchResults(TerminalNodeSearchResultsResult):
         allow_population_by_field_name = True
 
 
+class TerminalNodeResultOutput_Error(TerminalNodeErrorResult):
+    type: typing_extensions.Literal["ERROR"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 TerminalNodeResultOutput = typing.Union[
     TerminalNodeResultOutput_String,
     TerminalNodeResultOutput_Number,
     TerminalNodeResultOutput_Json,
     TerminalNodeResultOutput_ChatHistory,
     TerminalNodeResultOutput_SearchResults,
+    TerminalNodeResultOutput_Error,
 ]

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_error.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
+from .vellum_error_code_enum import VellumErrorCodeEnum
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class VellumError(pydantic.BaseModel):
+    message: str
+    code: VellumErrorCodeEnum
+
+    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/vellum_error_code_enum.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class VellumErrorCodeEnum(str, enum.Enum):
+    """
+    * `INVALID_REQUEST` - INVALID_REQUEST
+    * `PROVIDER_ERROR` - PROVIDER_ERROR
+    * `INTERNAL_SERVER_ERROR` - INTERNAL_SERVER_ERROR
+    """
+
+    INVALID_REQUEST = "INVALID_REQUEST"
+    PROVIDER_ERROR = "PROVIDER_ERROR"
+    INTERNAL_SERVER_ERROR = "INTERNAL_SERVER_ERROR"
+
+    def visit(
+        self,
+        invalid_request: typing.Callable[[], T_Result],
+        provider_error: typing.Callable[[], T_Result],
+        internal_server_error: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is VellumErrorCodeEnum.INVALID_REQUEST:
+            return invalid_request()
+        if self is VellumErrorCodeEnum.PROVIDER_ERROR:
+            return provider_error()
+        if self is VellumErrorCodeEnum.INTERNAL_SERVER_ERROR:
+            return internal_server_error()

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

@@ -13,6 +13,7 @@ class VellumVariableType(str, enum.Enum):
     * `JSON` - JSON
     * `CHAT_HISTORY` - CHAT_HISTORY
     * `SEARCH_RESULTS` - SEARCH_RESULTS
+    * `ERROR` - ERROR
     """
 
     STRING = "STRING"
@@ -20,6 +21,7 @@ class VellumVariableType(str, enum.Enum):
     JSON = "JSON"
     CHAT_HISTORY = "CHAT_HISTORY"
     SEARCH_RESULTS = "SEARCH_RESULTS"
+    ERROR = "ERROR"
 
     def visit(
         self,
@@ -28,6 +30,7 @@ class VellumVariableType(str, enum.Enum):
         json: typing.Callable[[], T_Result],
         chat_history: typing.Callable[[], T_Result],
         search_results: typing.Callable[[], T_Result],
+        error: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is VellumVariableType.STRING:
             return string()
@@ -39,3 +42,5 @@ class VellumVariableType(str, enum.Enum):
             return chat_history()
         if self is VellumVariableType.SEARCH_RESULTS:
             return search_results()
+        if self is VellumVariableType.ERROR:
+            return error()

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