vellum-ai 1.3.9__py3-none-any.whl → 1.3.11__py3-none-any.whl

vellum/client/core/client_wrapper.py

@@ -27,10 +27,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.3.9",
+            "User-Agent": "vellum-ai/1.3.11",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.3.9",
+            "X-Fern-SDK-Version": "1.3.11",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None:

vellum/client/tests/test_utils.py

@@ -0,0 +1,34 @@
+import uuid
+from unittest.mock import Mock
+from vellum.client.utils import convert_input_variable_to_uuid
+
+
+def test_convert_input_variable_to_uuid_with_valid_uuid():
+    """Test convert_input_variable_to_uuid with valid UUID passes through unchanged."""
+    uuid_value = "123e4567-e89b-12d3-a456-426614174000"
+
+    # GIVEN a valid UUID
+    # WHEN the function is called
+    result = convert_input_variable_to_uuid(uuid_value)
+
+    # THEN the result should be the original UUID unchanged
+    assert result == uuid_value
+
+
+def test_convert_input_variable_to_uuid_with_invalid_uuid_converts():
+    """Test convert_input_variable_to_uuid with invalid UUID gets converted to UUID."""
+    non_uuid_value = "some_variable_name"
+    executable_id = "test_executable_123"
+
+    # GIVEN an invalid UUID with context
+    mock_info = Mock()
+    mock_info.context = {"executable_id": executable_id}
+
+    # WHEN the function is called
+    result = convert_input_variable_to_uuid(non_uuid_value, mock_info)
+
+    # THEN it should return a different value
+    assert result != non_uuid_value
+
+    # AND it's a valid UUID
+    uuid.UUID(result)  # This will raise ValueError if not a valid UUID

vellum/client/types/variable_prompt_block.py

@@ -3,6 +3,7 @@
 import typing
 
 import pydantic
+from vellum.client.utils import convert_input_variable_to_uuid
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
 from .prompt_block_state import PromptBlockState
