vellum-ai 0.14.63__py3-none-any.whl → 1.11.2__py3-none-any.whl

vellum/client/types/workflow_result_event.py

@@ -1,21 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+from __future__ import annotations
+
 import datetime as dt
 import typing
-from .workflow_result_event_output_data import WorkflowResultEventOutputData
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .execution_vellum_value import ExecutionVellumValue
 from .workflow_event_error import WorkflowEventError
 from .workflow_output import WorkflowOutput
-from .execution_vellum_value import ExecutionVellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .workflow_result_event_output_data import WorkflowResultEventOutputData
+from .workflow_result_event_state import WorkflowResultEventState
 
 
 class WorkflowResultEvent(UniversalBaseModel):
     id: str
-    state: WorkflowNodeResultEventState
+    state: WorkflowResultEventState
     ts: dt.datetime
     output: typing.Optional[WorkflowResultEventOutputData] = None
     error: typing.Optional[WorkflowEventError] = None
@@ -30,3 +31,8 @@ class WorkflowResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowResultEvent)

vellum/client/types/workflow_result_event_output_data.py

@@ -1,14 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
-from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
-from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
-from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
-from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
+
 from .workflow_result_event_output_data_array import WorkflowResultEventOutputDataArray
-from .workflow_result_event_output_data_function_call import WorkflowResultEventOutputDataFunctionCall
+from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
 from .workflow_result_event_output_data_error import WorkflowResultEventOutputDataError
