vellum-ai 1.8.6__py3-none-any.whl → 1.9.0__py3-none-any.whl

vellum/workflows/triggers/base.py

@@ -200,6 +200,16 @@ class BaseTrigger(ABC, metaclass=BaseTriggerMeta):
 
     __id__: UUID
 
+    def __init__(self, **kwargs: Any):
+        """
+        Initialize trigger with event data.
+
+        Args:
+            **kwargs: Arbitrary keyword arguments passed during trigger instantiation.
+                     Subclasses may use these to populate trigger attributes.
+        """
+        self._event_data = kwargs
+
     @classmethod
     def attribute_references(cls) -> Dict[str, "TriggerAttributeReference[Any]"]:
         """Return class-level trigger attribute descriptors keyed by attribute name."""

vellum/workflows/triggers/integration.py

@@ -1,67 +1,186 @@
 from abc import ABC
-from typing import ClassVar, Optional
+from typing import Any, ClassVar, Dict
 
-from vellum.workflows.triggers.base import BaseTrigger
+from vellum.workflows.constants import VellumIntegrationProviderType
+from vellum.workflows.references.trigger import TriggerAttributeReference
+from vellum.workflows.triggers.base import BaseTrigger, BaseTriggerMeta
 
 
-class IntegrationTrigger(BaseTrigger, ABC):
+class IntegrationTriggerMeta(BaseTriggerMeta):
     """
-    Base class for integration-based triggers (Slack, Email, etc.).
+    Custom metaclass for IntegrationTrigger.
 
-    Integration triggers:
-    - Are initiated by external events (webhooks, API calls)
-    - Produce attributes that downstream nodes can reference
-    - Require configuration (auth, webhooks, etc.)
+    This metaclass extends BaseTriggerMeta to automatically convert type annotations
+    into TriggerAttributeReference objects during class creation. This enables trigger
+    attributes to be referenced in workflow graphs while maintaining type safety.
+    """
+
+    def __new__(mcs, name: str, bases: tuple, namespace: dict, **kwargs: Any) -> "IntegrationTriggerMeta":
+        """Create a new trigger class and set up attribute references."""
+        cls = super().__new__(mcs, name, bases, namespace, **kwargs)
+
+        # Process __annotations__ to create TriggerAttributeReference for each attribute
+        # Only process if class has Config and annotations
+        has_config = hasattr(cls, "Config") and "Config" in namespace
+        if has_config and hasattr(cls, "__annotations__"):
+            # Create TriggerAttributeReference for each annotated attribute
+            for attr_name, attr_type in cls.__annotations__.items():
+                # Skip special attributes and Config
+                if attr_name.startswith("_") or attr_name == "Config":
+                    continue
+
+                # Create reference with proper type
+                reference = TriggerAttributeReference(
+                    name=attr_name, types=(attr_type,), instance=None, trigger_class=cls
+                )
+                # Set as class attribute so it's directly accessible
+                setattr(cls, attr_name, reference)
+
+        return cls
+
+
+class IntegrationTrigger(BaseTrigger, ABC, metaclass=IntegrationTriggerMeta):
+    """
+    Base class for Vellum-managed integration triggers.
+
+    Subclasses define two types of attributes:
+    1. **Config class**: Specifies how the trigger is configured (provider, integration_name, slug)
+       - These are configuration details users shouldn't need to interact with directly
+    2. **Top-level type annotations**: Define the webhook event payload structure (message, user, channel, etc.)
+       - These become TriggerAttributeReference that can be referenced in workflow nodes
 
     Examples:
-        # Define an integration trigger
-        class MyIntegrationTrigger(IntegrationTrigger):
-            data: str
-
-            def __init__(self, event_data: dict):
-                super().__init__(event_data)
-                self.data = event_data.get("data", "")
-
-        # Use in workflow
-        class MyWorkflow(BaseWorkflow):
-            graph = MyIntegrationTrigger >> ProcessNode
-
-        # Reference trigger attributes in nodes
-        class ProcessNode(BaseNode):
-            class Outputs(BaseNode.Outputs):
-                result = MyIntegrationTrigger.data
-
-    Note:
-        Unlike ManualTrigger, integration triggers provide structured attributes
-        that downstream nodes can reference directly.
+        Create a Slack trigger:
+            >>> class SlackNewMessageTrigger(IntegrationTrigger):
+            ...     # Event attributes (webhook payload structure)
+            ...     message: str
+            ...     user: str
+            ...     channel: str
+            ...     timestamp: float
+            ...
+            ...     # Configuration (how trigger is set up)
+            ...     class Config(IntegrationTrigger.Config):
+            ...         provider = VellumIntegrationProviderType.COMPOSIO
+            ...         integration_name = "SLACK"
+            ...         slug = "slack_new_message"
+
+        Use in workflow graph:
+            >>> class MyWorkflow(BaseWorkflow):
+            ...     graph = SlackNewMessageTrigger >> ProcessMessageNode
+
+        Reference trigger attributes in nodes:
+            >>> class ProcessNode(BaseNode):
+            ...     class Outputs(BaseNode.Outputs):
+            ...         text = SlackNewMessageTrigger.message
+            ...         channel = SlackNewMessageTrigger.channel
+
+        Instantiate for testing:
+            >>> trigger = SlackNewMessageTrigger(
+            ...     message="Hello world",
+            ...     channel="C123456",
+            ...     user="U123",
+            ...     timestamp=1234567890.0,
+            ... )
+            >>> trigger.message
+            'Hello world'
     """
 