@@ -26,3 +27,9 @@ class VariablePromptBlock(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+    @pydantic.field_serializer("input_variable")
+    def serialize_input_variable(self, value: str, info: pydantic.SerializationInfo) -> str:
+        """Convert input_variable to UUID using executable_id from context."""
+        return convert_input_variable_to_uuid(value, info)

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/workflows/__init__.py

@@ -1,5 +1,7 @@
+from .inputs.dataset_row import DatasetRow
 from .workflows import BaseWorkflow
 
 __all__ = [
     "BaseWorkflow",
+    "DatasetRow",
 ]

vellum/workflows/errors/types.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from enum import Enum
 import logging
-from typing import Any, Dict
+from typing import Any, Dict, Optional
 
 from vellum.client.types.vellum_error import VellumError
 from vellum.client.types.vellum_error_code_enum import VellumErrorCodeEnum
@@ -30,6 +30,7 @@ class WorkflowErrorCode(Enum):
 class WorkflowError:
     message: str
     code: WorkflowErrorCode
+    raw_data: Optional[Dict[str, Any]] = None
 
     def __contains__(self, item: Any) -> bool:
         return item in self.message
@@ -39,6 +40,7 @@ _VELLUM_ERROR_CODE_TO_WORKFLOW_ERROR_CODE: Dict[VellumErrorCodeEnum, WorkflowErr
     "INVALID_REQUEST": WorkflowErrorCode.INVALID_INPUTS,
     "INVALID_INPUTS": WorkflowErrorCode.INVALID_INPUTS,
     "PROVIDER_ERROR": WorkflowErrorCode.PROVIDER_ERROR,
+    "PROVIDER_CREDENTIALS_UNAVAILABLE": WorkflowErrorCode.PROVIDER_CREDENTIALS_UNAVAILABLE,
     "REQUEST_TIMEOUT": WorkflowErrorCode.PROVIDER_ERROR,
     "INTERNAL_SERVER_ERROR": WorkflowErrorCode.INTERNAL_ERROR,
     "USER_DEFINED_ERROR": WorkflowErrorCode.USER_DEFINED_ERROR,
@@ -55,6 +57,7 @@ def vellum_error_to_workflow_error(error: VellumError) -> WorkflowError:
     return WorkflowError(
         message=error.message,
         code=workflow_error_code,
+        raw_data=error.raw_data or {},
     )
 
 

vellum/workflows/events/tests/test_event.py

@@ -260,6 +260,7 @@ mock_node_uuid = str(uuid4_from_hash(MockNode.__qualname__))
                     "error": {
                         "message": "Workflow failed",
                         "code": "USER_DEFINED_ERROR",
+                        "raw_data": None,
                     },
                     "stacktrace": None,
                 },

vellum/workflows/exceptions.py

@@ -1,10 +1,18 @@
+from typing import Any, Dict, Optional
+
 from vellum.workflows.errors import WorkflowError, WorkflowErrorCode
 
 
 class NodeException(Exception):
-    def __init__(self, message: str, code: WorkflowErrorCode = WorkflowErrorCode.INTERNAL_ERROR):
+    def __init__(
+        self,
+        message: str,
+        code: WorkflowErrorCode = WorkflowErrorCode.INTERNAL_ERROR,
+        raw_data: Optional[Dict[str, Any]] = None,
+    ):
         self.message = message
         self.code = code
+        self.raw_data = raw_data
         super().__init__(message)
 
     @property
@@ -12,11 +20,12 @@ class NodeException(Exception):
         return WorkflowError(
             message=self.message,
             code=self.code,
+            raw_data=self.raw_data,
         )
 
     @staticmethod
     def of(workflow_error: WorkflowError) -> "NodeException":
-        return NodeException(message=workflow_error.message, code=workflow_error.code)
+        return NodeException(message=workflow_error.message, code=workflow_error.code, raw_data=workflow_error.raw_data)
 
 
 class WorkflowInitializationException(Exception):

vellum/workflows/inputs/__init__.py

@@ -1,5 +1,7 @@
 from .base import BaseInputs
+from .dataset_row import DatasetRow
 
 __all__ = [
     "BaseInputs",
+    "DatasetRow",
 ]

vellum/workflows/inputs/dataset_row.py

@@ -0,0 +1,38 @@
+from typing import Any, Dict
+
+from pydantic import field_serializer
+
+from vellum.client.core.pydantic_utilities import UniversalBaseModel
+from vellum.workflows.inputs.base import BaseInputs
+
+
+class DatasetRow(UniversalBaseModel):
+    """
+    Universal base model representing a dataset row with a label and inputs.
+
+    Attributes:
+        label: String label for the dataset row
+        inputs: BaseInputs instance containing the input data
+    """
+
+    label: str
+    inputs: BaseInputs
+
+    @field_serializer("inputs")
+    def serialize_inputs(self, inputs: BaseInputs) -> Dict[str, Any]:
+        """
+        Custom serializer for BaseInputs that converts it to a dictionary.
+
+        Args:
+            inputs: BaseInputs instance to serialize
+
+        Returns:
+            Dictionary representation of the inputs
+        """
+        result = {}
+
+        for input_descriptor, value in inputs:
+            if not input_descriptor.name.startswith("__"):
+                result[input_descriptor.name] = value
+
+        return result

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

@@ -45,12 +45,14 @@ from vellum.workflows.nodes.displayable.bases.base_prompt_node import BasePrompt
 from vellum.workflows.nodes.displayable.bases.utils import process_additional_prompt_outputs
 from vellum.workflows.outputs import BaseOutput
 from vellum.workflows.types import MergeBehavior
-from vellum.workflows.types.definition import DeploymentDefinition
+from vellum.workflows.types.definition import DeploymentDefinition, MCPServer
 from vellum.workflows.types.generics import StateType, is_workflow_class
 from vellum.workflows.utils.functions import (
     compile_function_definition,
     compile_inline_workflow_function_definition,
+    compile_mcp_tool_definition,
     compile_workflow_deployment_function_definition,
+    get_mcp_tool_name,
 )
 from vellum.workflows.utils.pydantic_schema import normalize_json
 
@@ -140,6 +142,16 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
                     normalized_functions.append(compile_inline_workflow_function_definition(function))
                 elif callable(function):
                     normalized_functions.append(compile_function_definition(function))
+                elif isinstance(function, MCPServer):
+                    tool_definitions = compile_mcp_tool_definition(function)
+                    for tool_def in tool_definitions:
+                        normalized_functions.append(
+                            FunctionDefinition(
+                                name=get_mcp_tool_name(tool_def),
+                                description=tool_def.description,
+                                parameters=tool_def.parameters,
+                            )
+                        )
                 else:
                     raise NodeException(
                         message=f"`{function}` is not a valid function definition",

vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py

@@ -114,6 +114,7 @@ def test_inline_text_prompt_node__catch_provider_error(vellum_adhoc_prompt_clien
     expected_error = VellumError(
         message="OpenAI failed",
         code="PROVIDER_ERROR",
+        raw_data={"type": "ERROR", "error": {"message": "Something went wrong"}},
     )
 
     def generate_prompt_events(*args: Any, **kwargs: Any) -> Iterator[ExecutePromptEvent]:
@@ -144,6 +145,7 @@ def test_inline_text_prompt_node__catch_provider_error(vellum_adhoc_prompt_clien
             value=SdkVellumError(
                 message="OpenAI failed",
                 code=WorkflowErrorCode.PROVIDER_ERROR,
+                raw_data={"type": "ERROR", "error": {"message": "Something went wrong"}},
             ),
         )
         in outputs

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

@@ -18,14 +18,13 @@ from vellum.workflows.nodes.displayable.tool_calling_node.utils import (
     create_router_node,
     create_tool_prompt_node,
     get_function_name,
-    get_mcp_tool_name,
-    hydrate_mcp_tool_definitions,
 )
 from vellum.workflows.outputs.base import BaseOutput, BaseOutputs
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.types.core import EntityInputsInterface, Tool
 from vellum.workflows.types.definition import MCPServer
 from vellum.workflows.types.generics import StateType
+from vellum.workflows.utils.functions import compile_mcp_tool_definition, get_mcp_tool_name
 from vellum.workflows.workflows.event_filters import all_workflow_event_filter
 
 
@@ -162,7 +161,7 @@ class ToolCallingNode(BaseNode[StateType], Generic[StateType]):
         self._function_nodes = {}
         for function in self.functions:
             if isinstance(function, MCPServer):
-                tool_definitions = hydrate_mcp_tool_definitions(function)
+                tool_definitions = compile_mcp_tool_definition(function)
                 for tool_definition in tool_definitions:
                     function_name = get_mcp_tool_name(tool_definition)
 

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

@@ -33,6 +33,7 @@ from vellum.workflows.state.encoder import DefaultStateEncoder
 from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior, Tool, ToolBase
 from vellum.workflows.types.definition import ComposioToolDefinition, DeploymentDefinition, MCPServer, MCPToolDefinition
 from vellum.workflows.types.generics import is_workflow_class
+from vellum.workflows.utils.functions import compile_mcp_tool_definition, get_mcp_tool_name
 
 CHAT_HISTORY_VARIABLE = "chat_history"
 
@@ -303,26 +304,6 @@ def _hydrate_composio_tool_definition(tool_def: ComposioToolDefinition) -> Funct
         )
 
 
-def hydrate_mcp_tool_definitions(server_def: MCPServer) -> List[MCPToolDefinition]:
-    """Hydrate an MCPToolDefinition with detailed information from the MCP server.
-
-    We do tool discovery on the MCP server to get the tool definitions.
-
-    Args:
-        tool_def: The basic MCPToolDefinition to enhance
-
-    Returns:
-        MCPToolDefinition with detailed parameters and description
-    """
-    try:
-        mcp_service = MCPService()
-        return mcp_service.hydrate_tool_definitions(server_def)
-    except Exception as e:
-        # If hydration fails, log and return original
-        logger.warning(f"Failed to enhance MCP server '{server_def.name}': {e}")
-        return []
-
-
 def create_tool_prompt_node(
     ml_model: str,
     blocks: List[Union[PromptBlock, Dict[str, Any]]],
@@ -341,17 +322,6 @@ def create_tool_prompt_node(
                 # Get Composio tool details and hydrate the function definition
                 enhanced_function = _hydrate_composio_tool_definition(function)
                 prompt_functions.append(enhanced_function)
-            elif isinstance(function, MCPServer):
-                tool_functions: List[MCPToolDefinition] = hydrate_mcp_tool_definitions(function)
-                for tool_function in tool_functions:
-                    name = get_mcp_tool_name(tool_function)
-                    prompt_functions.append(
-                        FunctionDefinition(
-                            name=name,
-                            description=tool_function.description,
-                            parameters=tool_function.parameters,
-                        )
-                    )
             else:
                 prompt_functions.append(function)
     else:
@@ -440,7 +410,7 @@ def create_router_node(
                 port = create_port_condition(function_name)
                 setattr(Ports, function_name, port)
             elif isinstance(function, MCPServer):
-                tool_functions: List[MCPToolDefinition] = hydrate_mcp_tool_definitions(function)
+                tool_functions: List[MCPToolDefinition] = compile_mcp_tool_definition(function)
                 for tool_function in tool_functions:
                     name = get_mcp_tool_name(tool_function)
                     port = create_port_condition(name)
@@ -604,8 +574,3 @@ def get_function_name(function: ToolBase) -> str:
         return function.name  # type: ignore[return-value]
     else:
         return snake_case(function.__name__)
-
-
-def get_mcp_tool_name(tool_def: MCPToolDefinition) -> str:
-    server_name = snake_case(tool_def.server.name)
-    return f"{server_name}__{tool_def.name}"

vellum/workflows/sandbox.py

@@ -1,9 +1,10 @@
-from typing import Generic, Optional, Sequence
+from typing import Generic, Optional, Sequence, Union
 
 import dotenv
 
 from vellum.workflows.events.workflow import WorkflowEventStream
 from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.inputs.dataset_row import DatasetRow
 from vellum.workflows.logging import load_logger
 from vellum.workflows.types.generics import WorkflowType
 from vellum.workflows.workflows.event_filters import root_workflow_event_filter
@@ -14,7 +15,7 @@ class WorkflowSandboxRunner(Generic[WorkflowType]):
         self,
         workflow: WorkflowType,
         inputs: Optional[Sequence[BaseInputs]] = None,  # DEPRECATED - remove in v2.0.0
-        dataset: Optional[Sequence[BaseInputs]] = None,
+        dataset: Optional[Sequence[Union[BaseInputs, DatasetRow]]] = None,
     ):
         dotenv.load_dotenv()
         self._logger = load_logger()
@@ -50,6 +51,9 @@ class WorkflowSandboxRunner(Generic[WorkflowType]):
 
         selected_inputs = self._inputs[index]
 
+        if isinstance(selected_inputs, DatasetRow):
+            selected_inputs = selected_inputs.inputs
+
         events = self._workflow.stream(
             inputs=selected_inputs,
             event_filter=root_workflow_event_filter,

vellum/workflows/tests/test_dataset_row.py

@@ -0,0 +1,99 @@
+from typing import Optional
+
+from vellum.client.types.chat_message import ChatMessage
+from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.inputs.dataset_row import DatasetRow
+
+
+def test_dataset_row_serialization():
+    """
+    Test that DatasetRow can be properly serialized to JSON and back.
+    """
+
+    class TestInputs(BaseInputs):
+        message: str
+        count: int
+        optional_field: Optional[str] = None
+        chat_history: list[ChatMessage]
+
+    test_inputs = TestInputs(
+        message="Hello World",
+        count=42,
+        optional_field="test",
+        chat_history=[ChatMessage(text="Hello", role="USER"), ChatMessage(text="Hi there!", role="ASSISTANT")],
+    )
+    dataset_row = DatasetRow(label="test_label", inputs=test_inputs)
+
+    serialized_dict = dataset_row.model_dump()
+
+    assert "label" in serialized_dict
+    assert "inputs" in serialized_dict
+    assert serialized_dict["label"] == "test_label"
+
+    inputs_data = serialized_dict["inputs"]
+    assert inputs_data["message"] == "Hello World"
+    assert inputs_data["count"] == 42
+    assert inputs_data["optional_field"] == "test"
+    assert "chat_history" in inputs_data
+    assert len(inputs_data["chat_history"]) == 2
+    assert inputs_data["chat_history"][0]["text"] == "Hello"
+    assert inputs_data["chat_history"][0]["role"] == "USER"
+    assert inputs_data["chat_history"][1]["text"] == "Hi there!"
+    assert inputs_data["chat_history"][1]["role"] == "ASSISTANT"
+
+
+def test_dataset_row_dict_serialization():
+    """
+    Test that DatasetRow can be properly converted to dict.
+    """
+
+    class SimpleInputs(BaseInputs):
+        text: str
+
+    simple_inputs = SimpleInputs(text="sample text")
+    dataset_row = DatasetRow(label="simple_label", inputs=simple_inputs)
+
+    result_dict = dataset_row.model_dump()
+
+    assert result_dict["label"] == "simple_label"
+    assert result_dict["inputs"]["text"] == "sample text"
+
+
+def test_dataset_row_with_multiple_fields():
+    """
+    Test that DatasetRow works with BaseInputs that have multiple fields.
+    """
+
+    class MultiFieldInputs(BaseInputs):
+        text_field: str
+        number_field: int
+        optional_field: Optional[str] = None
+
+    multi_inputs = MultiFieldInputs(text_field="test_text", number_field=456, optional_field="optional_value")
+    dataset_row = DatasetRow(label="multi_field_test", inputs=multi_inputs)
+
+    result_dict = dataset_row.model_dump()
+
+    assert result_dict["label"] == "multi_field_test"
+    assert result_dict["inputs"]["text_field"] == "test_text"
+    assert result_dict["inputs"]["number_field"] == 456
+    assert result_dict["inputs"]["optional_field"] == "optional_value"
+
+
+def test_dataset_row_with_default_inputs():
+    """
+    Test that DatasetRow works with BaseInputs that have default values.
+    """
+
+    class InputsWithDefaults(BaseInputs):
+        required_field: str
+        optional_with_default: str = "default_value"
+
+    inputs_with_defaults = InputsWithDefaults(required_field="required_value")
+    dataset_row = DatasetRow(label="defaults_test", inputs=inputs_with_defaults)
+
+    serialized_dict = dataset_row.model_dump()
+
+    assert serialized_dict["label"] == "defaults_test"
+    assert serialized_dict["inputs"]["required_field"] == "required_value"
+    assert serialized_dict["inputs"]["optional_with_default"] == "default_value"

vellum/workflows/types/definition.py

@@ -2,12 +2,14 @@ import importlib
 import inspect
 from types import FrameType
 from uuid import UUID
-from typing import Annotated, Any, Dict, Literal, Optional, Union
+from typing import Annotated, Any, Dict, List, Literal, Optional, Union
 
-from pydantic import BeforeValidator
+from pydantic import BeforeValidator, SerializationInfo, model_serializer
 
+from vellum import Vellum
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
 from vellum.client.types.code_resource_definition import CodeResourceDefinition as ClientCodeResourceDefinition
+from vellum.client.types.vellum_variable import VellumVariable
 from vellum.workflows.constants import AuthorizationType
 from vellum.workflows.references.environment_variable import EnvironmentVariableReference
 
@@ -78,6 +80,11 @@ class DeploymentDefinition(UniversalBaseModel):
     deployment: str
     release_tag: str = "LATEST"
 
+    # hydrated fields
+    name: Optional[str] = None
+    description: Optional[str] = None
+    input_variables: Optional[List[VellumVariable]] = None
+
     def _is_uuid(self) -> bool:
         """Check if the deployment field is a valid UUID."""
         try:
@@ -100,6 +107,46 @@ class DeploymentDefinition(UniversalBaseModel):
             return self.deployment
         return None
 
+    def get_release_info(self, client: Vellum):
+        try:
+            release = client.workflow_deployments.retrieve_workflow_deployment_release(
+                self.deployment, self.release_tag
+            )
+        except Exception:
+            # If we fail to get the release info, we'll use the deployment name and description
+            return {
+                "name": self.deployment,
+                "description": f"Workflow Deployment for {self.deployment}",
+                "input_variables": [],
+            }
+
+        return {
+            "name": release.deployment.name,
+            "description": release.description or f"Workflow Deployment for {self.deployment}",
+            "input_variables": release.workflow_version.input_variables,
+        }
+
+    @model_serializer(mode="wrap")
+    def _serialize(self, handler, info: SerializationInfo):
+        """Allow Pydantic to serialize directly given a `client` in context.
+
+        Falls back to the default serialization when client is not provided.
+        """
+        context = info.context if info and hasattr(info, "context") else {}
+        client: Optional[Vellum] = context.get("client") if context else None
+
+        if client:
+            release_info = self.get_release_info(client)
+            return {
+                "type": "WORKFLOW_DEPLOYMENT",
+                "name": release_info["name"],
+                "description": release_info["description"],
+                "deployment": self.deployment,
+                "release_tag": self.release_tag,
+            }
+
+        return handler(self)
+
 
 class ComposioToolDefinition(UniversalBaseModel):
     """Represents a specific Composio action that can be used in Tool Calling Node"""

vellum/workflows/utils/functions.py

@@ -1,6 +1,19 @@
 import dataclasses
 import inspect
-from typing import TYPE_CHECKING, Annotated, Any, Callable, Dict, Literal, Optional, Type, Union, get_args, get_origin
+from typing import (
+    TYPE_CHECKING,
+    Annotated,
+    Any,
+    Callable,
+    Dict,
+    List,
+    Literal,
+    Optional,
+    Type,
+    Union,
+    get_args,
+    get_origin,
+)
 
 from pydantic import BaseModel
 from pydantic_core import PydanticUndefined
@@ -8,6 +21,8 @@ from pydash import snake_case
 
 from vellum import Vellum
 from vellum.client.types.function_definition import FunctionDefinition
+from vellum.workflows.integrations.mcp_service import MCPService
+from vellum.workflows.types.definition import MCPServer, MCPToolDefinition
 from vellum.workflows.utils.vellum_variables import vellum_variable_type_to_openapi_type
 
 if TYPE_CHECKING:
@@ -261,6 +276,30 @@ def compile_workflow_deployment_function_definition(
     )
 
 
+def get_mcp_tool_name(tool_def: MCPToolDefinition) -> str:
+    """Generate a unique name for an MCP tool by combining server and tool names."""
+    server_name = snake_case(tool_def.server.name)
+    return f"{server_name}__{tool_def.name}"
+
+
+def compile_mcp_tool_definition(server_def: MCPServer) -> List[MCPToolDefinition]:
+    """Hydrate an MCPToolDefinition with detailed information from the MCP server.
+
+    We do tool discovery on the MCP server to get the tool definitions.
+
+    Args:
+        tool_def: The basic MCPToolDefinition to enhance
+
+    Returns:
+        MCPToolDefinition with detailed parameters and description
+    """
+    try:
+        mcp_service = MCPService()
+        return mcp_service.hydrate_tool_definitions(server_def)
+    except Exception:
+        return []
+
+
 def use_tool_inputs(**inputs):
     """
     Decorator to specify which parameters of a tool function should be provided

{vellum_ai-1.3.9.dist-info → vellum_ai-1.3.11.dist-info}/METADATA

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

{vellum_ai-1.3.9.dist-info → vellum_ai-1.3.11.dist-info}/RECORD

@@ -98,10 +98,10 @@ vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_templating_node_serialization.py,sha256=ddPa8gNBYH2tWk92ymngY7M8n74J-8CEre50HISP_-g,7877
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_terminal_node_serialization.py,sha256=A7Ef8P1-Nyvsb97bumKT9W2R1LuZaY9IKFV-7iRueog,4010
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_composio_serialization.py,sha256=oVXCjkU0G56QJmqnd_xIwF3D9bhJwALFibM2wmRhwUk,3739
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py,sha256=1GiF3XgJ85vE05_nZFo66snNuqTt0p1c22gNI9c_BXY,26442
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py,sha256=UD1HU8aEAXSo43JGhjBGE2FGsknHj_zOsIHTo2TLEW0,26470
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_mcp_serialization.py,sha256=QhQbijeCnFeX1i3SMjHJg2WVAEt5JEO3dhFRv-mofdA,2458
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_parent_input.py,sha256=__LX4cuzbyZp_1wc-SI8X_J0tnhOkCEmRVUWLKI5aQM,4578
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py,sha256=VB5c6U0GRUxPm9FbzgKYIpoZfoIU9szAWUQJ5L-4Lug,10187
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py,sha256=tyRlb6ozcsvkeHOC9dsHc4jg3HrHQITDGadi8qbNGns,10215
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_workflow_deployment_serialization.py,sha256=XIZZr5POo2NLn2uEWm9EC3rejeBMoO4X-JtzTH6mvp4,4074
 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=exT7U-axwtYgFylagScflSQLJEND51qIAx2UATju6JM,6023
@@ -113,7 +113,7 @@ vellum_ee/workflows/display/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 vellum_ee/workflows/display/utils/auto_layout.py,sha256=f4GiLn_LazweupfqTpubcdtdfE_vrOcmZudSsnYIY9E,3906
 vellum_ee/workflows/display/utils/events.py,sha256=DE33uoKW78BZtITJ6L22dMZN3KR1BuZBVC98C_gIyzU,1943
 vellum_ee/workflows/display/utils/exceptions.py,sha256=LSwwxCYNxFkf5XMUcFkaZKpQ13OSrI7y_bpEUwbKVk0,169
-vellum_ee/workflows/display/utils/expressions.py,sha256=UoZKiZy2Rhx5Hp2e2J2ipSdQCIQfkAOL_tIxHqLCrAM,20299
+vellum_ee/workflows/display/utils/expressions.py,sha256=K_9CVZJkwkYtLEPwaOA3Sb0wMmi0KbmJUGNkpXsRSbE,19506
 vellum_ee/workflows/display/utils/registry.py,sha256=1qXiBTdsnro6FeCX0FGBEK7CIf6wa--Jt50iZ_nEp_M,3460
 vellum_ee/workflows/display/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/utils/tests/test_auto_layout.py,sha256=vfXI769418s9vda5Gb5NFBH747WMOwSgHRXeLCTLVm8,2356
@@ -155,7 +155,7 @@ vellum/client/README.md,sha256=flqu57ubZNTfpq60CdLtJC9gp4WEkyjb_n_eZ4OYf9w,6497
 vellum/client/__init__.py,sha256=T5Ht_w-Mk_9nzGqdadhQB8V20M0vYj7am06ut0A3P1o,73401
 vellum/client/core/__init__.py,sha256=lTcqUPXcx4112yLDd70RAPeqq6tu3eFMe1pKOqkW9JQ,1562
 vellum/client/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-vellum/client/core/client_wrapper.py,sha256=yiW8T6LXnqj5RVczNfHBZ2L3TMA_MWurdCwV6hFZPw8,2840
+vellum/client/core/client_wrapper.py,sha256=jUohtsg67ju5RCuUUwRLdh1g44h86JtrmqbDsIJDNts,2842
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/force_multipart.py,sha256=awxh5MtcRYe74ehY8U76jzv6fYM_w_D3Rur7KQQzSDk,429
@@ -250,6 +250,8 @@ vellum/client/resources/workspace_secrets/raw_client.py,sha256=ZfiNd1NisolmK07QP
 vellum/client/resources/workspaces/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 vellum/client/resources/workspaces/client.py,sha256=36KYa2FDu6h65q2GscUFOJs4qKeiOA6grOYoCc0Gi3E,2936
 vellum/client/resources/workspaces/raw_client.py,sha256=M3Ewk1ZfEZ44EeTvBtBNoNKi5whwfLY-1GR07SyfDTI,3517
+vellum/client/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vellum/client/tests/test_utils.py,sha256=zk8z45-2xrm9sZ2hq8PTqY8MXmXtPqMqYK0VBBX0GHg,1176
 vellum/client/types/__init__.py,sha256=ZzAl8sIsRT3hkRaCO_tlR9fjgVIM7sQCxmLO_S5WrIo,71915
 vellum/client/types/ad_hoc_execute_prompt_event.py,sha256=B69EesIH6fpNsdoiJaSG9zF1Sl17FnjoTu4CBkUSoHk,608
 vellum/client/types/ad_hoc_expand_meta.py,sha256=Kajcj3dKKed5e7uZibRnE3ZonK_bB2JPM-3aLjLfUp4,1295
@@ -794,7 +796,7 @@ vellum/client/types/token_overlapping_window_chunking_request.py,sha256=ez-hqu3q
 vellum/client/types/unit_enum.py,sha256=BKWRVp2WfHtGK4D6TsolhNJHGHfExzrRHkFn8H8QkwQ,113
 vellum/client/types/upload_document_response.py,sha256=C-JP0uAraD8waVwCNRM4FRCRglmv78m532gpy2Q9_Oc,613
 vellum/client/types/upsert_test_suite_test_case_request.py,sha256=VZGW1_Bw5cvU7RDg7MhT3A9C3ftyeYnAJaDMiil1FuQ,1978
-vellum/client/types/variable_prompt_block.py,sha256=417cXRq9V33iWTDWiX941VzOwhzfHvUHYPjgUev9Kf0,910
+vellum/client/types/variable_prompt_block.py,sha256=sGLrhpDIj7ThxYZXvE3DsMGdQn-F2uoZ_GfTn-pMKjE,1255
 vellum/client/types/vellum_audio.py,sha256=rgL5GPBKDYk0PPQX_2XkerVERBxra9OR3k_PP57ruTc,685
 vellum/client/types/vellum_audio_request.py,sha256=vtkd49BVlLrY9UF3Yk52P5sDbtdY7sY9_XeMBU_VDm0,692
 vellum/client/types/vellum_code_resource_definition.py,sha256=XdueTR342BDjevZ3ktJJI99RqRED4A5SUOyzPt2K6us,661
@@ -915,6 +917,7 @@ vellum/client/types/workflow_sandbox_parent_context.py,sha256=8FJ8vq7CVTg-OnOXlb
 vellum/client/types/workflow_stream_event.py,sha256=6dQx_D-UPfJVrIsSW6krmZKDKeP9kPojFHDgqy_58io,362
 vellum/client/types/workspace_read.py,sha256=NAVouD25ZHvldLrAvAOwL2w2f1tqZaSC05wZ--YvBCQ,661
 vellum/client/types/workspace_secret_read.py,sha256=qWabw1u5HZpuv77kAwtAQigj4CDB41csJ2wmXu5eJS8,678
+vellum/client/utils.py,sha256=b_ajWqSrXigPbUVOGTdzy_gL0iQo9r1OMUH8quuhjuA,885
 vellum/core/__init__.py,sha256=Iph1pJ7wkjHt7Xh1vUeLgMqFCNa0GYjtu3BhG8EhV8Y,136
 vellum/core/api_error.py,sha256=GDjkxQb8k4HqTthWpqwIVE7hLVXJQmRT8sip2mzB-8I,146
 vellum/core/client_wrapper.py,sha256=d3MqLPTOfMq8-AaHg3S7BkBczhyrbdpnkhR0FtOvV8A,151
@@ -1712,7 +1715,7 @@ vellum/utils/typing.py,sha256=wx_daFqD69cYkuJTVnvNrpjhqC3uuhbnyJ9_bIwC9OU,327
 vellum/utils/uuid.py,sha256=Ch6wWRgwICxLxJCTl5iE3EdRlZj2zADR-zUMUtjcMWM,214
 vellum/version.py,sha256=jq-1PlAYxN9AXuaZqbYk9ak27SgE2lw9Ia5gx1b1gVI,76
 vellum/workflows/README.md,sha256=hZdTKBIcsTKPofK68oPkBhyt0nnRh0csqC12k4FMHHA,3597
-vellum/workflows/__init__.py,sha256=CssPsbNvN6rDhoLuqpEv7MMKGa51vE6dvAh6U31Pcio,71
+vellum/workflows/__init__.py,sha256=gd5AiZqVTcvqelhysG0jOWYfC6pJKRAVhS7qwf0bHU4,132
 vellum/workflows/constants.py,sha256=xweiPRUSVEnGz9BJvpIWu96Gfok89QneARu4K7wj7f8,1358
 vellum/workflows/context.py,sha256=ViyIeMDhUv-MhnynLaXPlvlbYxRU45ySvYidCNSbFZU,2458
 vellum/workflows/descriptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1728,7 +1731,7 @@ vellum/workflows/emitters/vellum_emitter.py,sha256=ECBIRA48WS5rIJd1iWUfye7B5Up7u
 vellum/workflows/environment/__init__.py,sha256=TJz0m9dwIs6YOwCTeuN0HHsU-ecyjc1OJXx4AFy83EQ,121
 vellum/workflows/environment/environment.py,sha256=Ck3RPKXJvtMGx_toqYQQQF-ZwXm5ijVwJpEPTeIJ4_Q,471
 vellum/workflows/errors/__init__.py,sha256=tWGPu5xyAU8gRb8_bl0fL7OfU3wxQ9UH6qVwy4X4P_Q,113
-vellum/workflows/errors/types.py,sha256=nUWuniEfrhdtb-_2GzoDGlYnSJ_yuNUGjVkaKLNr-rM,4049
+vellum/workflows/errors/types.py,sha256=1LQzsEwCL-kqLGUxZcgJWahL-XNfIOwCz_5f_fWzLrM,4236
 vellum/workflows/events/__init__.py,sha256=V4mh766fyA70WvHelm9kfVZGrUgEKcJ9tJt8EepfQYU,832
 vellum/workflows/events/context.py,sha256=vCfMIPmz4j9Om36rRWa35A_JU_VccWWS52_mZkkqxak,3345
 vellum/workflows/events/node.py,sha256=yHVd-rX2E3qc2XLnZr0fW6uq4ZCMm34mnY2tzYceyOg,5884
@@ -1736,10 +1739,10 @@ vellum/workflows/events/relational_threads.py,sha256=zmLrBCBYpdpQV0snKH3HleST-_h
 vellum/workflows/events/stream.py,sha256=xhXJTZirFi0xad5neAQNogrIQ4h47fpnKbVC3vCM5Js,889
 vellum/workflows/events/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/events/tests/test_basic_workflow.py,sha256=Pj6orHsXz37jWC5FARi0Sx2Gjf99Owri2Cvr6Chb79k,1765
-vellum/workflows/events/tests/test_event.py,sha256=V26TNmbo2aL4sDvcY3nPxCWgcoS4ejF-7VuesMeD04U,18451
+vellum/workflows/events/tests/test_event.py,sha256=RGVGz0DXMTvwCgdjua8rpPpCuCLDM5-8ET_3q4yH8_8,18493
 vellum/workflows/events/types.py,sha256=mVrqAH9Hs9SpXm08Hcxdyap_ImQPwGsxRr56rSNMP34,5043
 vellum/workflows/events/workflow.py,sha256=kLSWFXiDZH0TELWoDjQ_kHVomFnw8MVVUPDGIzgyosw,8997
-vellum/workflows/exceptions.py,sha256=NiBiR3ggfmPxBVqD-H1SqmjI-7mIn0EStSN1BqApvCM,1213
+vellum/workflows/exceptions.py,sha256=Y7EULLN0kss8jxTLFNWEiOLMnyr1uKG83wiMvmtQ9Ak,1438
 vellum/workflows/executable.py,sha256=um-gLJMVYfGJwGJfZIPlCRHhHIYm6pn8PUEfeqrNx5k,218
 vellum/workflows/expressions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/expressions/accessor.py,sha256=3lu1-_-dBfZdJvtX-q66jbmRAZtqIfdsh_3_JNuzg1E,4462
@@ -1789,8 +1792,9 @@ vellum/workflows/graph/__init__.py,sha256=3sHlay5d_-uD7j3QJXiGl0WHFZZ_QScRvgyDhN
 vellum/workflows/graph/graph.py,sha256=vkpteMc2a61IFGHlrA50J4ntVj6m3agcyWrXGQEbjHc,11280
 vellum/workflows/graph/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/graph/tests/test_graph.py,sha256=0Pov0sCsxjzUDL9wy7xy9jFD-F2GsMJnZVEVFXzQGdM,15433
-vellum/workflows/inputs/__init__.py,sha256=AbFEteIYEvCb14fM3EK7bhM-40-6s494rSlIhQ4Dsss,62
+vellum/workflows/inputs/__init__.py,sha256=02pj0IbJkN1AxTreswK39cNi45tA8GWcAAdRJve4cuM,116
 vellum/workflows/inputs/base.py,sha256=w3owT5B3rLBmIj-v-jL2l-HD4yd3hXK9RmHVd557BpA,5126
+vellum/workflows/inputs/dataset_row.py,sha256=T8lcn9qyC7IY0w3EIfnn4AwZ3pYw9sf4kdIi0VkX_Sw,1033
 vellum/workflows/inputs/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/inputs/tests/test_inputs.py,sha256=lioA8917mFLYq7Ml69UNkqUjcWbbxkxnpIEJ4FBaYBk,2206
 vellum/workflows/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1840,7 +1844,7 @@ vellum/workflows/nodes/displayable/bases/api_node/tests/test_node.py,sha256=5C59
 vellum/workflows/nodes/displayable/bases/base_prompt_node/__init__.py,sha256=Org3xTvgp1pA0uUXFfnJr29D3HzCey2lEdYF4zbIUgo,70
 vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py,sha256=ea20icDM1HB942wkH-XtXNSNCBDcjeOiN3vowkHL4fs,4477
 vellum/workflows/nodes/displayable/bases/inline_prompt_node/__init__.py,sha256=Hl35IAoepRpE-j4cALaXVJIYTYOF3qszyVbxTj4kS1s,82
-vellum/workflows/nodes/displayable/bases/inline_prompt_node/node.py,sha256=TkVfwD5-GGWG32vtGMo4ZjuxWrjVYbK7tDWq0U9OBCM,17316
+vellum/workflows/nodes/displayable/bases/inline_prompt_node/node.py,sha256=eRhqxpkqcP5S4jx-vnnyHZvZAVQxKyCQ9kYxQCyWNvY,17921
 vellum/workflows/nodes/displayable/bases/inline_prompt_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/bases/inline_prompt_node/tests/test_inline_prompt_node.py,sha256=xc53wGwVqxBnN7eoyWkJ-RJ-FeUpHKekkKjViASHAFg,27495
 vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py,sha256=H9mM75EQpP6PUvsXCTbwjw4CqMMLf36m1G2XqiPEvH4,12139
@@ -1889,18 +1893,18 @@ vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py,sha256=ZB
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/test_node.py,sha256=c98nMPogZ6iN_pTvVUMTB3J72Hj--H-XVgvvRXhdSQE,19085
 vellum/workflows/nodes/displayable/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py,sha256=MHuIolSsrY9ziwoXWsye3XOODncL9DLZOkNYzQMLhRw,4696
+vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py,sha256=E6PQ9OKcDlsJbWoxVKWgiAPgz9p59g1ONMgNggfOeiI,4868
 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=Bjv-wZyFgNaVZb9KEMMZd9lFoLzbPEPjEMpANizMZw4,2413
 vellum/workflows/nodes/displayable/tool_calling_node/__init__.py,sha256=3n0-ysmFKsr40CVxPthc0rfJgqVJeZuUEsCmYudLVRg,117
-vellum/workflows/nodes/displayable/tool_calling_node/node.py,sha256=J4RNOggTx5nzovC0354SPGV-NkRpYnV51PMTYQ7aIQ8,8202
+vellum/workflows/nodes/displayable/tool_calling_node/node.py,sha256=Exw0_5-bzphKF42pKwHF-tH25s8MD6wSbGcp7jZuBFg,8236
 vellum/workflows/nodes/displayable/tool_calling_node/state.py,sha256=CcBVb_YtwfSSka4ze678k6-qwmzMSfjfVP8_Y95feSo,302
 vellum/workflows/nodes/displayable/tool_calling_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/tool_calling_node/tests/test_composio_service.py,sha256=in1fbEz5x1tx3uKv9YXdvOncsHucNL8Ro6Go7lBuuOQ,8962
 vellum/workflows/nodes/displayable/tool_calling_node/tests/test_node.py,sha256=GZoeybB9uM7ai8sBLAtUMHrMVgh-WrJDWrKZci6feDs,11892
 vellum/workflows/nodes/displayable/tool_calling_node/tests/test_utils.py,sha256=SIu5GCj4tIE4fz-cAcdULtQfqZIhrcc3Doo6TWLXBws,8804
-vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=c0fJA-de3yJKGTzKfjyZOkbVhIndSAhZqBZp_DpU1fg,24158
+vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=6XsvA77wF9MI6KBQNN1JMdpQwhK8pKUQtXgsPmq_I3Q,22837
 vellum/workflows/nodes/displayable/web_search_node/__init__.py,sha256=8FOnEP-n-U68cvxTlJW9wphIAGHq5aqjzLM-DoSSXnU,61
 vellum/workflows/nodes/displayable/web_search_node/node.py,sha256=NQYux2bOtuBF5E4tn-fXi5y3btURPRrNqMSM9MAZYI4,5091
 vellum/workflows/nodes/displayable/web_search_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1940,7 +1944,7 @@ vellum/workflows/resolvers/tests/test_resolver.py,sha256=jXkJBb9SwtoH__bBN-ECohp
 vellum/workflows/resolvers/types.py,sha256=Hndhlk69g6EKLh_LYg5ILepW5U_h_BYNllfzhS9k8p4,237
 vellum/workflows/runner/__init__.py,sha256=i1iG5sAhtpdsrlvwgH6B-m49JsINkiWyPWs8vyT-bqM,72
 vellum/workflows/runner/runner.py,sha256=lnVbqA1nSdWuyY1SZDpLDvnpLRQcQyWYOWrx3RIJpcg,41043
-vellum/workflows/sandbox.py,sha256=jwlFFQjHDwmbVoBah_Q3i8K_BrzOt-F6TXFauiyVyIk,3021
+vellum/workflows/sandbox.py,sha256=mezSZmilR_fwR8164n8CEfzlMeQ55IqfapHp4ftImvQ,3212
 vellum/workflows/state/__init__.py,sha256=yUUdR-_Vl7UiixNDYQZ-GEM_kJI9dnOia75TtuNEsnE,60
 vellum/workflows/state/base.py,sha256=m9fCqbZn21GshCVCjJTD1dPZEQjFrsMXqlg7tM9fIwM,24283
 vellum/workflows/state/context.py,sha256=khM30U1iDNts5Xp8LXa_WfpkITNITexrDUUFJ5wZ2W4,8445
@@ -1950,12 +1954,13 @@ vellum/workflows/state/store.py,sha256=uVe-oN73KwGV6M6YLhwZMMUQhzTQomsVfVnb8V91g
 vellum/workflows/state/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/state/tests/test_state.py,sha256=zEVFIY2any41X2BA5Us_qqKpzH5HRqmyrUJ04GTO0pU,7484
 vellum/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vellum/workflows/tests/test_dataset_row.py,sha256=qtz3cEcppCtr96co_8MmX4Jh1ro6bKZd_FGtRyFncRA,3370
 vellum/workflows/tests/test_sandbox.py,sha256=JKwaluI-lODQo7Ek9sjDstjL_WTdSqUlVik6ZVTfVOA,1826
 vellum/workflows/tests/test_undefined.py,sha256=zMCVliCXVNLrlC6hEGyOWDnQADJ2g83yc5FIM33zuo8,353
 vellum/workflows/types/__init__.py,sha256=KxUTMBGzuRCfiMqzzsykOeVvrrkaZmTTo1a7SLu8gRM,68
 vellum/workflows/types/code_execution_node_wrappers.py,sha256=fewX9bqF_4TZuK-gZYIn12s31-k03vHMGRpvFAPm11Y,3206
 vellum/workflows/types/core.py,sha256=TggDVs2lVya33xvu374EDhMC1b7RRlAAs0zWLaF46BA,1385
-vellum/workflows/types/definition.py,sha256=2vq3uGT-m994zRcla0yTUsMiPLKSDuzEZs7H6U9QbiE,4993
+vellum/workflows/types/definition.py,sha256=727tDk-XzLAlpK2_OuPICz2VnFLW5WZd-WcZQrd2lvY,6854
 vellum/workflows/types/generics.py,sha256=8jptbEx1fnJV0Lhj0MpCJOT6yNiEWeTOYOwrEAb5CRU,1576
 vellum/workflows/types/stack.py,sha256=h7NE0vXR7l9DevFBIzIAk1Zh59K-kECQtDTKOUunwMY,1314
 vellum/workflows/types/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1963,7 +1968,7 @@ vellum/workflows/types/tests/test_definition.py,sha256=rvDYjdJ1rvAv0qHBN7i7s-_WA
 vellum/workflows/types/tests/test_utils.py,sha256=UnZog59tR577mVwqZRqqWn2fScoOU1H6up0EzS8zYhw,2536
 vellum/workflows/types/utils.py,sha256=mTctHITBybpt4855x32oCKALBEcMNLn-9cCmfEKgJHQ,6498
 vellum/workflows/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/utils/functions.py,sha256=6WRRMb_XbxtvhUKOJq5ZChy0KKvlBaQCBiPhvecXT7I,10029
+vellum/workflows/utils/functions.py,sha256=z1V_HujCZI-pfOLIEwkd0_ZeeR1MNqq3fvgDke62NRo,11016
 vellum/workflows/utils/hmac.py,sha256=JJCczc6pyV6DuE1Oa0QVfYPUN_of3zEYmGFib3OZnrE,1135
 vellum/workflows/utils/names.py,sha256=QtHquoaGqRseu5gg2OcVGI2d_CMcEOvjb9KspwH4C-A,552
 vellum/workflows/utils/pydantic_schema.py,sha256=eR_bBtY-T0pttJP-ARwagSdCOnwPUtiT3cegm2lzDTQ,1310
@@ -1982,8 +1987,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=ptMntHzVyy8ZuzNgeTuk7hREgKQ5UBdgq8VJFSGaW4Y,20832
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-1.3.9.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-1.3.9.dist-info/METADATA,sha256=uOjA8lJ-F41325oLMigUN2TZ7A4Kuyi9d-fsQ6rRRB4,5547
-vellum_ai-1.3.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-1.3.9.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-1.3.9.dist-info/RECORD,,
+vellum_ai-1.3.11.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-1.3.11.dist-info/METADATA,sha256=OXKFjRVt9ymdiuFWO-WdvZ_lv_lWwiNYryNTEqoS054,5548
+vellum_ai-1.3.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-1.3.11.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-1.3.11.dist-info/RECORD,,

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

@@ -114,7 +114,7 @@ def test_serialize_workflow():
                                                 "block_type": "VARIABLE",
                                                 "state": None,
                                                 "cache_config": None,
-                                                "input_variable": "question",
+                                                "input_variable": "8eb8b551-9b48-43b3-861f-52adb5c585a8",
                                             }
                                         ],
                                     }

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

@@ -115,7 +115,7 @@ def test_serialize_workflow():
                                                 "block_type": "VARIABLE",
                                                 "state": None,
                                                 "cache_config": None,
-                                                "input_variable": "question",
+                                                "input_variable": "8eb8b551-9b48-43b3-861f-52adb5c585a8",
                                             }
                                         ],
                                     }

vellum_ee/workflows/display/utils/expressions.py

@@ -51,7 +51,6 @@ from vellum.workflows.references.state_value import StateValueReference
 from vellum.workflows.references.vellum_secret import VellumSecretReference
 from vellum.workflows.references.workflow_input import WorkflowInputReference
 from vellum.workflows.types.core import JsonArray, JsonObject
-from vellum.workflows.types.definition import DeploymentDefinition
 from vellum.workflows.types.generics import is_workflow_class
 from vellum.workflows.utils.functions import compile_function_definition
 from vellum.workflows.utils.uuids import uuid4_from_hash
@@ -425,29 +424,9 @@ def serialize_value(executable_id: UUID, display_context: "WorkflowDisplayContex
             },
         }
 
-    if isinstance(value, DeploymentDefinition):
-        workflow_deployment_release = display_context.client.workflow_deployments.retrieve_workflow_deployment_release(
-            value.deployment, value.release_tag
-        )
-        name = workflow_deployment_release.deployment.name or value.deployment
-        description = workflow_deployment_release.description or f"Workflow Deployment for {name}"
-
-        return {
-            "type": "CONSTANT_VALUE",
-            "value": {
-                "type": "JSON",
-                "value": {
-                    "type": "WORKFLOW_DEPLOYMENT",
-                    "name": name,
-                    "description": description,
-                    "deployment": value.deployment,
-                    "release_tag": value.release_tag,
-                },
-            },
-        }
-
     if isinstance(value, BaseModel):
-        dict_value = value.model_dump()
+        context = {"executable_id": executable_id, "client": display_context.client}
+        dict_value = value.model_dump(context=context)
         return serialize_value(executable_id, display_context, dict_value)
 
     if callable(value):