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

vellum/types/workflow_execution_actual_string_request.py

@@ -4,35 +4,31 @@ 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 WorkflowExecutionActualStringRequest(pydantic.BaseModel):
-    output_id: typing.Optional[str] = pydantic.Field(default=None)
+class WorkflowExecutionActualStringRequest(pydantic_v1.BaseModel):
+    output_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
     """
 
-    output_key: typing.Optional[str] = pydantic.Field(default=None)
+    output_key: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
     """
 
-    quality: typing.Optional[float] = pydantic.Field(default=None)
+    quality: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
     """
 
-    timestamp: typing.Optional[float] = pydantic.Field(default=None)
+    timestamp: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
     """
 
-    desired_output_value: typing.Optional[str] = pydantic.Field(default=None)
+    desired_output_value: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     Optionally provide the value that the output ideally should have been.
     """
@@ -48,5 +44,5 @@ class WorkflowExecutionActualStringRequest(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/workflow_execution_event_error_code.py

@@ -1,56 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class WorkflowExecutionEventErrorCode(str, enum.Enum):
-    """
-    - `WORKFLOW_INITIALIZATION` - WORKFLOW_INITIALIZATION
-    - `WORKFLOW_CANCELLED` - WORKFLOW_CANCELLED
-    - `NODE_EXECUTION_COUNT_LIMIT_REACHED` - NODE_EXECUTION_COUNT_LIMIT_REACHED
-    - `INTERNAL_SERVER_ERROR` - INTERNAL_SERVER_ERROR
-    - `NODE_EXECUTION` - NODE_EXECUTION
-    - `LLM_PROVIDER` - LLM_PROVIDER
-    - `INVALID_TEMPLATE` - INVALID_TEMPLATE
-    - `USER_DEFINED_ERROR` - USER_DEFINED_ERROR
-    """
-
-    WORKFLOW_INITIALIZATION = "WORKFLOW_INITIALIZATION"
-    WORKFLOW_CANCELLED = "WORKFLOW_CANCELLED"
-    NODE_EXECUTION_COUNT_LIMIT_REACHED = "NODE_EXECUTION_COUNT_LIMIT_REACHED"
-    INTERNAL_SERVER_ERROR = "INTERNAL_SERVER_ERROR"
-    NODE_EXECUTION = "NODE_EXECUTION"
-    LLM_PROVIDER = "LLM_PROVIDER"
-    INVALID_TEMPLATE = "INVALID_TEMPLATE"
-    USER_DEFINED_ERROR = "USER_DEFINED_ERROR"
-
-    def visit(
-        self,
-        workflow_initialization: typing.Callable[[], T_Result],
-        workflow_cancelled: typing.Callable[[], T_Result],
-        node_execution_count_limit_reached: typing.Callable[[], T_Result],
-        internal_server_error: typing.Callable[[], T_Result],
-        node_execution: typing.Callable[[], T_Result],
-        llm_provider: typing.Callable[[], T_Result],
-        invalid_template: typing.Callable[[], T_Result],
-        user_defined_error: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is WorkflowExecutionEventErrorCode.WORKFLOW_INITIALIZATION:
-            return workflow_initialization()
-        if self is WorkflowExecutionEventErrorCode.WORKFLOW_CANCELLED:
-            return workflow_cancelled()
-        if self is WorkflowExecutionEventErrorCode.NODE_EXECUTION_COUNT_LIMIT_REACHED:
-            return node_execution_count_limit_reached()
-        if self is WorkflowExecutionEventErrorCode.INTERNAL_SERVER_ERROR:
-            return internal_server_error()
-        if self is WorkflowExecutionEventErrorCode.NODE_EXECUTION:
-            return node_execution()
-        if self is WorkflowExecutionEventErrorCode.LLM_PROVIDER:
-            return llm_provider()
-        if self is WorkflowExecutionEventErrorCode.INVALID_TEMPLATE:
-            return invalid_template()
-        if self is WorkflowExecutionEventErrorCode.USER_DEFINED_ERROR:
-            return user_defined_error()
+WorkflowExecutionEventErrorCode = typing.Union[
+    typing.Literal[
+        "WORKFLOW_INITIALIZATION",
+        "WORKFLOW_CANCELLED",
+        "NODE_EXECUTION_COUNT_LIMIT_REACHED",
+        "INTERNAL_SERVER_ERROR",
+        "NODE_EXECUTION",
+        "LLM_PROVIDER",
+        "INVALID_TEMPLATE",
+        "USER_DEFINED_ERROR",
+    ],
+    typing.Any,
+]

vellum/types/workflow_execution_event_type.py

@@ -1,22 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class WorkflowExecutionEventType(str, enum.Enum):
-    """
-    - `NODE` - NODE
-    - `WORKFLOW` - WORKFLOW
-    """
-
-    NODE = "NODE"
-    WORKFLOW = "WORKFLOW"
-
-    def visit(self, node: typing.Callable[[], T_Result], workflow: typing.Callable[[], T_Result]) -> T_Result:
-        if self is WorkflowExecutionEventType.NODE:
-            return node()
-        if self is WorkflowExecutionEventType.WORKFLOW:
-            return workflow()
+WorkflowExecutionEventType = typing.Union[typing.Literal["NODE", "WORKFLOW"], typing.Any]

vellum/types/workflow_execution_node_result_event.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 .workflow_node_result_event import WorkflowNodeResultEvent
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowExecutionNodeResultEvent(pydantic.BaseModel):
+class WorkflowExecutionNodeResultEvent(pydantic_v1.BaseModel):
     """
     A NODE-level event emitted from the workflow's execution.
     """
@@ -33,5 +29,5 @@ class WorkflowExecutionNodeResultEvent(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/workflow_execution_workflow_result_event.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 .workflow_result_event import WorkflowResultEvent
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowExecutionWorkflowResultEvent(pydantic.BaseModel):
+class WorkflowExecutionWorkflowResultEvent(pydantic_v1.BaseModel):
     """
     A WORKFLOW-level event emitted from the workflow's execution.
     """
@@ -33,5 +29,5 @@ class WorkflowExecutionWorkflowResultEvent(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/workflow_node_result_event_state.py

@@ -1,36 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class WorkflowNodeResultEventState(str, enum.Enum):
-    """
-    - `INITIATED` - INITIATED
-    - `STREAMING` - STREAMING
-    - `FULFILLED` - FULFILLED
-    - `REJECTED` - REJECTED
-    """
-
-    INITIATED = "INITIATED"
-    STREAMING = "STREAMING"
-    FULFILLED = "FULFILLED"
-    REJECTED = "REJECTED"
-
-    def visit(
-        self,
-        initiated: typing.Callable[[], T_Result],
-        streaming: typing.Callable[[], T_Result],
-        fulfilled: typing.Callable[[], T_Result],
-        rejected: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is WorkflowNodeResultEventState.INITIATED:
-            return initiated()
-        if self is WorkflowNodeResultEventState.STREAMING:
-            return streaming()
-        if self is WorkflowNodeResultEventState.FULFILLED:
-            return fulfilled()
-        if self is WorkflowNodeResultEventState.REJECTED:
-            return rejected()
+WorkflowNodeResultEventState = typing.Union[
+    typing.Literal["INITIATED", "STREAMING", "FULFILLED", "REJECTED"], typing.Any
+]

vellum/types/workflow_output_array.py

@@ -4,21 +4,17 @@ 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 WorkflowOutputArray(pydantic.BaseModel):
+class WorkflowOutputArray(pydantic_v1.BaseModel):
     """
     An array output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputArray(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/workflow_output_chat_history.py

@@ -4,21 +4,17 @@ 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 WorkflowOutputChatHistory(pydantic.BaseModel):
+class WorkflowOutputChatHistory(pydantic_v1.BaseModel):
     """
     A chat history output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputChatHistory(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/workflow_output_error.py

@@ -4,21 +4,17 @@ 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 WorkflowOutputError(pydantic.BaseModel):
+class WorkflowOutputError(pydantic_v1.BaseModel):
     """
     An error output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputError(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/workflow_output_function_call.py

@@ -4,21 +4,17 @@ 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 WorkflowOutputFunctionCall(pydantic.BaseModel):
+class WorkflowOutputFunctionCall(pydantic_v1.BaseModel):
     """
     A function call output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputFunctionCall(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/workflow_output_image.py

@@ -4,21 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_image import VellumImage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputImage(pydantic.BaseModel):
+class WorkflowOutputImage(pydantic_v1.BaseModel):
     """
     An image output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputImage(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/workflow_output_json.py

@@ -4,20 +4,16 @@ 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 WorkflowOutputJson(pydantic.BaseModel):
+class WorkflowOutputJson(pydantic_v1.BaseModel):
     """
     A JSON output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -35,5 +31,5 @@ class WorkflowOutputJson(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/workflow_output_number.py

@@ -4,20 +4,16 @@ 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 WorkflowOutputNumber(pydantic.BaseModel):
+class WorkflowOutputNumber(pydantic_v1.BaseModel):
     """
     A number output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -35,5 +31,5 @@ class WorkflowOutputNumber(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/workflow_output_search_results.py

@@ -4,21 +4,17 @@ 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 WorkflowOutputSearchResults(pydantic.BaseModel):
+class WorkflowOutputSearchResults(pydantic_v1.BaseModel):
     """
     A search results output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputSearchResults(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/workflow_output_string.py

@@ -4,20 +4,16 @@ 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 WorkflowOutputString(pydantic.BaseModel):
+class WorkflowOutputString(pydantic_v1.BaseModel):
     """
     A string output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -35,5 +31,5 @@ class WorkflowOutputString(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/workflow_request_chat_history_input_request.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message_request import ChatMessageRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowRequestChatHistoryInputRequest(pydantic.BaseModel):
+class WorkflowRequestChatHistoryInputRequest(pydantic_v1.BaseModel):
     """
     The input for a chat history variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -35,5 +31,5 @@ class WorkflowRequestChatHistoryInputRequest(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/workflow_request_json_input_request.py

@@ -4,19 +4,15 @@ 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 WorkflowRequestJsonInputRequest(pydantic.BaseModel):
+class WorkflowRequestJsonInputRequest(pydantic_v1.BaseModel):
     """
     The input for a JSON variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -34,5 +30,5 @@ class WorkflowRequestJsonInputRequest(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/workflow_request_number_input_request.py

@@ -4,19 +4,15 @@ 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 WorkflowRequestNumberInputRequest(pydantic.BaseModel):
+class WorkflowRequestNumberInputRequest(pydantic_v1.BaseModel):
     """
     The input for a number variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -34,5 +30,5 @@ class WorkflowRequestNumberInputRequest(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/workflow_request_string_input_request.py

@@ -4,19 +4,15 @@ 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 WorkflowRequestStringInputRequest(pydantic.BaseModel):
+class WorkflowRequestStringInputRequest(pydantic_v1.BaseModel):
     """
     The input for a string variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -34,5 +30,5 @@ class WorkflowRequestStringInputRequest(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/workflow_result_event.py

@@ -4,19 +4,15 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .execution_vellum_value import ExecutionVellumValue
 from .workflow_event_error import WorkflowEventError
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 from .workflow_output import WorkflowOutput
 from .workflow_result_event_output_data import WorkflowResultEventOutputData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEvent(pydantic.BaseModel):
+class WorkflowResultEvent(pydantic_v1.BaseModel):
     id: str
     state: WorkflowNodeResultEventState
     ts: dt.datetime
@@ -36,5 +32,5 @@ class WorkflowResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}