vellum-ai 0.14.80__tar.gz → 0.14.82__tar.gz

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/PKG-INFO

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

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_cli/image_push.py

@@ -24,6 +24,16 @@ def image_push_command(
     logger = load_cli_logger()
     config = load_vellum_cli_config()
 
+    logger.info("Cleaning up unused Docker images...")
+    try:
+        subprocess.run(
+            ["docker", "image", "prune", "--all", "-f", "--filter", "label=image-type=python-workflow-runtime"],
+            check=True,
+        )
+        logger.info("Docker image pruning completed successfully")
+    except subprocess.CalledProcessError as e:
+        logger.warning(f"Docker image pruning failed: {e}")
+
     if source:
         logger.info(f"Building Docker image from Dockerfile: {source}")
 

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_cli/tests/test_image_push.py

@@ -48,6 +48,7 @@ def test_image_push__self_hosted_happy_path(mock_docker_from_env, mock_subproces
     mock_docker_from_env.return_value = mock_docker_client
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -100,6 +101,7 @@ def test_image_push__self_hosted_happy_path__workspace_option(
     mock_docker_from_env.return_value = mock_docker_client
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -189,6 +191,7 @@ def test_image_push_with_source_success(
     mock_docker_client.images.push.return_value = [b'{"status": "Pushed"}']
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(args="", returncode=0, stdout=b"Build successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
@@ -205,7 +208,7 @@ def test_image_push_with_source_success(
 
     assert result.exit_code == 0, result.output
 
-    build_call = mock_subprocess_run.call_args_list[0]
+    build_call = mock_subprocess_run.call_args_list[1]
     assert build_call[0][0] == [
         "docker",
         "buildx",
@@ -246,6 +249,7 @@ def test_image_push_with_source_build_fails(mock_subprocess_run, monkeypatch, mo
         f.write("FROM alpine:latest\n")
 
     mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(args="", returncode=1, stderr=b"Build failed: missing dependency"),
     ]
 
@@ -255,3 +259,85 @@ def test_image_push_with_source_build_fails(mock_subprocess_run, monkeypatch, mo
     assert result.exit_code == 1
     assert "Docker build failed" in result.output
     assert "Build failed: missing dependency" in result.output
+
+
+@pytest.mark.usefixtures("vellum_client", "info_log_level")
+def test_image_push_includes_docker_prune(mock_docker_from_env, mock_subprocess_run, monkeypatch):
+    """
+    Tests that image_push_command calls docker image prune before validation.
+    """
+    # GIVEN a self hosted vellum api URL env var
+    monkeypatch.setenv("VELLUM_API_URL", "mycompany.api.com")
+    monkeypatch.setenv("VELLUM_API_KEY", "123456abcdef")
+
+    # AND a mock Docker client
+    mock_docker_client = MagicMock()
+    mock_docker_from_env.return_value = mock_docker_client
+
+    mock_subprocess_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
+        subprocess.CompletedProcess(
+            args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
+        ),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"manifest"),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"sha256:hellosha"),
+    ]
+
+    # WHEN the user runs the image push command with any image
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["image", "push", "myrepo.net/myimage:latest"])
+
+    # THEN the command exits successfully
+    assert result.exit_code == 0, result.output
+
+    prune_call = mock_subprocess_run.call_args_list[0]
+    assert prune_call[0][0] == [
+        "docker",
+        "image",
+        "prune",
+        "--all",
+        "-f",
+        "--filter",
+        "label=image-type=python-workflow-runtime",
+    ]
+
+    # AND the success message includes pruning completion
+    assert "Docker image pruning completed successfully" in result.output
+
+
+@pytest.mark.usefixtures("vellum_client", "info_log_level")
+def test_image_push_continues_if_prune_fails(mock_docker_from_env, mock_subprocess_run, monkeypatch):
+    """
+    Tests that image_push_command continues if docker image prune fails.
+    """
+    # GIVEN a self hosted vellum api URL env var
+    monkeypatch.setenv("VELLUM_API_URL", "mycompany.api.com")
+    monkeypatch.setenv("VELLUM_API_KEY", "123456abcdef")
+
+    # AND a mock Docker client
+    mock_docker_client = MagicMock()
+    mock_docker_from_env.return_value = mock_docker_client
+
+    mock_subprocess_run.side_effect = [
+        subprocess.CalledProcessError(
+            1, ["docker", "image", "prune", "--all", "-f", "--filter", "label=image-type=python-workflow-runtime"]
+        ),
+        subprocess.CompletedProcess(
+            args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
+        ),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"manifest"),
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"sha256:hellosha"),
+    ]
+
+    # WHEN the user runs the image push command with any image
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["image", "push", "myrepo.net/myimage:latest"])
+
+    # THEN the command exits successfully despite pruning failure
+    assert result.exit_code == 0, result.output
+
+    # AND the warning message about pruning failure is shown
+    assert "Docker image pruning failed" in result.output
+
+    # AND the success message is still shown
+    assert "Image successfully pushed" in result.output

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_cli/tests/test_image_push_error_handling.py

@@ -18,6 +18,7 @@ def test_image_push_docker_service_token_401_error(mock_docker_from_env, mock_ru
     mock_docker_client.images.get.return_value.id = "test-image-id"
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -43,6 +44,7 @@ def test_image_push_docker_service_token_500_error(mock_docker_from_env, mock_ru
     mock_docker_client.images.get.return_value.id = "test-image-id"
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -70,6 +72,7 @@ def test_image_push_docker_service_token_other_error(mock_docker_from_env, mock_
     mock_docker_client.images.get.return_value.id = "test-image-id"
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -98,6 +101,7 @@ def test_image_push_container_image_401_error(mock_docker_from_env, mock_run, ve
     mock_docker_client.images.push.return_value = ["pushed"]
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -129,6 +133,7 @@ def test_image_push_container_image_500_error(mock_docker_from_env, mock_run, ve
     mock_docker_client.images.push.return_value = ["pushed"]
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),
@@ -162,6 +167,7 @@ def test_image_push_container_image_other_error(mock_docker_from_env, mock_run,
     mock_docker_client.images.push.return_value = ["pushed"]
 
     mock_run.side_effect = [
+        subprocess.CompletedProcess(args="", returncode=0, stdout=b"Pruning successful"),
         subprocess.CompletedProcess(
             args="", returncode=0, stdout=b'{"manifests": [{"platform": {"architecture": "amd64"}}]}'
         ),

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py

@@ -1,8 +1,7 @@
-from vellum.client.types.code_execution_package import CodeExecutionPackage
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
-from vellum.workflows.nodes.experimental.tool_calling_node.node import ToolCallingNode
+from vellum.workflows.nodes.displayable.tool_calling_node.node import ToolCallingNode
 from vellum.workflows.state.base import BaseState
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
@@ -123,105 +122,3 @@ def test_serialize_node__prompt_inputs__mixed_values():
             ],
         },
     }
-
-
-def test_serialize_node__function_configs():
-    # GIVEN a tool calling node with function packages
-    def foo():
-        pass
-
-    def bar():
-        pass
-
-    class MyToolCallingNode(ToolCallingNode):
-        functions = [foo, bar]
-        function_configs = {
-            "foo": {
-                "runtime": "PYTHON_3_11_6",
-                "packages": [
-                    CodeExecutionPackage(name="first_package", version="1.0.0", repository="test-repo"),
-                    CodeExecutionPackage(name="second_package", version="2.0.0", repository="test-repo"),
-                ],
-            },
-            "bar": {
-                "runtime": "PYTHON_3_11_6",
-                "packages": [CodeExecutionPackage(name="third_package", version="3.0.0")],
-            },
-        }
-
-    # AND a workflow with the tool calling node
-    class Workflow(BaseWorkflow):
-        graph = MyToolCallingNode
-
-    # WHEN the workflow is serialized
-    workflow_display = get_workflow_display(workflow_class=Workflow)
-    serialized_workflow: dict = workflow_display.serialize()
-
-    # THEN the node should properly serialize the function packages
-    my_tool_calling_node = next(
-        node
-        for node in serialized_workflow["workflow_raw_data"]["nodes"]
-        if node["id"] == str(MyToolCallingNode.__id__)
-    )
-
-    function_configs_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
-    )
-
-    assert function_configs_attribute == {
-        "id": "90cc5fc7-9fb3-450f-be5a-e90e7412a601",
-        "name": "function_configs",
-        "value": {
-            "type": "CONSTANT_VALUE",
-            "value": {
-                "type": "JSON",
-                "value": {
-                    "foo": {
-                        "runtime": "PYTHON_3_11_6",
-                        "packages": [
-                            {"version": "1.0.0", "name": "first_package", "repository": "test-repo"},
-                            {"version": "2.0.0", "name": "second_package", "repository": "test-repo"},
-                        ],
-                    },
-                    "bar": {
-                        "runtime": "PYTHON_3_11_6",
-                        "packages": [{"version": "3.0.0", "name": "third_package", "repository": None}],
-                    },
-                },
-            },
-        },
-    }
-
-
-def test_serialize_node__function_configs__none():
-    # GIVEN a tool calling node with no function configs
-    def foo():
-        pass
-
-    class MyToolCallingNode(ToolCallingNode):
-        functions = [foo]
-
-    # AND a workflow with the tool calling node
-    class Workflow(BaseWorkflow):
-        graph = MyToolCallingNode
-
-    # WHEN the workflow is serialized
-    workflow_display = get_workflow_display(workflow_class=Workflow)
-    serialized_workflow: dict = workflow_display.serialize()
-
-    # THEN the node should properly serialize the functions
-    my_tool_calling_node = next(
-        node
-        for node in serialized_workflow["workflow_raw_data"]["nodes"]
-        if node["id"] == str(MyToolCallingNode.__id__)
-    )
-
-    function_configs_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
-    )
-
-    assert function_configs_attribute == {
-        "id": "5f63f2aa-72d7-47ad-a552-630753418b7d",
-        "name": "function_configs",
-        "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-    }

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py

@@ -561,7 +561,7 @@ def test_serialize_workflow__try_wrapped():
             "code_input_id": "f2e8a4fa-b54e-41e9-b314-0e5443519ac7",
             "runtime_input_id": "19d64948-f22b-4103-a7f5-3add184b31cc",
             "output_type": "NUMBER",
-            "packages": [{"name": "openai", "version": "1.0.0"}],
+            "packages": [{"name": "openai", "version": "1.0.0", "repository": "test-repo"}],
             "output_id": "0fde9607-353f-42c2-85c4-20f720ebc1ec",
             "log_output_id": "7cac05e3-b7c3-475e-8df8-422b496c3398",
         },

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py

@@ -48,7 +48,7 @@ def test_serialize_workflow():
         },
         "base": {
             "name": "ToolCallingNode",
-            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+            "module": ["vellum", "workflows", "nodes", "displayable", "tool_calling_node", "node"],
         },
         "definition": {
             "name": "GetCurrentWeatherNode",
@@ -407,11 +407,6 @@ def test_serialize_workflow():
                     ],
                 },
             },
-            {
-                "id": "a4e3bc9f-7112-4d2f-94fb-7362a85db27a",
-                "name": "function_configs",
-                "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-            },
             {
                 "id": "1668419e-a193-43a5-8a97-3394e89bf278",
                 "name": "max_prompt_iterations",

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py

@@ -48,7 +48,7 @@ def test_serialize_workflow():
         },
         "base": {
             "name": "ToolCallingNode",
-            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+            "module": ["vellum", "workflows", "nodes", "displayable", "tool_calling_node", "node"],
         },
         "definition": {
             "name": "GetCurrentWeatherNode",
@@ -152,22 +152,6 @@ def test_serialize_workflow():
                     },
                 },
             },
-            {
-                "id": "a4e3bc9f-7112-4d2f-94fb-7362a85db27a",
-                "name": "function_configs",
-                "value": {
-                    "type": "CONSTANT_VALUE",
-                    "value": {
-                        "type": "JSON",
-                        "value": {
-                            "get_current_weather": {
-                                "runtime": "PYTHON_3_11_6",
-                                "packages": [{"version": "2.26.0", "name": "requests", "repository": None}],
-                            }
-                        },
-                    },
-                },
-            },
             {
                 "id": "0f6dc102-3460-4963-91fa-7ba85d65ef7a",
                 "name": "prompt_inputs",

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "0.14.80"
+version = "0.14.82"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/__init__.py

@@ -121,6 +121,7 @@ from .types import (
     ExecutionNumberVellumValue,
     ExecutionSearchResultsVellumValue,
     ExecutionStringVellumValue,
+    ExecutionThinkingVellumValue,
     ExecutionVellumValue,
     ExternalInputDescriptor,
     ExternalTestCaseExecution,
@@ -487,6 +488,8 @@ from .types import (
     TestSuiteTestCaseReplacedBulkResult,
     TestSuiteTestCaseReplacedBulkResultData,
     TestSuiteTestCaseUpsertBulkOperationRequest,
+    ThinkingVellumValue,
+    ThinkingVellumValueRequest,
     TokenOverlappingWindowChunkerConfig,
     TokenOverlappingWindowChunkerConfigRequest,
     TokenOverlappingWindowChunking,
@@ -754,6 +757,7 @@ __all__ = [
     "ExecutionNumberVellumValue",
     "ExecutionSearchResultsVellumValue",
     "ExecutionStringVellumValue",
+    "ExecutionThinkingVellumValue",
     "ExecutionVellumValue",
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
@@ -1127,6 +1131,8 @@ __all__ = [
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
     "TestSuiteTestCaseUpsertBulkOperationRequest",
+    "ThinkingVellumValue",
+    "ThinkingVellumValueRequest",
     "TokenOverlappingWindowChunkerConfig",
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/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.80",
+            "X-Fern-SDK-Version": "0.14.82",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/client/types/__init__.py

@@ -125,6 +125,7 @@ from .execution_json_vellum_value import ExecutionJsonVellumValue
 from .execution_number_vellum_value import ExecutionNumberVellumValue
 from .execution_search_results_vellum_value import ExecutionSearchResultsVellumValue
 from .execution_string_vellum_value import ExecutionStringVellumValue
+from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
 from .execution_vellum_value import ExecutionVellumValue
 from .external_input_descriptor import ExternalInputDescriptor
 from .external_test_case_execution import ExternalTestCaseExecution
@@ -511,6 +512,8 @@ from .test_suite_test_case_replace_bulk_operation_request import TestSuiteTestCa
 from .test_suite_test_case_replaced_bulk_result import TestSuiteTestCaseReplacedBulkResult
 from .test_suite_test_case_replaced_bulk_result_data import TestSuiteTestCaseReplacedBulkResultData
 from .test_suite_test_case_upsert_bulk_operation_request import TestSuiteTestCaseUpsertBulkOperationRequest
+from .thinking_vellum_value import ThinkingVellumValue
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
 from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunkerConfig
 from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
@@ -741,6 +744,7 @@ __all__ = [
     "ExecutionNumberVellumValue",
     "ExecutionSearchResultsVellumValue",
     "ExecutionStringVellumValue",
+    "ExecutionThinkingVellumValue",
     "ExecutionVellumValue",
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
@@ -1107,6 +1111,8 @@ __all__ = [
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
     "TestSuiteTestCaseUpsertBulkOperationRequest",
+    "ThinkingVellumValue",
+    "ThinkingVellumValueRequest",
     "TokenOverlappingWindowChunkerConfig",
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",

vellum_ai-0.14.82/src/vellum/client/types/execution_thinking_vellum_value.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+import typing
+from .string_vellum_value import StringVellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class ExecutionThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    id: str = pydantic.Field()
+    """
+    The variable's uniquely identifying internal id.
+    """
+
+    name: str
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValue]
+
+    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_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/client/types/execution_vellum_value.py

@@ -9,6 +9,7 @@ from .execution_search_results_vellum_value import ExecutionSearchResultsVellumV
 from .execution_error_vellum_value import ExecutionErrorVellumValue
 from .execution_array_vellum_value import ExecutionArrayVellumValue
 from .execution_function_call_vellum_value import ExecutionFunctionCallVellumValue
+from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
 
 ExecutionVellumValue = typing.Union[
     ExecutionStringVellumValue,
@@ -19,4 +20,5 @@ ExecutionVellumValue = typing.Union[
     ExecutionErrorVellumValue,
     ExecutionArrayVellumValue,
     ExecutionFunctionCallVellumValue,
+    ExecutionThinkingVellumValue,
 ]

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/client/types/prompt_output.py

@@ -5,5 +5,8 @@ from .string_vellum_value import StringVellumValue
 from .json_vellum_value import JsonVellumValue
 from .error_vellum_value import ErrorVellumValue
 from .function_call_vellum_value import FunctionCallVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 
-PromptOutput = typing.Union[StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue]
+PromptOutput = typing.Union[
+    StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue, ThinkingVellumValue
+]

vellum_ai-0.14.82/src/vellum/client/types/thinking_vellum_value.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .string_vellum_value import StringVellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValue]
+
+    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_ai-0.14.82/src/vellum/client/types/thinking_vellum_value_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .string_vellum_value_request import StringVellumValueRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ThinkingVellumValueRequest(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValueRequest]
+
+    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_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/client/types/vellum_value.py

@@ -12,6 +12,7 @@ from .function_call_vellum_value import FunctionCallVellumValue
 from .error_vellum_value import ErrorVellumValue
 from .chat_history_vellum_value import ChatHistoryVellumValue
 from .search_results_vellum_value import SearchResultsVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 import typing
 
 if typing.TYPE_CHECKING:
@@ -28,4 +29,5 @@ VellumValue = typing.Union[
     "ArrayVellumValue",
     ChatHistoryVellumValue,
     SearchResultsVellumValue,
+    ThinkingVellumValue,
 ]

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/client/types/vellum_value_request.py

@@ -12,6 +12,7 @@ from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .error_vellum_value_request import ErrorVellumValueRequest
 from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .search_results_vellum_value_request import SearchResultsVellumValueRequest
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
 import typing
 
 if typing.TYPE_CHECKING:
@@ -28,4 +29,5 @@ VellumValueRequest = typing.Union[
     "ArrayVellumValueRequest",
     ChatHistoryVellumValueRequest,
     SearchResultsVellumValueRequest,
+    ThinkingVellumValueRequest,
 ]

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/client/types/vellum_variable_type.py

@@ -16,6 +16,7 @@ VellumVariableType = typing.Union[
         "AUDIO",
         "DOCUMENT",
         "NULL",
+        "THINKING",
     ],
     typing.Any,
 ]

vellum_ai-0.14.82/src/vellum/types/execution_thinking_vellum_value.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.execution_thinking_vellum_value import *

vellum_ai-0.14.82/src/vellum/types/thinking_vellum_value.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.thinking_vellum_value import *

vellum_ai-0.14.82/src/vellum/types/thinking_vellum_value_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.thinking_vellum_value_request import *

{vellum_ai-0.14.80 → vellum_ai-0.14.82}/src/vellum/workflows/nodes/displayable/__init__.py

@@ -13,6 +13,7 @@ from .note_node import NoteNode
 from .prompt_deployment_node import PromptDeploymentNode
 from .search_node import SearchNode
 from .subworkflow_deployment_node import SubworkflowDeploymentNode
+from .tool_calling_node import ToolCallingNode
 
 __all__ = [
     "APINode",
@@ -29,5 +30,6 @@ __all__ = [
     "PromptDeploymentNode",
     "SearchNode",
     "TemplatingNode",
+    "ToolCallingNode",
     "FinalOutputNode",
 ]