vellum-ai 0.14.75__py3-none-any.whl → 0.14.77__py3-none-any.whl

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.75",
+            "X-Fern-SDK-Version": "0.14.77",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

vellum/workflows/descriptors/base.py

@@ -17,6 +17,7 @@ if TYPE_CHECKING:
     from vellum.workflows.expressions.greater_than_or_equal_to import GreaterThanOrEqualToExpression
     from vellum.workflows.expressions.in_ import InExpression
     from vellum.workflows.expressions.is_blank import IsBlankExpression
+    from vellum.workflows.expressions.is_error import IsErrorExpression
     from vellum.workflows.expressions.is_nil import IsNilExpression
     from vellum.workflows.expressions.is_not_blank import IsNotBlankExpression
     from vellum.workflows.expressions.is_not_nil import IsNotNilExpression
@@ -358,3 +359,8 @@ class BaseDescriptor(Generic[_T]):
         from vellum.workflows.expressions.parse_json import ParseJsonExpression
 
         return ParseJsonExpression(expression=self)
+
+    def is_error(self) -> "IsErrorExpression":
+        from vellum.workflows.expressions.is_error import IsErrorExpression
+
+        return IsErrorExpression(expression=self)

vellum/workflows/descriptors/tests/test_utils.py

@@ -1,5 +1,6 @@
 import pytest
 
+from vellum.client.types.vellum_error import VellumError
 from vellum.workflows.descriptors.utils import resolve_value
 from vellum.workflows.errors.types import WorkflowError, WorkflowErrorCode
 from vellum.workflows.nodes.bases.base import BaseNode
@@ -96,6 +97,27 @@ class DummyNode(BaseNode[FixtureState]):
             ).does_not_contain("test"),
             False,
         ),
+        (
+            ConstantValueReference(
+                WorkflowError(
+                    message="This is a test",
+                    code=WorkflowErrorCode.USER_DEFINED_ERROR,
+                )
+            ).is_error(),
+            True,
+        ),
+        (
+            ConstantValueReference(
+                VellumError(
+                    message="This is a test",
+                    code="USER_DEFINED_ERROR",
+                )
+            ).is_error(),
+            True,
+        ),
+        (FixtureState.alpha.is_error(), False),
+        (FixtureState.eta.is_error(), False),
+        (DummyNode.Outputs.empty.is_error(), False),
         (ConstantValueReference('{"foo": "bar"}').parse_json(), {"foo": "bar"}),
         (ConstantValueReference('{"foo": "bar"}').parse_json()["foo"], "bar"),
         (ConstantValueReference("[1, 2, 3]").parse_json(), [1, 2, 3]),
@@ -150,6 +172,11 @@ class DummyNode(BaseNode[FixtureState]):
         "list_index",
         "error_contains",
         "error_does_not_contain",
+        "is_error_on_workflow_error",
+        "is_error_on_vellum_error",
+        "is_error_on_value",
+        "is_error_on_null",
+        "is_error_on_undefined",
         "parse_json_constant",
         "parse_json_accessor",
         "parse_json_list",

vellum/workflows/expressions/is_error.py

@@ -0,0 +1,23 @@
+from typing import Generic, TypeVar, Union
+
+from vellum.client.types.vellum_error import VellumError
+from vellum.workflows.descriptors.base import BaseDescriptor
+from vellum.workflows.descriptors.utils import resolve_value
+from vellum.workflows.errors.types import WorkflowError
+from vellum.workflows.state.base import BaseState
+
+_T = TypeVar("_T")
+
+
+class IsErrorExpression(BaseDescriptor[bool], Generic[_T]):
+    def __init__(
+        self,
+        *,
+        expression: Union[BaseDescriptor[_T], _T],
+    ) -> None:
+        super().__init__(name=f"{expression} is error", types=(bool,))
+        self._expression = expression
+
+    def resolve(self, state: "BaseState") -> bool:
+        expression = resolve_value(self._expression, state)
+        return isinstance(expression, (VellumError, WorkflowError))

