vellum-ai 1.11.2__py3-none-any.whl → 1.13.5__py3-none-any.whl

vellum/client/resources/workflows/raw_client.py

@@ -11,12 +11,19 @@ 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 ...core.serialization import convert_and_respect_annotation_metadata
 from ...errors.bad_request_error import BadRequestError
+from ...errors.internal_server_error import InternalServerError
+from ...errors.not_found_error import NotFoundError
+from ...types.check_workflow_execution_status_response import CheckWorkflowExecutionStatusResponse
 from ...types.dataset_row_push_request import DatasetRowPushRequest
+from ...types.runner_config_request import RunnerConfigRequest
+from ...types.type_checker_enum import TypeCheckerEnum
 from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
 from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_response import WorkflowPushResponse
 from ...types.workflow_resolved_state import WorkflowResolvedState
+from ...types.workflow_sandbox_execute_node_response import WorkflowSandboxExecuteNodeResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -156,10 +163,160 @@ class RawWorkflowsClient:
             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_node(
+        self,
+        *,
+        files: typing.Dict[str, str],
+        node: str,
+        inputs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[WorkflowSandboxExecuteNodeResponse]:
+        """
+        Parameters
+        ----------
+        files : typing.Dict[str, str]
+
+        node : str
+
+        inputs : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[WorkflowSandboxExecuteNodeResponse]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/workflows/execute-node",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "files": files,
+                "node": node,
+                "inputs": inputs,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    WorkflowSandboxExecuteNodeResponse,
+                    parse_obj_as(
+                        type_=WorkflowSandboxExecuteNodeResponse,  # 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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
+    def workflow_execution_status(
+        self, execution_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[CheckWorkflowExecutionStatusResponse]:
+        """
+        Checks if a workflow execution is currently executing (not fulfilled, not rejected, and has no end time).
+        Uses the ClickHouse Prime summary materialized view.
+
+        Parameters
+        ----------
+        execution_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[CheckWorkflowExecutionStatusResponse]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/workflows/executions/{jsonable_encoder(execution_id)}/status",
+            base_url=self._client_wrapper.get_environment().predict,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CheckWorkflowExecutionStatusResponse,
+                    parse_obj_as(
+                        type_=CheckWorkflowExecutionStatusResponse,  # 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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
     def push(
         self,
         *,
-        exec_config: WorkflowPushExecConfig,
+        exec_config: typing.Optional[WorkflowPushExecConfig] = OMIT,
         workflow_sandbox_id: typing.Optional[str] = OMIT,
         deployment_config: typing.Optional[WorkflowPushDeploymentConfigRequest] = OMIT,
         artifact: typing.Optional[core.File] = OMIT,
@@ -171,8 +328,8 @@ class RawWorkflowsClient:
         """
         Parameters
         ----------
-        exec_config : WorkflowPushExecConfig
-            The execution configuration of the workflow.
+        exec_config : typing.Optional[WorkflowPushExecConfig]
+            The execution configuration of the workflow. If not provided, it will be derived from the artifact.
 
         workflow_sandbox_id : typing.Optional[str]
 
@@ -235,6 +392,8 @@ class RawWorkflowsClient:
         *,
         files: typing.Dict[str, typing.Optional[typing.Any]],
         module: typing.Optional[str] = OMIT,
+        runner_config: typing.Optional[RunnerConfigRequest] = OMIT,
+        type_checker: typing.Optional[TypeCheckerEnum] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[typing.Dict[str, typing.Optional[typing.Any]]]:
         """
@@ -246,6 +405,15 @@ class RawWorkflowsClient:
 
         module : typing.Optional[str]
 
+        runner_config : typing.Optional[RunnerConfigRequest]
+
+        type_checker : typing.Optional[TypeCheckerEnum]
+            Optional type checker to run during serialization. Supported values: mypy, zuban, default.
+
+            * `mypy` - Mypy
+            * `zuban` - Zuban
+            * `default` - Default
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -261,6 +429,10 @@ class RawWorkflowsClient:
             json={
                 "files": files,
                 "module": module,
+                "runner_config": convert_and_respect_annotation_metadata(
+                    object_=runner_config, annotation=typing.Optional[RunnerConfigRequest], direction="write"
+                ),
+                "type_checker": type_checker,
             },
             headers={
                 "content-type": "application/json",
@@ -419,10 +591,160 @@ class AsyncRawWorkflowsClient:
             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_node(
+        self,
+        *,
+        files: typing.Dict[str, str],
+        node: str,
+        inputs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[WorkflowSandboxExecuteNodeResponse]:
+        """
+        Parameters
+        ----------
+        files : typing.Dict[str, str]
+
+        node : str
+
+        inputs : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[WorkflowSandboxExecuteNodeResponse]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/workflows/execute-node",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "files": files,
+                "node": node,
+                "inputs": inputs,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    WorkflowSandboxExecuteNodeResponse,
+                    parse_obj_as(
+                        type_=WorkflowSandboxExecuteNodeResponse,  # 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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
+    async def workflow_execution_status(
+        self, execution_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[CheckWorkflowExecutionStatusResponse]:
+        """
+        Checks if a workflow execution is currently executing (not fulfilled, not rejected, and has no end time).
+        Uses the ClickHouse Prime summary materialized view.
+
+        Parameters
+        ----------
+        execution_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[CheckWorkflowExecutionStatusResponse]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/workflows/executions/{jsonable_encoder(execution_id)}/status",
+            base_url=self._client_wrapper.get_environment().predict,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CheckWorkflowExecutionStatusResponse,
+                    parse_obj_as(
+                        type_=CheckWorkflowExecutionStatusResponse,  # 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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
     async def push(
         self,
         *,
-        exec_config: WorkflowPushExecConfig,
+        exec_config: typing.Optional[WorkflowPushExecConfig] = OMIT,
         workflow_sandbox_id: typing.Optional[str] = OMIT,
         deployment_config: typing.Optional[WorkflowPushDeploymentConfigRequest] = OMIT,
         artifact: typing.Optional[core.File] = OMIT,
@@ -434,8 +756,8 @@ class AsyncRawWorkflowsClient:
         """
         Parameters
         ----------
-        exec_config : WorkflowPushExecConfig
-            The execution configuration of the workflow.
+        exec_config : typing.Optional[WorkflowPushExecConfig]
+            The execution configuration of the workflow. If not provided, it will be derived from the artifact.
 
         workflow_sandbox_id : typing.Optional[str]
 
@@ -498,6 +820,8 @@ class AsyncRawWorkflowsClient:
         *,
         files: typing.Dict[str, typing.Optional[typing.Any]],
         module: typing.Optional[str] = OMIT,
+        runner_config: typing.Optional[RunnerConfigRequest] = OMIT,
+        type_checker: typing.Optional[TypeCheckerEnum] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[typing.Dict[str, typing.Optional[typing.Any]]]:
         """
@@ -509,6 +833,15 @@ class AsyncRawWorkflowsClient:
 
         module : typing.Optional[str]
 
+        runner_config : typing.Optional[RunnerConfigRequest]
+
+        type_checker : typing.Optional[TypeCheckerEnum]
+            Optional type checker to run during serialization. Supported values: mypy, zuban, default.
+
+            * `mypy` - Mypy
+            * `zuban` - Zuban
+            * `default` - Default
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -524,6 +857,10 @@ class AsyncRawWorkflowsClient:
             json={
                 "files": files,
                 "module": module,
+                "runner_config": convert_and_respect_annotation_metadata(
+                    object_=runner_config, annotation=typing.Optional[RunnerConfigRequest], direction="write"
+                ),
+                "type_checker": type_checker,
             },
             headers={
                 "content-type": "application/json",

vellum/client/types/__init__.py

@@ -55,6 +55,8 @@ from .chat_message_content_request import ChatMessageContentRequest
 from .chat_message_prompt_block import ChatMessagePromptBlock
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
+from .check_workflow_execution_status_error import CheckWorkflowExecutionStatusError
+from .check_workflow_execution_status_response import CheckWorkflowExecutionStatusResponse
 from .code_execution_node_array_result import CodeExecutionNodeArrayResult
 from .code_execution_node_chat_history_result import CodeExecutionNodeChatHistoryResult
 from .code_execution_node_error_result import CodeExecutionNodeErrorResult
@@ -67,6 +69,7 @@ from .code_execution_node_result_output import CodeExecutionNodeResultOutput
 from .code_execution_node_search_results_result import CodeExecutionNodeSearchResultsResult
 from .code_execution_node_string_result import CodeExecutionNodeStringResult
 from .code_execution_package import CodeExecutionPackage
+from .code_execution_package_request import CodeExecutionPackageRequest
 from .code_execution_runtime import CodeExecutionRuntime
 from .code_executor_input import CodeExecutorInput
 from .code_executor_response import CodeExecutorResponse
@@ -308,6 +311,8 @@ from .node_execution_fulfilled_body import NodeExecutionFulfilledBody
 from .node_execution_fulfilled_event import NodeExecutionFulfilledEvent
 from .node_execution_initiated_body import NodeExecutionInitiatedBody
 from .node_execution_initiated_event import NodeExecutionInitiatedEvent
+from .node_execution_log_body import NodeExecutionLogBody
+from .node_execution_log_event import NodeExecutionLogEvent
 from .node_execution_paused_body import NodeExecutionPausedBody
 from .node_execution_paused_event import NodeExecutionPausedEvent
 from .node_execution_rejected_body import NodeExecutionRejectedBody
@@ -427,6 +432,7 @@ from .release_tag_source import ReleaseTagSource
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 from .rich_text_child_block import RichTextChildBlock
 from .rich_text_prompt_block import RichTextPromptBlock
+from .runner_config_request import RunnerConfigRequest
 from .sandbox_scenario import SandboxScenario
 from .scenario_input import ScenarioInput
 from .scenario_input_audio_variable_value import ScenarioInputAudioVariableValue
@@ -458,6 +464,7 @@ from .sentence_chunker_config import SentenceChunkerConfig
 from .sentence_chunker_config_request import SentenceChunkerConfigRequest
 from .sentence_chunking import SentenceChunking
 from .sentence_chunking_request import SentenceChunkingRequest
+from .severity_enum import SeverityEnum
 from .slim_composio_tool_definition import SlimComposioToolDefinition
 from .slim_deployment_read import SlimDeploymentRead
 from .slim_document import SlimDocument
@@ -603,6 +610,7 @@ from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunk
 from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
 from .token_overlapping_window_chunking_request import TokenOverlappingWindowChunkingRequest
+from .type_checker_enum import TypeCheckerEnum
 from .unit_enum import UnitEnum
 from .update_active_workspace_response import UpdateActiveWorkspaceResponse
 from .upload_document_response import UploadDocumentResponse
@@ -733,6 +741,7 @@ from .workflow_result_event_output_data_string import WorkflowResultEventOutputD
 from .workflow_result_event_state import WorkflowResultEventState
 from .workflow_sandbox_display_data import WorkflowSandboxDisplayData
 from .workflow_sandbox_example import WorkflowSandboxExample
+from .workflow_sandbox_execute_node_response import WorkflowSandboxExecuteNodeResponse
 from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
 from .workflow_stream_event import WorkflowStreamEvent
 from .workspace_display_config import WorkspaceDisplayConfig
@@ -785,6 +794,8 @@ __all__ = [
     "ChatMessagePromptBlock",
     "ChatMessageRequest",
     "ChatMessageRole",
+    "CheckWorkflowExecutionStatusError",
+    "CheckWorkflowExecutionStatusResponse",
     "CodeExecutionNodeArrayResult",
     "CodeExecutionNodeChatHistoryResult",
     "CodeExecutionNodeErrorResult",
@@ -797,6 +808,7 @@ __all__ = [
     "CodeExecutionNodeSearchResultsResult",
     "CodeExecutionNodeStringResult",
     "CodeExecutionPackage",
+    "CodeExecutionPackageRequest",
     "CodeExecutionRuntime",
     "CodeExecutorInput",
     "CodeExecutorResponse",
@@ -1034,6 +1046,8 @@ __all__ = [
     "NodeExecutionFulfilledEvent",
     "NodeExecutionInitiatedBody",
     "NodeExecutionInitiatedEvent",
+    "NodeExecutionLogBody",
+    "NodeExecutionLogEvent",
     "NodeExecutionPausedBody",
     "NodeExecutionPausedEvent",
     "NodeExecutionRejectedBody",
@@ -1153,6 +1167,7 @@ __all__ = [
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",
     "RichTextPromptBlock",
+    "RunnerConfigRequest",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputAudioVariableValue",
@@ -1184,6 +1199,7 @@ __all__ = [
     "SentenceChunkerConfigRequest",
     "SentenceChunking",
     "SentenceChunkingRequest",
+    "SeverityEnum",
     "SlimComposioToolDefinition",
     "SlimDeploymentRead",
     "SlimDocument",
@@ -1313,6 +1329,7 @@ __all__ = [
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",
     "TokenOverlappingWindowChunkingRequest",
+    "TypeCheckerEnum",
     "UnitEnum",
     "UpdateActiveWorkspaceResponse",
     "UploadDocumentResponse",
@@ -1443,6 +1460,7 @@ __all__ = [
     "WorkflowResultEventState",
     "WorkflowSandboxDisplayData",
     "WorkflowSandboxExample",
+    "WorkflowSandboxExecuteNodeResponse",
     "WorkflowSandboxParentContext",
     "WorkflowStreamEvent",
     "WorkspaceDisplayConfig",

vellum/client/types/api_actor_type_enum.py

@@ -3,5 +3,5 @@
 import typing
 
 ApiActorTypeEnum = typing.Union[
-    typing.Literal["WORKSPACE_API_KEY", "ENVIRONMENT_API_KEY", "JWT", "SERVICE_TOKEN"], typing.Any
+    typing.Literal["ANONYMOUS", "WORKSPACE_API_KEY", "ENVIRONMENT_API_KEY", "JWT", "SERVICE_TOKEN"], typing.Any
 ]

vellum/client/types/check_workflow_execution_status_error.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_execution_event_error_code import WorkflowExecutionEventErrorCode
+
+
+class CheckWorkflowExecutionStatusError(UniversalBaseModel):
+    message: typing.Optional[str] = None
+    code: typing.Optional[WorkflowExecutionEventErrorCode] = None
+
+    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/check_workflow_execution_status_response.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .check_workflow_execution_status_error import CheckWorkflowExecutionStatusError
+from .workflow_result_event_state import WorkflowResultEventState
+
+
+class CheckWorkflowExecutionStatusResponse(UniversalBaseModel):
+    """
+    Response serializer for workflow execution status check.
+    """
+
+    status: WorkflowResultEventState
+    outputs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    error: typing.Optional[CheckWorkflowExecutionStatusError] = None
+    execution_id: str
+    execution_detail_url: typing.Optional[str] = None
+
+    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/code_execution_package_request.py

@@ -0,0 +1,21 @@
+# 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 CodeExecutionPackageRequest(UniversalBaseModel):
+    version: str
+    name: str
+    repository: typing.Optional[str] = None
+
+    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_request.py

@@ -13,6 +13,11 @@ class ComposioExecuteToolRequest(UniversalBaseModel):
 
     provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
     arguments: typing.Dict[str, typing.Optional[typing.Any]]
+    toolkit_version: typing.Optional[str] = None
+    integration_name: typing.Optional[str] = None
+    integration: typing.Optional[str] = None
+    tool_name: typing.Optional[str] = None
+    tool: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/composio_tool_definition.py

@@ -19,6 +19,7 @@ class ComposioToolDefinition(UniversalBaseModel):
     description: str
     input_parameters: typing.Dict[str, typing.Optional[typing.Any]]
     output_parameters: typing.Dict[str, typing.Optional[typing.Any]]
+    toolkit_version: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/container_image_build_config.py

@@ -10,6 +10,7 @@ from .code_execution_package import CodeExecutionPackage
 class ContainerImageBuildConfig(UniversalBaseModel):
     packages: typing.List[CodeExecutionPackage]
     user_script: typing.Optional[str] = None
+    hotswappable_version: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/container_image_container_image_tag.py

@@ -10,6 +10,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 class ContainerImageContainerImageTag(UniversalBaseModel):
     name: str
     modified: dt.datetime
+    history_item_sha: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/dataset_row_push_request.py

@@ -7,8 +7,11 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class DatasetRowPushRequest(UniversalBaseModel):
+    id: typing.Optional[str] = None
     label: str
     inputs: typing.Dict[str, typing.Optional[typing.Any]]
+    mocks: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = None
+    workflow_trigger_id: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2