vellum-ai 0.3.23__py3-none-any.whl → 0.5.0__py3-none-any.whl

vellum/types/node_input_compiled_chat_history_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledChatHistoryValue(pydantic.BaseModel):
+class NodeInputCompiledChatHistoryValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.List[ChatMessage]] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledChatHistoryValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_error_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledErrorValue(pydantic.BaseModel):
+class NodeInputCompiledErrorValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[VellumError] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledErrorValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_function_call.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledFunctionCall(pydantic.BaseModel):
+class NodeInputCompiledFunctionCall(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[FunctionCall] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledFunctionCall(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_json_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledJsonValue(pydantic.BaseModel):
+class NodeInputCompiledJsonValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
@@ -27,5 +23,5 @@ class NodeInputCompiledJsonValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_number_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledNumberValue(pydantic.BaseModel):
+class NodeInputCompiledNumberValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[float] = None
@@ -27,5 +23,5 @@ class NodeInputCompiledNumberValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_search_results_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledSearchResultsValue(pydantic.BaseModel):
+class NodeInputCompiledSearchResultsValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.List[SearchResult]] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledSearchResultsValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_string_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledStringValue(pydantic.BaseModel):
+class NodeInputCompiledStringValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[str] = None
@@ -27,5 +23,5 @@ class NodeInputCompiledStringValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_array_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledArrayValue(pydantic.BaseModel):
+class NodeOutputCompiledArrayValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledArrayValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_chat_history_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledChatHistoryValue(pydantic.BaseModel):
+class NodeOutputCompiledChatHistoryValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.List[ChatMessage]] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledChatHistoryValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_error_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledErrorValue(pydantic.BaseModel):
+class NodeOutputCompiledErrorValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[VellumError] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledErrorValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_function_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledFunctionValue(pydantic.BaseModel):
+class NodeOutputCompiledFunctionValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[FunctionCall] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledFunctionValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_json_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledJsonValue(pydantic.BaseModel):
+class NodeOutputCompiledJsonValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
@@ -26,5 +22,5 @@ class NodeOutputCompiledJsonValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_number_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledNumberValue(pydantic.BaseModel):
+class NodeOutputCompiledNumberValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[float] = None
 
@@ -26,5 +22,5 @@ class NodeOutputCompiledNumberValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_search_results_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledSearchResultsValue(pydantic.BaseModel):
+class NodeOutputCompiledSearchResultsValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.List[SearchResult]] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledSearchResultsValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_string_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledStringValue(pydantic.BaseModel):
+class NodeOutputCompiledStringValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[str] = None
 
@@ -26,5 +22,5 @@ class NodeOutputCompiledStringValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/normalized_log_probs.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 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):
+class NormalizedLogProbs(pydantic_v1.BaseModel):
     tokens: typing.List[NormalizedTokenLogProbs]
     likelihood: typing.Optional[float] = None
 
@@ -27,5 +23,5 @@ class NormalizedLogProbs(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/normalized_token_log_probs.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NormalizedTokenLogProbs(pydantic.BaseModel):
+class NormalizedTokenLogProbs(pydantic_v1.BaseModel):
     token: str
     logprob: typing.Optional[float] = None
     top_logprobs: typing.Optional[typing.Dict[str, typing.Optional[float]]] = None
@@ -28,5 +24,5 @@ class NormalizedTokenLogProbs(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/number_variable_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NumberVariableValue(pydantic.BaseModel):
+class NumberVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[float] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class NumberVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_document_index_read_list.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .document_index_read import DocumentIndexRead
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedDocumentIndexReadList(pydantic.BaseModel):
+class PaginatedDocumentIndexReadList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedDocumentIndexReadList(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_slim_deployment_read_list.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .slim_deployment_read import SlimDeploymentRead
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedSlimDeploymentReadList(pydantic.BaseModel):
+class PaginatedSlimDeploymentReadList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedSlimDeploymentReadList(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_slim_document_list.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .slim_document import SlimDocument
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedSlimDocumentList(pydantic.BaseModel):
+class PaginatedSlimDocumentList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedSlimDocumentList(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_slim_workflow_deployment_list.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .slim_workflow_deployment import SlimWorkflowDeployment
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedSlimWorkflowDeploymentList(pydantic.BaseModel):
+class PaginatedSlimWorkflowDeploymentList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedSlimWorkflowDeploymentList(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_test_suite_run_execution_list.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_suite_run_execution import TestSuiteRunExecution
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedTestSuiteRunExecutionList(pydantic.BaseModel):
+class PaginatedTestSuiteRunExecutionList(pydantic_v1.BaseModel):
     count: int
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedTestSuiteRunExecutionList(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_test_suite_test_case_list.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_suite_test_case import TestSuiteTestCase
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedTestSuiteTestCaseList(pydantic.BaseModel):
+class PaginatedTestSuiteTestCaseList(pydantic_v1.BaseModel):
     count: int
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedTestSuiteTestCaseList(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/processing_failure_reason_enum.py

@@ -1,24 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class ProcessingFailureReasonEnum(str, enum.Enum):
-    """
-    - `EXCEEDED_CHARACTER_LIMIT` - Exceeded Character Limit
-    - `INVALID_FILE` - Invalid File
-    """
-
-    EXCEEDED_CHARACTER_LIMIT = "EXCEEDED_CHARACTER_LIMIT"
-    INVALID_FILE = "INVALID_FILE"
-
-    def visit(
-        self, exceeded_character_limit: typing.Callable[[], T_Result], invalid_file: typing.Callable[[], T_Result]
-    ) -> T_Result:
-        if self is ProcessingFailureReasonEnum.EXCEEDED_CHARACTER_LIMIT:
-            return exceeded_character_limit()
-        if self is ProcessingFailureReasonEnum.INVALID_FILE:
-            return invalid_file()
+ProcessingFailureReasonEnum = typing.Union[typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE"], typing.Any]

vellum/types/processing_state_enum.py

@@ -1,36 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class ProcessingStateEnum(str, enum.Enum):
-    """
-    - `QUEUED` - Queued
-    - `PROCESSING` - Processing
-    - `PROCESSED` - Processed
-    - `FAILED` - Failed
-    """
-
-    QUEUED = "QUEUED"
-    PROCESSING = "PROCESSING"
-    PROCESSED = "PROCESSED"
-    FAILED = "FAILED"
-
-    def visit(
-        self,
-        queued: typing.Callable[[], T_Result],
-        processing: typing.Callable[[], T_Result],
-        processed: typing.Callable[[], T_Result],
-        failed: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is ProcessingStateEnum.QUEUED:
-            return queued()
-        if self is ProcessingStateEnum.PROCESSING:
-            return processing()
-        if self is ProcessingStateEnum.PROCESSED:
-            return processed()
-        if self is ProcessingStateEnum.FAILED:
-            return failed()
+ProcessingStateEnum = typing.Union[typing.Literal["QUEUED", "PROCESSING", "PROCESSED", "FAILED"], typing.Any]