-    # Configuration that can be set at runtime
-    config: ClassVar[Optional[dict]] = None
+    class Config:
+        """
+        Configuration for IntegrationTrigger subclasses.
 
-    def __init__(self, event_data: dict):
+        Defines how the trigger connects to the integration provider. These settings
+        specify which integration and which specific trigger type to use.
         """
-        Initialize trigger with event data from external system.
 
-        Subclasses should override this method to parse external
-        event payloads (e.g., Slack webhooks, email notifications) and
-        populate trigger attributes.
+        provider: ClassVar[VellumIntegrationProviderType]
+        integration_name: ClassVar[str]
+        slug: ClassVar[str]
 
-        Args:
-            event_data: Raw event data from the external system
+    def __init_subclass__(cls, **kwargs: Any) -> None:
+        """Validate that subclasses define required Config class with all required fields."""
+        super().__init_subclass__(**kwargs)
+
+        # Skip validation for the base class itself
+        if cls.__name__ == "IntegrationTrigger":
+            return
+
+        # Require Config class with required fields
+        if not hasattr(cls, "Config") or cls.Config is IntegrationTrigger.Config:
+            raise TypeError(
+                f"{cls.__name__} must define a nested Config class. "
+                f"Example:\n"
+                f"  class {cls.__name__}(IntegrationTrigger):\n"
+                f"      message: str\n"
+                f"      class Config(IntegrationTrigger.Config):\n"
+                f"          provider = VellumIntegrationProviderType.COMPOSIO\n"
+                f"          integration_name = 'SLACK'\n"
+                f"          slug = 'slack_new_message'"
+            )
+
+        # Validate Config class has required fields
+        config_cls = cls.Config
+        required_fields = ["provider", "integration_name", "slug"]
+        for field in required_fields:
+            if not hasattr(config_cls, field):
+                raise TypeError(
+                    f"{cls.__name__}.Config must define '{field}'. " f"Required fields: {', '.join(required_fields)}"
+                )
+
+    def __init__(self, **kwargs: Any):
+        """
+        Initialize trigger with event data from the integration.
+
+        The trigger dynamically populates its attributes based on the kwargs
+        dictionary keys. Any key in kwargs becomes an accessible attribute.
 
         Examples:
-            >>> class MyTrigger(IntegrationTrigger):
-            ...     data: str
+            >>> class SlackTrigger(IntegrationTrigger):
+            ...     message: str
+            ...     channel: str
+            ...     user: str
             ...
-            ...     def __init__(self, event_data: dict):
-            ...         super().__init__(event_data)
-            ...         self.data = event_data.get("data", "")
-            >>>
-            >>> trigger = MyTrigger({"data": "hello"})
-            >>> state = workflow.get_default_state()
-            >>> trigger.bind_to_state(state)
-            >>> MyTrigger.data.resolve(state)
-            'hello'
+            ...     class Config(IntegrationTrigger.Config):
+            ...         provider = VellumIntegrationProviderType.COMPOSIO
+            ...         integration_name = "SLACK"
+            ...         slug = "slack_new_message"
+            >>> trigger = SlackTrigger(
+            ...     message="Hello",
+            ...     channel="C123",
+            ...     user="U456"
+            ... )
+            >>> trigger.message
+            'Hello'
+            >>> trigger.channel
+            'C123'
         """
-        self._event_data = event_data
+        super().__init__(**kwargs)
+
+        # Dynamically populate instance attributes from kwargs.
+        # This allows any key in kwargs to become an accessible attribute:
+        # kwargs={"message": "Hi"} → trigger.message == "Hi"
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def to_trigger_attribute_values(self) -> Dict["TriggerAttributeReference[Any]", Any]:
+        """
+        Materialize attribute descriptor/value pairs for this trigger instance.
+
+        For IntegrationTrigger, this includes all dynamic attributes from event_data.
+        """
+        attribute_values: Dict["TriggerAttributeReference[Any]", Any] = {}
+
+        # Unlike the base class which iterates over type(self) (predefined annotations),
+        # we iterate over event_data keys since our attributes are discovered dynamically
+        # from the actual event data received during workflow execution.
+        # The base class approach: for reference in type(self)
+        # Our approach: for attr_name in self._event_data.keys()
+        for attr_name in self._event_data.keys():
+            # Get the class-level reference for this attribute (created by __new__ from annotations)
+            # Unknown keys can appear in webhook payloads, so gracefully skip them if the
+            # trigger class doesn't expose a corresponding reference.
+            reference = getattr(type(self), attr_name, None)
+            if isinstance(reference, TriggerAttributeReference):
+                attribute_values[reference] = getattr(self, attr_name)
+
+        return attribute_values

vellum/workflows/triggers/tests/test_integration.py

@@ -1,5 +1,8 @@
 """Tests for IntegrationTrigger base class."""
 