+from .workflow_result_event_output_data_function_call import WorkflowResultEventOutputDataFunctionCall
+from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
+from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
+from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
+from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
 
 WorkflowResultEventOutputData = typing.Union[
     WorkflowResultEventOutputDataString,

vellum/client/types/workflow_result_event_output_data_array.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
-from .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataArray(UniversalBaseModel):
@@ -24,7 +24,7 @@ class WorkflowResultEventOutputDataArray(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValue]] = None
+    value: typing.Optional[typing.List["VellumValue"]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -34,3 +34,9 @@ class WorkflowResultEventOutputDataArray(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowResultEventOutputDataArray)

vellum/client/types/workflow_result_event_output_data_chat_history.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .chat_message import ChatMessage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataChatHistory(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_error.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .vellum_error import VellumError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataError(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_function_call.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .function_call import FunctionCall
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataFunctionCall(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_json.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataJson(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_number.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataNumber(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_search_results.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .search_result import SearchResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataSearchResults(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_string.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataString(UniversalBaseModel):

vellum/client/types/workflow_result_event_state.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+WorkflowResultEventState = typing.Union[
+    typing.Literal["INITIATED", "STREAMING", "FULFILLED", "REJECTED", "PENDING"], typing.Any
+]

vellum/client/types/workflow_sandbox_display_data.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_display_icon import WorkflowDisplayIcon
+
+
+class WorkflowSandboxDisplayData(UniversalBaseModel):
+    """
+    Information used to display this Workflow Sandbox.
+    """
+
+    icon: typing.Optional[WorkflowDisplayIcon] = pydantic.Field(default=None)
+    """
+    The icon associated with this Workflow Sandbox.
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_sandbox_example.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class WorkflowSandboxExample(UniversalBaseModel):
@@ -11,6 +11,8 @@ class WorkflowSandboxExample(UniversalBaseModel):
     label: str
     description: typing.Optional[str] = None
     icon_name: typing.Optional[str] = None
+    ui_image_url: typing.Optional[str] = None
+    code_image_url: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/workflow_sandbox_parent_context.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class WorkflowSandboxParentContext(UniversalBaseModel):
@@ -27,12 +27,15 @@ class WorkflowSandboxParentContext(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .api_request_parent_context import ApiRequestParentContext  # noqa: E402
-from .node_parent_context import NodeParentContext  # noqa: E402
-from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402
-from .span_link import SpanLink  # noqa: E402
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402
-from .workflow_parent_context import WorkflowParentContext  # noqa: E402
-from .parent_context import ParentContext  # noqa: E402
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
 
 update_forward_refs(WorkflowSandboxParentContext)

vellum/client/types/workflow_stream_event.py

@@ -1,7 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
+
 from .workflow_execution_node_result_event import WorkflowExecutionNodeResultEvent
+from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
 
 WorkflowStreamEvent = typing.Union[WorkflowExecutionWorkflowResultEvent, WorkflowExecutionNodeResultEvent]

vellum/client/types/workspace_display_config.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class WorkspaceDisplayConfig(UniversalBaseModel):
+    color: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workspace_read.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
 import datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workspace_display_config import WorkspaceDisplayConfig
+
 
 class WorkspaceRead(UniversalBaseModel):
     id: str
@@ -16,6 +17,7 @@ class WorkspaceRead(UniversalBaseModel):
 
     label: str
     created: dt.datetime
+    display_config: typing.Optional[WorkspaceDisplayConfig] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/workspace_secret_read.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
-from .secret_type_enum import SecretTypeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .secret_type_enum import SecretTypeEnum
 
 
 class WorkspaceSecretRead(UniversalBaseModel):

vellum/client/utils.py

@@ -0,0 +1,24 @@
+# Custom client utilities
+# This file is not generated by Fern and can be freely edited
+
+from uuid import UUID
+from typing import Optional
+from pydantic import SerializationInfo
+
+
+def convert_input_variable_to_uuid(value: str, info: Optional[SerializationInfo] = None) -> str:
+    """Convert input_variable to UUID using executable_id from context."""
+    # Check if value is already a valid UUID
+    try:
+        UUID(value)
+        return value  # Already a UUID, return as-is
+    except ValueError:
+        pass  # Not a UUID, continue with conversion logic
+
+    if info and hasattr(info, "context") and info.context:
+        executable_id = info.context.get("executable_id")
+        if executable_id:
+            from vellum.workflows.utils.uuids import uuid4_from_hash  # avoid circular import
+
+            return str(uuid4_from_hash(f"{executable_id}|{value}"))
+    return value

vellum/core/force_multipart.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.core.force_multipart import *

vellum/core/http_response.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.core.http_response import *

vellum/errors/misdirected_request_error.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.errors.misdirected_request_error import *

vellum/errors/too_many_requests_error.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.errors.too_many_requests_error import *

vellum/errors/unauthorized_error.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.errors.unauthorized_error import *

vellum/evaluations/resources.py

@@ -8,11 +8,7 @@ from typing import Callable, Generator, Iterable, List, cast
 
 from vellum import TestSuiteRunMetricNumberOutput, TestSuiteRunRead
 from vellum.client import OMIT, Vellum
-from vellum.evaluations.constants import DEFAULT_MAX_POLLING_DURATION_MS, DEFAULT_POLLING_INTERVAL_MS
-from vellum.evaluations.exceptions import TestSuiteRunResultsException
-from vellum.evaluations.utils.env import get_api_key
-from vellum.evaluations.utils.paginator import PaginatedResults, get_all_results
-from vellum.types import (
+from vellum.client.types import (
     ExternalTestCaseExecutionRequest,
     NamedTestCaseVariableValueRequest,
     TestCaseVariableValue,
@@ -22,6 +18,10 @@ from vellum.types import (
     TestSuiteRunMetricOutput,
     TestSuiteRunState,
 )
+from vellum.evaluations.constants import DEFAULT_MAX_POLLING_DURATION_MS, DEFAULT_POLLING_INTERVAL_MS
+from vellum.evaluations.exceptions import TestSuiteRunResultsException
+from vellum.evaluations.utils.env import get_api_key
+from vellum.evaluations.utils.paginator import PaginatedResults, get_all_results
 from vellum.utils.typing import cast_not_optional
 from vellum.utils.uuid import is_valid_uuid
 

vellum/plugins/pydantic.py

@@ -28,9 +28,12 @@ def import_base_descriptor():
     """
     We have to avoid importing from vellum.* in this file because it will cause a circular import.
     """
-    from vellum.workflows.descriptors.base import BaseDescriptor
+    try:
+        from vellum.workflows.descriptors.base import BaseDescriptor
 
-    return BaseDescriptor
+        return BaseDescriptor
+    except Exception:
+        return None
 
 
 # https://docs.pydantic.dev/2.8/concepts/plugins/#build-a-plugin
@@ -45,6 +48,7 @@ class OnValidatePython(ValidatePythonHandlerProtocol):
         from_attributes: Optional[bool] = None,
         context: Optional[Dict[str, Any]] = None,
         self_instance: Optional[Any] = None,
+        extra: Optional[Literal["allow", "ignore", "forbid"]] = None,
         allow_partial: Union[bool, Literal["off", "on", "trailing-strings"]] = False,
         by_alias: Optional[bool] = None,
         by_name: Optional[bool] = None,
@@ -60,6 +64,10 @@ class OnValidatePython(ValidatePythonHandlerProtocol):
         self.tracked_descriptors = {}
         BaseDescriptor = import_base_descriptor()
 
+        # If BaseDescriptor import failed, skip descriptor processing
+        if BaseDescriptor is None:
+            return
+
         for key, value in input.items():
             field_info = model_fields.get(key)
             if isinstance(value, BaseDescriptor) and (
@@ -71,7 +79,10 @@ class OnValidatePython(ValidatePythonHandlerProtocol):
                 self.tracked_descriptors[key] = value
                 # TODO: This does not yet work for descriptors that map to more complex types
                 # https://app.shortcut.com/vellum/story/4636
-                input[key] = value.types[0]()
+                if len(value.types) == 0:
+                    input[key] = dict()
+                else:
+                    input[key] = value.types[0]()
 
     def on_success(self, result: Any) -> None:
         if self.tracked_descriptors:

vellum/plugins/tests/test_pydantic.py

@@ -0,0 +1,30 @@
+from typing import Any
+
+from pydantic import BaseModel
+
+from vellum.workflows.expressions.accessor import AccessorExpression
+from vellum.workflows.references.constant import ConstantValueReference
+from vellum.workflows.state.base import BaseState
+
+
+class TestState(BaseState):
+    pass
+
+
+class TestModel(BaseModel):
+    field: Any
+
+
+def test_pydantic_plugin__empty_types_descriptor():
+    """
+    Test that the pydantic plugin handles BaseDescriptor with empty types gracefully.
+    """
+
+    base_ref = ConstantValueReference({"name": "test"})
+    accessor = AccessorExpression(base=base_ref, field="name")
+
+    model = TestModel(field=accessor)
+
+    assert model is not None
+    assert hasattr(model, "field")
+    assert model.field == accessor