vellum/workflows/nodes/experimental/tool_calling_node/tests/test_utils.py

@@ -46,4 +46,13 @@ def test_get_function_name_subworkflow_deployment():
 
     result = get_function_name(deployment_config)
 
-    assert result == "my-test-deployment"
+    assert result == "mytestdeployment"
+
+
+def test_get_function_name_subworkflow_deployment_uuid():
+    """Test subworkflow deployment with UUID."""
+    deployment_config = DeploymentDefinition(deployment="57f09beb-b463-40e0-bf9e-c972e664352f", release_tag="v1.0.0")
+
+    result = get_function_name(deployment_config)
+
+    assert result == "57f09bebb46340e0bf9ec972e664352f"

vellum/workflows/nodes/experimental/tool_calling_node/utils.py

@@ -156,7 +156,7 @@ def create_function_node(
         runtime: The runtime to use for code execution (default: "PYTHON_3_11_6")
     """
     if isinstance(function, DeploymentDefinition):
-        deployment = function.deployment
+        deployment = function.deployment_id or function.deployment_name
         release_tag = function.release_tag
 
         def execute_workflow_deployment_function(self) -> BaseNode.Outputs:
@@ -314,6 +314,7 @@ def main(arguments):
 
 def get_function_name(function: Tool) -> str:
     if isinstance(function, DeploymentDefinition):
-        return function.deployment
+        name = str(function.deployment_id or function.deployment_name)
+        return name.replace("-", "")
     else:
         return snake_case(function.__name__)

vellum/workflows/types/definition.py

@@ -75,3 +75,25 @@ VellumCodeResourceDefinition = Annotated[
 class DeploymentDefinition(UniversalBaseModel):
     deployment: str
     release_tag: str = "LATEST"
+
+    def _is_uuid(self) -> bool:
+        """Check if the deployment field is a valid UUID."""
+        try:
+            UUID(self.deployment)
+            return True
+        except ValueError:
+            return False
+
+    @property
+    def deployment_id(self) -> Optional[UUID]:
+        """Get the deployment ID if the deployment field is a UUID."""
+        if self._is_uuid():
+            return UUID(self.deployment)
+        return None
+
+    @property
+    def deployment_name(self) -> Optional[str]:
+        """Get the deployment name if the deployment field is not a UUID."""
+        if not self._is_uuid():
+            return self.deployment
+        return None

vellum/workflows/types/tests/test_definition.py

@@ -0,0 +1,33 @@
+import pytest
+from uuid import UUID
+
+from vellum.workflows.types.definition import DeploymentDefinition
+
+
+@pytest.mark.parametrize(
+    "deployment_value, expected_deployment_id, expected_deployment_name",
+    [
+        # Valid UUID string
+        (
+            "57f09beb-b463-40e0-bf9e-c972e664352f",
+            UUID("57f09beb-b463-40e0-bf9e-c972e664352f"),
+            None,
+        ),
+        # Name string
+        (
+            "tool-calling-subworkflow",
+            None,
+            "tool-calling-subworkflow",
+        ),
+    ],
+    ids=[
+        "valid_uuid",
+        "valid_name",
+    ],
+)
+def test_deployment_definition(deployment_value, expected_deployment_id, expected_deployment_name):
+    """Test that DeploymentDefinition properties correctly identify and extract UUID vs name."""
+    deployment = DeploymentDefinition(deployment=deployment_value)
+
+    assert deployment.deployment_id == expected_deployment_id
+    assert deployment.deployment_name == expected_deployment_name

vellum/workflows/utils/functions.py

@@ -201,7 +201,7 @@ def compile_workflow_deployment_function_definition(
     parameters = {"type": "object", "properties": properties, "required": required}
 
     return FunctionDefinition(
-        name=deployment,
+        name=deployment.replace("-", ""),
         description=description,
         parameters=parameters,
     )

{vellum_ai-0.14.75.dist-info → vellum_ai-0.14.77.dist-info}/METADATA

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

{vellum_ai-0.14.75.dist-info → vellum_ai-0.14.77.dist-info}/RECORD

@@ -10,15 +10,15 @@ vellum_cli/ping.py,sha256=p_BCCRjgPhng6JktuECtkDQLbhopt6JpmrtGoLnLJT8,1161
 vellum_cli/pull.py,sha256=udYyPlJ6VKDdh78rApNJOZgxHl82fcV6iGnRPSdX1LY,14750
 vellum_cli/push.py,sha256=ibAaf6zO41Qrgfholl18bCq8sWVYd2PDCiQsPkEYAFw,10296
 vellum_cli/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum_cli/tests/conftest.py,sha256=AFYZryKA2qnUuCPBxBKmHLFoPiE0WhBFFej9tNwSHdc,1526
+vellum_cli/tests/conftest.py,sha256=wx3PlJjVB0HRf5dr2b_idOIw27WPPl0J0FNbhIJJaVk,1689
 vellum_cli/tests/test_config.py,sha256=uvKGDc8BoVyT9_H0Z-g8469zVxomn6Oi3Zj-vK7O_wU,2631
-vellum_cli/tests/test_image_push.py,sha256=xXZjI-CDp166cmk5P1NTa-aogergeIbcdfK4_KBsoHc,9095
+vellum_cli/tests/test_image_push.py,sha256=qrlSZVMkPYc0ieqYaY2HCI8bBzwwL_Sx1G2nLb9_pSo,9199
 vellum_cli/tests/test_image_push_error_handling.py,sha256=_Wjfkn1orI2K4Ahzqz4u8T13or7NOX01K4BtcTuTIOM,7107
 vellum_cli/tests/test_init.py,sha256=8UOc_ThfouR4ja5cCl_URuLk7ohr9JXfCnG4yka1OUQ,18754
 vellum_cli/tests/test_main.py,sha256=qDZG-aQauPwBwM6A2DIu1494n47v3pL28XakTbLGZ-k,272
-vellum_cli/tests/test_ping.py,sha256=3ucVRThEmTadlV9LrJdCCrr1Ofj3rOjG6ue0BNR2UC0,2523
+vellum_cli/tests/test_ping.py,sha256=178EJHxPZtnnPMNXXynsQt8DIFhsrdc2bL17_YsG17M,2580
 vellum_cli/tests/test_pull.py,sha256=hxMbW_j0weDDrkzVGpvLpFcwNQdn-fxTv4wBHeYizzc,49904
-vellum_cli/tests/test_push.py,sha256=uyWknZ1CBkxxfKVdwcyv7fkGSxvvSNRdJegp5cV-_V4,35167
+vellum_cli/tests/test_push.py,sha256=rasQsxMYsSKgi_iaYz5cBD2ZBHusxhG9FAPx1Tn7mE4,35382
 vellum_ee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -144,7 +144,7 @@ vellum/client/README.md,sha256=CuGUYnaE0Imt0KqQ4sIPaUghCjLHkF3DdEvZWu14-8s,4807
 vellum/client/__init__.py,sha256=AYopGv2ZRVn3zsU8_km6KOvEHDbXiTPCVuYVI7bWvdA,120166
 vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/client/core/client_wrapper.py,sha256=cx7FO_qUE2Qlz3HNnJqi2W7mb70YPGjdNeHhflWGylY,1869
+vellum/client/core/client_wrapper.py,sha256=3d_rKaMktUFHpsEuOeVzjJA8WivXt89sMCtRJpKKYRc,1869
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -1507,9 +1507,9 @@ vellum/workflows/__init__.py,sha256=CssPsbNvN6rDhoLuqpEv7MMKGa51vE6dvAh6U31Pcio,
 vellum/workflows/constants.py,sha256=2yg4_uo5gpqViy3ZLSwfC8qTybleYCtOnhA4Rj6bacM,1310
 vellum/workflows/context.py,sha256=jvMuyeRluay8BQa7GX1TqUlmoHLCycAVYKkp87sfXSo,1644
 vellum/workflows/descriptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/descriptors/base.py,sha256=bvF3MWsc4Xyw5Z2s1A0fbsfMCebIbPYcGvbQ9uoa_Pg,14655
+vellum/workflows/descriptors/base.py,sha256=X47a4TClHknsnjs53DkiXnop_5uLGVor487oxhHuGo4,14902
 vellum/workflows/descriptors/exceptions.py,sha256=gUy4UD9JFUKSeQnQpeuDSLiRqWjWiIsxLahB7p_q3JY,54
-vellum/workflows/descriptors/tests/test_utils.py,sha256=xoojJMyG5WLG9xGtmUjirz3lDFCcDsAcxjrtbdG8dNE,6060
+vellum/workflows/descriptors/tests/test_utils.py,sha256=HJ5DoRz0sJvViGxyZ_FtytZjxN2J8xTkGtaVwCy6Q90,6928
 vellum/workflows/descriptors/utils.py,sha256=gmVXJjf2yWmvlYey41J2FZHeSou0JuCHKb3826K_Jok,3838
 vellum/workflows/edges/__init__.py,sha256=wSkmAnz9xyi4vZwtDbKxwlplt2skD7n3NsxkvR_pUus,50
 vellum/workflows/edges/edge.py,sha256=N0SnY3gKVuxImPAdCbPMPlHJIXbkQ3fwq_LbJRvVMFc,677
@@ -1544,6 +1544,7 @@ vellum/workflows/expressions/greater_than.py,sha256=1sbUH6Obf-VoBgs7ilIncwYBHYfX
 vellum/workflows/expressions/greater_than_or_equal_to.py,sha256=tsD5ZalB4SlryvEsvVtDkSr5Z13B2pABmHB8oHD8ojs,1276
 vellum/workflows/expressions/in_.py,sha256=RgiAIFntXGN4eWoOVBj1gqLymnBxSiw5hYD3TngF3dk,1123
 vellum/workflows/expressions/is_blank.py,sha256=vOOmK5poXmiNRVH7MR0feIFnL4rwKn7vmmTkJ9TcfVU,904
+vellum/workflows/expressions/is_error.py,sha256=YkgISyxo20Oko53rxkLeVLZx7HFXRk6mJpAzgXeZPRg,809
 vellum/workflows/expressions/is_nil.py,sha256=xCHwhKlm2UnfC-bVedmGgENCrzNtcn4ZeCYwNflVWbU,748
 vellum/workflows/expressions/is_not_blank.py,sha256=GJNTe8TKIbh4RwWPFuPwEQw0hbxg2MobHg8bcal4xWU,911
 vellum/workflows/expressions/is_not_nil.py,sha256=sVNWq_7GKExujpCB_bXEmRxm1tnj0GRDbFY4BtTV1Ew,769
@@ -1672,8 +1673,8 @@ vellum/workflows/nodes/experimental/tool_calling_node/__init__.py,sha256=S7OzT3I
 vellum/workflows/nodes/experimental/tool_calling_node/node.py,sha256=jwL1sbitmm1CpTOAEI0IIuc6VRr8d7yxUpS4Y5s9Bk8,5966
 vellum/workflows/nodes/experimental/tool_calling_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/experimental/tool_calling_node/tests/test_node.py,sha256=XK1H_QAT_nVFmFP442RYyPvpTfSgtU6kSGu3-OQPBNU,5072
-vellum/workflows/nodes/experimental/tool_calling_node/tests/test_utils.py,sha256=sF4ZfGK2uQNBVXC9yrnbFVzKIM-HWvXO1wak160MUTE,1386
-vellum/workflows/nodes/experimental/tool_calling_node/utils.py,sha256=ut3cdC7ezbxSvjMqHP0Km9yAVqOYoRztZVHShiq6zhk,12579
+vellum/workflows/nodes/experimental/tool_calling_node/tests/test_utils.py,sha256=-g90SdXscuikF7JP0lFGvSvPc8jl2vBuHwBeiYJIiXk,1719
+vellum/workflows/nodes/experimental/tool_calling_node/utils.py,sha256=U5JjMXg_NV_5XavydhOhsS4_EUoQ42KlEVBe_JiSR4g,12683
 vellum/workflows/nodes/mocks.py,sha256=a1FjWEIocseMfjzM-i8DNozpUsaW0IONRpZmXBoWlyc,10455
 vellum/workflows/nodes/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/tests/test_mocks.py,sha256=mfPvrs75PKcsNsbJLQAN6PDFoVqs9TmQxpdyFKDdO60,7837
@@ -1716,14 +1717,15 @@ vellum/workflows/tests/test_undefined.py,sha256=zMCVliCXVNLrlC6hEGyOWDnQADJ2g83y
 vellum/workflows/types/__init__.py,sha256=KxUTMBGzuRCfiMqzzsykOeVvrrkaZmTTo1a7SLu8gRM,68
 vellum/workflows/types/code_execution_node_wrappers.py,sha256=3MNIoFZKzVzNS5qFLVuDwMV17QJw72zo7NRf52yMq5A,3074
 vellum/workflows/types/core.py,sha256=iLJkMKf417kjwRncWdT_qsfJ-qBv5x58um7SfrydJbs,1266
-vellum/workflows/types/definition.py,sha256=guuCHZkto8bkknoMcjfXRhSaDuiNjx3SNkutPf1makc,2506
+vellum/workflows/types/definition.py,sha256=WSTi7DfwgIUMaHaNl0jV_9thw_wsLbzt5WRElJebaHw,3172
 vellum/workflows/types/generics.py,sha256=8jptbEx1fnJV0Lhj0MpCJOT6yNiEWeTOYOwrEAb5CRU,1576
 vellum/workflows/types/stack.py,sha256=h7NE0vXR7l9DevFBIzIAk1Zh59K-kECQtDTKOUunwMY,1314
 vellum/workflows/types/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vellum/workflows/types/tests/test_definition.py,sha256=5wh_WEnE51epkoo-4PE-JbPlg8OGJUNlaBVWa9TcNSw,993
 vellum/workflows/types/tests/test_utils.py,sha256=UnZog59tR577mVwqZRqqWn2fScoOU1H6up0EzS8zYhw,2536
 vellum/workflows/types/utils.py,sha256=axxHbPLsnjhEOnMZrc5YarFd-P2bnsacBDQGNCvY8OY,6367
 vellum/workflows/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/utils/functions.py,sha256=W4t-IXvCoIn_oVY_wEkph1xjpd_xpGzBwExWqgJXeho,7168
+vellum/workflows/utils/functions.py,sha256=ZN0rrIBF4R_KNt1CbRPVNGR36xEMUa1T7FkgZioou-Y,7185
 vellum/workflows/utils/names.py,sha256=QLUqfJ1tmSEeUwBKTTiv_Qk3QGbInC2RSmlXfGXc8Wo,380
 vellum/workflows/utils/pydantic_schema.py,sha256=eR_bBtY-T0pttJP-ARwagSdCOnwPUtiT3cegm2lzDTQ,1310
 vellum/workflows/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1740,8 +1742,8 @@ vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnad
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=fROqff6AZpCIzaSwOKSdtYy4XR0UZQ6ejxL3RJOSJVs,20447
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-0.14.75.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.14.75.dist-info/METADATA,sha256=Wclzf2hGrdv0YvzbG76xaKSH1JyLVRISdFarQai4Lks,5556
-vellum_ai-0.14.75.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-0.14.75.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.14.75.dist-info/RECORD,,
+vellum_ai-0.14.77.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.14.77.dist-info/METADATA,sha256=6DQyChVvOW8sRMyNEo1bB9d073wwWqAovQzhVZjigiU,5556
+vellum_ai-0.14.77.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-0.14.77.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.14.77.dist-info/RECORD,,

vellum_cli/tests/conftest.py

@@ -58,3 +58,9 @@ def mock_module(request) -> Generator[MockModuleResult, None, None]:
 
     os.chdir(current_dir)
     shutil.rmtree(temp_dir)
+
+
+@pytest.fixture
+def info_log_level(monkeypatch):
+    """Set log level to INFO for tests that request this fixture"""
+    monkeypatch.setenv("LOG_LEVEL", "INFO")

vellum_cli/tests/test_image_push.py

@@ -37,7 +37,7 @@ def mock_subprocess_run(mocker):
     return mocker.patch("subprocess.run")
 
 
-@pytest.mark.usefixtures("vellum_client")
+@pytest.mark.usefixtures("vellum_client", "info_log_level")
 def test_image_push__self_hosted_happy_path(mock_docker_from_env, mock_subprocess_run, monkeypatch):
     # GIVEN a self hosted vellum api URL env var
     monkeypatch.setenv("VELLUM_API_URL", "mycompany.api.com")
@@ -66,6 +66,7 @@ def test_image_push__self_hosted_happy_path(mock_docker_from_env, mock_subproces
     assert "Image successfully pushed" in result.output
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_image_push__self_hosted_happy_path__workspace_option(
     mock_docker_from_env, mock_subprocess_run, mock_httpx_transport, mock_temp_dir
 ):
@@ -172,6 +173,7 @@ def test_image_push__self_hosted_blocks_repo(mock_docker_from_env, monkeypatch):
     assert "For adding images to your self hosted install you must include" in result.output
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_image_push_with_source_success(
     mock_docker_from_env, mock_subprocess_run, vellum_client, monkeypatch, mock_temp_dir
 ):

vellum_cli/tests/test_ping.py

@@ -1,3 +1,4 @@
+import pytest
 from datetime import datetime
 
 from click.testing import CliRunner
@@ -8,6 +9,7 @@ from vellum.client.types.workspace_read import WorkspaceRead
 from vellum_cli import main as cli_main
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_ping__happy_path(vellum_client):
     # GIVEN a cli
     runner = CliRunner()

vellum_cli/tests/test_push.py

@@ -140,6 +140,7 @@ def test_push__happy_path(mock_module, vellum_client, base_command):
     assert extracted_files["workflow.py"] == workflow_py_file_content
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_push__verify_default_url_in_raw_httpx_transport(mock_module, mock_httpx_transport):
     # GIVEN a single workflow configured
     module = mock_module.module
@@ -396,6 +397,7 @@ def test_push__deployment(mock_module, vellum_client, base_command):
     assert extracted_files["workflow.py"] == workflow_py_file_content
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_push__dry_run_option_returns_report(mock_module, vellum_client):
     # GIVEN a single workflow configured
     temp_dir = mock_module.temp_dir
@@ -459,6 +461,7 @@ class ExampleWorkflow(BaseWorkflow):
     assert "iterable_item_added" in result.output
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_push__dry_run_option_no_errors_returns_success(mock_module, vellum_client):
     """Test that dry-run returns exit code 0 when there are no errors or diffs"""
     # GIVEN a workflow module with a valid workflow (using the same pattern as happy path test)
@@ -490,6 +493,7 @@ def test_push__dry_run_option_no_errors_returns_success(mock_module, vellum_clie
     assert "## Proposed Diffs" in result.output
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_push__strict_option_returns_diffs(mock_module, vellum_client):
     # GIVEN a single workflow configured
     temp_dir = mock_module.temp_dir
@@ -733,6 +737,7 @@ MY_OTHER_VELLUM_API_KEY=aaabbbcccddd
         }
 
 
+@pytest.mark.usefixtures("info_log_level")
 def test_push__workspace_option__uses_different_api_url_env(mock_module, mock_httpx_transport):
     # GIVEN a single workflow configured
     temp_dir = mock_module.temp_dir