vellum-ai 1.5.1__py3-none-any.whl → 1.5.3__py3-none-any.whl

vellum/__init__.py

@@ -348,6 +348,8 @@ from .client.types import (
     PdfSearchResultMetaSourceRequest,
     PlainTextPromptBlock,
     Price,
+    PrivateVectorizer,
+    PrivateVectorizerRequest,
     ProcessingFailureReasonEnum,
     PromptBlock,
     PromptBlockState,
@@ -1073,6 +1075,8 @@ __all__ = [
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlock",
     "Price",
+    "PrivateVectorizer",
+    "PrivateVectorizerRequest",
     "ProcessingFailureReasonEnum",
     "PromptBlock",
     "PromptBlockState",

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.5.1",
+            "User-Agent": "vellum-ai/1.5.3",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.5.1",
+            "X-Fern-SDK-Version": "1.5.3",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None:

vellum/client/reference.md

@@ -3122,6 +3122,14 @@ client.document_indexes.retrieve(
 <dl>
 <dd>
 
+**mask_indexing_config:** `typing.Optional[bool]` — Whether to mask the indexing configuration in the response
+    
+</dd>
+</dl>
+
+<dl>
+<dd>
+
 **request_options:** `typing.Optional[RequestOptions]` — Request-specific configuration.
     
 </dd>
@@ -5810,6 +5818,14 @@ client.workflow_deployments.list_workflow_deployment_event_executions(
 <dl>
 <dd>
 
+**ordering:** `typing.Optional[str]` 
+    
+</dd>
+</dl>
+
+<dl>
+<dd>
+
 **request_options:** `typing.Optional[RequestOptions]` — Request-specific configuration.
     
 </dd>

vellum/client/resources/document_indexes/client.py

@@ -163,7 +163,13 @@ class DocumentIndexesClient:
         )
         return _response.data
 
-    def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DocumentIndexRead:
+    def retrieve(
+        self,
+        id: str,
+        *,
+        mask_indexing_config: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DocumentIndexRead:
         """
         Used to retrieve a Document Index given its ID or name.
 
@@ -172,6 +178,9 @@ class DocumentIndexesClient:
         id : str
             Either the Document Index's ID or its unique name
 
+        mask_indexing_config : typing.Optional[bool]
+            Whether to mask the indexing configuration in the response
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -192,7 +201,9 @@ class DocumentIndexesClient:
             id="id",
         )
         """
-        _response = self._raw_client.retrieve(id, request_options=request_options)
+        _response = self._raw_client.retrieve(
+            id, mask_indexing_config=mask_indexing_config, request_options=request_options
+        )
         return _response.data
 
     def update(
@@ -558,7 +569,13 @@ class AsyncDocumentIndexesClient:
         )
         return _response.data
 
-    async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DocumentIndexRead:
+    async def retrieve(
+        self,
+        id: str,
+        *,
+        mask_indexing_config: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DocumentIndexRead:
         """
         Used to retrieve a Document Index given its ID or name.
 
@@ -567,6 +584,9 @@ class AsyncDocumentIndexesClient:
         id : str
             Either the Document Index's ID or its unique name
 
+        mask_indexing_config : typing.Optional[bool]
+            Whether to mask the indexing configuration in the response
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -595,7 +615,9 @@ class AsyncDocumentIndexesClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.retrieve(id, request_options=request_options)
+        _response = await self._raw_client.retrieve(
+            id, mask_indexing_config=mask_indexing_config, request_options=request_options
+        )
         return _response.data
 
     async def update(

vellum/client/resources/document_indexes/raw_client.py

@@ -168,7 +168,11 @@ class RawDocumentIndexesClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     def retrieve(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        id: str,
+        *,
+        mask_indexing_config: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[DocumentIndexRead]:
         """
         Used to retrieve a Document Index given its ID or name.
@@ -178,6 +182,9 @@ class RawDocumentIndexesClient:
         id : str
             Either the Document Index's ID or its unique name
 
+        mask_indexing_config : typing.Optional[bool]
+            Whether to mask the indexing configuration in the response
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -190,6 +197,9 @@ class RawDocumentIndexesClient:
             f"v1/document-indexes/{jsonable_encoder(id)}",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
+            params={
+                "mask_indexing_config": mask_indexing_config,
+            },
             request_options=request_options,
         )
         try:
@@ -580,7 +590,11 @@ class AsyncRawDocumentIndexesClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     async def retrieve(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        id: str,
+        *,
+        mask_indexing_config: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[DocumentIndexRead]:
         """
         Used to retrieve a Document Index given its ID or name.
@@ -590,6 +604,9 @@ class AsyncRawDocumentIndexesClient:
         id : str
             Either the Document Index's ID or its unique name
 
+        mask_indexing_config : typing.Optional[bool]
+            Whether to mask the indexing configuration in the response
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -602,6 +619,9 @@ class AsyncRawDocumentIndexesClient:
             f"v1/document-indexes/{jsonable_encoder(id)}",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
+            params={
+                "mask_indexing_config": mask_indexing_config,
+            },
             request_options=request_options,
         )
         try:

vellum/client/resources/workflow_deployments/client.py

@@ -127,6 +127,7 @@ class WorkflowDeploymentsClient:
         filters: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkflowDeploymentEventExecutionsResponse:
         """
@@ -142,6 +143,8 @@ class WorkflowDeploymentsClient:
         offset : typing.Optional[int]
             The initial index from which to return the executions.
 
+        ordering : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -163,7 +166,7 @@ class WorkflowDeploymentsClient:
         )
         """
         _response = self._raw_client.list_workflow_deployment_event_executions(
-            id, filters=filters, limit=limit, offset=offset, request_options=request_options
+            id, filters=filters, limit=limit, offset=offset, ordering=ordering, request_options=request_options
         )
         return _response.data
 
@@ -546,6 +549,7 @@ class AsyncWorkflowDeploymentsClient:
         filters: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkflowDeploymentEventExecutionsResponse:
         """
@@ -561,6 +565,8 @@ class AsyncWorkflowDeploymentsClient:
         offset : typing.Optional[int]
             The initial index from which to return the executions.
 
+        ordering : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -590,7 +596,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.list_workflow_deployment_event_executions(
-            id, filters=filters, limit=limit, offset=offset, request_options=request_options
+            id, filters=filters, limit=limit, offset=offset, ordering=ordering, request_options=request_options
         )
         return _response.data
 

vellum/client/resources/workflow_deployments/raw_client.py

@@ -136,6 +136,7 @@ class RawWorkflowDeploymentsClient:
         filters: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[WorkflowDeploymentEventExecutionsResponse]:
         """
@@ -151,6 +152,8 @@ class RawWorkflowDeploymentsClient:
         offset : typing.Optional[int]
             The initial index from which to return the executions.
 
+        ordering : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -167,6 +170,7 @@ class RawWorkflowDeploymentsClient:
                 "filters": filters,
                 "limit": limit,
                 "offset": offset,
+                "ordering": ordering,
             },
             request_options=request_options,
         )
@@ -589,6 +593,7 @@ class AsyncRawWorkflowDeploymentsClient:
         filters: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[WorkflowDeploymentEventExecutionsResponse]:
         """
@@ -604,6 +609,8 @@ class AsyncRawWorkflowDeploymentsClient:
         offset : typing.Optional[int]
             The initial index from which to return the executions.
 
+        ordering : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -620,6 +627,7 @@ class AsyncRawWorkflowDeploymentsClient:
                 "filters": filters,
                 "limit": limit,
                 "offset": offset,
+                "ordering": ordering,
             },
             request_options=request_options,
         )

vellum/client/types/__init__.py

@@ -356,6 +356,8 @@ from .pdf_search_result_meta_source import PdfSearchResultMetaSource
 from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequest
 from .plain_text_prompt_block import PlainTextPromptBlock
 from .price import Price
+from .private_vectorizer import PrivateVectorizer
+from .private_vectorizer_request import PrivateVectorizerRequest
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .prompt_block import PromptBlock
 from .prompt_block_state import PromptBlockState
@@ -1044,6 +1046,8 @@ __all__ = [
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlock",
     "Price",
+    "PrivateVectorizer",
+    "PrivateVectorizerRequest",
     "ProcessingFailureReasonEnum",
     "PromptBlock",
     "PromptBlockState",

vellum/client/types/indexing_config_vectorizer.py

@@ -19,6 +19,7 @@ from .hkunlp_instructor_xl_vectorizer import HkunlpInstructorXlVectorizer
 from .open_ai_vectorizer_text_embedding_3_large import OpenAiVectorizerTextEmbedding3Large
 from .open_ai_vectorizer_text_embedding_3_small import OpenAiVectorizerTextEmbedding3Small
 from .open_ai_vectorizer_text_embedding_ada_002 import OpenAiVectorizerTextEmbeddingAda002
+from .private_vectorizer import PrivateVectorizer
 
 IndexingConfigVectorizer = typing.Union[
     OpenAiVectorizerTextEmbedding3Small,
@@ -32,4 +33,5 @@ IndexingConfigVectorizer = typing.Union[
     GoogleVertexAiVectorizerTextMultilingualEmbedding002,
     GoogleVertexAiVectorizerGeminiEmbedding001,
     FastEmbedVectorizerBaaiBgeSmallEnV15,
+    PrivateVectorizer,
 ]

vellum/client/types/indexing_config_vectorizer_request.py

@@ -19,6 +19,7 @@ from .hkunlp_instructor_xl_vectorizer_request import HkunlpInstructorXlVectorize
 from .open_ai_vectorizer_text_embedding_3_large_request import OpenAiVectorizerTextEmbedding3LargeRequest
 from .open_ai_vectorizer_text_embedding_3_small_request import OpenAiVectorizerTextEmbedding3SmallRequest
 from .open_ai_vectorizer_text_embedding_ada_002_request import OpenAiVectorizerTextEmbeddingAda002Request
+from .private_vectorizer_request import PrivateVectorizerRequest
 
 IndexingConfigVectorizerRequest = typing.Union[
     OpenAiVectorizerTextEmbedding3SmallRequest,
@@ -32,4 +33,5 @@ IndexingConfigVectorizerRequest = typing.Union[
     GoogleVertexAiVectorizerTextMultilingualEmbedding002Request,
     GoogleVertexAiVectorizerGeminiEmbedding001Request,
     FastEmbedVectorizerBaaiBgeSmallEnV15Request,
+    PrivateVectorizerRequest,
 ]

vellum/client/types/private_vectorizer.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class PrivateVectorizer(UniversalBaseModel):
+    """
+    Serializer for private vectorizer.
+    """
+
+    model_name: typing.Literal["private-vectorizer"] = "private-vectorizer"
+
+    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/private_vectorizer_request.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class PrivateVectorizerRequest(UniversalBaseModel):
+    """
+    Serializer for private vectorizer.
+    """
+
+    model_name: typing.Literal["private-vectorizer"] = "private-vectorizer"
+
+    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_error_code_enum.py

@@ -7,7 +7,7 @@ VellumErrorCodeEnum = typing.Union[
         "INVALID_REQUEST",
         "INVALID_INPUTS",
         "PROVIDER_ERROR",
-        "PROVIDER_CREDENTIALS_AVAILABLE",
+        "PROVIDER_CREDENTIALS_UNAVAILABLE",
         "REQUEST_TIMEOUT",
         "INTERNAL_SERVER_ERROR",
         "USER_DEFINED_ERROR",

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

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

vellum/workflows/integrations/tests/test_vellum_integration_service.py

@@ -1,225 +1,206 @@
 import pytest
 from unittest import mock
 
+from vellum.workflows.constants import VellumIntegrationProviderType
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.integrations.vellum_integration_service import VellumIntegrationService
+from vellum.workflows.types.definition import VellumIntegrationToolDetails
 
 
-def test_vellum_integration_service_get_tool_definition_success():
+def test_vellum_integration_service_get_tool_definition_success(vellum_client):
     """Test that tool definitions are successfully retrieved from Vellum API"""
-    with mock.patch(
-        "vellum.workflows.integrations.vellum_integration_service.create_vellum_client"
-    ) as mock_create_client:
-        # GIVEN a mock Vellum client configured to return a tool definition
-        mock_client = mock.MagicMock()
-        mock_create_client.return_value = mock_client
-
-        mock_response = mock.MagicMock()
-        mock_response.name = "GITHUB_CREATE_AN_ISSUE"
-        mock_response.description = "Create a new issue in a GitHub repository"
-        mock_response.parameters = {
-            "type": "object",
-            "properties": {
-                "repo": {"type": "string", "description": "Repository name"},
-                "title": {"type": "string", "description": "Issue title"},
-                "body": {"type": "string", "description": "Issue body"},
-            },
-            "required": ["repo", "title"],
-        }
-        mock_response.provider = "COMPOSIO"
-
-        mock_client.integrations.retrieve_integration_tool_definition.return_value = mock_response
-
-        # WHEN we request a tool definition
-        service = VellumIntegrationService()
-        result = service.get_tool_definition(
-            integration="GITHUB",
-            provider="COMPOSIO",
-            tool_name="GITHUB_CREATE_AN_ISSUE",
-        )
+    mock_client = vellum_client
+    mock_client.integrations = mock.MagicMock()
+
+    mock_response = mock.MagicMock()
+    mock_response.name = "GITHUB_CREATE_AN_ISSUE"
+    mock_response.description = "Create a new issue in a GitHub repository"
+    mock_response.parameters = {
+        "type": "object",
+        "properties": {
+            "repo": {"type": "string", "description": "Repository name"},
+            "title": {"type": "string", "description": "Issue title"},
+            "body": {"type": "string", "description": "Issue body"},
+        },
+        "required": ["repo", "title"],
+    }
+    mock_response.provider = "COMPOSIO"
+
+    mock_client.integrations.retrieve_integration_tool_definition.return_value = mock_response
+
+    # WHEN we request a tool definition
+    service = VellumIntegrationService(client=mock_client)
+    result = service.get_tool_definition(
+        integration="GITHUB",
+        provider="COMPOSIO",
+        tool_name="GITHUB_CREATE_AN_ISSUE",
+    )
+
+    # THEN the tool definition should be returned with all expected fields
+    assert isinstance(result, VellumIntegrationToolDetails)
+    assert result.name == "GITHUB_CREATE_AN_ISSUE"
+    assert result.description == "Create a new issue in a GitHub repository"
+    assert result.provider == VellumIntegrationProviderType.COMPOSIO
+    # Parameters should now be included in the tool details
+    assert result.parameters is not None
+    assert result.parameters["type"] == "object"
+    assert "properties" in result.parameters
+    assert "repo" in result.parameters["properties"]
+    assert "title" in result.parameters["properties"]
+
+    # AND the API should have been called with the correct parameters
+    mock_client.integrations.retrieve_integration_tool_definition.assert_called_once_with(
+        integration="GITHUB",
+        provider="COMPOSIO",
+        tool_name="GITHUB_CREATE_AN_ISSUE",
+    )
+
+
+def test_vellum_integration_service_get_tool_definition_api_error(vellum_client):
+    """Test that API errors are properly handled when retrieving tool definitions"""
+    mock_client = vellum_client
+    mock_client.integrations = mock.MagicMock()
+    mock_client.integrations.retrieve_integration_tool_definition.side_effect = Exception("Tool not found")
 
-        # THEN the tool definition should be returned with all expected fields
-        assert result["name"] == "GITHUB_CREATE_AN_ISSUE"
-        assert result["description"] == "Create a new issue in a GitHub repository"
-        assert result["provider"] == "COMPOSIO"
-        assert "properties" in result["parameters"]
-        assert "repo" in result["parameters"]["properties"]
+    # WHEN we attempt to get a tool definition for an invalid tool
+    service = VellumIntegrationService(client=mock_client)
 
-        # AND the API should have been called with the correct parameters
-        mock_client.integrations.retrieve_integration_tool_definition.assert_called_once_with(
+    # THEN it should raise a NodeException with appropriate error message
+    with pytest.raises(NodeException) as exc_info:
+        service.get_tool_definition(
             integration="GITHUB",
             provider="COMPOSIO",
-            tool_name="GITHUB_CREATE_AN_ISSUE",
+            tool_name="INVALID_TOOL",
         )
 
-
-def test_vellum_integration_service_get_tool_definition_api_error():
-    """Test that API errors are properly handled when retrieving tool definitions"""
-    with mock.patch(
-        "vellum.workflows.integrations.vellum_integration_service.create_vellum_client"
-    ) as mock_create_client:
-        # GIVEN a mock client that raises an exception when retrieving tool definitions
-        mock_client = mock.MagicMock()
-        mock_create_client.return_value = mock_client
-
-        mock_client.integrations.retrieve_integration_tool_definition.side_effect = Exception("Tool not found")
-
-        # WHEN we attempt to get a tool definition for an invalid tool
-        service = VellumIntegrationService()
-
-        # THEN it should raise a NodeException with appropriate error message
-        with pytest.raises(NodeException) as exc_info:
-            service.get_tool_definition(
-                integration="GITHUB",
-                provider="COMPOSIO",
-                tool_name="INVALID_TOOL",
-            )
-
-        assert "Failed to retrieve tool definition" in str(exc_info.value)
-        assert "Tool not found" in str(exc_info.value)
+    assert "Failed to retrieve tool definition" in str(exc_info.value)
+    assert "Tool not found" in str(exc_info.value)
 
 
-def test_vellum_integration_service_execute_tool_success():
+def test_vellum_integration_service_execute_tool_success(vellum_client):
     """Test that tools are successfully executed via Vellum API"""
-    with mock.patch(
-        "vellum.workflows.integrations.vellum_integration_service.create_vellum_client"
-    ) as mock_create_client:
-        # GIVEN a mock client configured to return successful execution results
-        mock_client = mock.MagicMock()
-        mock_create_client.return_value = mock_client
-
-        mock_response = mock.MagicMock()
-        mock_response.data = {
-            "success": True,
-            "issue_id": 123,
-            "issue_url": "https://github.com/user/repo/issues/123",
-        }
-
-        mock_client.integrations.execute_integration_tool.return_value = mock_response
-
-        # WHEN we execute a tool with valid arguments
-        service = VellumIntegrationService()
-        result = service.execute_tool(
-            integration="GITHUB",
-            provider="COMPOSIO",
-            tool_name="GITHUB_CREATE_AN_ISSUE",
-            arguments={
-                "repo": "user/repo",
-                "title": "Test Issue",
-                "body": "Test body",
-            },
-        )
+    mock_client = vellum_client
+    mock_client.integrations = mock.MagicMock()
+
+    mock_response = mock.MagicMock()
+    mock_response.data = {
+        "success": True,
+        "issue_id": 123,
+        "issue_url": "https://github.com/user/repo/issues/123",
+    }
+
+    mock_client.integrations.execute_integration_tool.return_value = mock_response
+
+    # WHEN we execute a tool with valid arguments
+    service = VellumIntegrationService(client=mock_client)
+    result = service.execute_tool(
+        integration="GITHUB",
+        provider="COMPOSIO",
+        tool_name="GITHUB_CREATE_AN_ISSUE",
+        arguments={
+            "repo": "user/repo",
+            "title": "Test Issue",
+            "body": "Test body",
+        },
+    )
+
+    # THEN the execution result should contain expected data
+    assert result["success"] is True
+    assert result["issue_id"] == 123
+    assert result["issue_url"] == "https://github.com/user/repo/issues/123"
+
+    # AND the API should have been called with correct parameters
+    mock_client.integrations.execute_integration_tool.assert_called_once_with(
+        integration="GITHUB",
+        provider="COMPOSIO",
+        tool_name="GITHUB_CREATE_AN_ISSUE",
+        arguments={
+            "repo": "user/repo",
+            "title": "Test Issue",
+            "body": "Test body",
+        },
+    )
+
+
+def test_vellum_integration_service_execute_tool_api_error(vellum_client):
+    """Test that execution errors are properly handled"""
+    mock_client = vellum_client
+    mock_client.integrations = mock.MagicMock()
+    mock_client.integrations.execute_integration_tool.side_effect = Exception("Authentication failed")
 
-        # THEN the execution result should contain expected data
-        assert result["success"] is True
-        assert result["issue_id"] == 123
-        assert result["issue_url"] == "https://github.com/user/repo/issues/123"
+    # WHEN we attempt to execute a tool that encounters an error
+    service = VellumIntegrationService(client=mock_client)
 
-        # AND the API should have been called with correct parameters
-        mock_client.integrations.execute_integration_tool.assert_called_once_with(
+    # THEN it should raise a NodeException with appropriate error message
+    with pytest.raises(NodeException) as exc_info:
+        service.execute_tool(
             integration="GITHUB",
             provider="COMPOSIO",
             tool_name="GITHUB_CREATE_AN_ISSUE",
-            arguments={
-                "repo": "user/repo",
-                "title": "Test Issue",
-                "body": "Test body",
-            },
+            arguments={"repo": "user/repo"},
         )
 
+    assert "Failed to execute tool" in str(exc_info.value)
+    assert "Authentication failed" in str(exc_info.value)
 
-def test_vellum_integration_service_execute_tool_api_error():
-    """Test that execution errors are properly handled"""
-    with mock.patch(
-        "vellum.workflows.integrations.vellum_integration_service.create_vellum_client"
-    ) as mock_create_client:
-        # GIVEN a mock client that raises an exception during tool execution
-        mock_client = mock.MagicMock()
-        mock_create_client.return_value = mock_client
-
-        mock_client.integrations.execute_integration_tool.side_effect = Exception("Authentication failed")
-
-        # WHEN we attempt to execute a tool that encounters an error
-        service = VellumIntegrationService()
 
-        # THEN it should raise a NodeException with appropriate error message
-        with pytest.raises(NodeException) as exc_info:
-            service.execute_tool(
-                integration="GITHUB",
-                provider="COMPOSIO",
-                tool_name="GITHUB_CREATE_AN_ISSUE",
-                arguments={"repo": "user/repo"},
-            )
-
-        assert "Failed to execute tool" in str(exc_info.value)
-        assert "Authentication failed" in str(exc_info.value)
-
-
-def test_vellum_integration_service_execute_tool_empty_response():
+def test_vellum_integration_service_execute_tool_empty_response(vellum_client):
     """Test that empty response data is handled gracefully"""
-    with mock.patch(
-        "vellum.workflows.integrations.vellum_integration_service.create_vellum_client"
-    ) as mock_create_client:
-        # GIVEN a mock client that returns an empty response
-        mock_client = mock.MagicMock()
-        mock_create_client.return_value = mock_client
-
-        mock_response = mock.MagicMock()
-        mock_response.data = {}
-
-        mock_client.integrations.execute_integration_tool.return_value = mock_response
-
-        # WHEN we execute a tool that returns empty data
-        service = VellumIntegrationService()
-        result = service.execute_tool(
-            integration="SLACK",
-            provider="COMPOSIO",
-            tool_name="SLACK_SEND_MESSAGE",
-            arguments={
-                "channel": "#general",
-                "message": "Hello, world!",
-            },
-        )
+    mock_client = vellum_client
+    mock_client.integrations = mock.MagicMock()
 
-        # THEN an empty dictionary should be returned without errors
-        assert result == {}
+    mock_response = mock.MagicMock()
+    mock_response.data = {}
 
+    mock_client.integrations.execute_integration_tool.return_value = mock_response
 
-def test_vellum_integration_service_multiple_tool_executions():
-    """Test that the service handles multiple sequential tool executions"""
-    with mock.patch(
-        "vellum.workflows.integrations.vellum_integration_service.create_vellum_client"
-    ) as mock_create_client:
-        # GIVEN a mock client configured to return different responses for each call
-        mock_client = mock.MagicMock()
-        mock_create_client.return_value = mock_client
-
-        responses = [
-            mock.MagicMock(data={"result": "first"}),
-            mock.MagicMock(data={"result": "second"}),
-        ]
-        mock_client.integrations.execute_integration_tool.side_effect = responses
-
-        # WHEN we execute multiple tools in sequence
-        service = VellumIntegrationService()
-
-        result1 = service.execute_tool(
-            integration="GITHUB",
-            provider="COMPOSIO",
-            tool_name="TOOL_1",
-            arguments={"arg": "val1"},
-        )
+    # WHEN we execute a tool that returns empty data
+    service = VellumIntegrationService(client=mock_client)
+    result = service.execute_tool(
+        integration="SLACK",
+        provider="COMPOSIO",
+        tool_name="SLACK_SEND_MESSAGE",
+        arguments={
+            "channel": "#general",
+            "message": "Hello, world!",
+        },
+    )
 
-        result2 = service.execute_tool(
-            integration="SLACK",
-            provider="COMPOSIO",
-            tool_name="TOOL_2",
-            arguments={"arg": "val2"},
-        )
+    # THEN an empty dictionary should be returned without errors
+    assert result == {}
 
-        # THEN each tool execution should return its respective result
-        assert result1["result"] == "first"
-        assert result2["result"] == "second"
 
-        # AND the API should have been called twice
-        assert mock_client.integrations.execute_integration_tool.call_count == 2
+def test_vellum_integration_service_multiple_tool_executions(vellum_client):
+    """Test that the service handles multiple sequential tool executions"""
+    mock_client = vellum_client
+    mock_client.integrations = mock.MagicMock()
+
+    responses = [
+        mock.MagicMock(data={"result": "first"}),
+        mock.MagicMock(data={"result": "second"}),
+    ]
+    mock_client.integrations.execute_integration_tool.side_effect = responses
+
+    # WHEN we execute multiple tools in sequence
+    service = VellumIntegrationService(client=mock_client)
+
+    result1 = service.execute_tool(
+        integration="GITHUB",
+        provider="COMPOSIO",
+        tool_name="TOOL_1",
+        arguments={"arg": "val1"},
+    )
+
+    result2 = service.execute_tool(
+        integration="SLACK",
+        provider="COMPOSIO",
+        tool_name="TOOL_2",
+        arguments={"arg": "val2"},
+    )
+
+    # THEN each tool execution should return its respective result
+    assert result1["result"] == "first"
+    assert result2["result"] == "second"
+
+    # AND the API should have been called twice
+    assert mock_client.integrations.execute_integration_tool.call_count == 2

vellum/workflows/integrations/vellum_integration_service.py

@@ -1,7 +1,9 @@
-from typing import Any, Dict
+from typing import Any, Dict, Optional
 
+from vellum.workflows.constants import VellumIntegrationProviderType
 from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
+from vellum.workflows.types.definition import VellumIntegrationToolDetails
 from vellum.workflows.vellum_client import create_vellum_client
 
 
@@ -13,16 +15,16 @@ class VellumIntegrationService:
     own integration infrastructure.
     """
 
-    def __init__(self) -> None:
+    def __init__(self, client: Optional[Any] = None) -> None:
         """Initialize the VellumIntegrationService with a Vellum client."""
-        self._client = create_vellum_client()
+        self._client = client or create_vellum_client()
 
     def get_tool_definition(
         self,
         integration: str,
         provider: str,
         tool_name: str,
-    ) -> Dict[str, Any]:
+    ) -> VellumIntegrationToolDetails:
         """Retrieve a tool definition from Vellum integrations.
 
         Args:
@@ -31,7 +33,7 @@ class VellumIntegrationService:
             tool_name: The tool's unique name as specified by the provider
 
         Returns:
-            Dict containing the tool definition with name, description, and parameters
+            VellumIntegrationToolDetails containing the tool definition with parameters
 
         Raises:
             NodeException: If the tool definition cannot be retrieved
@@ -43,13 +45,13 @@ class VellumIntegrationService:
                 tool_name=tool_name,
             )
 
-            # Convert the response to a dict format matching what's expected
-            return {
-                "name": response.name,
-                "description": response.description,
-                "parameters": response.parameters,
-                "provider": response.provider,
-            }
+            return VellumIntegrationToolDetails(
+                provider=VellumIntegrationProviderType(response.provider),
+                integration=integration,
+                name=response.name,
+                description=response.description,
+                parameters=response.parameters,
+            )
         except Exception as e:
             error_message = f"Failed to retrieve tool definition for {tool_name}: {str(e)}"
             raise NodeException(

vellum/workflows/types/definition.py

@@ -177,6 +177,17 @@ class VellumIntegrationToolDefinition(UniversalBaseModel):
     description: str
 
 
+class VellumIntegrationToolDetails(VellumIntegrationToolDefinition):
+    """Extended version of VellumIntegrationToolDefinition with runtime parameters.
+
+    This class includes the parameters field which is populated during compilation
+    from the Vellum integrations API response. It inherits all fields from the base
+    VellumIntegrationToolDefinition class.
+    """
+
+    parameters: Optional[Dict[str, Any]] = None
+
+
 class MCPServer(UniversalBaseModel):
     type: Literal["MCP_SERVER"] = "MCP_SERVER"
     name: str

vellum/workflows/utils/functions.py

@@ -335,13 +335,15 @@ def compile_vellum_integration_tool_definition(tool_def: VellumIntegrationToolDe
     try:
         service = VellumIntegrationService()
         tool_details = service.get_tool_definition(
-            integration=tool_def.integration, provider=tool_def.provider.value, tool_name=tool_def.name
+            integration=tool_def.integration,
+            provider=tool_def.provider.value,
+            tool_name=tool_def.name,
         )
 
         return FunctionDefinition(
             name=tool_def.name,
-            description=tool_details.get("description", tool_def.description),
-            parameters=tool_details.get("parameters", {}),
+            description=tool_details.description,
+            parameters=tool_details.parameters or {},
         )
     except Exception:
         # Fallback for service failures

{vellum_ai-1.5.1.dist-info → vellum_ai-1.5.3.dist-info}/METADATA

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

{vellum_ai-1.5.1.dist-info → vellum_ai-1.5.3.dist-info}/RECORD

@@ -22,9 +22,9 @@ vellum_cli/tests/test_ping.py,sha256=b3aQLd-N59_8w2rRiWqwpB1rlHaKEYVbAj1Y3hi7A-g
 vellum_cli/tests/test_pull.py,sha256=e2XHzcHIx9k-FyuNAl7wMSNsSSebPGyP6U05JGcddFs,49447
 vellum_cli/tests/test_push.py,sha256=2MjkNKr_9Guv5Exjsm3L1BeVXmPkKUcCSiKnp90HgW4,41996
 vellum_ee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum_ee/assets/node-definitions.json,sha256=Sn4fy0oUUCd_mj7vRNvQI5XR88A7-HKbe9pWaVnbz28,24293
+vellum_ee/assets/node-definitions.json,sha256=UDx8gBZ4Vj853aY7CSHR8JipVjZSQiKdkOKwKqwwVKw,25499
 vellum_ee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum_ee/scripts/generate_node_definitions.py,sha256=1pVf2EA3LeX4kLKC2hSanyk2ZZDGVZ_lb83TE416XP8,3274
+vellum_ee/scripts/generate_node_definitions.py,sha256=FOYQsXIqU45I0OAcsyZUGODF9JK44yunf58rR6YaAdA,3303
 vellum_ee/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/base.py,sha256=R3f2T8FlZrXn2FawAmpVuLB3fKFWw11mCUulWAyIKA0,1912
@@ -48,7 +48,7 @@ vellum_ee/workflows/display/nodes/vellum/final_output_node.py,sha256=z4oeTgKnMGV
 vellum_ee/workflows/display/nodes/vellum/guardrail_node.py,sha256=9_AslWjzj4RHH2sq3SIaq9FU0NCg7ex5TIWrNMybqXg,2173
 vellum_ee/workflows/display/nodes/vellum/inline_prompt_node.py,sha256=muyLv3KMIsUnnXhiPbPhw5B0TO1Z8LUwytpVQKlz4tM,11906
 vellum_ee/workflows/display/nodes/vellum/inline_subworkflow_node.py,sha256=m20_ZZ3Au0ZCpI3TNC9xh54ld1X13CNq-T51VOtP23k,6434
-vellum_ee/workflows/display/nodes/vellum/map_node.py,sha256=kwLqptup7bzYUDkGDbpcJPMMusMezsYrG5rSUYl5TlQ,3750
+vellum_ee/workflows/display/nodes/vellum/map_node.py,sha256=AW5JuHXtmtcJKeRiZKd7iB1re_D7G7jl_OlaZs8nUl0,4219
 vellum_ee/workflows/display/nodes/vellum/merge_node.py,sha256=xMHaPfTSZWYprQenlHm2g47u0a5O9Me_dhAjfqo8nKQ,3116
 vellum_ee/workflows/display/nodes/vellum/note_node.py,sha256=6PcAFA_EJn7vEMdqgoRjYTLHwnXCrJv80B10zuUx4jE,1026
 vellum_ee/workflows/display/nodes/vellum/prompt_deployment_node.py,sha256=uo4YZRV48iX4peGAA1xkGniSPrepywvW_gS7codt-VQ,3378
@@ -153,12 +153,12 @@ vellum_ee/workflows/tests/test_registry.py,sha256=B8xRIuEyLWfSqrYoPldNQXhKPfe50P
 vellum_ee/workflows/tests/test_serialize_module.py,sha256=d4ZpMd3oIxiq-sBXeSQESS6ix6-1P6rdCRFqBEReJIU,2882
 vellum_ee/workflows/tests/test_server.py,sha256=dXFBraU99Y6cKp2aBhLFXQTScSRcE9WaWjo1z9piqdU,23344
 vellum_ee/workflows/tests/test_virtual_files.py,sha256=TJEcMR0v2S8CkloXNmCHA0QW0K6pYNGaIjraJz7sFvY,2762
-vellum/__init__.py,sha256=QWxRjUIOceGP7ARsXZVhWrwJDrbtyoZmXeAtJ8uud5A,47952
+vellum/__init__.py,sha256=eR-V_gunexfXJhjtd2ayuyvACUUcNgGRf1lTGPJlp3w,48062
 vellum/client/README.md,sha256=flqu57ubZNTfpq60CdLtJC9gp4WEkyjb_n_eZ4OYf9w,6497
 vellum/client/__init__.py,sha256=-dZaD_0KtlkpQ-pULNNWcarC5xXlGMcGc2nBKLIyRlA,73661
 vellum/client/core/__init__.py,sha256=lTcqUPXcx4112yLDd70RAPeqq6tu3eFMe1pKOqkW9JQ,1562
 vellum/client/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-vellum/client/core/client_wrapper.py,sha256=HGgXu4osF39E2IYXXZKqZ_UBngm-WseBYRAmezfoveA,2840
+vellum/client/core/client_wrapper.py,sha256=bnEtzZoUPqygbLirwbvuRYuZY_jk14h67pGBxT4CwNs,2840
 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
@@ -179,7 +179,7 @@ vellum/client/errors/not_found_error.py,sha256=YrqVM0oc3qkQbFbmmm6xr300VGfUNxMSy
 vellum/client/errors/too_many_requests_error.py,sha256=SJJemdgUlQHV_VpxK8UfFNexgZebNT5_MTOeQs6oVgc,397
 vellum/client/errors/unauthorized_error.py,sha256=waPl5Swiqsk3FQK-Lljzx4KDh4RPZ0wL6BLHjM8onQ8,394
 vellum/client/raw_client.py,sha256=cmMR0t87iUYvkIE9L4g0dcCmw3uUQNze9rD9CBv5rzs,113481
-vellum/client/reference.md,sha256=Cmp2G_L5Xrw2y_R6NzOhJj4w8e1qdkM8bhHURgkY9yo,98843
+vellum/client/reference.md,sha256=i1hKF-inDAc3CqodGJM0FdkgR10zO4nhHJDLdGjjkw4,99052
 vellum/client/resources/__init__.py,sha256=AlEiBj5gJLcaAJhAa-X1vtES-uujc6c83p6_C55U6FE,1647
 vellum/client/resources/ad_hoc/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 vellum/client/resources/ad_hoc/client.py,sha256=v5I_YzJaaPezsE4KVuMSUXJISstKuJ_9-VUeXakIJhw,14353
@@ -194,8 +194,8 @@ vellum/client/resources/deployments/types/__init__.py,sha256=IjycypLdKnmutjjli7a
 vellum/client/resources/deployments/types/deployments_list_request_status.py,sha256=CxlQD16KZXme7x31YYCe_3aAgEueutDTeJo5A4Au-aU,174
 vellum/client/resources/deployments/types/list_deployment_release_tags_request_source.py,sha256=hRGgWMYZL9uKCmD_2dU8-u9RCPUUGItpNn1tUY-NXKY,180
 vellum/client/resources/document_indexes/__init__.py,sha256=cCCSLBe93_1w6-u6Q-IYM0XXS57PW9Xw14myKqdcaqc,185
-vellum/client/resources/document_indexes/client.py,sha256=cMJo6xEPnHW7-rX-3vOb4-jupdH_M3u9oA6uuxaTIO8,24213
-vellum/client/resources/document_indexes/raw_client.py,sha256=HOBjCtuP1slPMIShqkDv0uoDl2bbu7CQrGG3paLDFN4,32063
+vellum/client/resources/document_indexes/client.py,sha256=pH9slCzMzC-x403hRwQQwcJB0spFFXBv3Hp7rwlIUK0,24791
+vellum/client/resources/document_indexes/raw_client.py,sha256=jD717kJH4j5kdXx9rOp37uyjGAV5FiBBARYHdoWua-s,32679
 vellum/client/resources/document_indexes/types/__init__.py,sha256=HaAObnALQo1B9vq9-6NxBcr5z-IRjqaPojE2Q6thkPM,216
 vellum/client/resources/document_indexes/types/document_indexes_list_request_status.py,sha256=sfUEB0cvOSmlE2iITqnMVyHv05Zy2fWP4QjCIYqMg0M,178
 vellum/client/resources/documents/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
@@ -234,8 +234,8 @@ vellum/client/resources/test_suites/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-C
 vellum/client/resources/test_suites/client.py,sha256=xo197bXk84jj6gEieRmJuhqBbQ8IB-VMNLki7gdqhlo,20753
 vellum/client/resources/test_suites/raw_client.py,sha256=XfDqmJa7fngPsHAirTSSNBMHy2O4mKOaNS16IL567L8,22335
 vellum/client/resources/workflow_deployments/__init__.py,sha256=MVnGG7CkZA7F8p__MGerey22DTg6oYqfay77mdlDjBc,271
-vellum/client/resources/workflow_deployments/client.py,sha256=OjISxlNMMw241kmQN5uD3Cp9cOtQYJgbGh_yABG9J4o,27114
-vellum/client/resources/workflow_deployments/raw_client.py,sha256=ypDXg7CzASIiTsThqJZhTmgpKuDQERgNJenFPWAnugU,37281
+vellum/client/resources/workflow_deployments/client.py,sha256=AolU-CJ7gKKTkVekNp9qZM5EyQQlsQyDtwJplwHIHLs,27328
+vellum/client/resources/workflow_deployments/raw_client.py,sha256=LvOUJmHV7EVk734DQGH65XbTtIPGLFzWzaOaXLHU7Ss,37533
 vellum/client/resources/workflow_deployments/types/__init__.py,sha256=2Lrjd-cX_h9gZEgxsF7e9nhCWUcTNXQyEaOjytC1NEM,360
 vellum/client/resources/workflow_deployments/types/list_workflow_release_tags_request_source.py,sha256=LPETHLX9Ygha_JRT9oWZAZR6clv-W1tTelXzktkTBX8,178
 vellum/client/resources/workflow_deployments/types/workflow_deployments_list_request_status.py,sha256=FXVkVmGM6DZ2RpTGnZXWJYiVlLQ-K5fDtX3WMaBPaWk,182
@@ -258,7 +258,7 @@ vellum/client/resources/workspaces/client.py,sha256=36KYa2FDu6h65q2GscUFOJs4qKei
 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=n4NG8ONAsDGYTCAfh1BtPvO7DGKyov_L_fMKkGPshqA,72678
+vellum/client/types/__init__.py,sha256=KhRC4aJBoLP3SMWUNlbLiqmpCB8Wh_5AUPhGXZRQi7c,72850
 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
 vellum/client/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=5kD6ZcbU8P8ynK0lMD8Mh7vHzvQt06ziMyphvWYg6FU,968
@@ -463,8 +463,8 @@ vellum/client/types/image_input_request.py,sha256=8WEE7LaFJ0M8ncxmts6yKTNAwGT-C1
 vellum/client/types/image_prompt_block.py,sha256=5VUNnsbmvK3RkWFq5PFq6_SleHR4XyHYFzKWBgpmwJ0,972
 vellum/client/types/image_vellum_value.py,sha256=aYtApnz5PeTYia0Mwmxi3LQ2F6KAoy3vDYzovzaEqzs,712
 vellum/client/types/image_vellum_value_request.py,sha256=1y5MyppljIcx748SLcC6N84RPNS2nGPdueuLZNrekxk,741
-vellum/client/types/indexing_config_vectorizer.py,sha256=nTHf22XEYxZpjzZg29IR2ow4r6zpgJqW1UiY6Ht-T8U,1827
-vellum/client/types/indexing_config_vectorizer_request.py,sha256=-ofzJtUpBZ1uykyE_io1wXtMwQdmVbDXJ0EfyD21S3s,2076
+vellum/client/types/indexing_config_vectorizer.py,sha256=rK0JnE9lCCXiB71wkT5FCyky-_WYkgNyL0VrFM5UImg,1900
+vellum/client/types/indexing_config_vectorizer_request.py,sha256=KGaJ4Z74RdjU27o4zIK_Hp32LE82-nFy1V-Tu_2UW8U,2171
 vellum/client/types/indexing_state_enum.py,sha256=KWYMz5DwJnVhu3ZlSDdqiC5MtiTIdrxE4EvwFYiel1U,213
 vellum/client/types/initiated_ad_hoc_execute_prompt_event.py,sha256=S2pwVWNs0sJKNfCLPd3N0xlYw1Tl67Js1waB5hSML5E,879
 vellum/client/types/initiated_execute_prompt_event.py,sha256=yrNgIGqJEfe2cxoE-3zIOFa6j976Qo2Hcbm-vU6wogs,857
@@ -601,6 +601,8 @@ vellum/client/types/pdf_search_result_meta_source.py,sha256=KaYx-xvNtOn_ADRDBbb4
 vellum/client/types/pdf_search_result_meta_source_request.py,sha256=Fh2EUxWyhdP7yW2CUNvSTSZo8EcukgogALr4HpppHvQ,1097
 vellum/client/types/plain_text_prompt_block.py,sha256=vqZESoqj6P1IyHFmRAk2kmdU3ktsM_852crRCBcYV64,894
 vellum/client/types/price.py,sha256=f-j-74UUDuX2c-IQxXH78KV8L-jLi6sdsHWVLRKqBy4,574
+vellum/client/types/private_vectorizer.py,sha256=byAlExo-LWDBOMXrgMalUiV-IgWf1yk1LN9MmJ22p4o,642
+vellum/client/types/private_vectorizer_request.py,sha256=ToGDgT3V_wsr75vpTYFVrhnrHcbzBe03QkadvWL7REg,649
 vellum/client/types/processing_failure_reason_enum.py,sha256=cQ3Rmryo7U0Lkh-XZzWIDI9Re-oVh1GZb2cDgMR5sL8,224
 vellum/client/types/prompt_block.py,sha256=950JeKeNKZ0DQXwCD-Sy9SDMtiR7F-BqCrJZoxZt7JM,886
 vellum/client/types/prompt_block_state.py,sha256=BRAzTYARoSU36IVZGWMeeqhl5fgFMXCyhJ8rCbfB-f0,163
@@ -815,7 +817,7 @@ vellum/client/types/vellum_code_resource_definition.py,sha256=XdueTR342BDjevZ3kt
 vellum/client/types/vellum_document.py,sha256=qwXqMS2Eud2a5KmF8QHhU_vJzDX0g5cesrCpmBqREsA,604
 vellum/client/types/vellum_document_request.py,sha256=P9vA7ZDNeaHNlMqyzfl-ZD4bpdf-xA5mH8R1QuOAmOY,611
 vellum/client/types/vellum_error.py,sha256=G4WSD-w_skoDDnsAt-TtImt-hZT-Sc8LjHvERBUVnhE,691
-vellum/client/types/vellum_error_code_enum.py,sha256=_Z1p2ExKwGLmNU5Hq-8MZ328LStQWbevipjYkBF-GSU,402
+vellum/client/types/vellum_error_code_enum.py,sha256=kaqf7PNViS1cELE571Ql2-Xueh058MPzq_jvqLiMH48,404
 vellum/client/types/vellum_error_request.py,sha256=7l4Ux6wj3C9BdSXUPBrtxECsAirmvaLU42Y23VqncBU,698
 vellum/client/types/vellum_image.py,sha256=LAGUYBDsT0bmMOqgbaeCTCy2w4zAeHEyUIgPtmdjjJ4,601
 vellum/client/types/vellum_image_request.py,sha256=6DoI2AdJIG8NShHSslpHvsFUw5PwIMconjlHSipOP5Q,608
@@ -1396,6 +1398,8 @@ vellum/types/pdf_search_result_meta_source.py,sha256=tkM53z99Zky8ifkcfj1HoS7k-sc
 vellum/types/pdf_search_result_meta_source_request.py,sha256=TW9FWTdqZi-SuERKkjMdrKBgQEq3RA4W9nwefm8it6k,175
 vellum/types/plain_text_prompt_block.py,sha256=K5tGXMDCVTLDIDOL17TjLvZD6pMaHnRtcSYfXOrzQMM,161
 vellum/types/price.py,sha256=TTMB0kw-FD2OtTVWMPPB12zay8ncWtnAq48mg1Bwmz0,143
+vellum/types/private_vectorizer.py,sha256=MjfxUfZCwY-Yt7ksER7w4RfVqtIToJK080U5iq-o-b4,156
+vellum/types/private_vectorizer_request.py,sha256=sy498k6SmUSPGgaDfrQgb5NN-ebxE68MeGODv7DNXLA,164
 vellum/types/processing_failure_reason_enum.py,sha256=Pz9H8oNn8H6lLLVuLWsuRYzOiQkVXRAW_KeT3mPd5z8,168
 vellum/types/prompt_block.py,sha256=WU91uuWAvel_8EFjoBSJodOZdAVlZAFwQYCrT4Mt-dE,150
 vellum/types/prompt_block_state.py,sha256=tKqNrZnHWjvfGS_6oIUTpdCPGxvRJa31Le6qWL_3198,156
@@ -1824,8 +1828,8 @@ vellum/workflows/integrations/composio_service.py,sha256=rSliaZtNiBcDSvDxz9k5i1K
 vellum/workflows/integrations/mcp_service.py,sha256=9DYb8dg2_kgc1UOu830kxhaFlt9yTbhKPhK3L6kb1t4,9831
 vellum/workflows/integrations/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/integrations/tests/test_mcp_service.py,sha256=q_DYrDkIqI4sQBNgID4YdbM4e9tneLVWY8YmI4R26d8,8859
-vellum/workflows/integrations/tests/test_vellum_integration_service.py,sha256=5Cw9fqw2DfIlD1h9xjso2WdocyhELVG8RMpMttUiqQ4,8903
-vellum/workflows/integrations/vellum_integration_service.py,sha256=0pD__eFjhKpmkF110tW2L6btr9CCUOKHMo5mBU7Ykrs,3405
+vellum/workflows/integrations/tests/test_vellum_integration_service.py,sha256=qGDV1Tx3cWoaazjfnkv8MutLYscwvSZazRqtnz3EPhY,7607
+vellum/workflows/integrations/vellum_integration_service.py,sha256=AzF2QkEcLspTV5rySA2GToirNpMmTiY07OZx-6Vq3IA,3625
 vellum/workflows/logging.py,sha256=_a217XogktV4Ncz6xKFz7WfYmZAzkfVRVuC0rWob8ls,437
 vellum/workflows/nodes/__init__.py,sha256=zymtc3_iW2rFmMR-sayTLuN6ZsAw8VnJweWPsjQk2-Q,1197
 vellum/workflows/nodes/bases/__init__.py,sha256=cniHuz_RXdJ4TQgD8CBzoiKDiPxg62ErdVpCbWICX64,58
@@ -1985,7 +1989,7 @@ vellum/workflows/tests/test_undefined.py,sha256=zMCVliCXVNLrlC6hEGyOWDnQADJ2g83y
 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=yKm3sE02ult969q80DTmawiwYqodVjcAW-zlaUIgIv4,1495
-vellum/workflows/types/definition.py,sha256=rVoiXhj7xcQS793qjt2gdv64ywfQrRvujURjIWeC6gA,7240
+vellum/workflows/types/definition.py,sha256=_wzpa0CtS60IIEPfhrLXmj2y9datz5XqJZhLYpml-NI,7663
 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
@@ -1993,7 +1997,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=9KqkCOpuTEncjAxGHfcMGIsrD8-yFQp8NgzvU3w5Rd8,13066
+vellum/workflows/utils/functions.py,sha256=_esuDTYKEORjBoC0YGT3ANO--OswaKu6s-_42_kzlwU,13057
 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
@@ -2012,8 +2016,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=Boa-_m9ii2Qsa1RvVM-VYniF7zCpzGgEGy-OnPZkrHg,23941
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-1.5.1.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-1.5.1.dist-info/METADATA,sha256=Ptz4NIBFWJ5foDFhe-2vq_L8oNt9HKbtTxgf0qN_Ei4,5547
-vellum_ai-1.5.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-1.5.1.dist-info/entry_points.txt,sha256=xVavzAKN4iF_NbmhWOlOkHluka0YLkbN_pFQ9pW3gLI,117
-vellum_ai-1.5.1.dist-info/RECORD,,
+vellum_ai-1.5.3.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-1.5.3.dist-info/METADATA,sha256=TAWtp2asXx1CHMu82sElBaUAF2B9fF8IMK5dZvohhCQ,5547
+vellum_ai-1.5.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-1.5.3.dist-info/entry_points.txt,sha256=xVavzAKN4iF_NbmhWOlOkHluka0YLkbN_pFQ9pW3gLI,117
+vellum_ai-1.5.3.dist-info/RECORD,,

vellum_ee/assets/node-definitions.json

@@ -407,6 +407,51 @@
         }
       ]
     },
+    {
+      "id": "527a10bc-30b2-4e36-91d6-91c8e865bac8",
+      "display_data": {
+        "position": {
+          "x": 0.0,
+          "y": 0.0
+        },
+        "comment": {
+          "value": "\n    Used to map over a list of items and execute a Subworkflow on each iteration.\n\n    items: List[MapNodeItemType] - The items to map over\n    max_concurrency: Optional[int] = None - The maximum number of concurrent subworkflow executions\n    subworkflow: Type[\"BaseWorkflow\"] - The Subworkflow to execute\n    ",
+          "expanded": true
+        }
+      },
+      "base": {
+        "name": "BaseAdornmentNode",
+        "module": [
+          "vellum",
+          "workflows",
+          "nodes",
+          "bases",
+          "base_adornment_node"
+        ]
+      },
+      "definition": {
+        "name": "MapNode",
+        "module": [
+          "vellum",
+          "workflows",
+          "nodes",
+          "core",
+          "map_node",
+          "node"
+        ]
+      },
+      "trigger": {
+        "id": "3a702aac-7b68-47d3-92e1-264883e9532c",
+        "merge_behavior": "AWAIT_ATTRIBUTES"
+      },
+      "ports": [
+        {
+          "id": "0ae85239-a159-48af-b39c-2484eac901df",
+          "name": "default",
+          "type": "DEFAULT"
+        }
+      ]
+    },
     {
       "id": "7426f273-a43d-4448-a2d2-76d0ee0d069c",
       "display_data": {
@@ -809,25 +854,21 @@
     }
   ],
   "errors": [
-    {
-      "node": "MapNode",
-      "error": "Expected NodeReference items to have an instance"
-    },
     {
       "node": "SubworkflowDeploymentNode",
-      "error": "headers: {'server': 'gunicorn', 'date': 'Tue, 23 Sep 2025 21:40:16 GMT', 'content-type': 'application/json', 'allow': 'GET, PUT, PATCH, DELETE, HEAD, OPTIONS', 'x-frame-options': 'DENY', 'content-length': '58', 'vary': 'Accept-Language, Origin', 'content-language': 'en', 'strict-transport-security': 'max-age=60; includeSubDomains; preload', 'x-content-type-options': 'nosniff', 'referrer-policy': 'same-origin', 'cross-origin-opener-policy': 'same-origin', 'via': '1.1 google', 'alt-svc': 'h3=\":443\"; ma=2592000,h3-29=\":443\"; ma=2592000'}, status_code: 403, body: {'detail': 'Authentication credentials were not provided.'}"
+      "error": "ApiError: headers: {'server': 'gunicorn', 'date': 'Wed, 24 Sep 2025 04:58:53 GMT', 'content-type': 'application/json', 'allow': 'GET, PUT, PATCH, DELETE, HEAD, OPTIONS', 'x-frame-options': 'DENY', 'content-length': '58', 'vary': 'Accept-Language, Origin', 'content-language': 'en', 'strict-transport-security': 'max-age=60; includeSubDomains; preload', 'x-content-type-options': 'nosniff', 'referrer-policy': 'same-origin', 'cross-origin-opener-policy': 'same-origin', 'via': '1.1 google', 'alt-svc': 'h3=\":443\"; ma=2592000,h3-29=\":443\"; ma=2592000'}, status_code: 403, body: {'detail': 'Authentication credentials were not provided.'}"
     },
     {
       "node": "PromptDeploymentNode",
-      "error": "headers: {'server': 'gunicorn', 'date': 'Tue, 23 Sep 2025 21:40:16 GMT', 'content-type': 'application/json', 'allow': 'GET, PUT, PATCH, DELETE, HEAD, OPTIONS', 'x-frame-options': 'DENY', 'content-length': '58', 'vary': 'Accept-Language, Origin', 'content-language': 'en', 'strict-transport-security': 'max-age=60; includeSubDomains; preload', 'x-content-type-options': 'nosniff', 'referrer-policy': 'same-origin', 'cross-origin-opener-policy': 'same-origin', 'via': '1.1 google', 'alt-svc': 'h3=\":443\"; ma=2592000,h3-29=\":443\"; ma=2592000'}, status_code: 403, body: {'detail': 'Authentication credentials were not provided.'}"
+      "error": "ApiError: headers: {'server': 'gunicorn', 'date': 'Wed, 24 Sep 2025 04:58:53 GMT', 'content-type': 'application/json', 'allow': 'GET, PUT, PATCH, DELETE, HEAD, OPTIONS', 'x-frame-options': 'DENY', 'content-length': '58', 'vary': 'Accept-Language, Origin', 'content-language': 'en', 'strict-transport-security': 'max-age=60; includeSubDomains; preload', 'x-content-type-options': 'nosniff', 'referrer-policy': 'same-origin', 'cross-origin-opener-policy': 'same-origin', 'via': '1.1 google', 'alt-svc': 'h3=\":443\"; ma=2592000,h3-29=\":443\"; ma=2592000'}, status_code: 403, body: {'detail': 'Authentication credentials were not provided.'}"
     },
     {
       "node": "SearchNode",
-      "error": "Expected NodeReference query to have an instance"
+      "error": "ValueError: Expected NodeReference query to have an instance"
     },
     {
       "node": "TemplatingNode",
-      "error": "TemplatingNode.Outputs.result"
+      "error": "KeyError: TemplatingNode.Outputs.result"
     }
   ]
 }

vellum_ee/scripts/generate_node_definitions.py

@@ -70,7 +70,7 @@ def main() -> None:
                 display_instance = display_class()
                 display_instance.serialize(display_context)
             except Exception as e:
-                errors.append({"node": node_class.__name__, "error": str(e)})
+                errors.append({"node": node_class.__name__, "error": f"{e.__class__.__name__}: {str(e)}"})
 
     result = {"nodes": successful_nodes, "errors": errors}
 

vellum_ee/workflows/display/nodes/vellum/map_node.py

@@ -1,7 +1,8 @@
 from uuid import UUID
-from typing import Generic, Optional, TypeVar, cast
+from typing import Generic, Optional, Type, TypeVar, cast
 
 from vellum.workflows.nodes import MapNode
+from vellum.workflows.state.base import BaseState
 from vellum.workflows.types.core import JsonObject
 from vellum.workflows.workflows.base import BaseWorkflow
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
@@ -22,12 +23,19 @@ class BaseMapNodeDisplay(BaseAdornmentNodeDisplay[_MapNodeType], Generic[_MapNod
         node = self._node
         node_id = self.node_id
 
-        subworkflow = cast(type[BaseWorkflow], raise_if_descriptor(node.subworkflow))
+        subworkflow_value = raise_if_descriptor(node.subworkflow)
+        subworkflow = (
+            cast(type[BaseWorkflow], subworkflow_value)
+            if subworkflow_value is not None
+            else self._default_workflow_class()
+        )
+
+        items = raise_if_descriptor(node.items)
 
         items_node_input = create_node_input(
             node_id=node_id,
             input_name="items",
-            value=node.items,
+            value=items or [],
             display_context=display_context,
             input_id=self.node_input_ids_by_name.get("items"),
         )
@@ -81,3 +89,9 @@ class BaseMapNodeDisplay(BaseAdornmentNodeDisplay[_MapNodeType], Generic[_MapNod
             },
             **self.serialize_generic_fields(display_context),
         }
+
+    def _default_workflow_class(self) -> Type[BaseWorkflow]:
+        class MapNodeSubworkflow(BaseWorkflow[MapNode.SubworkflowInputs, BaseState]):
+            pass
+
+        return MapNodeSubworkflow