vellum-ai 1.4.2__py3-none-any.whl → 1.5.1__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/descriptors/utils.py

@@ -33,6 +33,9 @@ def resolve_value(
     from the `_T` generic.
     """
 
+    if memo is not None and path in memo:
+        return cast(_T, memo[path])
+
     if inspect.isclass(value):
         return cast(_T, value)
 

vellum/workflows/emitters/vellum_emitter.py

@@ -126,7 +126,10 @@ class VellumEmitter(BaseWorkflowEmitter):
             return
 
         client = self._context.vellum_client
-        request_options = RequestOptions(timeout_in_seconds=self._timeout, max_retries=self._max_retries)
+        if self._timeout is not None:
+            request_options = RequestOptions(timeout_in_seconds=int(self._timeout), max_retries=self._max_retries)
+        else:
+            request_options = RequestOptions(max_retries=self._max_retries)
 
         client.events.create(
             # The API accepts a ClientWorkflowEvent but our SDK emits an SDKWorkflowEvent. These shapes are

vellum/workflows/integrations/__init__.py

@@ -0,0 +1,5 @@
+from .composio_service import ComposioService
+from .mcp_service import MCPService
+from .vellum_integration_service import VellumIntegrationService
+
+__all__ = ["ComposioService", "MCPService", "VellumIntegrationService"]