vellum-ai 0.14.49__tar.gz → 0.14.50__tar.gz

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.14.49
+Version: 0.14.50
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/ee/vellum_ee/workflows/display/nodes/base_node_display.py

@@ -44,18 +44,40 @@ if TYPE_CHECKING:
 _NodeDisplayAttrType = TypeVar("_NodeDisplayAttrType")
 
 
+def _get_node_input_ids_by_ref(node_class: Type[BaseNode], path: str, inst: Any):
+    if isinstance(inst, dict):
+        node_input_ids_by_name: Dict[str, UUID] = {}
+        for key, value in inst.items():
+            node_input_ids_by_name.update(_get_node_input_ids_by_ref(node_class, f"{path}.{key}", value))
+        return node_input_ids_by_name
+
+    return {path: uuid4_from_hash(f"{node_class.__id__}|{path}")}
+
+
 class BaseNodeDisplayMeta(type):
     def __new__(mcs, name: str, bases: Tuple[Type, ...], dct: Dict[str, Any]) -> Any:
         cls = cast(Type["BaseNodeDisplay"], super().__new__(mcs, name, bases, dct))
+        # This cast shouldn't be necessary, but it's a workaround for a mypy bug
+        node_class = cast(Type[BaseNode], cls.infer_node_class() if name != "BaseNodeDisplay" else BaseNode)
 
         if not dct.get("output_display"):
-            node_class = cls.infer_node_class()
             cls.output_display = {
                 ref: NodeOutputDisplay(id=node_class.__output_ids__[ref.name], name=ref.name)
                 for ref in node_class.Outputs
                 if ref.name in node_class.__output_ids__
             }
 
+        if not dct.get("node_input_ids_by_name"):
+            node_input_ids_by_name: Dict[str, UUID] = {}
+            for ref in node_class:
+                if ref not in cls.__serializable_inputs__:
+                    continue
+
+                node_input_ids_by_name.update(_get_node_input_ids_by_ref(node_class, ref.name, ref.instance))
+
+            if node_input_ids_by_name:
+                cls.node_input_ids_by_name = node_input_ids_by_name
+
         return cls.__annotate_node__()
 
     def __annotate_node__(cls):

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/ee/vellum_ee/workflows/display/nodes/get_node_display_class.py

@@ -1,9 +1,7 @@
 import types
-from uuid import UUID
-from typing import TYPE_CHECKING, Any, Dict, Generic, Type, TypeVar
+from typing import TYPE_CHECKING, Generic, Type, TypeVar
 
 from vellum.workflows.types.generics import NodeType
-from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.utils.registry import get_from_node_display_registry
 
 if TYPE_CHECKING:
@@ -22,30 +20,9 @@ def get_node_display_class(node_class: Type[NodeType]) -> Type["BaseNodeDisplay"
     # `base_node_display_class` is always a Generic class, so it's safe to index into it
     NodeDisplayBaseClass = base_node_display_class[_NodeClassType]  # type: ignore[index]
 
-    def _get_node_input_ids_by_ref(path: str, inst: Any):
-        if isinstance(inst, dict):
-            node_input_ids_by_name: Dict[str, UUID] = {}
-            for key, value in inst.items():
-                node_input_ids_by_name.update(_get_node_input_ids_by_ref(f"{path}.{key}", value))
-            return node_input_ids_by_name
-
-        return {path: uuid4_from_hash(f"{node_class.__id__}|{path}")}
-
-    def exec_body(ns: Dict):
-        node_input_ids_by_name: Dict[str, UUID] = {}
-        for ref in node_class:
-            if ref not in base_node_display_class.__serializable_inputs__:
-                continue
-
-            node_input_ids_by_name.update(_get_node_input_ids_by_ref(ref.name, ref.instance))
-
-        if node_input_ids_by_name:
-            ns["node_input_ids_by_name"] = node_input_ids_by_name
-
     NodeDisplayClass = types.new_class(
         f"{node_class.__name__}Display",
         bases=(NodeDisplayBaseClass, Generic[_NodeClassType]),
-        exec_body=exec_body,
     )
 
     return NodeDisplayClass

vellum_ai-0.14.50/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py

@@ -0,0 +1,177 @@
+from deepdiff import DeepDiff
+
+from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
+
+from tests.workflows.basic_tool_calling_node.workflow import BasicToolCallingNodeWorkflow
+
+
+def test_serialize_workflow():
+    # GIVEN a Workflow that uses a generic node
+    # WHEN we serialize it
+    workflow_display = get_workflow_display(workflow_class=BasicToolCallingNodeWorkflow)
+
+    serialized_workflow: dict = workflow_display.serialize()
+    # THEN we should get a serialized representation of the Workflow
+    assert serialized_workflow.keys() == {
+        "workflow_raw_data",
+        "input_variables",
+        "state_variables",
+        "output_variables",
+    }
+
+    # AND its input variables should be what we expect
+    input_variables = serialized_workflow["input_variables"]
+    assert len(input_variables) == 0
+
+    # AND its output variables should be what we expect
+    output_variables = serialized_workflow["output_variables"]
+    assert len(output_variables) == 2
+    assert not DeepDiff(
+        [
+            {"id": "8e7c0147-930d-4b7f-b6b1-6d79641cd3eb", "key": "text", "type": "STRING"},
+            {"id": "01a07e6d-7269-4f45-8b44-ef0227a2e88d", "key": "chat_history", "type": "CHAT_HISTORY"},
+        ],
+        output_variables,
+        ignore_order=True,
+    )
+
+    # AND its raw data should be what we expect
+    workflow_raw_data = serialized_workflow["workflow_raw_data"]
+    tool_calling_node = workflow_raw_data["nodes"][1]
+    assert tool_calling_node == {
+        "id": "21f29cac-da87-495f-bba1-093d423f4e46",
+        "label": "GetCurrentWeatherNode",
+        "type": "GENERIC",
+        "display_data": {
+            "position": {"x": 0.0, "y": 0.0},
+            "comment": {"value": "\n    A tool calling node that calls the get_current_weather function.\n    "},
+        },
+        "base": {
+            "name": "ToolCallingNode",
+            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+        },
+        "definition": {
+            "name": "GetCurrentWeatherNode",
+            "module": ["tests", "workflows", "basic_tool_calling_node", "workflow"],
+        },
+        "trigger": {"id": "2414743b-b1dd-4552-8abf-9b7481df9762", "merge_behavior": "AWAIT_ATTRIBUTES"},
+        "ports": [{"id": "3cd6d78c-9dad-42aa-ad38-31f67057c379", "name": "default", "type": "DEFAULT"}],
+        "adornments": None,
+        "attributes": [
+            {
+                "id": "44420e39-966f-4c59-bdf8-6365a61c5d2a",
+                "name": "ml_model",
+                "value": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "gpt-4o-mini"}},
+            },
+            {
+                "id": "669cfb4b-8c25-460e-8952-b63d91302cbc",
+                "name": "blocks",
+                "value": {
+                    "type": "CONSTANT_VALUE",
+                    "value": {
+                        "type": "JSON",
+                        "value": [
+                            {
+                                "block_type": "CHAT_MESSAGE",
+                                "state": None,
+                                "cache_config": None,
+                                "chat_role": "SYSTEM",
+                                "chat_source": None,
+                                "chat_message_unterminated": None,
+                                "blocks": [
+                                    {
+                                        "block_type": "RICH_TEXT",
+                                        "state": None,
+                                        "cache_config": None,
+                                        "blocks": [
+                                            {
+                                                "block_type": "PLAIN_TEXT",
+                                                "state": None,
+                                                "cache_config": None,
+                                                "text": "You are a weather expert",
+                                            }
+                                        ],
+                                    }
+                                ],
+                            },
+                            {
+                                "block_type": "CHAT_MESSAGE",
+                                "state": None,
+                                "cache_config": None,
+                                "chat_role": "USER",
+                                "chat_source": None,
+                                "chat_message_unterminated": None,
+                                "blocks": [
+                                    {
+                                        "block_type": "RICH_TEXT",
+                                        "state": None,
+                                        "cache_config": None,
+                                        "blocks": [
+                                            {
+                                                "block_type": "VARIABLE",
+                                                "state": None,
+                                                "cache_config": None,
+                                                "input_variable": "question",
+                                            }
+                                        ],
+                                    }
+                                ],
+                            },
+                        ],
+                    },
+                },
+            },
+            {
+                "id": "78324739-ff89-47a5-902b-10da0cb95c6d",
+                "name": "functions",
+                "value": {
+                    "type": "CONSTANT_VALUE",
+                    "value": {
+                        "type": "JSON",
+                        "value": [
+                            {
+                                "state": None,
+                                "cache_config": None,
+                                "name": "get_current_weather",
+                                "description": None,
+                                "parameters": {
+                                    "type": "object",
+                                    "properties": {"location": {"type": "string"}, "unit": {"type": "string"}},
+                                    "required": ["location", "unit"],
+                                },
+                                "forced": None,
+                                "strict": None,
+                            }
+                        ],
+                    },
+                },
+            },
+            {
+                "id": "0f6dc102-3460-4963-91fa-7ba85d65ef7a",
+                "name": "prompt_inputs",
+                "value": {
+                    "type": "CONSTANT_VALUE",
+                    "value": {"type": "JSON", "value": {"question": "What's the weather like in San Francisco?"}},
+                },
+            },
+            {
+                "id": "5c041b7d-732c-4773-a93a-32211f2af0b3",
+                "name": "max_tool_calls",
+                "value": {"type": "CONSTANT_VALUE", "value": {"type": "NUMBER", "value": 1.0}},
+            },
+        ],
+        "outputs": [
+            {
+                "id": "e62bc785-a914-4066-b79e-8c89a5d0ec6c",
+                "name": "text",
+                "type": "STRING",
+                "value": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": ""}},
+            },
+            {
+                "id": "4674f1d9-e3af-411f-8a55-40a3a3ab5394",
+                "name": "chat_history",
+                "type": "CHAT_HISTORY",
+                "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": []}},
+            },
+        ],
+    }

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/ee/vellum_ee/workflows/display/utils/expressions.py