+from typing import Any
+
+from vellum.workflows.constants import VellumIntegrationProviderType
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.references.trigger import TriggerAttributeReference
 from vellum.workflows.state.base import BaseState
@@ -9,7 +12,7 @@ from vellum.workflows.triggers.integration import IntegrationTrigger
 def test_integration_trigger__can_be_instantiated_as_base():
     """IntegrationTrigger can be instantiated as a base class."""
     # WHEN we instantiate IntegrationTrigger directly
-    trigger = IntegrationTrigger({"test": "data"})
+    trigger = IntegrationTrigger(test="data")
 
     # THEN it creates an instance with event data stored
     assert trigger._event_data == {"test": "data"}
@@ -20,7 +23,10 @@ def test_integration_trigger__can_be_instantiated():
 
     # GIVEN IntegrationTrigger with concrete implementation
     class TestTrigger(IntegrationTrigger):
-        pass
+        class Config(IntegrationTrigger.Config):
+            provider = VellumIntegrationProviderType.COMPOSIO
+            integration_name = "test"
+            slug = "test"
 
     # THEN it can be instantiated (even though base is ABC, concrete subclasses work)
     assert TestTrigger is not None
@@ -33,12 +39,17 @@ def test_integration_trigger__can_be_subclassed():
     class TestTrigger(IntegrationTrigger):
         data: str
 
-        def __init__(self, event_data: dict):
-            super().__init__(event_data)
-            self.data = event_data.get("data", "")
+        class Config(IntegrationTrigger.Config):
+            provider = VellumIntegrationProviderType.COMPOSIO
+            integration_name = "test"
+            slug = "test"
+
+        def __init__(self, **kwargs: Any):
+            super().__init__(**kwargs)
+            self.data = kwargs.get("data", "")
 
     # WHEN we create a trigger instance
-    result = TestTrigger({"data": "test"})
+    result = TestTrigger(data="test")
 
     # THEN it returns the expected trigger instance with populated attributes
     assert result.data == "test"
@@ -50,9 +61,14 @@ def test_integration_trigger__attribute_reference():
     class TestTrigger(IntegrationTrigger):
         value: str
 
-        def __init__(self, event_data: dict):
-            super().__init__(event_data)
-            self.value = event_data.get("value", "")
+        class Config(IntegrationTrigger.Config):
+            provider = VellumIntegrationProviderType.COMPOSIO
+            integration_name = "test"
+            slug = "test"
+
+        def __init__(self, **kwargs: Any):
+            super().__init__(**kwargs)
+            self.value = kwargs.get("value", "")
 
     reference = TestTrigger.value
     assert isinstance(reference, TriggerAttributeReference)
@@ -61,7 +77,7 @@ def test_integration_trigger__attribute_reference():
     assert reference == TestTrigger.attribute_references()["value"]
 
     state = BaseState()
-    trigger = TestTrigger({"value": "data"})
+    trigger = TestTrigger(value="data")
     trigger.bind_to_state(state)
     assert reference.resolve(state) == "data"
 
@@ -73,9 +89,14 @@ def test_integration_trigger__graph_syntax():
     class TestTrigger(IntegrationTrigger):
         value: str
 
-        def __init__(self, event_data: dict):
-            super().__init__(event_data)
-            self.value = event_data.get("value", "")
+        class Config(IntegrationTrigger.Config):
+            provider = VellumIntegrationProviderType.COMPOSIO
+            integration_name = "test"
+            slug = "test"
+
+        def __init__(self, **kwargs: Any):
+            super().__init__(**kwargs)
+            self.value = kwargs.get("value", "")
 
     class TestNode(BaseNode):
         pass
@@ -97,9 +118,14 @@ def test_integration_trigger__multiple_entrypoints():
     class TestTrigger(IntegrationTrigger):
         msg: str
 
-        def __init__(self, event_data: dict):
-            super().__init__(event_data)
-            self.msg = event_data.get("msg", "")
+        class Config(IntegrationTrigger.Config):
+            provider = VellumIntegrationProviderType.COMPOSIO
+            integration_name = "test"
+            slug = "test"
+
+        def __init__(self, **kwargs: Any):
+            super().__init__(**kwargs)
+            self.msg = kwargs.get("msg", "")
 
     class NodeA(BaseNode):
         pass
@@ -118,9 +144,12 @@ def test_integration_trigger__multiple_entrypoints():
 
 
 def test_integration_trigger__config_attribute():
-    """IntegrationTrigger has optional config attribute."""
+    """IntegrationTrigger requires Config class with provider, integration_name, and slug."""
 
     # GIVEN IntegrationTrigger
-    # THEN it has a config class variable
-    assert hasattr(IntegrationTrigger, "config")
-    assert IntegrationTrigger.config is None
+    # THEN it has a Config class with type annotations for required fields
+    assert hasattr(IntegrationTrigger, "Config")
+    assert hasattr(IntegrationTrigger.Config, "__annotations__")
+    assert "provider" in IntegrationTrigger.Config.__annotations__
+    assert "integration_name" in IntegrationTrigger.Config.__annotations__
+    assert "slug" in IntegrationTrigger.Config.__annotations__

vellum/workflows/workflows/base.py

