vellum-ai 0.14.15__py3-none-any.whl → 0.14.17__py3-none-any.whl

vellum/__init__.py

@@ -296,6 +296,7 @@ from .types import (
     RejectedExecuteWorkflowWorkflowResultEvent,
     RejectedPromptExecutionMeta,
     RejectedWorkflowNodeResultEvent,
+    Release,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
     RichTextChildBlock,
@@ -844,6 +845,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
+    "Release",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",

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.15",
+            "X-Fern-SDK-Version": "0.14.17",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/client/resources/document_indexes/client.py

@@ -10,7 +10,6 @@ from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...types.document_index_indexing_config_request import DocumentIndexIndexingConfigRequest
 from ...types.entity_status import EntityStatus
-from ...types.environment_enum import EnvironmentEnum
 from ...types.document_index_read import DocumentIndexRead
 from ...core.serialization import convert_and_respect_annotation_metadata
 from ...core.jsonable_encoder import jsonable_encoder
@@ -108,7 +107,6 @@ class DocumentIndexesClient:
         name: str,
         indexing_config: DocumentIndexIndexingConfigRequest,
         status: typing.Optional[EntityStatus] = OMIT,
-        environment: typing.Optional[EnvironmentEnum] = OMIT,
         copy_documents_from_index_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
@@ -131,13 +129,6 @@ class DocumentIndexesClient:
             * `ACTIVE` - Active
             * `ARCHIVED` - Archived
 
-        environment : typing.Optional[EnvironmentEnum]
-            The environment this document index is used in
-
-            * `DEVELOPMENT` - Development
-            * `STAGING` - Staging
-            * `PRODUCTION` - Production
-
         copy_documents_from_index_id : typing.Optional[str]
             Optionally specify the id of a document index from which you'd like to copy and re-index its documents into this newly created index
 
@@ -191,7 +182,6 @@ class DocumentIndexesClient:
                 "label": label,
                 "name": name,
                 "status": status,
-                "environment": environment,
                 "indexing_config": convert_and_respect_annotation_metadata(
                     object_=indexing_config, annotation=DocumentIndexIndexingConfigRequest, direction="write"
                 ),
@@ -268,7 +258,6 @@ class DocumentIndexesClient:
         *,
         label: str,
         status: typing.Optional[EntityStatus] = OMIT,
-        environment: typing.Optional[EnvironmentEnum] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
@@ -288,13 +277,6 @@ class DocumentIndexesClient:
             * `ACTIVE` - Active
             * `ARCHIVED` - Archived
 
-        environment : typing.Optional[EnvironmentEnum]
-            The environment this document index is used in
-
-            * `DEVELOPMENT` - Development
-            * `STAGING` - Staging
-            * `PRODUCTION` - Production
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -322,7 +304,6 @@ class DocumentIndexesClient:
             json={
                 "label": label,
                 "status": status,
-                "environment": environment,
             },
             request_options=request_options,
             omit=OMIT,
@@ -388,7 +369,6 @@ class DocumentIndexesClient:
         *,
         label: typing.Optional[str] = OMIT,
         status: typing.Optional[EntityStatus] = OMIT,
-        environment: typing.Optional[EnvironmentEnum] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
@@ -408,13 +388,6 @@ class DocumentIndexesClient:
             * `ACTIVE` - Active
             * `ARCHIVED` - Archived
 
-        environment : typing.Optional[EnvironmentEnum]
-            The environment this document index is used in
-
-            * `DEVELOPMENT` - Development
-            * `STAGING` - Staging
-            * `PRODUCTION` - Production
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -441,7 +414,6 @@ class DocumentIndexesClient:
             json={
                 "label": label,
                 "status": status,
-                "environment": environment,
             },
             request_options=request_options,
             omit=OMIT,
@@ -651,7 +623,6 @@ class AsyncDocumentIndexesClient:
         name: str,
         indexing_config: DocumentIndexIndexingConfigRequest,
         status: typing.Optional[EntityStatus] = OMIT,
-        environment: typing.Optional[EnvironmentEnum] = OMIT,
         copy_documents_from_index_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
@@ -674,13 +645,6 @@ class AsyncDocumentIndexesClient:
             * `ACTIVE` - Active
             * `ARCHIVED` - Archived
 
-        environment : typing.Optional[EnvironmentEnum]
-            The environment this document index is used in
-
-            * `DEVELOPMENT` - Development
-            * `STAGING` - Staging
-            * `PRODUCTION` - Production
-
         copy_documents_from_index_id : typing.Optional[str]
             Optionally specify the id of a document index from which you'd like to copy and re-index its documents into this newly created index
 
@@ -742,7 +706,6 @@ class AsyncDocumentIndexesClient:
                 "label": label,
                 "name": name,
                 "status": status,
-                "environment": environment,
                 "indexing_config": convert_and_respect_annotation_metadata(
                     object_=indexing_config, annotation=DocumentIndexIndexingConfigRequest, direction="write"
                 ),
@@ -827,7 +790,6 @@ class AsyncDocumentIndexesClient:
         *,
         label: str,
         status: typing.Optional[EntityStatus] = OMIT,
-        environment: typing.Optional[EnvironmentEnum] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
@@ -847,13 +809,6 @@ class AsyncDocumentIndexesClient:
             * `ACTIVE` - Active
             * `ARCHIVED` - Archived
 
-        environment : typing.Optional[EnvironmentEnum]
-            The environment this document index is used in
-
-            * `DEVELOPMENT` - Development
-            * `STAGING` - Staging
-            * `PRODUCTION` - Production
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -889,7 +844,6 @@ class AsyncDocumentIndexesClient:
             json={
                 "label": label,
                 "status": status,
-                "environment": environment,
             },
             request_options=request_options,
             omit=OMIT,
@@ -963,7 +917,6 @@ class AsyncDocumentIndexesClient:
         *,
         label: typing.Optional[str] = OMIT,
         status: typing.Optional[EntityStatus] = OMIT,
-        environment: typing.Optional[EnvironmentEnum] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
@@ -983,13 +936,6 @@ class AsyncDocumentIndexesClient:
             * `ACTIVE` - Active
             * `ARCHIVED` - Archived
 
-        environment : typing.Optional[EnvironmentEnum]
-            The environment this document index is used in
-
-            * `DEVELOPMENT` - Development
-            * `STAGING` - Staging
-            * `PRODUCTION` - Production
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1024,7 +970,6 @@ class AsyncDocumentIndexesClient:
             json={
                 "label": label,
                 "status": status,
-                "environment": environment,
             },
             request_options=request_options,
             omit=OMIT,

vellum/client/types/__init__.py

@@ -304,6 +304,7 @@ from .rejected_execute_prompt_response import RejectedExecutePromptResponse
 from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
+from .release import Release
 from .release_tag_source import ReleaseTagSource
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 from .rich_text_child_block import RichTextChildBlock
@@ -825,6 +826,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
+    "Release",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",

vellum/client/types/document_index_read.py

@@ -5,7 +5,6 @@ import datetime as dt
 import pydantic
 import typing
 from .entity_status import EntityStatus
-from .environment_enum import EnvironmentEnum
 from .document_index_indexing_config import DocumentIndexIndexingConfig
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
@@ -31,15 +30,6 @@ class DocumentIndexRead(UniversalBaseModel):
     * `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
-    """
-    The environment this document index is used in
-    
-    * `DEVELOPMENT` - Development
-    * `STAGING` - Staging
-    * `PRODUCTION` - Production
-    """
-
     indexing_config: DocumentIndexIndexingConfig
 
     if IS_PYDANTIC_V2:

vellum/client/types/release.py

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

vellum/client/types/workflow_release_tag_read.py

@@ -4,6 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import pydantic
 from .release_tag_source import ReleaseTagSource
 from .workflow_release_tag_workflow_deployment_history_item import WorkflowReleaseTagWorkflowDeploymentHistoryItem
+from .release import Release
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
@@ -24,7 +25,12 @@ class WorkflowReleaseTagRead(UniversalBaseModel):
 
     history_item: WorkflowReleaseTagWorkflowDeploymentHistoryItem = pydantic.Field()
     """
-    The Workflow Deployment History Item that this Release Tag is associated with
+    Deprecated. Reference the `release` field instead.
+    """
+
+    release: Release = pydantic.Field()
+    """
+    The Release that this Release Tag points to.
     """
 
     if IS_PYDANTIC_V2:

vellum/plugins/pydantic.py

@@ -1,17 +1,27 @@
 from functools import lru_cache
-from typing import Any, Dict, Literal, Optional, Tuple, Union
+from typing import Any, Dict, Literal, NamedTuple, Optional, Tuple, Union
+from typing_extensions import TypeAlias
 
 from pydantic.fields import FieldInfo
 from pydantic.plugin import (
     PydanticPluginProtocol,
-    SchemaKind,
-    SchemaTypePath,
     ValidateJsonHandlerProtocol,
     ValidatePythonHandlerProtocol,
     ValidateStringsHandlerProtocol,
 )
 from pydantic_core import CoreSchema
 
+# Redefined manually instead of imported from pydantic to support versions < 2.5
+SchemaKind: TypeAlias = Literal["BaseModel", "TypeAdapter", "dataclass", "create_model", "validate_call"]
+
+
+# Redefined manually instead of imported from pydantic to support versions < 2.5
+class SchemaTypePath(NamedTuple):
+    """Path defining where `schema_type` was defined, or where `TypeAdapter` was called."""
+
+    module: str
+    name: str
+
 
 @lru_cache(maxsize=1)
 def import_base_descriptor():
@@ -81,7 +91,7 @@ class VellumPydanticPlugin(PydanticPluginProtocol):
         self,
         schema: CoreSchema,
         schema_type: Any,
-        schema_type_path: SchemaTypePath,
+        schema_type_path: SchemaTypePath,  # type: ignore
         schema_kind: SchemaKind,
         config: Any,
         plugin_settings: Dict[str, Any],

vellum/prompts/blocks/compilation.py

@@ -3,11 +3,13 @@ from typing import Sequence, Union, cast
 
 from vellum import (
     ChatMessage,
+    DocumentVellumValue,
     JsonVellumValue,
     PromptBlock,
     PromptRequestInput,
     RichTextPromptBlock,
     StringVellumValue,
+    VellumDocument,
     VellumVariable,
 )
 from vellum.client.types.audio_vellum_value import AudioVellumValue
@@ -159,6 +161,18 @@ def compile_prompt_blocks(
                 cache_config=block.cache_config,
             )
             compiled_blocks.append(audio_block)
+
+        elif block.block_type == "DOCUMENT":
+            document_block = CompiledValuePromptBlock(
+                content=DocumentVellumValue(
+                    value=VellumDocument(
+                        src=block.src,
+                        metadata=block.metadata,
+                    )
+                ),
+                cache_config=block.cache_config,
+            )
+            compiled_blocks.append(document_block)
         else:
             raise PromptCompilationError(f"Unknown block_type: {block.block_type}")
 

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

vellum/workflows/nodes/bases/base.py

@@ -293,14 +293,14 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ALL:
                 """
-                A node utilizing an AWAIT_ALL merge strategy will only be considered ready for the Nth time
-                when all of its dependencies have been executed N times.
+                A node utilizing an AWAIT_ALL merge strategy will only be considered ready
+                when all of its dependencies have invoked this node.
                 """
-                current_node_execution_count = state.meta.node_execution_cache.get_execution_count(cls.node_class)
-                return all(
-                    state.meta.node_execution_cache.get_execution_count(dep) == current_node_execution_count + 1
-                    for dep in dependencies
-                )
+                # Check if all dependencies have invoked this node
+                dependencies_invoked = state.meta.node_execution_cache._dependencies_invoked.get(node_span_id, set())
+                all_deps_invoked = all(dep in dependencies_invoked for dep in dependencies)
+
+                return all_deps_invoked
 
             raise NodeException(
                 message="Invalid Trigger Node Specification",

vellum/workflows/nodes/bases/base_adornment_node.py

@@ -73,3 +73,5 @@ class BaseAdornmentNode(
         # Subclasses of BaseAdornableNode can override this method to provider their own
         # approach to annotating the outputs class based on the `subworkflow.Outputs`
         setattr(outputs_class, reference.name, reference)
+        if cls.__wrapped_node__:
+            cls.__output_ids__[reference.name] = cls.__wrapped_node__.__output_ids__[reference.name]

vellum/workflows/nodes/core/retry_node/node.py

@@ -85,7 +85,7 @@ Message: {terminal_event.error.message}""",
     @classmethod
     def wrap(
         cls,
-        max_attempts: int,
+        max_attempts: int = 3,
         delay: Optional[float] = None,
         retry_on_error_code: Optional[WorkflowErrorCode] = None,
         retry_on_condition: Optional[BaseDescriptor] = None,

vellum/workflows/nodes/core/try_node/node.py

@@ -102,4 +102,4 @@ Message: {event.error.message}""",
         if reference.name == "error":
             raise ValueError("`error` is a reserved name for TryNode.Outputs")
 
-        setattr(outputs_class, reference.name, reference)
+        super().__annotate_outputs_class__(outputs_class, reference)

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

@@ -28,6 +28,9 @@ class BasePromptNode(BaseNode, Generic[StateType]):
     def _get_prompt_event_stream(self) -> Union[Iterator[AdHocExecutePromptEvent], Iterator[ExecutePromptEvent]]:
         pass
 
+    def _validate(self) -> None:
+        pass
+
     def run(self) -> Iterator[BaseOutput]:
         outputs = yield from self._process_prompt_event_stream()
         if outputs is None:
@@ -37,6 +40,7 @@ class BasePromptNode(BaseNode, Generic[StateType]):
             )
 
     def _process_prompt_event_stream(self) -> Generator[BaseOutput, None, Optional[List[PromptOutput]]]:
+        self._validate()
         try:
             prompt_event_stream = self._get_prompt_event_stream()
         except ApiError as e:

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

@@ -1,6 +1,6 @@
 import json
 from uuid import uuid4
-from typing import Callable, ClassVar, Generic, Iterator, List, Optional, Tuple, Union
+from typing import Callable, ClassVar, Generic, Iterator, List, Optional, Set, Tuple, Union
 
 from vellum import (
     AdHocExecutePromptEvent,
@@ -18,6 +18,7 @@ from vellum import (
 from vellum.client import RequestOptions
 from vellum.client.types.chat_message_request import ChatMessageRequest
 from vellum.client.types.prompt_settings import PromptSettings
+from vellum.client.types.rich_text_child_block import RichTextChildBlock
 from vellum.workflows.constants import OMIT
 from vellum.workflows.context import get_execution_context
 from vellum.workflows.errors import WorkflowErrorCode
@@ -59,6 +60,31 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
     class Trigger(BasePromptNode.Trigger):
         merge_behavior = MergeBehavior.AWAIT_ANY
 
+    def _extract_required_input_variables(self, blocks: Union[List[PromptBlock], List[RichTextChildBlock]]) -> Set[str]:
+        required_variables = set()
+
+        for block in blocks:
+            if block.block_type == "VARIABLE":
+                required_variables.add(block.input_variable)
+            elif block.block_type == "CHAT_MESSAGE" and block.blocks:
+                required_variables.update(self._extract_required_input_variables(block.blocks))
+            elif block.block_type == "RICH_TEXT" and block.blocks:
+                required_variables.update(self._extract_required_input_variables(block.blocks))
+
+        return required_variables
+
+    def _validate(self) -> None:
+        required_variables = self._extract_required_input_variables(self.blocks)
+        provided_variables = set(self.prompt_inputs.keys() if self.prompt_inputs else set())
+
+        missing_variables = required_variables - provided_variables
+        if missing_variables:
+            missing_vars_str = ", ".join(f"'{var}'" for var in missing_variables)
+            raise NodeException(
+                message=f"Missing required input variables by VariablePromptBlock: {missing_vars_str}",
+                code=WorkflowErrorCode.INVALID_INPUTS,
+            )
+
     def _get_prompt_event_stream(self) -> Iterator[AdHocExecutePromptEvent]:
         input_variables, input_values = self._compile_prompt_inputs()
         current_context = get_execution_context()

vellum/workflows/nodes/displayable/bases/inline_prompt_node/tests/test_inline_prompt_node.py

@@ -0,0 +1,182 @@
+import pytest
+from uuid import uuid4
+from typing import Any, Iterator, List
+
+from vellum import (
+    ChatMessagePromptBlock,
+    JinjaPromptBlock,
+    PlainTextPromptBlock,
+    PromptBlock,
+    RichTextPromptBlock,
+    VariablePromptBlock,
+)
+from vellum.client.types.execute_prompt_event import ExecutePromptEvent
+from vellum.client.types.fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
+from vellum.client.types.initiated_execute_prompt_event import InitiatedExecutePromptEvent
+from vellum.client.types.prompt_output import PromptOutput
+from vellum.client.types.prompt_request_string_input import PromptRequestStringInput
+from vellum.client.types.string_vellum_value import StringVellumValue
+from vellum.workflows.errors import WorkflowErrorCode
+from vellum.workflows.exceptions import NodeException
+from vellum.workflows.nodes.displayable.bases.inline_prompt_node import BaseInlinePromptNode
+
+
+def test_validation_with_missing_variables():
+    """Test that validation correctly identifies missing variables."""
+    test_blocks: List[PromptBlock] = [
+        VariablePromptBlock(input_variable="required_var1"),
+        VariablePromptBlock(input_variable="required_var2"),
+        RichTextPromptBlock(
+            blocks=[
+                PlainTextPromptBlock(text="Some text"),
+                VariablePromptBlock(input_variable="required_var3"),
+            ],
+        ),
+        JinjaPromptBlock(template="Template without variables"),
+        ChatMessagePromptBlock(
+            chat_role="USER",
+            blocks=[
+                RichTextPromptBlock(
+                    blocks=[
+                        PlainTextPromptBlock(text="Nested text"),
+                        VariablePromptBlock(input_variable="required_var4"),
+                    ],
+                ),
+            ],
+        ),
+    ]
+
+    # GIVEN a BaseInlinePromptNode
+    class TestNode(BaseInlinePromptNode):
+        ml_model = "test-model"
+        blocks = test_blocks
+        prompt_inputs = {
+            "required_var1": "value1",
+            # required_var2 is missing
+            # required_var3 is missing
+            # required_var4 is missing
+        }
+
+    # WHEN the node is run
+    node = TestNode()
+    with pytest.raises(NodeException) as excinfo:
+        list(node.run())
+
+    # THEN the node raises the correct NodeException
+    assert excinfo.value.code == WorkflowErrorCode.INVALID_INPUTS
+    assert "required_var2" in str(excinfo.value)
+    assert "required_var3" in str(excinfo.value)
+    assert "required_var4" in str(excinfo.value)
+
+
+def test_validation_with_all_variables_provided(vellum_adhoc_prompt_client):
+    """Test that validation passes when all variables are provided."""
+    test_blocks: List[PromptBlock] = [
+        VariablePromptBlock(input_variable="required_var1"),
+        VariablePromptBlock(input_variable="required_var2"),
+        RichTextPromptBlock(
+            blocks=[
+                PlainTextPromptBlock(text="Some text"),
+                VariablePromptBlock(input_variable="required_var3"),
+            ],
+        ),
+        JinjaPromptBlock(template="Template without variables"),
+        ChatMessagePromptBlock(
+            chat_role="USER",
+            blocks=[
+                RichTextPromptBlock(
+                    blocks=[
+                        PlainTextPromptBlock(text="Nested text"),
+                        VariablePromptBlock(input_variable="required_var4"),
+                    ],
+                ),
+            ],
+        ),
+    ]
+
+    # GIVEN a BaseInlinePromptNode
+    class TestNode(BaseInlinePromptNode):
+        ml_model = "test-model"
+        blocks = test_blocks
+        prompt_inputs = {
+            "required_var1": "value1",
+            "required_var2": "value2",
+            "required_var3": "value3",
+            "required_var4": "value4",
+        }
+
+    expected_outputs: List[PromptOutput] = [
+        StringVellumValue(value="Test response"),
+    ]
+
+    def generate_prompt_events(*args: Any, **kwargs: Any) -> Iterator[ExecutePromptEvent]:
+        execution_id = str(uuid4())
+        events: List[ExecutePromptEvent] = [
+            InitiatedExecutePromptEvent(execution_id=execution_id),
+            FulfilledExecutePromptEvent(
+                execution_id=execution_id,
+                outputs=expected_outputs,
+            ),
+        ]
+        yield from events
+
+    vellum_adhoc_prompt_client.adhoc_execute_prompt_stream.side_effect = generate_prompt_events
+
+    # WHEN the node is run
+    node = TestNode()
+    list(node.run())
+
+    # THEN the prompt is executed with the correct inputs
+    mock_api = vellum_adhoc_prompt_client.adhoc_execute_prompt_stream
+    assert mock_api.call_count == 1
+    assert mock_api.call_args.kwargs["input_values"] == [
+        PromptRequestStringInput(key="required_var1", type="STRING", value="value1"),
+        PromptRequestStringInput(key="required_var2", type="STRING", value="value2"),
+        PromptRequestStringInput(key="required_var3", type="STRING", value="value3"),
+        PromptRequestStringInput(key="required_var4", type="STRING", value="value4"),
+    ]
+
+
+def test_validation_with_extra_variables(vellum_adhoc_prompt_client):
+    """Test that validation passes when extra variables are provided."""
+    test_blocks: List[PromptBlock] = [
+        VariablePromptBlock(input_variable="required_var"),
+    ]
+
+    # GIVEN a BaseInlinePromptNode
+    class TestNode(BaseInlinePromptNode):
+        ml_model = "test-model"
+        blocks = test_blocks
+        prompt_inputs = {
+            "required_var": "value",
+            "extra_var": "extra_value",  # This is not required
+        }
+
+    expected_outputs: List[PromptOutput] = [
+        StringVellumValue(value="Test response"),
+    ]
+
+    def generate_prompt_events(*args: Any, **kwargs: Any) -> Iterator[ExecutePromptEvent]:
+        execution_id = str(uuid4())
+        events: List[ExecutePromptEvent] = [
+            InitiatedExecutePromptEvent(execution_id=execution_id),
+            FulfilledExecutePromptEvent(
+                execution_id=execution_id,
+                outputs=expected_outputs,
+            ),
+        ]
+        yield from events
+
+    vellum_adhoc_prompt_client.adhoc_execute_prompt_stream.side_effect = generate_prompt_events
+
+    # WHEN the node is run
+    node = TestNode()
+    list(node.run())
+
+    # THEN the prompt is executed with the correct inputs
+    mock_api = vellum_adhoc_prompt_client.adhoc_execute_prompt_stream
+    assert mock_api.call_count == 1
+    assert mock_api.call_args.kwargs["input_values"] == [
+        PromptRequestStringInput(key="required_var", type="STRING", value="value"),
+        PromptRequestStringInput(key="extra_var", type="STRING", value="extra_value"),
+    ]