vellum-ai 0.14.80__py3-none-any.whl → 0.14.82__py3-none-any.whl

vellum/__init__.py

@@ -121,6 +121,7 @@ from .types import (
     ExecutionNumberVellumValue,
     ExecutionSearchResultsVellumValue,
     ExecutionStringVellumValue,
+    ExecutionThinkingVellumValue,
     ExecutionVellumValue,
     ExternalInputDescriptor,
     ExternalTestCaseExecution,
@@ -487,6 +488,8 @@ from .types import (
     TestSuiteTestCaseReplacedBulkResult,
     TestSuiteTestCaseReplacedBulkResultData,
     TestSuiteTestCaseUpsertBulkOperationRequest,
+    ThinkingVellumValue,
+    ThinkingVellumValueRequest,
     TokenOverlappingWindowChunkerConfig,
     TokenOverlappingWindowChunkerConfigRequest,
     TokenOverlappingWindowChunking,
@@ -754,6 +757,7 @@ __all__ = [
     "ExecutionNumberVellumValue",
     "ExecutionSearchResultsVellumValue",
     "ExecutionStringVellumValue",
+    "ExecutionThinkingVellumValue",
     "ExecutionVellumValue",
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
@@ -1127,6 +1131,8 @@ __all__ = [
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
     "TestSuiteTestCaseUpsertBulkOperationRequest",
+    "ThinkingVellumValue",
+    "ThinkingVellumValueRequest",
     "TokenOverlappingWindowChunkerConfig",
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",

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.80",
+            "X-Fern-SDK-Version": "0.14.82",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

vellum/client/types/__init__.py

@@ -125,6 +125,7 @@ from .execution_json_vellum_value import ExecutionJsonVellumValue
 from .execution_number_vellum_value import ExecutionNumberVellumValue
 from .execution_search_results_vellum_value import ExecutionSearchResultsVellumValue
 from .execution_string_vellum_value import ExecutionStringVellumValue
+from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
 from .execution_vellum_value import ExecutionVellumValue
 from .external_input_descriptor import ExternalInputDescriptor
 from .external_test_case_execution import ExternalTestCaseExecution
@@ -511,6 +512,8 @@ from .test_suite_test_case_replace_bulk_operation_request import TestSuiteTestCa
 from .test_suite_test_case_replaced_bulk_result import TestSuiteTestCaseReplacedBulkResult
 from .test_suite_test_case_replaced_bulk_result_data import TestSuiteTestCaseReplacedBulkResultData
 from .test_suite_test_case_upsert_bulk_operation_request import TestSuiteTestCaseUpsertBulkOperationRequest
+from .thinking_vellum_value import ThinkingVellumValue
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
 from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunkerConfig
 from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
@@ -741,6 +744,7 @@ __all__ = [
     "ExecutionNumberVellumValue",
     "ExecutionSearchResultsVellumValue",
     "ExecutionStringVellumValue",
+    "ExecutionThinkingVellumValue",
     "ExecutionVellumValue",
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
@@ -1107,6 +1111,8 @@ __all__ = [
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
     "TestSuiteTestCaseUpsertBulkOperationRequest",
+    "ThinkingVellumValue",
+    "ThinkingVellumValueRequest",
     "TokenOverlappingWindowChunkerConfig",
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",

vellum/client/types/execution_thinking_vellum_value.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+import typing
+from .string_vellum_value import StringVellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class ExecutionThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    id: str = pydantic.Field()
+    """
+    The variable's uniquely identifying internal id.
+    """
+
+    name: str
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValue]
+
+    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/execution_vellum_value.py

@@ -9,6 +9,7 @@ from .execution_search_results_vellum_value import ExecutionSearchResultsVellumV
 from .execution_error_vellum_value import ExecutionErrorVellumValue
 from .execution_array_vellum_value import ExecutionArrayVellumValue
 from .execution_function_call_vellum_value import ExecutionFunctionCallVellumValue
+from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
 
 ExecutionVellumValue = typing.Union[
     ExecutionStringVellumValue,
@@ -19,4 +20,5 @@ ExecutionVellumValue = typing.Union[
     ExecutionErrorVellumValue,
     ExecutionArrayVellumValue,
     ExecutionFunctionCallVellumValue,
+    ExecutionThinkingVellumValue,
 ]

vellum/client/types/prompt_output.py

@@ -5,5 +5,8 @@ from .string_vellum_value import StringVellumValue
 from .json_vellum_value import JsonVellumValue
 from .error_vellum_value import ErrorVellumValue
 from .function_call_vellum_value import FunctionCallVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 
-PromptOutput = typing.Union[StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue]
+PromptOutput = typing.Union[
+    StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue, ThinkingVellumValue
+]

vellum/client/types/thinking_vellum_value.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .string_vellum_value import StringVellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValue]
+
+    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/thinking_vellum_value_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .string_vellum_value_request import StringVellumValueRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ThinkingVellumValueRequest(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValueRequest]
+
+    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/vellum_value.py

@@ -12,6 +12,7 @@ from .function_call_vellum_value import FunctionCallVellumValue
 from .error_vellum_value import ErrorVellumValue
 from .chat_history_vellum_value import ChatHistoryVellumValue
 from .search_results_vellum_value import SearchResultsVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 import typing
 
 if typing.TYPE_CHECKING:
@@ -28,4 +29,5 @@ VellumValue = typing.Union[
     "ArrayVellumValue",
     ChatHistoryVellumValue,
     SearchResultsVellumValue,
+    ThinkingVellumValue,
 ]

vellum/client/types/vellum_value_request.py

@@ -12,6 +12,7 @@ from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .error_vellum_value_request import ErrorVellumValueRequest
 from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .search_results_vellum_value_request import SearchResultsVellumValueRequest
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
 import typing
 
 if typing.TYPE_CHECKING:
@@ -28,4 +29,5 @@ VellumValueRequest = typing.Union[
     "ArrayVellumValueRequest",
     ChatHistoryVellumValueRequest,
     SearchResultsVellumValueRequest,
+    ThinkingVellumValueRequest,
 ]

vellum/client/types/vellum_variable_type.py

@@ -16,6 +16,7 @@ VellumVariableType = typing.Union[
         "AUDIO",
         "DOCUMENT",
         "NULL",
+        "THINKING",
     ],
     typing.Any,
 ]

vellum/types/execution_thinking_vellum_value.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.execution_thinking_vellum_value import *

vellum/types/thinking_vellum_value.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.thinking_vellum_value import *

vellum/types/thinking_vellum_value_request.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.thinking_vellum_value_request import *

vellum/workflows/nodes/displayable/__init__.py

@@ -13,6 +13,7 @@ from .note_node import NoteNode
 from .prompt_deployment_node import PromptDeploymentNode
 from .search_node import SearchNode
 from .subworkflow_deployment_node import SubworkflowDeploymentNode
+from .tool_calling_node import ToolCallingNode
 
 __all__ = [
     "APINode",
@@ -29,5 +30,6 @@ __all__ = [
     "PromptDeploymentNode",
     "SearchNode",
     "TemplatingNode",
+    "ToolCallingNode",
     "FinalOutputNode",
 ]

vellum/workflows/nodes/displayable/code_execution_node/tests/test_node.py

@@ -109,6 +109,7 @@ def main(word: str) -> int:
             CodeExecutionPackage(
                 name="openai",
                 version="1.0.0",
+                repository="test-repo",
             )
         ]
 
@@ -150,6 +151,7 @@ def main(word: str) -> int:
             CodeExecutionPackage(
                 name="openai",
                 version="1.0.0",
+                repository="test-repo",
             )
         ],
         request_options=None,

vellum/workflows/nodes/displayable/inline_prompt_node/node.py

@@ -63,6 +63,9 @@ class InlinePromptNode(BaseInlinePromptNode[StateType]):
                 json_output = output.value
             elif output.type == "FUNCTION_CALL":
                 string_outputs.append(output.value.model_dump_json(indent=4))
+            elif output.type == "THINKING":
+                thinking_strings = [item.value for item in output.value if item.value is not None]
+                string_outputs.append("\n".join(thinking_strings))
             else:
                 string_outputs.append(output.value.message)
 

vellum/workflows/nodes/displayable/prompt_deployment_node/node.py

@@ -65,6 +65,9 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
                 string_outputs.append(json.dumps(output.value, indent=4))
             elif output.type == "FUNCTION_CALL":
                 string_outputs.append(output.value.model_dump_json(indent=4))
+            elif output.type == "THINKING":
+                thinking_strings = [item.value for item in output.value if item.value is not None]
+                string_outputs.append("\n".join(thinking_strings))
             else:
                 string_outputs.append(output.value.message)
 

vellum/workflows/nodes/displayable/tool_calling_node/__init__.py

@@ -0,0 +1,3 @@
+from vellum.workflows.nodes.displayable.tool_calling_node.node import ToolCallingNode
+
+__all__ = ["ToolCallingNode"]

vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/node.py

@@ -1,16 +1,13 @@
-from collections.abc import Sequence
-from typing import Any, ClassVar, Dict, List, Optional, cast
+from typing import ClassVar, List, Optional
 
 from vellum import ChatMessage, PromptBlock
-from vellum.client.types.code_execution_package import CodeExecutionPackage
-from vellum.client.types.code_execution_runtime import CodeExecutionRuntime
 from vellum.workflows.context import execution_context, get_parent_context
 from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.graph.graph import Graph
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import (
+from vellum.workflows.nodes.displayable.tool_calling_node.utils import (
     create_function_node,
     create_tool_router_node,
     get_function_name,
@@ -19,7 +16,6 @@ from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.types.core import EntityInputsInterface, Tool
-from vellum.workflows.workflows.base import BaseWorkflow
 
 
 class ToolCallingNode(BaseNode):
@@ -31,7 +27,6 @@ class ToolCallingNode(BaseNode):
         blocks: List[PromptBlock] - The prompt blocks to use (same format as InlinePromptNode)
         functions: List[Tool] - The functions that can be called
         prompt_inputs: Optional[EntityInputsInterface] - Mapping of input variable names to values
-        function_configs: Optional[Dict[str, Dict[str, Any]]] - Mapping of function names to their configuration
         max_prompt_iterations: Optional[int] - Maximum number of prompt iterations before stopping
     """
 
@@ -39,7 +34,6 @@ class ToolCallingNode(BaseNode):
     blocks: ClassVar[List[PromptBlock]] = []
     functions: ClassVar[List[Tool]] = []
     prompt_inputs: ClassVar[Optional[EntityInputsInterface]] = None
-    function_configs: ClassVar[Optional[Dict[str, Dict[str, Any]]]] = None
     max_prompt_iterations: ClassVar[Optional[int]] = 5
 
     class Outputs(BaseOutputs):
@@ -69,6 +63,8 @@ class ToolCallingNode(BaseNode):
                 chat_history: List[ChatMessage] = []
                 prompt_iterations: int = 0
 
+            from vellum.workflows.workflows.base import BaseWorkflow
+
             class ToolCallingWorkflow(BaseWorkflow[BaseInputs, ToolCallingState]):
                 graph = self._graph
 
@@ -112,23 +108,10 @@ class ToolCallingNode(BaseNode):
         self._function_nodes = {}
         for function in self.functions:
             function_name = get_function_name(function)
-            # Get configuration for this function
-            config = {}
-            if callable(function) and self.function_configs and function.__name__ in self.function_configs:
-                config = self.function_configs[function.__name__]
-
-            packages = config.get("packages", None)
-            if packages is not None:
-                packages = cast(Sequence[CodeExecutionPackage], packages)
-
-            runtime_raw = config.get("runtime", "PYTHON_3_11_6")
-            runtime = cast(CodeExecutionRuntime, runtime_raw)
 
             self._function_nodes[function_name] = create_function_node(
                 function=function,
                 tool_router_node=self.tool_router_node,
-                packages=packages,
-                runtime=runtime,
             )
 
         graph_set = set()

vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/tests/test_node.py

@@ -8,7 +8,7 @@ from vellum.client.types.string_chat_message_content import StringChatMessageCon
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import create_function_node, create_tool_router_node
+from vellum.workflows.nodes.displayable.tool_calling_node.utils import create_function_node, create_tool_router_node
 from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState, StateMeta
 from vellum.workflows.state.context import WorkflowContext

vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/tests/test_utils.py

@@ -1,7 +1,7 @@
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import get_function_name
+from vellum.workflows.nodes.displayable.tool_calling_node.utils import get_function_name
 from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState
 from vellum.workflows.types.definition import DeploymentDefinition

vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/utils.py

@@ -1,14 +1,9 @@
-from collections.abc import Sequence
-import inspect
 import json
-import types
 from typing import Any, Iterator, List, Optional, Type, cast
 
 from pydash import snake_case
 
 from vellum import ChatMessage, PromptBlock
-from vellum.client.types.code_execution_package import CodeExecutionPackage
-from vellum.client.types.code_execution_runtime import CodeExecutionRuntime
 from vellum.client.types.function_call_chat_message_content import FunctionCallChatMessageContent
 from vellum.client.types.function_call_chat_message_content_value import FunctionCallChatMessageContentValue
 from vellum.client.types.string_chat_message_content import StringChatMessageContent
@@ -16,13 +11,11 @@ from vellum.client.types.variable_prompt_block import VariablePromptBlock
 from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.displayable.code_execution_node.node import CodeExecutionNode
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
 from vellum.workflows.nodes.displayable.subworkflow_deployment_node.node import SubworkflowDeploymentNode
 from vellum.workflows.outputs.base import BaseOutput
 from vellum.workflows.ports.port import Port
 from vellum.workflows.references.lazy import LazyReference
-from vellum.workflows.state.base import BaseState
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.state.encoder import DefaultStateEncoder
 from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior, Tool
@@ -141,19 +134,16 @@ def create_tool_router_node(
 def create_function_node(
     function: Tool,
     tool_router_node: Type[ToolRouterNode],
-    packages: Optional[Sequence[CodeExecutionPackage]] = None,
-    runtime: CodeExecutionRuntime = "PYTHON_3_11_6",
 ) -> Type[FunctionNode]:
     """
     Create a FunctionNode class for a given function.
 
     For workflow functions: BaseNode
-    For regular functions: CodeExecutionNode with embedded function
+    For regular functions: BaseNode with direct function call
+
     Args:
         function: The function to create a node for
         tool_router_node: The tool router node class
-        packages: Optional list of packages to install for code execution (only used for regular functions)
-        runtime: The runtime to use for code execution (default: "PYTHON_3_11_6")
     """
     if isinstance(function, DeploymentDefinition):
         deployment = function.deployment_id or function.deployment_name
@@ -250,24 +240,8 @@ def create_function_node(
             },
         )
     else:
-        # For regular functions, use CodeExecutionNode approach
-        # function tool must be put in another file (no need to have the same name)
-        source_path = inspect.getmodule(function)
-        if source_path is not None:
-            function_source = inspect.getsource(source_path)
-        else:
-            function_source = f"<source code not available for {function.__name__}>"
-        function_name = function.__name__
-
-        code = f'''
-{function_source}
-
-def main(arguments):
-    """Main function that calls the original function with the provided arguments."""
-    return {function_name}(**arguments)
-'''
-
-        def execute_code_execution_function(self) -> BaseNode.Outputs:
+        # For regular functions, call them directly
+        def execute_regular_function(self) -> BaseNode.Outputs:
             # Get the function call from the tool router output
             function_call_output = self.state.meta.node_outputs.get(tool_router_node.Outputs.results)
             if function_call_output and len(function_call_output) > 0:
@@ -276,42 +250,33 @@ def main(arguments):
             else:
                 arguments = {}
 
-            self.code_inputs = {"arguments": arguments}
-
-            outputs = base_class.run(self)
+            # Call the function directly
+            try:
+                result = function(**arguments)
+            except Exception as e:
+                raise NodeException(
+                    message=f"Error executing function '{function.__name__}': {str(e)}",
+                    code=WorkflowErrorCode.NODE_EXECUTION,
+                )
 
+            # Add the result to the chat history
             self.state.chat_history.append(
                 ChatMessage(
                     role="FUNCTION",
-                    content=StringChatMessageContent(value=json.dumps(outputs.result, cls=DefaultStateEncoder)),
+                    content=StringChatMessageContent(value=json.dumps(result, cls=DefaultStateEncoder)),
                 )
             )
 
             return self.Outputs()
 
-        # Create the properly typed base class with explicit type annotation
-        def get_function_output_type() -> Type:
-            return function.__annotations__.get("return", Any)
-
-        output_type = get_function_output_type()
-
-        base_class: Type[CodeExecutionNode] = CodeExecutionNode[BaseState, output_type]  # type: ignore[valid-type]
-
-        # Create the class with basic attributes
-        node = types.new_class(
-            f"CodeExecutionNode_{function.__name__}",
-            (base_class,),
-            {},
-            lambda ns: ns.update(
-                {
-                    "code": code,
-                    "code_inputs": {},  # No inputs needed since we handle function call extraction in run()
-                    "run": execute_code_execution_function,
-                    "runtime": runtime,
-                    "packages": packages,
-                    "__module__": __name__,
-                }
-            ),
+        # Create BaseNode for regular functions
+        node = type(
+            f"FunctionNode_{function.__name__}",
+            (FunctionNode,),
+            {
+                "run": execute_regular_function,
+                "__module__": __name__,
+            },
         )
 
     return node

vellum/workflows/nodes/experimental/__init__.py

@@ -1,3 +1,3 @@
-from .tool_calling_node import ToolCallingNode
+from ..displayable.tool_calling_node import ToolCallingNode
 
 __all__ = ["ToolCallingNode"]

vellum/workflows/nodes/experimental/tool_calling_node/__init__.py

@@ -1,3 +1,3 @@
-from vellum.workflows.nodes.experimental.tool_calling_node.node import ToolCallingNode
+from vellum.workflows.nodes.displayable.tool_calling_node import ToolCallingNode
 
 __all__ = ["ToolCallingNode"]

vellum/workflows/state/encoder.py

@@ -68,10 +68,10 @@ class DefaultStateEncoder(JSONEncoder):
             function_definition = compile_function_definition(obj)
             source_path = inspect.getsourcefile(obj)
             if source_path is not None:
-                with open(source_path, "r") as f:
+                with open(source_path) as f:
                     source_code = f.read()
             else:
-                source_code = f"<source code not available for {obj.__name__}>"
+                source_code = f"# Error: Source code not available for {obj.__name__}"
 
             return {
                 "type": "CODE_EXECUTION",

{vellum_ai-0.14.80.dist-info → vellum_ai-0.14.82.dist-info}/METADATA

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

{vellum_ai-0.14.80.dist-info → vellum_ai-0.14.82.dist-info}/RECORD

@@ -3,7 +3,7 @@ vellum_cli/README.md,sha256=2NudRoLzWxNKqnuVy1JuQ7DerIaxWGYkrH8kMd-asIE,90
 vellum_cli/__init__.py,sha256=50dYVkBzGUS1i-UKf_9zv1JMif1lLClriNUSs-Z7i4E,12678
 vellum_cli/aliased_group.py,sha256=ugW498j0yv4ALJ8vS9MsO7ctDW7Jlir9j6nE_uHAP8c,3363
 vellum_cli/config.py,sha256=v5BmZ-t_v4Jmqd7KVuQMZF2pRI-rbMspSkVYXIRoTmI,9448
-vellum_cli/image_push.py,sha256=rPeSfTlFpdTzwTdnXRL82OVd61qMoGFGLxnLjZLU98Q,10596
+vellum_cli/image_push.py,sha256=eeOBqKtKkPu6Kgm_jQCVCivogzAcdlIlkv61-QxH67c,11006
 vellum_cli/init.py,sha256=WpnMXPItPmh0f0bBGIer3p-e5gu8DUGwSArT_FuoMEw,5093
 vellum_cli/logger.py,sha256=dcM_OmgqXLo93vDYswO5ylyUQQcTfnA5GTd5tbIt3wM,1446
 vellum_cli/ping.py,sha256=p_BCCRjgPhng6JktuECtkDQLbhopt6JpmrtGoLnLJT8,1161
@@ -12,8 +12,8 @@ vellum_cli/push.py,sha256=sUo0d7lm7saGxQciH-JW2FB_gWfP2gwoEHO1wGnNQR0,10933
 vellum_cli/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_cli/tests/conftest.py,sha256=wx3PlJjVB0HRf5dr2b_idOIw27WPPl0J0FNbhIJJaVk,1689
 vellum_cli/tests/test_config.py,sha256=uvKGDc8BoVyT9_H0Z-g8469zVxomn6Oi3Zj-vK7O_wU,2631
-vellum_cli/tests/test_image_push.py,sha256=qrlSZVMkPYc0ieqYaY2HCI8bBzwwL_Sx1G2nLb9_pSo,9199
-vellum_cli/tests/test_image_push_error_handling.py,sha256=_Wjfkn1orI2K4Ahzqz4u8T13or7NOX01K4BtcTuTIOM,7107
+vellum_cli/tests/test_image_push.py,sha256=X0YOPdoaAnWtK9IQTxaN_wWpw08-5G3v76k1Wu53JpU,12801
+vellum_cli/tests/test_image_push_error_handling.py,sha256=QRH0eYNEEIkD2-EVFQWYexOKlhMB6puh1GouovrE-VU,7647
 vellum_cli/tests/test_init.py,sha256=8UOc_ThfouR4ja5cCl_URuLk7ohr9JXfCnG4yka1OUQ,18754
 vellum_cli/tests/test_main.py,sha256=qDZG-aQauPwBwM6A2DIu1494n47v3pL28XakTbLGZ-k,272
 vellum_cli/tests/test_ping.py,sha256=178EJHxPZtnnPMNXXynsQt8DIFhsrdc2bL17_YsG17M,2580
@@ -62,7 +62,7 @@ vellum_ee/workflows/display/nodes/vellum/tests/test_retry_node.py,sha256=h93ysol
 vellum_ee/workflows/display/nodes/vellum/tests/test_search_node.py,sha256=KvByxgbUkVyfPIVxTUBUk6a92JiJMi8eReZWxzfPExU,3864
 vellum_ee/workflows/display/nodes/vellum/tests/test_subworkflow_deployment_node.py,sha256=BUzHJgjdWnPeZxjFjHfDBKnbFjYjnbXPjc-1hne1B2Y,3965
 vellum_ee/workflows/display/nodes/vellum/tests/test_templating_node.py,sha256=LSk2gx9TpGXbAqKe8dggQW8yJZqj-Cf0EGJFeGGlEcw,3321
-vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py,sha256=YMoLNWknmSsFj7MtprenOYuXTX1VWUQ2bXZs_p76gBg,8163
+vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py,sha256=bSMWCwD-niSx3MQPtgsLXjIf6zynQxZdpmeaOM6H3QA,4580
 vellum_ee/workflows/display/nodes/vellum/tests/test_try_node.py,sha256=Khjsb53PKpZuyhKoRMgKAL45eGp5hZqXvHmVeQWRw4w,2289
 vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py,sha256=3LS1O4DGPWit05oj_ubeW8AlHGnoBxdUMferGQuAiZs,4851
 vellum_ee/workflows/display/nodes/vellum/try_node.py,sha256=z9Omo676RRc7mQjLoL7hjiHhUj0OWVLhrrb97YTN4QA,4086
@@ -78,7 +78,7 @@ vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_outp
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_ports_serialization.py,sha256=PkSgghJDz0fpDB72HHPjLjo8LkZk-HpUkCQzRLX-iVw,40611
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_trigger_serialization.py,sha256=dsJr8I9AdPwMOGszirfNDzZP2Ychd94aAKuPXAzknMk,4632
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py,sha256=Q3QOzzF7KE_9EukJ9yNFgYL7Qio_vTqDVd-6kc40wd4,15834
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py,sha256=SrpLTehz__k9DFGMTWQfTKxuzDxY2rLi8TJ27-h0UKE,29681
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py,sha256=B4DC2e5nqEk5NZeUwR4vKmBoEyToHkAuq4cCqbsidFo,29708
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py,sha256=cBHGuY-DeyqQlLbpRZrrQyDh7em8rPkB7bV9Vi0EFoE,53608
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_default_state_serialization.py,sha256=E1ww97BFoGbnRkxf84TScat5NQhP8nLVrdaOlpGnSKU,8615
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_error_node_serialization.py,sha256=w1BYEZXIMjWmcKGQXzhelHnKFlEAXmsgjkI9fcgqKXA,5850
@@ -93,8 +93,8 @@ vellum_ee/workflows/display/tests/workflow_serialization/test_basic_search_node_
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_deployment_serialization.py,sha256=XWrhHg_acLsRHwjstBAii9Pmes9oXFtAUWSAVF1oSBc,11225
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_templating_node_serialization.py,sha256=V8b6gKghLlO7PJI8xeNdnfn8aII0W_IFQvSQBQM62UQ,7721
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_terminal_node_serialization.py,sha256=hDWtKXmGI1CKhTwTNqpu_d5RkE5n7SolMLtgd87KqTI,3856
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py,sha256=6q-lTGMp5HbQba3NsHUFSit8_zEBxIVPGE8yCw4SVx0,25720
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py,sha256=aY76l9s_2nezx7hCfUZsPYFzvYMy-Rjd73G0DlL6s6A,9348
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py,sha256=Nu5XZeGbqev638Osd9Vu17JE74ZWAHsrhYi9Uk-xLr8,25489
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py,sha256=PR2bLdtl0QuA9WbvZGNBpPeQy1xOV6UeALKpJSRSCEY,8712
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_workflow_deployment_serialization.py,sha256=mova0sPD3evHiHIN1O0VynxlCp-uOcEIKve5Pd_oCDg,4069
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_try_node_serialization.py,sha256=pLCyMScV88DTBXRH7jXaXOEA1GBq8NIipCUFwIAWnwI,2771
 vellum_ee/workflows/display/tests/workflow_serialization/test_complex_terminal_node_serialization.py,sha256=J4ouI8KxbMfxQP2Zq_9cWMGYgbjCWmKzjCJEtnSJb0I,5829
@@ -139,12 +139,12 @@ vellum_ee/workflows/tests/test_display_meta.py,sha256=PkXJVnMZs9GNooDkd59n4YTBAX
 vellum_ee/workflows/tests/test_serialize_module.py,sha256=qcHbl6YaKtJebQJvw0OXCVlJXLkj6sjkbSMu05zXI0U,1874
 vellum_ee/workflows/tests/test_server.py,sha256=SsOkS6sGO7uGC4mxvk4iv8AtcXs058P9hgFHzTWmpII,14519
 vellum_ee/workflows/tests/test_virtual_files.py,sha256=TJEcMR0v2S8CkloXNmCHA0QW0K6pYNGaIjraJz7sFvY,2762
-vellum/__init__.py,sha256=GQqR0tDU6_sjyk9pP-0qqV7Rrk1p5b8ObQKR5Xnr4o8,42510
+vellum/__init__.py,sha256=I8qQZ9GWQ_u-rMZlCWj-Zl2Z_YM1NGY3fAG5BrPeZb8,42698
 vellum/client/README.md,sha256=CuGUYnaE0Imt0KqQ4sIPaUghCjLHkF3DdEvZWu14-8s,4807
 vellum/client/__init__.py,sha256=AYopGv2ZRVn3zsU8_km6KOvEHDbXiTPCVuYVI7bWvdA,120166
 vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/client/core/client_wrapper.py,sha256=28--LC1-hrUv3Yo5HQtt5VDbxX6iu7PRzaZPVD9oylU,1869
+vellum/client/core/client_wrapper.py,sha256=xroZld9rUgZBqi4t2PSZNmSpW0hkjl8W5dkudogh9PI,1869
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -212,7 +212,7 @@ vellum/client/resources/workspace_secrets/__init__.py,sha256=FTtvy8EDg9nNNg9WCat
 vellum/client/resources/workspace_secrets/client.py,sha256=zlBdbeTP6sqvtyl_DlrpfG-W5hSP7tJ1NYLSygi4CLU,8205
 vellum/client/resources/workspaces/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/client/resources/workspaces/client.py,sha256=RthwzN1o-Jxwg5yyNNodavFyNUSxfLoTv26w3mRR5g8,3595
-vellum/client/types/__init__.py,sha256=DPbl8CjtbEKqhlsrTZxdRgx2vvJ9ZVvD20RO2AERRGM,64415
+vellum/client/types/__init__.py,sha256=gl1Lb4alZrG7SqoSibiKJY4I5sSCcqH-QH560_vwGUw,64711
 vellum/client/types/ad_hoc_execute_prompt_event.py,sha256=bCjujA2XsOgyF3bRZbcEqV2rOIymRgsLoIRtZpB14xg,607
 vellum/client/types/ad_hoc_expand_meta.py,sha256=1gv-NCsy_6xBYupLvZH979yf2VMdxAU-l0y0ynMKZaw,1331
 vellum/client/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=oDG60TpwK1YNSKhRsBbiP2O3ZF9PKR-M9chGIfKw4R4,1004
@@ -330,7 +330,8 @@ vellum/client/types/execution_json_vellum_value.py,sha256=oGY3CsJBKeOuEexmITfRYc
 vellum/client/types/execution_number_vellum_value.py,sha256=b2TpqyafRHCdl6EhgctNgUSLU-JBdouU6OgM8Jk_O78,809
 vellum/client/types/execution_search_results_vellum_value.py,sha256=HkxoXaUF6pMbfXd5wLk5VKmcXed2IRfEzkxsoGpwmg0,898
 vellum/client/types/execution_string_vellum_value.py,sha256=4w0ottwB5F2NL3uEXBBggP7XkcdE_D2lGmEobkXWY7o,807
-vellum/client/types/execution_vellum_value.py,sha256=PpbrbJdSioqaqT2gZ6mx2Ht0Oy8XGyxBDcLFfPqUBIA,980
+vellum/client/types/execution_thinking_vellum_value.py,sha256=n66gPYgiDpOGhOniQlRE6jDUOBnPkHnfOvz8RVyzw1g,879
+vellum/client/types/execution_vellum_value.py,sha256=gJ4UWA4KKzWGJZpEZGQV8Efqh27PmyWz6RZSsbtNux8,1088
 vellum/client/types/external_input_descriptor.py,sha256=ErOW2OfFMz1FDGmVY6NgiUBPsleaWhdJBekwFp4ru7o,805
 vellum/client/types/external_test_case_execution.py,sha256=TkO1CQcEI8LA7sdYJfAqhbdkj27sXEkF8VL7zBeDBM4,877
 vellum/client/types/external_test_case_execution_request.py,sha256=xd80GZg4FbZP6FZRIhZL9V1uFetI3bcCj-wzanco5tQ,921
@@ -524,7 +525,7 @@ vellum/client/types/prompt_execution_meta.py,sha256=3hhMZgdAR5mKfnh2e_eVN3oKfT0E
 vellum/client/types/prompt_node_execution_meta.py,sha256=IyWH__nCp5uwS0N32b2ZEsA-Fv7AZDB4nnlRZayU2Gc,888
 vellum/client/types/prompt_node_result.py,sha256=3jewO-nPodoXTq_5RxgwhKfDZrvoPjRZ_vUXLeqiuHY,749
 vellum/client/types/prompt_node_result_data.py,sha256=fNOxBfK3ablDBxkUWVVstJMYaGdHGgu27WxP87E6UQ4,872
-vellum/client/types/prompt_output.py,sha256=NpDGJNIYIivzQJnBeoJLpJlCk7gqBESLwv5Qtn_20qQ,398
+vellum/client/types/prompt_output.py,sha256=YRHDKRB2Yn3GS7pzY-cH17gtjcvew0vP0hls_tzWF3o,480
 vellum/client/types/prompt_parameters.py,sha256=Vkwh4zI9gX1DuGQxrWiUUa1TshTfnPlS7_yRrziD5qg,1046
 vellum/client/types/prompt_push_response.py,sha256=Un61pv4kVH0Omd8OKVfTk1yZ1RWrwuquAgl1vanb04Y,601
 vellum/client/types/prompt_request_chat_history_input.py,sha256=DB2io5piMSyA89f5lnIVYO4MLZoNALNSufx8Y-oOwOE,790
@@ -696,6 +697,8 @@ vellum/client/types/test_suite_test_case_replace_bulk_operation_request.py,sha25
 vellum/client/types/test_suite_test_case_replaced_bulk_result.py,sha256=BIlXI7udygWrwtyRhCl8hmExHbkAl9lI8s3sm1G5iGc,1019
 vellum/client/types/test_suite_test_case_replaced_bulk_result_data.py,sha256=ORmcUvwzvRLRaoFhxdXFIKzPxOI6PU1kESl0R6rsJuw,651
 vellum/client/types/test_suite_test_case_upsert_bulk_operation_request.py,sha256=PrKuqePiXBQv6iLAxsk4xQg29KGdOlqMDhIVdGNxuz4,1071
+vellum/client/types/thinking_vellum_value.py,sha256=XLeSzCwbO9kVek6zdMn6pQChLKDYFeyabE9OiFeoq_I,755
+vellum/client/types/thinking_vellum_value_request.py,sha256=n6FqEZs5TKhhNZqe9yau_Vcn55ktXxFQO5sWa0aRNbg,784
 vellum/client/types/token_overlapping_window_chunker_config.py,sha256=_8vR9AiZQmb5OA3OojbjuTOGiGNTS9EY0vXrmej_TM0,731
 vellum/client/types/token_overlapping_window_chunker_config_request.py,sha256=O58w5om6EsCgZeqM7n3KSzwo1PqINyHWln46EFW4Inc,738
 vellum/client/types/token_overlapping_window_chunking.py,sha256=TghiPKWZg3Eg_UzGI9VmjQgVPZFABrnhfsz4iPLEem8,889
@@ -719,14 +722,14 @@ vellum/client/types/vellum_sdk_error.py,sha256=1AyCu_sUa8gx5wUJg5EtDYbvs9uQPhZfH
 vellum/client/types/vellum_sdk_error_code_enum.py,sha256=a2K6XJvl72tNn82zyx5QpzYGgLfYWVBKY-o8jeDqKoM,504
 vellum/client/types/vellum_secret.py,sha256=jru3nBCquZHZ3wEdpgKZNmy2WFh26PVKrTNaXeVSsZQ,555
 vellum/client/types/vellum_span.py,sha256=V5P1z-OFqgQQAZjek2toYudlh-fCu0NTt5uKuKm9hd0,258
-vellum/client/types/vellum_value.py,sha256=Un9AeNlx0QZA9-eIGgA5nEdZU0_0vHfibiafJR0q8V4,1049
+vellum/client/types/vellum_value.py,sha256=eMg1StXImnx5ow09AbQmjz3YfoJndCYP23LU5y1M0I8,1129
 vellum/client/types/vellum_value_logical_condition_group_request.py,sha256=zdcBnm-qQbHHexNdM17WjMXNVRCPF8Vp7Sbs0a1d8Qw,1270
 vellum/client/types/vellum_value_logical_condition_request.py,sha256=c2gnVEmsrsJSRbx7DGlv8lBpgLRlOHSW1RNo-akbyWo,966
 vellum/client/types/vellum_value_logical_expression_request.py,sha256=vjNsI1NUAkwxLwIXJM_DVXTObyAM63gOfHj6aHw7osc,479
-vellum/client/types/vellum_value_request.py,sha256=WAUV4RnyAG6t0k94QTY1FE4F8HvpqhptDDGcCinwjis,1298
+vellum/client/types/vellum_value_request.py,sha256=zmu0DFiXuCkt7PxbhJc7giM_U0GPoMPjN5xE9LKhqhA,1400
 vellum/client/types/vellum_variable.py,sha256=BDcccISJsycsrW3E5A5RTIOfxS_83ofkleOPP1upqW4,959
 vellum/client/types/vellum_variable_extensions.py,sha256=PsrRo0STOKhxrkSFRrOXCPlf1x5Uxpy3vVMJz02O20E,685
-vellum/client/types/vellum_variable_type.py,sha256=XUBG699lv7iXtpG3vUeBmTGfKE9u164IuJfzHeXWKJ8,391
+vellum/client/types/vellum_variable_type.py,sha256=epYV-PY0NkvUntKzgzqelWMq9Dzmh7Y32c19GB_2mh0,411
 vellum/client/types/vellum_workflow_execution_event.py,sha256=H8rP3_6a6LTvemoHfsmI2THpTRf8PYrZQcjoeTzf-44,934
 vellum/client/types/workflow_deployment_event_executions_response.py,sha256=_X9hfsfpCf7DaQpfg83AAcbadZxEVi1NKNf2PZ7PRY8,1190
 vellum/client/types/workflow_deployment_history_item.py,sha256=w5yqAtmYWw8kTljYEVBa8-fLxKsrBA8Img7IC3GjZG4,1226
@@ -1017,6 +1020,7 @@ vellum/types/execution_json_vellum_value.py,sha256=AvcmBTxlf41ivq5KeMU8MRHAgRYxl
 vellum/types/execution_number_vellum_value.py,sha256=rs7w_72tN3TFVpx-tlD-IGrap25xWkPQ7o4yyUlqGk4,167
 vellum/types/execution_search_results_vellum_value.py,sha256=of3CXBuKUjHW0ZU3SeOe5XZKrSNz7HFnKwBVaPa7Lzg,175
 vellum/types/execution_string_vellum_value.py,sha256=MY47d3IJQj9PPThKsCZUFS8y80f3so3kDDaSW5pJYsA,167
+vellum/types/execution_thinking_vellum_value.py,sha256=Mx9C-vjS0LCGK60CARSyuDI2bHPJFA9AOoSKSfnICG0,169
 vellum/types/execution_vellum_value.py,sha256=VRe64r540rkdVWjWijd7Y5JsYM_9zw09QUUuorKHeXU,160
 vellum/types/external_input_descriptor.py,sha256=-r6xPo7-1Ioq4XZ2nwcrg8OBoyVItKr-BeXspJwDWDQ,163
 vellum/types/external_test_case_execution.py,sha256=Av4H59SBBCZ9CZtT-eKD3n05lgljTqdtSZSjtsDpbc4,166
@@ -1383,6 +1387,8 @@ vellum/types/test_suite_test_case_replace_bulk_operation_request.py,sha256=CvjbM
 vellum/types/test_suite_test_case_replaced_bulk_result.py,sha256=Aj-4gz999ma3K8q5uF6uMueQ1VomCwn9921oZE8lres,179
 vellum/types/test_suite_test_case_replaced_bulk_result_data.py,sha256=4qxVM1fTuPIo_sndTuV1_cSA9AR1wVfi5bFwIzLzdvA,184
 vellum/types/test_suite_test_case_upsert_bulk_operation_request.py,sha256=xDvR6dJbgCtVT-zwvIE-bO42ySq-3QLlXtwxK30s2LY,188
+vellum/types/thinking_vellum_value.py,sha256=-aDVQ7sFbAkbf2Eab-z_27fSf91pzqkIfS_65eT4Cks,159
+vellum/types/thinking_vellum_value_request.py,sha256=cxfwr_RWpepYUp3BHfq0Ui9I8XPqERCLosnpZpLfsNk,167
 vellum/types/token_overlapping_window_chunker_config.py,sha256=NFoh8PAQCFXMnmHJ-FHZIbJjdZoVvrSaFsZ-5UOvldI,177
 vellum/types/token_overlapping_window_chunker_config_request.py,sha256=C1_P8TQXCRkKD6zcy2O7l2eMaPH_atBMdQOW8ZB-HCg,185
 vellum/types/token_overlapping_window_chunking.py,sha256=fgVxqPWWRIwTyb1_MZxwmjbIMeCzEKBeWFbdk9cw4PU,171
@@ -1603,7 +1609,7 @@ vellum/workflows/nodes/core/try_node/__init__.py,sha256=JVD4DrldTIqFQQFrubs9KtWC
 vellum/workflows/nodes/core/try_node/node.py,sha256=XdyOvlwQ3m4h0-_WNtaBl2t_CdlzPXclulkLOtUcX3E,4388
 vellum/workflows/nodes/core/try_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/core/try_node/tests/test_node.py,sha256=h6eUc3SggvhzBWlOD0PrPUlkoCSQHwjqYn81VkxSIxU,4948
-vellum/workflows/nodes/displayable/__init__.py,sha256=6F_4DlSwvHuilWnIalp8iDjjDXl0Nmz4QzJV2PYe5RI,1023
+vellum/workflows/nodes/displayable/__init__.py,sha256=zH7SFotr4i8sO-r5_k53yPipQwDouDmHxTfCE6dXAoU,1093
 vellum/workflows/nodes/displayable/api_node/__init__.py,sha256=MoxdQSnidIj1Nf_d-hTxlOxcZXaZnsWFDbE-PkTK24o,56
 vellum/workflows/nodes/displayable/api_node/node.py,sha256=kGvKPxXj4XcAsNLtvE1RBGI58MiSNTkQcQU2g2JHC3E,2922
 vellum/workflows/nodes/displayable/api_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1629,7 +1635,7 @@ vellum/workflows/nodes/displayable/code_execution_node/node.py,sha256=Qh4SPafzdR
 vellum/workflows/nodes/displayable/code_execution_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/code_execution_node/tests/fixtures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/code_execution_node/tests/fixtures/main.py,sha256=5QsbmkzSlSbcbWTG_JmIqcP-JNJzOPTKxGzdHos19W4,79
-vellum/workflows/nodes/displayable/code_execution_node/tests/test_node.py,sha256=47W5fPLpDcaKhCVuauKOwcivx1NfhpbsRMPrs-WBnlY,38441
+vellum/workflows/nodes/displayable/code_execution_node/tests/test_node.py,sha256=0lxmU-XXie6qu35qAEfZJ3vKZ1XGvWV4eq9y2RMelOE,38521
 vellum/workflows/nodes/displayable/code_execution_node/utils.py,sha256=VRTKms59vrSR9mDk99cojParZVAP4lzjEeDwDNXU1tk,3837
 vellum/workflows/nodes/displayable/conditional_node/__init__.py,sha256=AS_EIqFdU1F9t8aLmbZU-rLh9ry6LCJ0uj0D8F0L5Uw,72
 vellum/workflows/nodes/displayable/conditional_node/node.py,sha256=71ZUNfTiD7t2Kai2ypw0tmv1lSf1brQaHAQD-SeUrGE,1101
@@ -1644,7 +1650,7 @@ vellum/workflows/nodes/displayable/guardrail_node/test_node.py,sha256=SAGv6hSFcB
 vellum/workflows/nodes/displayable/guardrail_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/guardrail_node/tests/test_node.py,sha256=X2pd6TI8miYxIa7rgvs1pHTEreyWcf77EyR0_Jsa700,2055
 vellum/workflows/nodes/displayable/inline_prompt_node/__init__.py,sha256=gSUOoEZLlrx35-tQhSAd3An8WDwBqyiQh-sIebLU9wU,74
-vellum/workflows/nodes/displayable/inline_prompt_node/node.py,sha256=ST_V1Z3-7lzZ__Ntfm6iry70u9D3GD2a2vxsl2d4Q_c,2843
+vellum/workflows/nodes/displayable/inline_prompt_node/node.py,sha256=ApPcHc5beGL2ReONT6MF-A4DGWZFCdKJ4ha-204bRoU,3053
 vellum/workflows/nodes/displayable/inline_prompt_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/inline_prompt_node/tests/test_node.py,sha256=bBHs90mV5SZ3rJPAL0wx4WWyawUA406LgMPOdvpZC_A,10923
 vellum/workflows/nodes/displayable/merge_node/__init__.py,sha256=J8IC08dSH7P76wKlNuxe1sn7toNGtSQdFirUbtPDEs0,60
@@ -1652,7 +1658,7 @@ vellum/workflows/nodes/displayable/merge_node/node.py,sha256=nZtGGVAvY4fvGg8vwV6
 vellum/workflows/nodes/displayable/note_node/__init__.py,sha256=KWA3P4fyYJ-fOTky8qNGlcOotQ-HeHJ9AjZt6mRQmCE,58
 vellum/workflows/nodes/displayable/note_node/node.py,sha256=sIN1VBQ7zeT3GhN0kupXbFfdpvgedWV79k4woJNp5IQ,394
 vellum/workflows/nodes/displayable/prompt_deployment_node/__init__.py,sha256=krX1Hds-TSVYZsx0wJFX4wsAKkEFYOX1ifwRGiIM-EA,82
-vellum/workflows/nodes/displayable/prompt_deployment_node/node.py,sha256=eUiQYdqJTrWhVcUgGAPJYEnRk6S71Yrzu5-c-XcVFs4,3243
+vellum/workflows/nodes/displayable/prompt_deployment_node/node.py,sha256=KBMylYcEIoqTG438s5I-rv4uRejB8-Yc_4QA7XyALy0,3453
 vellum/workflows/nodes/displayable/prompt_deployment_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/prompt_deployment_node/tests/test_node.py,sha256=c_nuuqrwiIjgj4qIbVypfDuOc-3TlgO6CbXFqQl2Nqw,19725
 vellum/workflows/nodes/displayable/search_node/__init__.py,sha256=hpBpvbrDYf43DElRZFLzieSn8weXiwNiiNOJurERQbs,62
@@ -1668,16 +1674,17 @@ vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py,sha256=
 vellum/workflows/nodes/displayable/tests/test_search_node_error_handling.py,sha256=8aw8hDFL0ZXThvAa7yxrJN026EYGD4-Q1si3Phu9-_0,6307
 vellum/workflows/nodes/displayable/tests/test_search_node_wth_text_output.py,sha256=VepO5z1277c1y5N6LLIC31nnWD1aak2m5oPFplfJHHs,6935
 vellum/workflows/nodes/displayable/tests/test_text_prompt_deployment_node.py,sha256=dc3EEn1sOICpr3GdS8eyeFtExaGwWWcw9eHSdkRhQJU,2584
+vellum/workflows/nodes/displayable/tool_calling_node/__init__.py,sha256=3n0-ysmFKsr40CVxPthc0rfJgqVJeZuUEsCmYudLVRg,117
+vellum/workflows/nodes/displayable/tool_calling_node/node.py,sha256=PPu8JPDHUAHU95d73WDwq75sfsUH-I47BdnOuZXkIm8,4976
+vellum/workflows/nodes/displayable/tool_calling_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vellum/workflows/nodes/displayable/tool_calling_node/tests/test_node.py,sha256=X9kl8qntn0cQOOvyn0noDayhU0hXenqYJ0XO1FkOjAs,5071
+vellum/workflows/nodes/displayable/tool_calling_node/tests/test_utils.py,sha256=eu6WTyENhGLg9pGp_j69rysZjf_qiQXske1YdZn9PzU,1718
+vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=f2hCLciTqjHXj9tVdnnTAVY0BLUCMBoH0emQeOJCB7I,11309
 vellum/workflows/nodes/experimental/README.md,sha256=eF6DfIL8t-HbF9-mcofOMymKrraiBHDLKTlnBa51ZiE,284
-vellum/workflows/nodes/experimental/__init__.py,sha256=_tpZGWAZLydcKxfrj1-plrZeTajskVhUr1A6mHoSaWM,78
+vellum/workflows/nodes/experimental/__init__.py,sha256=k7VQEyvgEdnrEZ-icXx3fiByPnyMOnMXNGGuqacyyik,91
 vellum/workflows/nodes/experimental/openai_chat_completion_node/__init__.py,sha256=lsyD9laR9p7kx5-BXGH2gUTM242UhKy8SMV0SR6S2iE,90
 vellum/workflows/nodes/experimental/openai_chat_completion_node/node.py,sha256=cKI2Ls25L-JVt4z4a2ozQa-YBeVy21Z7BQ32Sj7iBPE,10460
-vellum/workflows/nodes/experimental/tool_calling_node/__init__.py,sha256=S7OzT3I4cyOU5Beoz87nPwCejCMP2FsHBFL8OcVmxJ4,118
-vellum/workflows/nodes/experimental/tool_calling_node/node.py,sha256=jwL1sbitmm1CpTOAEI0IIuc6VRr8d7yxUpS4Y5s9Bk8,5966
-vellum/workflows/nodes/experimental/tool_calling_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/nodes/experimental/tool_calling_node/tests/test_node.py,sha256=XK1H_QAT_nVFmFP442RYyPvpTfSgtU6kSGu3-OQPBNU,5072
-vellum/workflows/nodes/experimental/tool_calling_node/tests/test_utils.py,sha256=-g90SdXscuikF7JP0lFGvSvPc8jl2vBuHwBeiYJIiXk,1719
-vellum/workflows/nodes/experimental/tool_calling_node/utils.py,sha256=t18ye2wY9jtTjKXJs8oChNGI5xCdxlUUyVYbRJfJLFI,12959
+vellum/workflows/nodes/experimental/tool_calling_node/__init__.py,sha256=8R5HQtDDfo0hgFX6VdM45-FiyxK-IMTI9B9LL4TRUbw,112
 vellum/workflows/nodes/mocks.py,sha256=a1FjWEIocseMfjzM-i8DNozpUsaW0IONRpZmXBoWlyc,10455
 vellum/workflows/nodes/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/tests/test_mocks.py,sha256=mfPvrs75PKcsNsbJLQAN6PDFoVqs9TmQxpdyFKDdO60,7837
@@ -1710,7 +1717,7 @@ vellum/workflows/sandbox.py,sha256=GVJzVjMuYzOBnSrboB0_6MMRZWBluAyQ2o7syeaeBd0,2
 vellum/workflows/state/__init__.py,sha256=yUUdR-_Vl7UiixNDYQZ-GEM_kJI9dnOia75TtuNEsnE,60
 vellum/workflows/state/base.py,sha256=WIMJYyuHUrP4zt0Nudk66HAK1L6GgGmsU_GQp7BGE2U,22189
 vellum/workflows/state/context.py,sha256=KOAI1wEGn8dGmhmAemJaf4SZbitP3jpIBcwKfznQaRE,3076
-vellum/workflows/state/encoder.py,sha256=xtKpDAR5qWbxIN61dz3X_5_D47MTrbvvas8n15jys18,2881
+vellum/workflows/state/encoder.py,sha256=sV80h-lAo66isD2ezK1_YLzj75Aa5dAtF2YA1c3BYKo,2883
 vellum/workflows/state/store.py,sha256=uVe-oN73KwGV6M6YLhwZMMUQhzTQomsVfVnb8V91gVo,1147
 vellum/workflows/state/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/state/tests/test_state.py,sha256=YOiC9qZAzkdiqb7nRarNWeDwxo7xHv3y3czlHl81ezg,6741
@@ -1745,8 +1752,8 @@ vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnad
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=fROqff6AZpCIzaSwOKSdtYy4XR0UZQ6ejxL3RJOSJVs,20447
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-0.14.80.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.14.80.dist-info/METADATA,sha256=AOpjIQ_DNiOeABYs3Y-0JAxvRnzf2jeZLJ0aQZMBcD0,5556
-vellum_ai-0.14.80.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-0.14.80.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.14.80.dist-info/RECORD,,
+vellum_ai-0.14.82.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.14.82.dist-info/METADATA,sha256=GhPQ5ikBeUMKXXz9PYx9SzIWCkT1puQg6cdTzveC7Zk,5556
+vellum_ai-0.14.82.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-0.14.82.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.14.82.dist-info/RECORD,,

vellum_cli/image_push.py

@@ -24,6 +24,16 @@ def image_push_command(
     logger = load_cli_logger()
     config = load_vellum_cli_config()
 
+    logger.info("Cleaning up unused Docker images...")
+    try:
+        subprocess.run(
+            ["docker", "image", "prune", "--all", "-f", "--filter", "label=image-type=python-workflow-runtime"],
+            check=True,
+        )
+        logger.info("Docker image pruning completed successfully")
+    except subprocess.CalledProcessError as e:
+        logger.warning(f"Docker image pruning failed: {e}")
+
     if source:
         logger.info(f"Building Docker image from Dockerfile: {source}")
 

vellum_cli/tests/test_image_push.py

@@ -48,6 +48,7 @@ def test_image_push__self_hosted_happy_path(mock_docker_from_env, mock_subproces
     mock_docker_from_env.return_value = mock_docker_client
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -100,6 +101,7 @@ def test_image_push__self_hosted_happy_path__workspace_option(
     mock_docker_from_env.return_value = mock_docker_client
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -189,6 +191,7 @@ def test_image_push_with_source_success(
     mock_docker_client.images.push.return_value = [b'{"status": "Pushed"}']
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(args="", returncode=0, stdout=b"Build successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
@@ -205,7 +208,7 @@ def test_image_push_with_source_success(
 
     assert result.exit_code == 0, result.output
 
-    build_call = mock_subprocess_run.call_args_list[0]
+    build_call = mock_subprocess_run.call_args_list[1]
     assert build_call[0][0] == [
         "docker",
         "buildx",
@@ -246,6 +249,7 @@ def test_image_push_with_source_build_fails(mock_subprocess_run, monkeypatch, mo
         f.write("FROM alpine:latest\n")
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(args="", returncode=1, stderr=b"Build failed: missing dependency"),
     ]
 
@@ -255,3 +259,85 @@ def test_image_push_with_source_build_fails(mock_subprocess_run, monkeypatch, mo
     assert result.exit_code == 1
     assert "Docker build failed" in result.output
     assert "Build failed: missing dependency" in result.output
+
+
+@pytest.mark.usefixtures("vellum_client", "info_log_level")
+def test_image_push_includes_docker_prune(mock_docker_from_env, mock_subprocess_run, monkeypatch):
+    """
+    Tests that image_push_command calls docker image prune before validation.
+    """
+    # GIVEN a self hosted vellum api URL env var
+    monkeypatch.setenv("VELLUM_API_URL", "mycompany.api.com")
+    monkeypatch.setenv("VELLUM_API_KEY", "123456abcdef")
+
+    # AND a mock Docker client
+    mock_docker_client = MagicMock()
+    mock_docker_from_env.return_value = mock_docker_client
+
+    mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
+        subprocess.CompletedProcess(
+            args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
+        ),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"manifest"),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"sha256:hellosha"),
+    ]
+
+    # WHEN the user runs the image push command with any image
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["image", "push", "myrepo.net/myimage:latest"])
+
+    # THEN the command exits successfully
+    assert result.exit_code == 0, result.output
+
+    prune_call = mock_subprocess_run.call_args_list[0]
+    assert prune_call[0][0] == [
+        "docker",
+        "image",
+        "prune",
+        "--all",
+        "-f",
+        "--filter",
+        "label=image-type=python-workflow-runtime",
+    ]
+
+    # AND the success message includes pruning completion
+    assert "Docker image pruning completed successfully" in result.output
+
+
+@pytest.mark.usefixtures("vellum_client", "info_log_level")
+def test_image_push_continues_if_prune_fails(mock_docker_from_env, mock_subprocess_run, monkeypatch):
+    """
+    Tests that image_push_command continues if docker image prune fails.
+    """
+    # GIVEN a self hosted vellum api URL env var
+    monkeypatch.setenv("VELLUM_API_URL", "mycompany.api.com")
+    monkeypatch.setenv("VELLUM_API_KEY", "123456abcdef")
+
+    # AND a mock Docker client
+    mock_docker_client = MagicMock()
+    mock_docker_from_env.return_value = mock_docker_client
+
+    mock_subprocess_run.side_effect = [
+        subprocess.CalledProcessError(
+            1, ["docker", "image", "prune", "--all", "-f", "--filter", "label=image-type=python-workflow-runtime"]
+        ),
+        subprocess.CompletedProcess(
+            args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
+        ),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"manifest"),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"sha256:hellosha"),
+    ]
+
+    # WHEN the user runs the image push command with any image
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["image", "push", "myrepo.net/myimage:latest"])
+
+    # THEN the command exits successfully despite pruning failure
+    assert result.exit_code == 0, result.output
+
+    # AND the warning message about pruning failure is shown
+    assert "Docker image pruning failed" in result.output
+
+    # AND the success message is still shown
+    assert "Image successfully pushed" in result.output

vellum_cli/tests/test_image_push_error_handling.py

@@ -18,6 +18,7 @@ def test_image_push_docker_service_token_401_error(mock_docker_from_env, mock_ru
     mock_docker_client.images.get.return_value.id = "test-image-id"
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -43,6 +44,7 @@ def test_image_push_docker_service_token_500_error(mock_docker_from_env, mock_ru
     mock_docker_client.images.get.return_value.id = "test-image-id"
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -70,6 +72,7 @@ def test_image_push_docker_service_token_other_error(mock_docker_from_env, mock_
     mock_docker_client.images.get.return_value.id = "test-image-id"
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -98,6 +101,7 @@ def test_image_push_container_image_401_error(mock_docker_from_env, mock_run, ve
     mock_docker_client.images.push.return_value = ["pushed"]
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -129,6 +133,7 @@ def test_image_push_container_image_500_error(mock_docker_from_env, mock_run, ve
     mock_docker_client.images.push.return_value = ["pushed"]
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -162,6 +167,7 @@ def test_image_push_container_image_other_error(mock_docker_from_env, mock_run,
     mock_docker_client.images.push.return_value = ["pushed"]
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),

vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py

@@ -1,8 +1,7 @@
-from vellum.client.types.code_execution_package import CodeExecutionPackage
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
-from vellum.workflows.nodes.experimental.tool_calling_node.node import ToolCallingNode
+from vellum.workflows.nodes.displayable.tool_calling_node.node import ToolCallingNode
 from vellum.workflows.state.base import BaseState
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
@@ -123,105 +122,3 @@ def test_serialize_node__prompt_inputs__mixed_values():
             ],
         },
     }
-
-
-def test_serialize_node__function_configs():
-    # GIVEN a tool calling node with function packages
-    def foo():
-        pass
-
-    def bar():
-        pass
-
-    class MyToolCallingNode(ToolCallingNode):
-        functions = [foo, bar]
-        function_configs = {
-            "foo": {
-                "runtime": "PYTHON_3_11_6",
-                "packages": [
-                    CodeExecutionPackage(name="first_package", version="1.0.0", repository="test-repo"),
-                    CodeExecutionPackage(name="second_package", version="2.0.0", repository="test-repo"),
-                ],
-            },
-            "bar": {
-                "runtime": "PYTHON_3_11_6",
-                "packages": [CodeExecutionPackage(name="third_package", version="3.0.0")],
-            },
-        }
-
-    # AND a workflow with the tool calling node
-    class Workflow(BaseWorkflow):
-        graph = MyToolCallingNode
-
-    # WHEN the workflow is serialized
-    workflow_display = get_workflow_display(workflow_class=Workflow)
-    serialized_workflow: dict = workflow_display.serialize()
-
-    # THEN the node should properly serialize the function packages
-    my_tool_calling_node = next(
-        node
-        for node in serialized_workflow["workflow_raw_data"]["nodes"]
-        if node["id"] == str(MyToolCallingNode.__id__)
-    )
-
-    function_configs_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
-    )
-
-    assert function_configs_attribute == {
-        "id": "90cc5fc7-9fb3-450f-be5a-e90e7412a601",
-        "name": "function_configs",
-        "value": {
-            "type": "CONSTANT_VALUE",
-            "value": {
-                "type": "JSON",
-                "value": {
-                    "foo": {
-                        "runtime": "PYTHON_3_11_6",
-                        "packages": [
-                            {"version": "1.0.0", "name": "first_package", "repository": "test-repo"},
-                            {"version": "2.0.0", "name": "second_package", "repository": "test-repo"},
-                        ],
-                    },
-                    "bar": {
-                        "runtime": "PYTHON_3_11_6",
-                        "packages": [{"version": "3.0.0", "name": "third_package", "repository": None}],
-                    },
-                },
-            },
-        },
-    }
-
-
-def test_serialize_node__function_configs__none():
-    # GIVEN a tool calling node with no function configs
-    def foo():
-        pass
-
-    class MyToolCallingNode(ToolCallingNode):
-        functions = [foo]
-
-    # AND a workflow with the tool calling node
-    class Workflow(BaseWorkflow):
-        graph = MyToolCallingNode
-
-    # WHEN the workflow is serialized
-    workflow_display = get_workflow_display(workflow_class=Workflow)
-    serialized_workflow: dict = workflow_display.serialize()
-
-    # THEN the node should properly serialize the functions
-    my_tool_calling_node = next(
-        node
-        for node in serialized_workflow["workflow_raw_data"]["nodes"]
-        if node["id"] == str(MyToolCallingNode.__id__)
-    )
-
-    function_configs_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
-    )
-
-    assert function_configs_attribute == {
-        "id": "5f63f2aa-72d7-47ad-a552-630753418b7d",
-        "name": "function_configs",
-        "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-    }

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py

@@ -561,7 +561,7 @@ def test_serialize_workflow__try_wrapped():
             "code_input_id": "f2e8a4fa-b54e-41e9-b314-0e5443519ac7",
             "runtime_input_id": "19d64948-f22b-4103-a7f5-3add184b31cc",
             "output_type": "NUMBER",
-            "packages": [{"name": "openai", "version": "1.0.0"}],
+            "packages": [{"name": "openai", "version": "1.0.0", "repository": "test-repo"}],
             "output_id": "0fde9607-353f-42c2-85c4-20f720ebc1ec",
             "log_output_id": "7cac05e3-b7c3-475e-8df8-422b496c3398",
         },

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py

@@ -48,7 +48,7 @@ def test_serialize_workflow():
         },
         "base": {
             "name": "ToolCallingNode",
-            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+            "module": ["vellum", "workflows", "nodes", "displayable", "tool_calling_node", "node"],
         },
         "definition": {
             "name": "GetCurrentWeatherNode",
@@ -407,11 +407,6 @@ def test_serialize_workflow():
                     ],
                 },
             },
-            {
-                "id": "a4e3bc9f-7112-4d2f-94fb-7362a85db27a",
-                "name": "function_configs",
-                "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-            },
             {
                 "id": "1668419e-a193-43a5-8a97-3394e89bf278",
                 "name": "max_prompt_iterations",

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py

@@ -48,7 +48,7 @@ def test_serialize_workflow():
         },
         "base": {
             "name": "ToolCallingNode",
-            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+            "module": ["vellum", "workflows", "nodes", "displayable", "tool_calling_node", "node"],
         },
         "definition": {
             "name": "GetCurrentWeatherNode",
@@ -152,22 +152,6 @@ def test_serialize_workflow():
                     },
                 },
             },
-            {
-                "id": "a4e3bc9f-7112-4d2f-94fb-7362a85db27a",
-                "name": "function_configs",
-                "value": {
-                    "type": "CONSTANT_VALUE",
-                    "value": {
-                        "type": "JSON",
-                        "value": {
-                            "get_current_weather": {
-                                "runtime": "PYTHON_3_11_6",
-                                "packages": [{"version": "2.26.0", "name": "requests", "repository": None}],
-                            }
-                        },
-                    },
-                },
-            },
             {
                 "id": "0f6dc102-3460-4963-91fa-7ba85d65ef7a",
                 "name": "prompt_inputs",