@@ -714,6 +714,50 @@ class BaseWorkflow(Generic[InputsType, StateType], BaseExecutable, metaclass=_Ba
 
         return state_class(**state)
 
+    @classmethod
+    def deserialize_trigger(cls, trigger_id: Optional[UUID], inputs: dict) -> Union[InputsType, BaseTrigger]:
+        """
+        Deserialize a trigger from a trigger_id and inputs dict.
+
+        If trigger_id is None, returns an instance of the workflow's Inputs class.
+        Otherwise, finds a trigger class that matches the trigger_id and creates an instance of that.
+
+        Parameters
+        ----------
+        trigger_id: Optional[UUID]
+            The UUID of the trigger class to instantiate. If None, returns workflow Inputs.
+
+        inputs: dict
+            The inputs to pass to the trigger or Inputs constructor.
+
+        Returns
+        -------
+        Union[InputsType, BaseTrigger]
+            Either an instance of the workflow's Inputs class (if trigger_id is None)
+            or an instance of the matching trigger class.
+
+        Raises
+        ------
+        WorkflowInitializationException
+            If trigger_id is provided but no matching trigger class is found in the workflow.
+        """
+        if trigger_id is None:
+            inputs_class = cls.get_inputs_class()
+            return inputs_class(**inputs)
+
+        trigger_classes = []
+        for subgraph in cls.get_subgraphs():
+            for trigger_class in subgraph.triggers:
+                if trigger_class.__id__ == trigger_id:
+                    return trigger_class(**inputs)
+
+                trigger_classes.append(trigger_class)
+
+        raise WorkflowInitializationException(
+            message=f"No trigger class found with id {trigger_id} in workflow {cls.__name__}. "
+            f"Available trigger classes: {[trigger_class.__name__ for trigger_class in trigger_classes]}"
+        )
+
     @staticmethod
     def load_from_module(module_path: str) -> Type["BaseWorkflow"]:
         workflow_path = f"{module_path}.workflow"

{vellum_ai-1.8.6.dist-info → vellum_ai-1.9.0.dist-info}/METADATA

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

{vellum_ai-1.8.6.dist-info → vellum_ai-1.9.0.dist-info}/RECORD

@@ -110,10 +110,10 @@ vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_try_node_serialization.py,sha256=pLCyMScV88DTBXRH7jXaXOEA1GBq8NIipCUFwIAWnwI,2771
 vellum_ee/workflows/display/tests/workflow_serialization/test_complex_terminal_node_serialization.py,sha256=exT7U-axwtYgFylagScflSQLJEND51qIAx2UATju6JM,6023
 vellum_ee/workflows/display/tests/workflow_serialization/test_final_output_node_map_reference_serialization.py,sha256=vl3pxUJlrYRA8zzFJ-gRm7fe-5fviLNSIsUC7imnMqk,3502
+vellum_ee/workflows/display/tests/workflow_serialization/test_integration_trigger_serialization.py,sha256=BmJJi1FGf1t6illUhN3YVnGmS-rzDo8omadRKFA4aXQ,8983
 vellum_ee/workflows/display/tests/workflow_serialization/test_list_vellum_document_serialization.py,sha256=ZRcDhOSVKFHvt_rBkNSL7j3VLeWKQbH-KRoJWrtWD2s,2193
 vellum_ee/workflows/display/tests/workflow_serialization/test_manual_trigger_serialization.py,sha256=QjyxK94uPWaBmCmCWxQwuW0OopR06mVcsIjmsnr8UwE,3743
 vellum_ee/workflows/display/tests/workflow_serialization/test_terminal_node_any_serialization.py,sha256=4WAmSEJZlDBLPhsD1f4GwY9ahB9F6qJKGnL6j7ZYlzQ,1740
-vellum_ee/workflows/display/tests/workflow_serialization/test_vellum_integration_trigger_serialization.py,sha256=eG7_D76ErR7IrCpWRG4gLKoKGV7VfBL1buezNJ_wkZc,9038
 vellum_ee/workflows/display/tests/workflow_serialization/test_web_search_node_serialization.py,sha256=vbDFBrWUPeeW7cxjNA6SXrsHlYcbOAhlQ4C45Vdnr1c,3428
 vellum_ee/workflows/display/tests/workflow_serialization/test_workflow_input_parameterization_error.py,sha256=vAdmn3YTBDpo55znbydQxsgg9ASqHcvsUPwiBR_7wfo,1461
 vellum_ee/workflows/display/types.py,sha256=LgRIZeEtV7bQe-nvrC4A0T6vMooSWT1rFtw-uTn45BQ,3752
@@ -130,11 +130,11 @@ vellum_ee/workflows/display/utils/tests/test_expressions.py,sha256=s8aHAwuuJS1_W
 vellum_ee/workflows/display/utils/vellum.py,sha256=Bt7kdLdXoBsHn5dVEY2uKcF542VL09jwu8J_30rl2vk,6413
 vellum_ee/workflows/display/vellum.py,sha256=J2mdJZ1sdLW535DDUkq_Vm8Z572vhuxHxVZF9deKSdk,391
 vellum_ee/workflows/display/workflows/__init__.py,sha256=JTB9ObEV3l4gGGdtfBHwVJtTTKC22uj-a-XjTVwXCyA,148
-vellum_ee/workflows/display/workflows/base_workflow_display.py,sha256=YA65D_PqJ52HkkhaLFfuE8kigaltd5IAFTlqLPtZRxs,52353
+vellum_ee/workflows/display/workflows/base_workflow_display.py,sha256=J3iOlEBMGkPxRtWpOjNiggwxptId8sAAKmO-up2c9Mw,52318
 vellum_ee/workflows/display/workflows/get_vellum_workflow_display_class.py,sha256=gxz76AeCqgAZ9D2lZeTiZzxY9eMgn3qOSfVgiqYcOh8,2028
 vellum_ee/workflows/display/workflows/tests/test_workflow_display.py,sha256=lg-c_3P3ldtqWq2VFsk_2Mkn3pVdXWuT59QpH7QwXVs,39764
 vellum_ee/workflows/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum_ee/workflows/server/virtual_file_loader.py,sha256=LYp29fZIemLV05GJVJWOvCZUr_HQPMVuO66Mf8N9AkU,5194
+vellum_ee/workflows/server/virtual_file_loader.py,sha256=oc5H5jlK2AqgrWAR716yWnVQRqJnbQHOPDQxAA4HMTc,6217
 vellum_ee/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/tests/local_files/__init__.py,sha256=UyP6kKkRqr9cTKHQF4MVLdLk5MM9GGcLuqxXsQGm22Y,51
 vellum_ee/workflows/tests/local_files/base_class.py,sha256=UuiC7J68MVr6A4949QYiBpXOLdsvFG_Cw1muEPiHT6I,298
@@ -157,14 +157,14 @@ vellum_ee/workflows/tests/local_workflow/workflow.py,sha256=A4qOzOPNwePYxWbcAgIP
 vellum_ee/workflows/tests/test_display_meta.py,sha256=PkXJVnMZs9GNooDkd59n4YTBAX3XGPQWeSSVbhehVFM,5112
 vellum_ee/workflows/tests/test_registry.py,sha256=B8xRIuEyLWfSqrYoPldNQXhKPfe50PllvtAZoI8-uPs,6066
 vellum_ee/workflows/tests/test_serialize_module.py,sha256=thzGsNzYCRXXdaC5yk1ZjtXrIO6uPdSnzdapKLCOsC8,6241
-vellum_ee/workflows/tests/test_server.py,sha256=abparWA7ViR0pkK4dFMdnlhKnmvd0BmSPJf4GJQXyUA,28283
-vellum_ee/workflows/tests/test_virtual_files.py,sha256=2K1jEHbWmYe9o3PnwEp9ZmVAbJUKpgd3yDlcWOl6bVg,4481
+vellum_ee/workflows/tests/test_server.py,sha256=6lXKu2HoX_UqYctelyAjycc5ZhD47F9PtsGV-rwwBq0,31251
+vellum_ee/workflows/tests/test_virtual_files.py,sha256=zErjlpAIv-gUd5C-NXsiu3DHrbxEWzvb3bNJW3LcGIA,4250
 vellum/__init__.py,sha256=i3_LxyUG6tOzzBIGg9KVV9SEWyZSQecpFqXonkVya5g,50006
 vellum/client/README.md,sha256=flqu57ubZNTfpq60CdLtJC9gp4WEkyjb_n_eZ4OYf9w,6497
 vellum/client/__init__.py,sha256=-nugZzQKoUJsStXe6PnOD__8kbDLKKokceDgpGxQ_q0,74576
 vellum/client/core/__init__.py,sha256=lTcqUPXcx4112yLDd70RAPeqq6tu3eFMe1pKOqkW9JQ,1562
 vellum/client/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-vellum/client/core/client_wrapper.py,sha256=mwe_YpJ1tIU1dr4fBWp5874l6UJJHXcHOvnarq2iHwU,2840
+vellum/client/core/client_wrapper.py,sha256=tCmMeMuphx-X1KzY8qTAmP7LOyqJR-JLSYZFBcsuOWU,2840
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/force_multipart.py,sha256=awxh5MtcRYe74ehY8U76jzv6fYM_w_D3Rur7KQQzSDk,429
@@ -499,7 +499,7 @@ vellum/client/types/integration.py,sha256=WqfHEskiMzNxuy5mmcXTARPQ4F3r99Jf-g-wVl
 vellum/client/types/integration_auth_config_integration.py,sha256=W3DrzVPwLrqm0wCYVF-sHyQPoKNxoliFgFJWEwW_Yqc,710
 vellum/client/types/integration_auth_config_integration_credential.py,sha256=lUNuJ1GyFps3M85Mff1C-SAeEacCMkJdmayFfYrTFXA,547
 vellum/client/types/integration_credential_access_type.py,sha256=sWkuDjW3aD7ApZ1xQ7Bu8g5kCIwqUqzm1TDliUQWeRI,178
-vellum/client/types/integration_name.py,sha256=5YatDLwjK5QttKIAgwc662WVJZyfVmK7D7DIqkmNZdw,1226
+vellum/client/types/integration_name.py,sha256=XLczMR8e4nOICkbVSHkh0THKrTIVCtnx_YcsqAHFZ7Q,1247
 vellum/client/types/integration_provider.py,sha256=lIh3yPyPEzmSAu8L4Gsd-iDkmDSNobo0_TB75zMtIXk,129
 vellum/client/types/integration_read.py,sha256=sUNCS01TIlHPJHEH3ZheIbPi-CplbFQ5XAV1QtOO1Gg,1035
 vellum/client/types/integration_trigger_context.py,sha256=PqzeyLBq8EK8GrEf5k-b-xQNDG0nYbyDI506UyTwA2I,1669
@@ -1900,7 +1900,7 @@ vellum/workflows/graph/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 vellum/workflows/graph/tests/test_graph.py,sha256=OLGkEQh-B1CucwSf4nOP79RFXB4R2isHo3LFlr-24yA,20132
 vellum/workflows/inputs/__init__.py,sha256=02pj0IbJkN1AxTreswK39cNi45tA8GWcAAdRJve4cuM,116
 vellum/workflows/inputs/base.py,sha256=79Dw3jiSnCVFeySg6rMFPIqPfErmGKY0XPhpW0dSY-E,5454
-vellum/workflows/inputs/dataset_row.py,sha256=Szdb_RKysCEBRfy4B5LoXoPiTZPEFp_0_RqRZUZRGbU,1076
+vellum/workflows/inputs/dataset_row.py,sha256=6T9RftdvNK0hnxbvmUftYg1OVmeU8eLuBIPZj2g0yQ8,1213
 vellum/workflows/inputs/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/inputs/tests/test_inputs.py,sha256=E6I2pEfGjCeQg0GAsjvKFEjO32bQ3H9VeRnLv3ry9O0,2828
 vellum/workflows/integrations/__init__.py,sha256=KoA7tGX_UiImu6erJKJ5tmLFpNFEqblsWFGPoWNsS1U,220
@@ -1976,9 +1976,9 @@ vellum/workflows/nodes/displayable/conditional_node/__init__.py,sha256=AS_EIqFdU
 vellum/workflows/nodes/displayable/conditional_node/node.py,sha256=2g32hWosE3zwVaK2UPFnVEer1Nbn04s3friF2rGztmE,1195
 vellum/workflows/nodes/displayable/conftest.py,sha256=K2kLM2JGAfcrmmd92u8DXInUO5klFdggPWblg5RVcx4,5729
 vellum/workflows/nodes/displayable/final_output_node/__init__.py,sha256=G7VXM4OWpubvSJtVkGmMNeqgb9GkM7qZT838eL18XU4,72
-vellum/workflows/nodes/displayable/final_output_node/node.py,sha256=ypetAGBaTjVozKqAos-C1LG5HHZ3iUekw7v2mnnUlOw,5175
+vellum/workflows/nodes/displayable/final_output_node/node.py,sha256=ITFmtuffQj4A01Bpbb8ntKQyknswKF5rfyT5MjnFinA,5236
 vellum/workflows/nodes/displayable/final_output_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/nodes/displayable/final_output_node/tests/test_node.py,sha256=Drj7tMNEBDzHIYQ5iLkIK0zHHj-NxiswCtefVbfDFRo,3418
+vellum/workflows/nodes/displayable/final_output_node/tests/test_node.py,sha256=pMfmoUtARDW68IbxhhVqtcqL2LSunJSjYKAaucQoLXA,4478
 vellum/workflows/nodes/displayable/guardrail_node/__init__.py,sha256=Ab5eXmOoBhyV4dMWdzh32HLUmnPIBEK_zFCT38C4Fng,68
 vellum/workflows/nodes/displayable/guardrail_node/node.py,sha256=axYUojar_kdB3gi4LG3g9euJ8VkOxNtiFxJNI46v-SQ,5869
 vellum/workflows/nodes/displayable/guardrail_node/test_node.py,sha256=SAGv6hSFcBwQkudn1VxtaKNsXSXWWELl3eK05zM6tS0,5410
@@ -2000,6 +2000,10 @@ vellum/workflows/nodes/displayable/search_node/__init__.py,sha256=hpBpvbrDYf43DE
 vellum/workflows/nodes/displayable/search_node/node.py,sha256=hbPsZhyXfq9dx0mfBKOyivluDmV2iXmDTT7loEyu9nI,2057
 vellum/workflows/nodes/displayable/search_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/search_node/tests/test_node.py,sha256=YXgIIAJHVQxrfyJ0gxeJC0fAJaic10_zbqvsS8hyZSc,9368
+vellum/workflows/nodes/displayable/set_state_node/__init__.py,sha256=9Qi3gXOg4koi3sADuDXMhA3_JTYBDx4Bd2hRhaoQ4XM,66
+vellum/workflows/nodes/displayable/set_state_node/node.py,sha256=PvIiJQdBmCuEO3dZ5azSIv3db7pArIci0VY2ZpWS240,2721
+vellum/workflows/nodes/displayable/set_state_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vellum/workflows/nodes/displayable/set_state_node/tests/test_node.py,sha256=dhhYASzfR2xeAxcFTzgaEcirZ0WIzL461TYeXuIqYag,6543
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/__init__.py,sha256=9yYM6001YZeqI1VOk1QuEM_yrffk_EdsO7qaPzINKds,92
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py,sha256=_4aJxlnSEQKUlL9DAXYdrv1ic0tC7IyvYV389HHGyvE,16281
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -2056,7 +2060,7 @@ vellum/workflows/resolvers/tests/test_resolver.py,sha256=PnUGzsulo1It_LjjhHsRNiI
 vellum/workflows/resolvers/types.py,sha256=Hndhlk69g6EKLh_LYg5ILepW5U_h_BYNllfzhS9k8p4,237
 vellum/workflows/runner/__init__.py,sha256=i1iG5sAhtpdsrlvwgH6B-m49JsINkiWyPWs8vyT-bqM,72
 vellum/workflows/runner/runner.py,sha256=mMJzRT1iMsJ8YLYcbTsrKH8l2VDxqlC-sD8xY7pIoGY,54834
-vellum/workflows/sandbox.py,sha256=mezSZmilR_fwR8164n8CEfzlMeQ55IqfapHp4ftImvQ,3212
+vellum/workflows/sandbox.py,sha256=W_jHJX0-Na1S_eCnwNsceEmlk7MiC-KOh7Hc2yHLWoA,3592
 vellum/workflows/state/__init__.py,sha256=yUUdR-_Vl7UiixNDYQZ-GEM_kJI9dnOia75TtuNEsnE,60
 vellum/workflows/state/base.py,sha256=8Zr7UIM_eC0O2N6w_1gYqyrjgJ9D9z-VqLFnBETEF7Q,26753
 vellum/workflows/state/context.py,sha256=L2Xqt8jL1RJCCuSVyKakuZQdgb_YBVKSW6n_3nSAcLE,10974
@@ -2066,18 +2070,17 @@ vellum/workflows/state/store.py,sha256=uVe-oN73KwGV6M6YLhwZMMUQhzTQomsVfVnb8V91g
 vellum/workflows/state/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/state/tests/test_state.py,sha256=zEVFIY2any41X2BA5Us_qqKpzH5HRqmyrUJ04GTO0pU,7484
 vellum/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/tests/test_dataset_row.py,sha256=S8aIiYU9TRzJ8GTl5qCjnJ-fuHdxatHJFGLlKTVHPr4,4174
-vellum/workflows/tests/test_sandbox.py,sha256=JKwaluI-lODQo7Ek9sjDstjL_WTdSqUlVik6ZVTfVOA,1826
+vellum/workflows/tests/test_dataset_row.py,sha256=eHJ52VEQ661QUBrBQxvrD75zHYv6KuaKVXpqggklPaw,4840
+vellum/workflows/tests/test_sandbox.py,sha256=flKIwW4BaOepm9WVLAsVil50SzJE6sJSxKS-XURRaqo,2998
 vellum/workflows/tests/test_undefined.py,sha256=zMCVliCXVNLrlC6hEGyOWDnQADJ2g83yc5FIM33zuo8,353
-vellum/workflows/tests/triggers/test_vellum_integration_trigger.py,sha256=24_wePSpjt7VdyjlLG23W_HHT9EtnsIrI-TW0zcVjY4,5172
-vellum/workflows/triggers/__init__.py,sha256=TPFTcfctEL-XP0GpQ6EWc-0XWolU3trab2S6VKKe9Bs,441
-vellum/workflows/triggers/base.py,sha256=xcai_kVu74hIH55RSEBNBvlrA3QfDeUzJjdg6oqqAcg,8683
-vellum/workflows/triggers/integration.py,sha256=hAWQMoIubosKgM56rrsAJVhPnc4MjA5YBPRpDgl6J08,2221
+vellum/workflows/tests/triggers/test_integration_trigger.py,sha256=3PEJs4RY4gFjZsfCEEJo8NBK-9NDIvIV1Ig68WCF4Uc,5020
+vellum/workflows/triggers/__init__.py,sha256=dNcF16tEEKSrEkAVmKDdlbpMYUwmagB7LPAWbVo4LUs,331
+vellum/workflows/triggers/base.py,sha256=T_IZ8-yo-S7Z9Oa2UFHMfE123gwdztyQEj-J5c1Phdg,9005
+vellum/workflows/triggers/integration.py,sha256=_coibUaWnHVwjdTyh9hHW4FhBrZW9NlJlOek5b3auos,8107
 vellum/workflows/triggers/manual.py,sha256=PgbZ92gcK25yz6REXm98zWic1QBfhxLKfGCeHpZEUx4,1266
 vellum/workflows/triggers/schedule.py,sha256=3XY519njXWvbBGoxOyQju9iAMVx8fYmce2eSD13bfWM,437
 vellum/workflows/triggers/tests/__init__.py,sha256=R8lag_iCRyulijHMK4e3Gf6YVB5NplfvwZeTkaRj8gQ,30
-vellum/workflows/triggers/tests/test_integration.py,sha256=q7EeftNwgI2EcnV8GwHTxCQ_SyZKHl2dIPCQOQ7yxSM,4046
-vellum/workflows/triggers/vellum_integration.py,sha256=AvqdAXRy5XWbFJtiyVLG0E82PslQrCex_srTGctPhFc,8419
+vellum/workflows/triggers/tests/test_integration.py,sha256=ENQp3oJ9CzgPEOxLXaWvBKgS-gp0mcYv1v6y4PyIiQQ,5246
 vellum/workflows/types/__init__.py,sha256=fZ3Xxly7YSsu4kCIYD5aYpYucNM97zTyInb9CA24mf0,102
 vellum/workflows/types/code_execution_node_wrappers.py,sha256=fewX9bqF_4TZuK-gZYIn12s31-k03vHMGRpvFAPm11Y,3206
 vellum/workflows/types/core.py,sha256=R7snCd7ci4tiRuHi5ALGh_5DIIF0T9eze3sf6EnJN-c,1126
@@ -2103,14 +2106,14 @@ vellum/workflows/utils/vellum_variables.py,sha256=X3lZn-EoWengRWBWRhTNW7hqbj7LkV
 vellum/workflows/utils/zip.py,sha256=HVg_YZLmBOTXKaDV3Xhaf3V6sYnfqqZXQ8CpuafkbPY,1181
 vellum/workflows/vellum_client.py,sha256=3iDR7VV_NgLSm1iZQCKDvrmfEaX1bOJiU15QrxyHpv0,1237
 vellum/workflows/workflows/__init__.py,sha256=KY45TqvavCCvXIkyCFMEc0dc6jTMOUci93U2DUrlZYc,66
-vellum/workflows/workflows/base.py,sha256=sxtMwfE1kYgdvZkxcFO8eM86Cw3Wp8ONdnOuDjI8PzQ,33573
+vellum/workflows/workflows/base.py,sha256=aiOe-f33iK5-aaLrHANr2zkO9o7Oj550oM0hfWKD1Oo,35285
 vellum/workflows/workflows/event_filters.py,sha256=OzaS1y_z1f7H4f4M914HttAfAuTiN0jXUmo1TUQagCY,2504
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=Boa-_m9ii2Qsa1RvVM-VYniF7zCpzGgEGy-OnPZkrHg,23941
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
 vellum/workflows/workflows/tests/test_event_filters.py,sha256=CPsgtn2F8QMuNMxN5MB6IwTY0y_8JWBCZsio75vxp6c,3638
-vellum_ai-1.8.6.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-1.8.6.dist-info/METADATA,sha256=kt1BJsAjvhgy9zxyX1L7VPkJK6b4Bs7obNJm6tuqJJI,5547
-vellum_ai-1.8.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-1.8.6.dist-info/entry_points.txt,sha256=xVavzAKN4iF_NbmhWOlOkHluka0YLkbN_pFQ9pW3gLI,117
-vellum_ai-1.8.6.dist-info/RECORD,,
+vellum_ai-1.9.0.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-1.9.0.dist-info/METADATA,sha256=0gzlisJU0BzjSlBsIPV2XjhXbSV4zMdm_HuCR7NjDPY,5547
+vellum_ai-1.9.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-1.9.0.dist-info/entry_points.txt,sha256=xVavzAKN4iF_NbmhWOlOkHluka0YLkbN_pFQ9pW3gLI,117
+vellum_ai-1.9.0.dist-info/RECORD,,

vellum_ee/workflows/display/tests/workflow_serialization/{test_vellum_integration_trigger_serialization.py → test_integration_trigger_serialization.py}

@@ -1,18 +1,18 @@
-"""Tests for VellumIntegrationTrigger serialization."""
+"""Tests for IntegrationTrigger serialization."""
 
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.state.base import BaseState
-from vellum.workflows.triggers.vellum_integration import VellumIntegrationTrigger
+from vellum.workflows.triggers.integration import IntegrationTrigger
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
 
 def test_vellum_integration_trigger_serialization():
-    """VellumIntegrationTrigger subclass serializes with class name and module path."""
+    """IntegrationTrigger subclass serializes with class name and module path."""
 
-    # Create a custom VellumIntegrationTrigger subclass
-    class SlackMessageTrigger(VellumIntegrationTrigger):
+    # Create a custom IntegrationTrigger subclass
+    class SlackMessageTrigger(IntegrationTrigger):
         """Custom Slack message trigger."""
 
         message: str
@@ -71,7 +71,7 @@ def test_vellum_integration_trigger_serialization():
 def test_vellum_integration_trigger_id_consistency():
     """Validates trigger and attribute IDs match between definitions and references."""
 
-    class SlackMessageTrigger(VellumIntegrationTrigger):
+    class SlackMessageTrigger(IntegrationTrigger):
         message: str
         channel: str
 
@@ -147,7 +147,7 @@ def test_vellum_integration_trigger_id_consistency():
 def test_trigger_module_paths_are_canonical():
     """Validates trigger module_path and class_name for consistent codegen."""
 
-    class TestSlackTrigger(VellumIntegrationTrigger):
+    class TestSlackTrigger(IntegrationTrigger):
         message: str
 
         class Config:
@@ -184,7 +184,7 @@ def test_trigger_module_paths_are_canonical():
 def test_integration_trigger_no_entrypoint_node():
     """IntegrationTrigger-only workflows use trigger ID in edges."""
 
-    class SlackMessageTrigger(VellumIntegrationTrigger):
+    class SlackMessageTrigger(IntegrationTrigger):
         message: str
 
         class Config: