vellum-ai 1.4.1__py3-none-any.whl → 1.5.0__py3-none-any.whl

vellum/client/resources/integrations/raw_client.py

@@ -0,0 +1,267 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...errors.bad_request_error import BadRequestError
+from ...types.components_schemas_composio_execute_tool_response import ComponentsSchemasComposioExecuteToolResponse
+from ...types.components_schemas_composio_tool_definition import ComponentsSchemasComposioToolDefinition
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawIntegrationsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def retrieve_integration_tool_definition(
+        self,
+        integration: str,
+        provider: str,
+        tool_name: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[ComponentsSchemasComposioToolDefinition]:
+        """
+        Parameters
+        ----------
+        integration : str
+            The integration name
+
+        provider : str
+            The integration provider name
+
+        tool_name : str
+            The tool's unique name, as specified by the integration provider
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ComponentsSchemasComposioToolDefinition]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ComponentsSchemasComposioToolDefinition,
+                    parse_obj_as(
+                        type_=ComponentsSchemasComposioToolDefinition,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def execute_integration_tool(
+        self,
+        integration: str,
+        provider: str,
+        tool_name: str,
+        *,
+        arguments: typing.Dict[str, typing.Optional[typing.Any]],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[ComponentsSchemasComposioExecuteToolResponse]:
+        """
+        Parameters
+        ----------
+        integration : str
+            The integration name
+
+        provider : str
+            The integration provider name
+
+        tool_name : str
+            The tool's unique name, as specified by the integration provider
+
+        arguments : typing.Dict[str, typing.Optional[typing.Any]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ComponentsSchemasComposioExecuteToolResponse]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}/execute",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "arguments": arguments,
+                "provider": "COMPOSIO",
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ComponentsSchemasComposioExecuteToolResponse,
+                    parse_obj_as(
+                        type_=ComponentsSchemasComposioExecuteToolResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawIntegrationsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def retrieve_integration_tool_definition(
+        self,
+        integration: str,
+        provider: str,
+        tool_name: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[ComponentsSchemasComposioToolDefinition]:
+        """
+        Parameters
+        ----------
+        integration : str
+            The integration name
+
+        provider : str
+            The integration provider name
+
+        tool_name : str
+            The tool's unique name, as specified by the integration provider
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ComponentsSchemasComposioToolDefinition]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ComponentsSchemasComposioToolDefinition,
+                    parse_obj_as(
+                        type_=ComponentsSchemasComposioToolDefinition,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def execute_integration_tool(
+        self,
+        integration: str,
+        provider: str,
+        tool_name: str,
+        *,
+        arguments: typing.Dict[str, typing.Optional[typing.Any]],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[ComponentsSchemasComposioExecuteToolResponse]:
+        """
+        Parameters
+        ----------
+        integration : str
+            The integration name
+
+        provider : str
+            The integration provider name
+
+        tool_name : str
+            The tool's unique name, as specified by the integration provider
+
+        arguments : typing.Dict[str, typing.Optional[typing.Any]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ComponentsSchemasComposioExecuteToolResponse]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}/execute",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "arguments": arguments,
+                "provider": "COMPOSIO",
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ComponentsSchemasComposioExecuteToolResponse,
+                    parse_obj_as(
+                        type_=ComponentsSchemasComposioExecuteToolResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

vellum/client/types/__init__.py

@@ -72,9 +72,15 @@ from .code_executor_secret_input import CodeExecutorSecretInput
 from .code_resource_definition import CodeResourceDefinition
 from .compile_prompt_deployment_expand_meta_request import CompilePromptDeploymentExpandMetaRequest
 from .compile_prompt_meta import CompilePromptMeta
+from .components_schemas_composio_execute_tool_request import ComponentsSchemasComposioExecuteToolRequest
+from .components_schemas_composio_execute_tool_response import ComponentsSchemasComposioExecuteToolResponse
+from .components_schemas_composio_tool_definition import ComponentsSchemasComposioToolDefinition
 from .components_schemas_pdf_search_result_meta_source import ComponentsSchemasPdfSearchResultMetaSource
 from .components_schemas_pdf_search_result_meta_source_request import ComponentsSchemasPdfSearchResultMetaSourceRequest
 from .components_schemas_prompt_version_build_config_sandbox import ComponentsSchemasPromptVersionBuildConfigSandbox
+from .composio_execute_tool_request import ComposioExecuteToolRequest
+from .composio_execute_tool_response import ComposioExecuteToolResponse
+from .composio_tool_definition import ComposioToolDefinition
 from .condition_combinator import ConditionCombinator
 from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
@@ -758,9 +764,15 @@ __all__ = [
     "CodeResourceDefinition",
     "CompilePromptDeploymentExpandMetaRequest",
     "CompilePromptMeta",
+    "ComponentsSchemasComposioExecuteToolRequest",
+    "ComponentsSchemasComposioExecuteToolResponse",
+    "ComponentsSchemasComposioToolDefinition",
     "ComponentsSchemasPdfSearchResultMetaSource",
     "ComponentsSchemasPdfSearchResultMetaSourceRequest",
     "ComponentsSchemasPromptVersionBuildConfigSandbox",
+    "ComposioExecuteToolRequest",
+    "ComposioExecuteToolResponse",
+    "ComposioToolDefinition",
     "ConditionCombinator",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",

vellum/client/types/components_schemas_composio_execute_tool_request.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_execute_tool_request import ComposioExecuteToolRequest
+
+ComponentsSchemasComposioExecuteToolRequest = ComposioExecuteToolRequest

vellum/client/types/components_schemas_composio_execute_tool_response.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_execute_tool_response import ComposioExecuteToolResponse
+
+ComponentsSchemasComposioExecuteToolResponse = ComposioExecuteToolResponse

vellum/client/types/components_schemas_composio_tool_definition.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_tool_definition import ComposioToolDefinition
+
+ComponentsSchemasComposioToolDefinition = ComposioToolDefinition

vellum/client/types/composio_execute_tool_request.py

@@ -0,0 +1,24 @@
+# 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 ComposioExecuteToolRequest(UniversalBaseModel):
+    """
+    Payload for executing a Composio tool with provider id and tool arguments.
+    """
+
+    provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    arguments: typing.Dict[str, typing.Optional[typing.Any]]
+
+    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/composio_execute_tool_response.py

@@ -0,0 +1,24 @@
+# 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 ComposioExecuteToolResponse(UniversalBaseModel):
+    """
+    Response payload with provider id and execution output from a Composio tool.
+    """
+
+    provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    data: typing.Dict[str, typing.Optional[typing.Any]]
+
+    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/composio_tool_definition.py

@@ -0,0 +1,26 @@
+# 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 ComposioToolDefinition(UniversalBaseModel):
+    """
+    Serializer for Composio tool definition response.
+    """
+
+    provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    name: str
+    description: str
+    parameters: typing.Dict[str, typing.Optional[typing.Any]]
+
+    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,9 +7,11 @@ VellumErrorCodeEnum = typing.Union[
         "INVALID_REQUEST",
         "INVALID_INPUTS",
         "PROVIDER_ERROR",
+        "PROVIDER_CREDENTIALS_AVAILABLE",
         "REQUEST_TIMEOUT",
         "INTERNAL_SERVER_ERROR",
         "USER_DEFINED_ERROR",
+        "WORKFLOW_CANCELLED",
     ],
     typing.Any,
 ]

vellum/client/types/vellum_sdk_error.py

@@ -10,6 +10,7 @@ from .vellum_sdk_error_code_enum import VellumSdkErrorCodeEnum
 class VellumSdkError(UniversalBaseModel):
     message: str
     code: VellumSdkErrorCodeEnum
+    raw_data: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/workflow_event_error.py

@@ -10,6 +10,7 @@ from .workflow_execution_event_error_code import WorkflowExecutionEventErrorCode
 class WorkflowEventError(UniversalBaseModel):
     message: str
     code: WorkflowExecutionEventErrorCode
+    raw_data: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     stacktrace: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:

vellum/resources/integrations/__init__.py

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

vellum/resources/integrations/client.py

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

vellum/resources/integrations/raw_client.py

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

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

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

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

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

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

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

vellum/workflows/constants.py

@@ -58,3 +58,7 @@ class APIRequestMethod(Enum):
 class AuthorizationType(Enum):
     BEARER_TOKEN = "BEARER_TOKEN"
     API_KEY = "API_KEY"
+
+
+class VellumIntegrationProviderType(Enum):
+    COMPOSIO = "COMPOSIO"

vellum/workflows/emitters/base.py

@@ -29,3 +29,11 @@ class BaseWorkflowEmitter(ABC):
     @abstractmethod
     def snapshot_state(self, state: BaseState) -> None:
         pass
+
+    @abstractmethod
+    def join(self) -> None:
+        """
+        Wait for any background threads or timers used by this emitter to complete.
+        This ensures all pending work is finished before the workflow terminates.
+        """
+        pass

vellum/workflows/emitters/vellum_emitter.py

@@ -135,3 +135,13 @@ class VellumEmitter(BaseWorkflowEmitter):
             request=events,  # type: ignore[arg-type]
             request_options=request_options,
         )
+
+    def join(self) -> None:
+        """
+        Wait for any background threads or timers used by this emitter to complete.
+        This ensures all pending work is finished before the workflow terminates.
+        """
+        self._flush_events()
+
+        if self._debounce_timer and self._debounce_timer.is_alive():
+            self._debounce_timer.join()

vellum/workflows/inputs/dataset_row.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict
 
-from pydantic import field_serializer
+from pydantic import Field, field_serializer
 
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
 from vellum.workflows.inputs.base import BaseInputs
@@ -16,7 +16,7 @@ class DatasetRow(UniversalBaseModel):
     """
 
     label: str
-    inputs: BaseInputs
+    inputs: BaseInputs = Field(default_factory=BaseInputs)
 
     @field_serializer("inputs")
     def serialize_inputs(self, inputs: BaseInputs) -> Dict[str, Any]:

vellum/workflows/nodes/bases/base.py

@@ -1,4 +1,4 @@
-from abc import ABC, ABCMeta
+from abc import ABC, ABCMeta, abstractmethod
 from dataclasses import field
 from functools import cached_property, reduce
 import inspect
@@ -215,6 +215,17 @@ class BaseNodeMeta(ABCMeta):
                 yield attr_value
                 yielded_attr_names.add(attr_name)
 
+    @abstractmethod
+    def __validate__(cls) -> None:
+        """
+        Validates the node.
+        Subclasses can override this method to implement their specific validation logic.
+        Called during serialization or explicit validation.
+
+        Default implementation performs no validation.
+        """
+        pass
+
 
 class _BaseNodeTriggerMeta(type):
     def __eq__(self, other: Any) -> bool:

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

@@ -112,4 +112,10 @@ class BasePromptNode(BaseNode[StateType], Generic[StateType]):
         if not target_node_output:
             return False
 
+        if not isinstance(target_node_output.instance, OutputReference):
+            return False
+
+        if target_node_output.instance.name != "text":
+            return False
+
         return True

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

@@ -45,9 +45,15 @@ 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, MCPServer
+from vellum.workflows.types.definition import (
+    ComposioToolDefinition,
+    DeploymentDefinition,
+    MCPServer,
+    VellumIntegrationToolDefinition,
+)
 from vellum.workflows.types.generics import StateType, is_workflow_class
 from vellum.workflows.utils.functions import (
+    compile_composio_tool_definition,
     compile_function_definition,
     compile_inline_workflow_function_definition,
     compile_mcp_tool_definition,
@@ -134,7 +140,7 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
                 elif isinstance(function, DeploymentDefinition):
                     normalized_functions.append(
                         compile_workflow_deployment_function_definition(
-                            function.model_dump(),
+                            function,
                             vellum_client=self._context.vellum_client,
                         )
                     )
@@ -142,6 +148,14 @@ 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, ComposioToolDefinition):
+                    normalized_functions.append(compile_composio_tool_definition(function))
+                elif isinstance(function, VellumIntegrationToolDefinition):
+                    # TODO: Implement compile_vellum_integration_tool_definition
+                    raise NotImplementedError(
+                        "VellumIntegrationToolDefinition support coming soon. "
+                        "This will be implemented when compile_vellum_integration_tool_definition is created."
+                    )
                 elif isinstance(function, MCPServer):
                     tool_definitions = compile_mcp_tool_definition(function)
                     for tool_def in tool_definitions: