vellum-ai 0.14.72__py3-none-any.whl → 0.14.73__py3-none-any.whl

vellum/__init__.py

@@ -34,6 +34,7 @@ from .types import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
+    BuildStatusEnum,
     ChatHistoryInput,
     ChatHistoryInputRequest,
     ChatHistoryVellumValue,
@@ -69,6 +70,7 @@ from .types import (
     ConditionCombinator,
     ConditionalNodeResult,
     ConditionalNodeResultData,
+    ContainerImageBuildConfig,
     ContainerImageContainerImageTag,
     ContainerImageRead,
     CreateTestSuiteTestCaseRequest,
@@ -127,6 +129,8 @@ from .types import (
     FastEmbedVectorizerBaaiBgeSmallEnV15Request,
     FinishReasonEnum,
     FolderEntity,
+    FolderEntityDataset,
+    FolderEntityDatasetData,
     FolderEntityDocumentIndex,
     FolderEntityDocumentIndexData,
     FolderEntityFolder,
@@ -661,6 +665,7 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
+    "BuildStatusEnum",
     "ChatHistoryInput",
     "ChatHistoryInputRequest",
     "ChatHistoryVellumValue",
@@ -696,6 +701,7 @@ __all__ = [
     "ConditionCombinator",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
+    "ContainerImageBuildConfig",
     "ContainerImageContainerImageTag",
     "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
@@ -757,6 +763,8 @@ __all__ = [
     "FinishReasonEnum",
     "FolderEntitiesListRequestEntityStatus",
     "FolderEntity",
+    "FolderEntityDataset",
+    "FolderEntityDatasetData",
     "FolderEntityDocumentIndex",
     "FolderEntityDocumentIndexData",
     "FolderEntityFolder",

vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.14.72",
+            "X-Fern-SDK-Version": "0.14.73",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

vellum/client/core/serialization.py

@@ -184,6 +184,7 @@ def _convert_mapping(
             converted_object[_alias_key(key, type_, direction, aliases_to_field_names)] = (
                 convert_and_respect_annotation_metadata(object_=value, annotation=type_, direction=direction)
             )
+
     return converted_object
 
 

vellum/client/types/__init__.py

@@ -38,6 +38,7 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1 import
 from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
 )
+from .build_status_enum import BuildStatusEnum
 from .chat_history_input import ChatHistoryInput
 from .chat_history_input_request import ChatHistoryInputRequest
 from .chat_history_vellum_value import ChatHistoryVellumValue
@@ -73,6 +74,7 @@ from .components_schemas_prompt_version_build_config_sandbox import ComponentsSc
 from .condition_combinator import ConditionCombinator
 from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
+from .container_image_build_config import ContainerImageBuildConfig
 from .container_image_container_image_tag import ContainerImageContainerImageTag
 from .container_image_read import ContainerImageRead
 from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
@@ -131,6 +133,8 @@ from .fast_embed_vectorizer_baai_bge_small_en_v_15 import FastEmbedVectorizerBaa
 from .fast_embed_vectorizer_baai_bge_small_en_v_15_request import FastEmbedVectorizerBaaiBgeSmallEnV15Request
 from .finish_reason_enum import FinishReasonEnum
 from .folder_entity import FolderEntity
+from .folder_entity_dataset import FolderEntityDataset
+from .folder_entity_dataset_data import FolderEntityDatasetData
 from .folder_entity_document_index import FolderEntityDocumentIndex
 from .folder_entity_document_index_data import FolderEntityDocumentIndexData
 from .folder_entity_folder import FolderEntityFolder
@@ -650,6 +654,7 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
+    "BuildStatusEnum",
     "ChatHistoryInput",
     "ChatHistoryInputRequest",
     "ChatHistoryVellumValue",
@@ -685,6 +690,7 @@ __all__ = [
     "ConditionCombinator",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
+    "ContainerImageBuildConfig",
     "ContainerImageContainerImageTag",
     "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
@@ -743,6 +749,8 @@ __all__ = [
     "FastEmbedVectorizerBaaiBgeSmallEnV15Request",
     "FinishReasonEnum",
     "FolderEntity",
+    "FolderEntityDataset",
+    "FolderEntityDatasetData",
     "FolderEntityDocumentIndex",
     "FolderEntityDocumentIndexData",
     "FolderEntityFolder",

vellum/client/types/build_status_enum.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+BuildStatusEnum = typing.Union[typing.Literal["QUEUED", "BUILDING", "AVAILABLE", "FAILED", "UNKNOWN"], typing.Any]

vellum/client/types/container_image_build_config.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .code_execution_package import CodeExecutionPackage
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ContainerImageBuildConfig(UniversalBaseModel):
+    packages: typing.List[CodeExecutionPackage]
+
+    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/container_image_read.py

@@ -5,6 +5,8 @@ from .entity_visibility import EntityVisibility
 import datetime as dt
 import typing
 from .container_image_container_image_tag import ContainerImageContainerImageTag
+from .build_status_enum import BuildStatusEnum
+from .container_image_build_config import ContainerImageBuildConfig
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -18,6 +20,8 @@ class ContainerImageRead(UniversalBaseModel):
     repository: str
     sha: str
     tags: typing.List[ContainerImageContainerImageTag]
+    build_status: typing.Optional[BuildStatusEnum] = None
+    build_config: typing.Optional[ContainerImageBuildConfig] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/execute_api_response.py

@@ -2,9 +2,9 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing_extensions
+import typing
 from .execute_api_response_json import ExecuteApiResponseJson
 from ..core.serialization import FieldMetadata
-import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -12,7 +12,7 @@ import pydantic
 class ExecuteApiResponse(UniversalBaseModel):
     status_code: int
     text: str
-    json_: typing_extensions.Annotated[ExecuteApiResponseJson, FieldMetadata(alias="json")]
+    json_: typing_extensions.Annotated[typing.Optional[ExecuteApiResponseJson], FieldMetadata(alias="json")] = None
     headers: typing.Dict[str, str]
 
     if IS_PYDANTIC_V2:

vellum/client/types/folder_entity.py

@@ -6,6 +6,7 @@ from .folder_entity_prompt_sandbox import FolderEntityPromptSandbox
 from .folder_entity_workflow_sandbox import FolderEntityWorkflowSandbox
 from .folder_entity_document_index import FolderEntityDocumentIndex
 from .folder_entity_test_suite import FolderEntityTestSuite
+from .folder_entity_dataset import FolderEntityDataset
 
 FolderEntity = typing.Union[
     FolderEntityFolder,
@@ -13,4 +14,5 @@ FolderEntity = typing.Union[
     FolderEntityWorkflowSandbox,
     FolderEntityDocumentIndex,
     FolderEntityTestSuite,
+    FolderEntityDataset,
 ]

vellum/client/types/folder_entity_dataset.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity_dataset_data import FolderEntityDatasetData
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityDataset(UniversalBaseModel):
+    """
+    A slim representation of a Dataset, as it exists within a Folder.
+    """
+
+    id: str
+    type: typing.Literal["DATASET"] = "DATASET"
+    data: FolderEntityDatasetData
+
+    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/folder_entity_dataset_data.py

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

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

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

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

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

vellum/workflows/nodes/core/retry_node/tests/test_node.py

@@ -44,7 +44,7 @@ def test_retry_node__retry_on_error_code__missed():
 
         def run(self) -> Outputs:
             if self.attempt_number < 3:
-                raise Exception("This will not be retried")
+                raise NodeException(message="This will not be retried", code=WorkflowErrorCode.INTERNAL_ERROR)
 
             return self.Outputs(execution_count=self.attempt_number)
 

vellum/workflows/nodes/displayable/api_node/node.py

@@ -31,6 +31,8 @@ class APINode(BaseAPINode):
         merge_behavior = MergeBehavior.AWAIT_ANY
 
     def run(self) -> BaseAPINode.Outputs:
+        self._validate()
+
         headers = self.headers or {}
         header_overrides = {}
         bearer_token = None

vellum/workflows/nodes/displayable/api_node/tests/test_api_node.py

@@ -3,6 +3,7 @@ import pytest
 from vellum import ExecuteApiResponse, VellumSecret as ClientVellumSecret
 from vellum.client.core.api_error import ApiError
 from vellum.workflows.constants import APIRequestMethod, AuthorizationType
+from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes import APINode
 from vellum.workflows.types.core import VellumSecret
@@ -190,3 +191,45 @@ def test_api_node__detects_client_environment_urls__legacy_does_not_override_hea
     # AND the vellum API should have been called with the correct headers
     assert mock_response.last_request
     assert mock_response.last_request.headers["X_API_KEY"] == "vellum-api-key-5678"
+
+
+def test_api_node_raises_error_for_empty_url():
+    class APINodeWithEmptyURL(APINode):
+        method = APIRequestMethod.GET
+        url = ""
+
+    node = APINodeWithEmptyURL()
+
+    with pytest.raises(NodeException) as excinfo:
+        node.run()
+
+    assert excinfo.value.code == WorkflowErrorCode.INVALID_INPUTS
+    assert "URL is required and must be a non-empty string" in str(excinfo.value)
+
+
+def test_api_node_raises_error_for_none_url():
+    class APINodeWithNoneURL(APINode):
+        method = APIRequestMethod.GET
+        url = None  # type: ignore
+
+    node = APINodeWithNoneURL()
+
+    with pytest.raises(NodeException) as excinfo:
+        node.run()
+
+    assert excinfo.value.code == WorkflowErrorCode.INVALID_INPUTS
+    assert "URL is required and must be a non-empty string" in str(excinfo.value)
+
+
+def test_api_node_raises_error_for_whitespace_url():
+    class APINodeWithWhitespaceURL(APINode):
+        method = APIRequestMethod.GET
+        url = "   "
+
+    node = APINodeWithWhitespaceURL()
+
+    with pytest.raises(NodeException) as excinfo:
+        node.run()
+
+    assert excinfo.value.code == WorkflowErrorCode.INVALID_INPUTS
+    assert "URL is required and must be a non-empty string" in str(excinfo.value)

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

@@ -40,6 +40,10 @@ class BaseAPINode(BaseNode, Generic[StateType]):
         status_code: int
         text: str
 
+    def _validate(self) -> None:
+        if not self.url or not isinstance(self.url, str) or not self.url.strip():
+            raise NodeException("URL is required and must be a non-empty string", code=WorkflowErrorCode.INVALID_INPUTS)
+
     def run(self) -> Outputs:
         return self._run(method=self.method, url=self.url, data=self.data, json=self.json, headers=self.headers)
 
@@ -52,6 +56,8 @@ class BaseAPINode(BaseNode, Generic[StateType]):
         headers: Any = None,
         bearer_token: Optional[VellumSecret] = None,
     ) -> Outputs:
+        self._validate()
+
         vellum_instance = False
         for header in headers or {}:
             if isinstance(headers[header], VellumSecret):

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

@@ -30,8 +30,14 @@ from vellum.workflows.nodes.displayable.bases.base_prompt_node import BasePrompt
 from vellum.workflows.nodes.displayable.bases.inline_prompt_node.constants import DEFAULT_PROMPT_PARAMETERS
 from vellum.workflows.outputs import BaseOutput
 from vellum.workflows.types import MergeBehavior
+from vellum.workflows.types.definition import DeploymentDefinition
 from vellum.workflows.types.generics import StateType, is_workflow_class
-from vellum.workflows.utils.functions import compile_function_definition, compile_workflow_function_definition
+from vellum.workflows.utils.functions import (
+    compile_deployment_workflow_function_definition,
+    compile_function_definition,
+    compile_inline_workflow_function_definition,
+)
+from vellum.workflows.utils.pydantic_schema import normalize_json
 
 
 class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
@@ -93,6 +99,8 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
         execution_context = get_execution_context()
         request_options = self.request_options or RequestOptions()
 
+        processed_parameters = self._process_parameters(self.parameters)
+
         request_options["additional_body_parameters"] = {
             "execution_context": execution_context.model_dump(mode="json"),
             **request_options.get("additional_body_parameters", {}),
@@ -105,8 +113,15 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
             for function in self.functions:
                 if isinstance(function, FunctionDefinition):
                     normalized_functions.append(function)
+                elif isinstance(function, DeploymentDefinition):
+                    normalized_functions.append(
+                        compile_deployment_workflow_function_definition(
+                            function.model_dump(),
+                            vellum_client=self._context.vellum_client,
+                        )
+                    )
                 elif is_workflow_class(function):
-                    normalized_functions.append(compile_workflow_function_definition(function))
+                    normalized_functions.append(compile_inline_workflow_function_definition(function))
                 else:
                     normalized_functions.append(compile_function_definition(function))
 
@@ -117,7 +132,7 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
                 ml_model=self.ml_model,
                 input_values=input_values,
                 input_variables=input_variables,
-                parameters=self.parameters,
+                parameters=processed_parameters,
                 blocks=self.blocks,
                 settings=self.settings,
                 functions=normalized_functions,
@@ -130,7 +145,7 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
                 ml_model=self.ml_model,
                 input_values=input_values,
                 input_variables=input_variables,
-                parameters=self.parameters,
+                parameters=processed_parameters,
                 blocks=self.blocks,
                 settings=self.settings,
                 functions=normalized_functions,
@@ -267,3 +282,14 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
                 )
 
         return input_variables, input_values
+
+    def _process_parameters(self, parameters: PromptParameters) -> PromptParameters:
+        """
+        Process parameters to recursively convert any Pydantic models to JSON schema dictionaries.
+        """
+        if not parameters.custom_parameters:
+            return parameters
+
+        processed_custom_params = normalize_json(parameters.custom_parameters)
+
+        return parameters.model_copy(update={"custom_parameters": processed_custom_params})

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

@@ -4,6 +4,8 @@ from unittest import mock
 from uuid import uuid4
 from typing import Any, Iterator, List
 
+from pydantic import BaseModel
+
 from vellum import (
     AdHocExecutePromptEvent,
     ChatMessagePromptBlock,
@@ -192,10 +194,26 @@ def test_validation_with_extra_variables(vellum_adhoc_prompt_client):
     ]
 
 
+class TestPydanticModel(BaseModel):
+    result: str
+    confidence: float = 0.9
+
+
 @pytest.mark.parametrize(
-    "custom_parameters,test_description",
+    "custom_parameters,expected_custom_parameters,test_description",
     [
         (
+            {
+                "json_mode": False,
+                "json_schema": {
+                    "name": "get_result",
+                    "schema": {
+                        "type": "object",
+                        "required": ["result"],
+                        "properties": {"result": {"type": "string", "description": ""}},
+                    },
+                },
+            },
             {
                 "json_mode": False,
                 "json_schema": {
@@ -209,13 +227,34 @@ def test_validation_with_extra_variables(vellum_adhoc_prompt_client):
             },
             "with json_schema configured",
         ),
+        (
+            {
+                "json_mode": False,
+                "json_schema": {"name": "get_result_pydantic", "schema": TestPydanticModel},
+            },
+            {
+                "json_mode": False,
+                "json_schema": {
+                    "name": "get_result_pydantic",
+                    "schema": {
+                        "type": "object",
+                        "properties": {"result": {"type": "string"}, "confidence": {"type": "number", "default": 0.9}},
+                        "required": ["result"],
+                    },
+                },
+            },
+            "with json_schema configured using Pydantic model",
+        ),
         (
+            {},
             {},
             "without json_mode or json_schema configured",
         ),
     ],
 )
-def test_inline_prompt_node__json_output(vellum_adhoc_prompt_client, custom_parameters, test_description):
+def test_inline_prompt_node__json_output(
+    vellum_adhoc_prompt_client, custom_parameters, expected_custom_parameters, test_description
+):
     """Confirm that InlinePromptNodes output the expected JSON when run."""
 
     # GIVEN a node that subclasses InlinePromptNode
@@ -281,6 +320,7 @@ def test_inline_prompt_node__json_output(vellum_adhoc_prompt_client, custom_para
     assert json_output.value == expected_json
 
     # AND we should have made the expected call to Vellum search
+
     vellum_adhoc_prompt_client.adhoc_execute_prompt_stream.assert_called_once_with(
         blocks=[],
         expand_meta=None,
@@ -297,7 +337,7 @@ def test_inline_prompt_node__json_output(vellum_adhoc_prompt_client, custom_para
             frequency_penalty=0.0,
             presence_penalty=0.0,
             logit_bias=None,
-            custom_parameters=custom_parameters,
+            custom_parameters=expected_custom_parameters,
         ),
         request_options=mock.ANY,
         settings=None,