@@ -253,7 +253,7 @@ def serialize_value(display_context: "WorkflowDisplayContext", value: Any) -> Js
                 "type": "CONSTANT_VALUE",
                 "value": {
                     "type": "JSON",
-                    "items": constant_values,
+                    "value": constant_values,
                 },
             }
         else:

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/ee/vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -153,20 +153,8 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
             "definition": None,
         }
 
-        # Add all the nodes in the workflow
-        for node in self._workflow.get_nodes():
-            node_display = self.display_context.node_displays[node]
-
-            try:
-                serialized_node = node_display.serialize(self.display_context)
-            except NotImplementedError as e:
-                self.add_error(e)
-                continue
-
-            serialized_nodes[node_display.node_id] = serialized_node
-
-        # Add all unused nodes in the workflow
-        for node in self._workflow.get_unused_nodes():
+        # Add all the nodes in the workflows
+        for node in self._workflow.get_all_nodes():
             node_display = self.display_context.node_displays[node]
 
             try:
@@ -417,16 +405,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
 
         port_displays: PortDisplays = {}
 
-        for node in self._workflow.get_nodes():
-            self._enrich_node_displays(
-                node=node,
-                node_displays=node_displays,
-                global_node_displays=global_node_displays,
-                global_node_output_displays=global_node_output_displays,
-                port_displays=port_displays,
-            )
-
-        for node in self._workflow.get_unused_nodes():
+        for node in self._workflow.get_all_nodes():
             self._enrich_node_displays(
                 node=node,
                 node_displays=node_displays,

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/ee/vellum_ee/workflows/display/workflows/tests/test_workflow_display.py

@@ -466,7 +466,7 @@ def test_serialize_workflow__array_values():
     assert "value" in array_output
     assert array_output["value"] == {
         "type": "CONSTANT_VALUE",
-        "value": {"type": "JSON", "items": ["item1", "item2", "item3"]},
+        "value": {"type": "JSON", "value": ["item1", "item2", "item3"]},
     }
 
     nested_array_outputs = [val for val in outputs if isinstance(val, dict) and val["name"] == "nested_array_value"]
@@ -477,7 +477,7 @@ def test_serialize_workflow__array_values():
     assert "value" in nested_array_output
     assert nested_array_output["value"] == {
         "type": "CONSTANT_VALUE",
-        "value": {"type": "JSON", "items": [["item1", "item2", "item3"], ["item4", "item5", "item6"]]},
+        "value": {"type": "JSON", "value": [["item1", "item2", "item3"], ["item4", "item5", "item6"]]},
     }
 
     mixed_array_outputs = [val for val in outputs if isinstance(val, dict) and val["name"] == "mixed_array_value"]
@@ -488,7 +488,7 @@ def test_serialize_workflow__array_values():
     assert "value" in mixed_array_output
     assert mixed_array_output["value"] == {
         "type": "CONSTANT_VALUE",
-        "value": {"type": "JSON", "items": [["item1"], "item2", "item3"]},
+        "value": {"type": "JSON", "value": [["item1"], "item2", "item3"]},
     }
 
 

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "0.14.49"
+version = "0.14.50"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/src/vellum/__init__.py

@@ -544,7 +544,9 @@ from .types import (
     WorkflowExecutionSpanAttributes,
     WorkflowExecutionStreamingBody,
     WorkflowExecutionStreamingEvent,
-    WorkflowExecutionUsageCalculationFulfilledBody,
+    WorkflowExecutionUsageCalculationError,
+    WorkflowExecutionUsageCalculationErrorCodeEnum,
+    WorkflowExecutionUsageResult,
     WorkflowExecutionViewOnlineEvalMetricResult,
     WorkflowExecutionWorkflowResultEvent,
     WorkflowExpandMetaRequest,
@@ -1177,7 +1179,9 @@ __all__ = [
     "WorkflowExecutionSpanAttributes",
     "WorkflowExecutionStreamingBody",
     "WorkflowExecutionStreamingEvent",
-    "WorkflowExecutionUsageCalculationFulfilledBody",
+    "WorkflowExecutionUsageCalculationError",
+    "WorkflowExecutionUsageCalculationErrorCodeEnum",
+    "WorkflowExecutionUsageResult",
     "WorkflowExecutionViewOnlineEvalMetricResult",
     "WorkflowExecutionWorkflowResultEvent",
     "WorkflowExpandMetaRequest",

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/src/vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.14.49",
+            "X-Fern-SDK-Version": "0.14.50",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/src/vellum/client/types/__init__.py

@@ -568,7 +568,9 @@ from .workflow_execution_span import WorkflowExecutionSpan
 from .workflow_execution_span_attributes import WorkflowExecutionSpanAttributes
 from .workflow_execution_streaming_body import WorkflowExecutionStreamingBody
 from .workflow_execution_streaming_event import WorkflowExecutionStreamingEvent
-from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
+from .workflow_execution_usage_calculation_error import WorkflowExecutionUsageCalculationError
+from .workflow_execution_usage_calculation_error_code_enum import WorkflowExecutionUsageCalculationErrorCodeEnum
+from .workflow_execution_usage_result import WorkflowExecutionUsageResult
 from .workflow_execution_view_online_eval_metric_result import WorkflowExecutionViewOnlineEvalMetricResult
 from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
 from .workflow_expand_meta_request import WorkflowExpandMetaRequest
@@ -1154,7 +1156,9 @@ __all__ = [
     "WorkflowExecutionSpanAttributes",
     "WorkflowExecutionStreamingBody",
     "WorkflowExecutionStreamingEvent",
-    "WorkflowExecutionUsageCalculationFulfilledBody",
+    "WorkflowExecutionUsageCalculationError",
+    "WorkflowExecutionUsageCalculationErrorCodeEnum",
+    "WorkflowExecutionUsageResult",
     "WorkflowExecutionViewOnlineEvalMetricResult",
     "WorkflowExecutionWorkflowResultEvent",
     "WorkflowExpandMetaRequest",

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/src/vellum/client/types/deployment_read.py

@@ -50,7 +50,7 @@ class DeploymentRead(UniversalBaseModel):
 
     active_model_version_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
-    Deprecated. This now always returns a null value.
+    Deprecated. This now always returns an empty array.
     """
 
     last_deployed_history_item_id: str = pydantic.Field()

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/src/vellum/client/types/slim_workflow_execution_read.py

@@ -15,7 +15,7 @@ from .execution_vellum_value import ExecutionVellumValue
 from .workflow_error import WorkflowError
 from .workflow_execution_actual import WorkflowExecutionActual
 from .workflow_execution_view_online_eval_metric_result import WorkflowExecutionViewOnlineEvalMetricResult
-from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
+from .workflow_execution_usage_result import WorkflowExecutionUsageResult
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -30,7 +30,7 @@ class SlimWorkflowExecutionRead(UniversalBaseModel):
     error: typing.Optional[WorkflowError] = None
     latest_actual: typing.Optional[WorkflowExecutionActual] = None
     metric_results: typing.List[WorkflowExecutionViewOnlineEvalMetricResult]
-    usage_results: typing.List[WorkflowExecutionUsageCalculationFulfilledBody]
+    usage_results: typing.Optional[typing.List[WorkflowExecutionUsageResult]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

{vellum_ai-0.14.49 → vellum_ai-0.14.50}/src/vellum/client/types/workflow_event_execution_read.py

@@ -15,7 +15,7 @@ from .execution_vellum_value import ExecutionVellumValue
 from .workflow_error import WorkflowError
 from .workflow_execution_actual import WorkflowExecutionActual
 from .workflow_execution_view_online_eval_metric_result import WorkflowExecutionViewOnlineEvalMetricResult
-from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
+from .workflow_execution_usage_result import WorkflowExecutionUsageResult
 from .vellum_span import VellumSpan
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
@@ -31,7 +31,7 @@ class WorkflowEventExecutionRead(UniversalBaseModel):
     error: typing.Optional[WorkflowError] = None
     latest_actual: typing.Optional[WorkflowExecutionActual] = None
     metric_results: typing.List[WorkflowExecutionViewOnlineEvalMetricResult]
-    usage_results: typing.List[WorkflowExecutionUsageCalculationFulfilledBody]
+    usage_results: typing.Optional[typing.List[WorkflowExecutionUsageResult]] = None
     spans: typing.List[VellumSpan]
 
     if IS_PYDANTIC_V2:

vellum_ai-0.14.50/src/vellum/client/types/workflow_execution_usage_calculation_error.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .workflow_execution_usage_calculation_error_code_enum import WorkflowExecutionUsageCalculationErrorCodeEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class WorkflowExecutionUsageCalculationError(UniversalBaseModel):
+    code: WorkflowExecutionUsageCalculationErrorCodeEnum
+    message: str
+
+    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_ai-0.14.50/src/vellum/client/types/workflow_execution_usage_calculation_error_code_enum.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+WorkflowExecutionUsageCalculationErrorCodeEnum = typing.Union[
+    typing.Literal["UNKNOWN", "DEPENDENCIES_FAILED", "NO_USAGE_CALCULATED", "INTERNAL_SERVER_ERROR"], typing.Any
+]

vellum_ai-0.14.49/src/vellum/client/types/workflow_execution_usage_calculation_fulfilled_body.py → vellum_ai-0.14.50/src/vellum/client/types/workflow_execution_usage_result.py

@@ -4,13 +4,15 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import typing
 from .ml_model_usage_wrapper import MlModelUsageWrapper
 from .price import Price
+from .workflow_execution_usage_calculation_error import WorkflowExecutionUsageCalculationError
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class WorkflowExecutionUsageCalculationFulfilledBody(UniversalBaseModel):
-    usage: typing.List[MlModelUsageWrapper]
-    cost: typing.List[Price]
+class WorkflowExecutionUsageResult(UniversalBaseModel):
+    usage: typing.Optional[typing.List[MlModelUsageWrapper]] = None
+    cost: typing.Optional[typing.List[Price]] = None
+    error: typing.Optional[WorkflowExecutionUsageCalculationError] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum_ai-0.14.50/src/vellum/types/workflow_execution_usage_calculation_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.types.workflow_execution_usage_calculation_error import *

vellum_ai-0.14.50/src/vellum/types/workflow_execution_usage_calculation_error_code_enum.py

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

vellum_ai-0.14.50/src/vellum/types/workflow_execution_usage_result.py

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