vellum-ai 1.8.5__tar.gz → 1.8.6__tar.gz

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/PKG-INFO

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

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/ee/vellum_ee/workflows/display/utils/expressions.py

@@ -57,7 +57,7 @@ from vellum.workflows.references.workflow_input import WorkflowInputReference
 from vellum.workflows.types.core import JsonArray, JsonObject
 from vellum.workflows.types.generics import is_workflow_class
 from vellum.workflows.utils.functions import compile_function_definition
-from vellum.workflows.utils.uuids import get_trigger_id, uuid4_from_hash
+from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.utils.exceptions import InvalidInputReferenceError, UnsupportedSerializationException
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileLoader
 
@@ -353,9 +353,8 @@ def serialize_value(executable_id: UUID, display_context: "WorkflowDisplayContex
         }
 
     if isinstance(value, TriggerAttributeReference):
-        # Generate trigger ID using get_trigger_id to ensure consistency with trigger definitions
         trigger_class = value.trigger_class
-        trigger_id = get_trigger_id(trigger_class)
+        trigger_id = trigger_class.__id__
 
         return {
             "type": "TRIGGER_ATTRIBUTE",

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/ee/vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -30,7 +30,7 @@ from vellum.workflows.triggers.vellum_integration import VellumIntegrationTrigge
 from vellum.workflows.types.core import Json, JsonArray, JsonObject
 from vellum.workflows.types.generics import WorkflowType
 from vellum.workflows.types.utils import get_original_base
-from vellum.workflows.utils.uuids import get_trigger_id, uuid4_from_hash
+from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum.workflows.utils.vellum_variables import primitive_type_to_vellum_variable_type
 from vellum.workflows.vellum_client import create_vellum_client
 from vellum_ee.workflows.display.base import (
@@ -202,7 +202,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
         if has_manual_trigger:
             # ManualTrigger: use trigger ID for ENTRYPOINT node (backward compatibility)
             trigger_class = manual_trigger_edges[0].trigger_class
-            entrypoint_node_id = get_trigger_id(trigger_class)
+            entrypoint_node_id = trigger_class.__id__
             entrypoint_node_source_handle_id = self.display_context.workflow_display.entrypoint_node_source_handle_id
 
             # Add ENTRYPOINT node for ManualTrigger workflows
@@ -386,7 +386,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
         if has_only_integration_trigger:
             # Use trigger ID directly as sourceNodeId (no ENTRYPOINT node)
             trigger_class = integration_trigger_edges[0].trigger_class
-            trigger_id = get_trigger_id(trigger_class)
+            trigger_id = trigger_class.__id__
             trigger_source_handle_id = trigger_id  # Use trigger ID as handle ID
 
             for target_node, entrypoint_display in self.display_context.entrypoint_displays.items():
@@ -550,7 +550,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
                 )
 
         # Return as a list with a single trigger object matching Django schema
-        trigger_id = get_trigger_id(trigger_class)
+        trigger_id = trigger_class.__id__
 
         # Serialize trigger attributes like node outputs
         attribute_references = trigger_class.attribute_references().values()

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/ee/vellum_ee/workflows/server/virtual_file_loader.py

@@ -30,7 +30,11 @@ class VirtualFileLoader(importlib.abc.Loader):
 
         if module_info:
             file_path, code = module_info
-            compiled = compile(code, file_path, "exec")
+            namespaced_path = (
+                file_path if file_path.startswith(f"{self.namespace}/") else f"{self.namespace}/{file_path}"
+            )
+            module.__file__ = namespaced_path
+            compiled = compile(code, namespaced_path, "exec")
             exec(compiled, module.__dict__)
 
     def get_source(self, fullname):
@@ -58,6 +62,18 @@ class VirtualFileLoader(importlib.abc.Loader):
             if code is not None:
                 return file_path, code
 
+            relative_name = self._to_relative(fullname)
+            if relative_name is not None and self._is_package_directory(relative_name):
+                return self._generate_init_content(relative_name)
+
+        return None
+
+    def _to_relative(self, fullname: str) -> Optional[str]:
+        """Convert a fully qualified module name to a relative name without the namespace prefix."""
+        if fullname.startswith(self.namespace + "."):
+            return fullname[len(self.namespace) + 1 :]
+        elif fullname == self.namespace:
+            return ""
         return None
 
     def _get_file_path(self, fullname):
@@ -67,6 +83,34 @@ class VirtualFileLoader(importlib.abc.Loader):
         file_key_name = re.sub(r"^" + re.escape(self.namespace) + r"/", "", file_path)
         return self.files.get(file_key_name)
 
+    def _is_package_directory(self, fullname: str) -> bool:
+        """Check if directory contains .py files that should be treated as a package."""
+        directory_prefix = fullname.replace(".", "/") + "/"
+
+        # Exclude top-level display directory from auto-generation as it typically has
+        # specific __init__.py content that shouldn't be replaced with empty files.
+        if directory_prefix == "display/":
+            return False
+
+        for file_path in self.files.keys():
+            if file_path.startswith(directory_prefix):
+                if file_path.endswith(".py") and not file_path.endswith("__init__.py"):
+                    return True
+                remaining_path = file_path[len(directory_prefix) :]
+                if "/" in remaining_path:
+                    return True
+
+        return False
+
+    def _generate_init_content(self, fullname: str) -> tuple[str, str]:
+        """Auto-generate empty __init__.py content to mark directory as a package."""
+        directory_prefix = fullname.replace(".", "/") + "/"
+        file_path = directory_prefix + "__init__.py"
+
+        code = ""
+
+        return file_path, code
+
 
 class VirtualFileFinder(BaseWorkflowFinder):
     def __init__(self, files: dict[str, str], namespace: str, source_module: Optional[str] = None):

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/ee/vellum_ee/workflows/tests/test_server.py

@@ -34,13 +34,6 @@ from .nodes.start_node import StartNode
 
 class Workflow(BaseWorkflow):
     graph = StartNode
-""",
-        "nodes/__init__.py": """\
-from .start_node import StartNode
-
-__all__ = [
-    "StartNode",
-]
 """,
         "nodes/start_node.py": """\
 from vellum.workflows.nodes import BaseNode
@@ -98,11 +91,6 @@ class Workflow(BaseWorkflow):
 
     class Outputs(BaseWorkflow.Outputs):
        final_output = CodeExecutionNode.Outputs.result
-""",
-        "nodes/__init__.py": """
-from .code_execution_node import CodeExecutionNode
-
-__all__ = ["CodeExecutionNode"]
 """,
         "nodes/code_execution_node/__init__.py": """
 from typing import Any
@@ -174,11 +162,6 @@ class Workflow(BaseWorkflow):
 
     class Outputs(BaseWorkflow.Outputs):
        final_output = CodeExecutionNode.Outputs.result
-""",
-        "nodes/__init__.py": """
-from .code_execution_node import CodeExecutionNode
-
-__all__ = ["CodeExecutionNode"]
 """,
         "nodes/code_execution_node/__init__.py": """
 from typing import Union
@@ -250,11 +233,6 @@ class Workflow(BaseWorkflow):
 
     class Outputs(BaseWorkflow.Outputs):
        final_output = CodeExecutionNode.Outputs.result
-""",
-        "nodes/__init__.py": """
-from .code_execution_node import CodeExecutionNode
-
-__all__ = ["CodeExecutionNode"]
 """,
         "nodes/code_execution_node/__init__.py": """
 from typing import Union
@@ -326,11 +304,6 @@ class Workflow(BaseWorkflow):
 
     class Outputs(BaseWorkflow.Outputs):
        final_output = Subworkflow.Outputs.result
-""",
-        "nodes/__init__.py": """
-from .subworkflow import Subworkflow
-
-__all__ = ["Subworkflow"]
 """,
         "nodes/subworkflow/__init__.py": """
 from vellum.workflows.nodes.displayable import InlineSubworkflowNode
@@ -339,11 +312,6 @@ from .workflow import SubworkflowWorkflow
 
 class Subworkflow(InlineSubworkflowNode):
     subworkflow = SubworkflowWorkflow
-""",
-        "nodes/subworkflow/nodes/__init__.py": """
-from .code_execution_node import CodeExecutionNode
-
-__all__ = ["CodeExecutionNode"]
 """,
         "nodes/subworkflow/nodes/code_execution_node/__init__.py": """
 from typing import Union
@@ -424,11 +392,6 @@ class Workflow(BaseWorkflow):
     
     class Outputs(BaseWorkflow.Outputs):  # noqa: W293
        results = MapNode.Outputs.final_output
-""",
-        "nodes/__init__.py": """
-from .map_node import MapNode
-
-__all__ = ["MapNode"]
 """,
         "nodes/map_node/__init__.py": """
 from vellum.workflows.nodes.core.map_node import MapNode as BaseMapNode
@@ -439,11 +402,6 @@ class MapNode(BaseMapNode):
     items = ["foo", "bar", "baz"]
     subworkflow = MapNodeWorkflow
     max_concurrency = 4
-""",
-        "nodes/map_node/nodes/__init__.py": """
-from .code_execution_node import CodeExecutionNode
-
-__all__ = ["CodeExecutionNode"]
 """,
         "nodes/map_node/nodes/code_execution_node/__init__.py": """
 from typing import Union
@@ -520,7 +478,6 @@ from .nodes.broken_node import BrokenNode
 class Workflow(BaseWorkflow):
     graph = BrokenNode
 """,
-        "nodes/__init__.py": "",
         "nodes/broken_node.py": """\
 from vellum.workflows.nodes import BaseNode
 
@@ -559,7 +516,6 @@ from .nodes.broken_node import BrokenNode
 class Workflow(BaseWorkflow):
     graph = BrokenNode
 """,
-        "nodes/__init__.py": "",
         "nodes/broken_node.py": """\
 from vellum.workflows.nodes import BaseNode
 
@@ -819,19 +775,9 @@ class TestSubworkflowDeploymentNode(SubworkflowDeploymentNode):
     files = {
         "__init__.py": "",
         "workflow.py": parent_workflow_code,
-        "nodes/__init__.py": """
-from .subworkflow_deployment_node import TestSubworkflowDeploymentNode
-
-__all__ = ["TestSubworkflowDeploymentNode"]
-""",
         "nodes/subworkflow_deployment_node.py": parent_node_code,
         f"{expected_prefix}/__init__.py": "",
         f"{expected_prefix}/workflow.py": mock_workflow_code,
-        f"{expected_prefix}/nodes/__init__.py": """
-from .test_node import TestNode
-
-__all__ = ["TestNode"]
-""",
         f"{expected_prefix}/nodes/test_node.py": test_node_code,
     }
 
@@ -919,11 +865,6 @@ class TestSubworkflowDeploymentNode(SubworkflowDeploymentNode):
     subworkflow_files = {
         "__init__.py": "",
         "workflow.py": mock_workflow_code,
-        "nodes/__init__.py": """
-from .test_node import TestNode
-
-__all__ = ["TestNode"]
-""",
         "nodes/test_node.py": test_node_code,
     }
 
@@ -931,11 +872,6 @@ __all__ = ["TestNode"]
         "__init__.py": "",
         "inputs.py": inputs_code,
         "workflow.py": parent_workflow_code,
-        "nodes/__init__.py": """
-from .subworkflow_deployment_node import TestSubworkflowDeploymentNode
-
-__all__ = ["TestSubworkflowDeploymentNode"]
-""",
         "nodes/subworkflow_deployment_node.py": parent_node_code,
     }
 

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/ee/vellum_ee/workflows/tests/test_virtual_files.py

@@ -80,3 +80,54 @@ def test_base_class_dynamic_import(files):
         assert instance.id
     except Exception as e:
         pytest.fail(f"Failed to create an instance of BaseClass: {e}")
+
+
+def test_display_directory_not_auto_generated():
+    """
+    Test that the top-level display directory is NOT auto-generated with empty __init__.py.
+    Display directories typically have specific __init__.py content (e.g., "from .workflow import *")
+    that should not be replaced with empty auto-generated files.
+    """
+    # GIVEN a workflow with display/workflow.py but NO display/__init__.py
+    files = {
+        "__init__.py": "",
+        "workflow.py": """\
+from vellum.workflows import BaseWorkflow
+from vellum.workflows.nodes.bases import BaseNode
+
+class StartNode(BaseNode):
+    pass
+
+class Workflow(BaseWorkflow):
+    graph = StartNode
+""",
+        "display/workflow.py": """\
+from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
+
+class WorkflowDisplay(BaseWorkflowDisplay):
+    pass
+""",
+        # Note: NO "display/__init__.py" in files dict
+    }
+
+    namespace = str(uuid4())
+
+    # AND the virtual file loader is registered
+    sys.meta_path.append(VirtualFileFinder(files, namespace))
+
+    try:
+        # WHEN we try to resolve display/__init__.py
+        import importlib.util
+
+        spec = importlib.util.find_spec(f"{namespace}.display")
+
+        # THEN the spec should be None because we don't want to auto-generate display/__init__.py
+        # If the spec exists, it means an empty __init__.py was auto-generated (BAD)
+        assert spec is None, (
+            "display directory should NOT have auto-generated __init__.py. "
+            "Display directories require specific __init__.py content that shouldn't be empty."
+        )
+
+    finally:
+        # Clean up
+        sys.meta_path = [finder for finder in sys.meta_path if not isinstance(finder, VirtualFileFinder)]

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/pyproject.toml

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

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/__init__.py

@@ -13,6 +13,7 @@ from .client.types import (
     AdHocRejectedPromptExecutionMeta,
     AdHocStreamingPromptExecutionMeta,
     AddOpenaiApiKeyEnum,
+    ApiActorTypeEnum,
     ApiNodeResult,
     ApiNodeResultData,
     ApiRequestParentContext,
@@ -233,6 +234,7 @@ from .client.types import (
     IntegrationName,
     IntegrationProvider,
     IntegrationRead,
+    IntegrationTriggerContext,
     InvokedPort,
     IterationStateEnum,
     JinjaPromptBlock,
@@ -425,6 +427,7 @@ from .client.types import (
     ScenarioInputJsonVariableValue,
     ScenarioInputStringVariableValue,
     ScenarioInputVideoVariableValue,
+    ScheduledTriggerContext,
     SearchFiltersRequest,
     SearchNodeResult,
     SearchNodeResultData,
@@ -756,6 +759,7 @@ __all__ = [
     "AdHocRejectedPromptExecutionMeta",
     "AdHocStreamingPromptExecutionMeta",
     "AddOpenaiApiKeyEnum",
+    "ApiActorTypeEnum",
     "ApiNodeResult",
     "ApiNodeResultData",
     "ApiRequestParentContext",
@@ -982,6 +986,7 @@ __all__ = [
     "IntegrationName",
     "IntegrationProvider",
     "IntegrationRead",
+    "IntegrationTriggerContext",
     "InternalServerError",
     "InvokedPort",
     "IterationStateEnum",
@@ -1179,6 +1184,7 @@ __all__ = [
     "ScenarioInputJsonVariableValue",
     "ScenarioInputStringVariableValue",
     "ScenarioInputVideoVariableValue",
+    "ScheduledTriggerContext",
     "SearchFiltersRequest",
     "SearchNodeResult",
     "SearchNodeResultData",

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/core/client_wrapper.py

@@ -27,10 +27,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.8.5",
+            "User-Agent": "vellum-ai/1.8.6",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.8.5",
+            "X-Fern-SDK-Version": "1.8.6",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None:

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/__init__.py

@@ -9,6 +9,7 @@ from .ad_hoc_initiated_prompt_execution_meta import AdHocInitiatedPromptExecutio
 from .ad_hoc_rejected_prompt_execution_meta import AdHocRejectedPromptExecutionMeta
 from .ad_hoc_streaming_prompt_execution_meta import AdHocStreamingPromptExecutionMeta
 from .add_openai_api_key_enum import AddOpenaiApiKeyEnum
+from .api_actor_type_enum import ApiActorTypeEnum
 from .api_node_result import ApiNodeResult
 from .api_node_result_data import ApiNodeResultData
 from .api_request_parent_context import ApiRequestParentContext
@@ -241,6 +242,7 @@ from .integration_credential_access_type import IntegrationCredentialAccessType
 from .integration_name import IntegrationName
 from .integration_provider import IntegrationProvider
 from .integration_read import IntegrationRead
+from .integration_trigger_context import IntegrationTriggerContext
 from .invoked_port import InvokedPort
 from .iteration_state_enum import IterationStateEnum
 from .jinja_prompt_block import JinjaPromptBlock
@@ -433,6 +435,7 @@ from .scenario_input_image_variable_value import ScenarioInputImageVariableValue
 from .scenario_input_json_variable_value import ScenarioInputJsonVariableValue
 from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
 from .scenario_input_video_variable_value import ScenarioInputVideoVariableValue
+from .scheduled_trigger_context import ScheduledTriggerContext
 from .search_filters_request import SearchFiltersRequest
 from .search_node_result import SearchNodeResult
 from .search_node_result_data import SearchNodeResultData
@@ -735,6 +738,7 @@ __all__ = [
     "AdHocRejectedPromptExecutionMeta",
     "AdHocStreamingPromptExecutionMeta",
     "AddOpenaiApiKeyEnum",
+    "ApiActorTypeEnum",
     "ApiNodeResult",
     "ApiNodeResultData",
     "ApiRequestParentContext",
@@ -955,6 +959,7 @@ __all__ = [
     "IntegrationName",
     "IntegrationProvider",
     "IntegrationRead",
+    "IntegrationTriggerContext",
     "InvokedPort",
     "IterationStateEnum",
     "JinjaPromptBlock",
@@ -1147,6 +1152,7 @@ __all__ = [
     "ScenarioInputJsonVariableValue",
     "ScenarioInputStringVariableValue",
     "ScenarioInputVideoVariableValue",
+    "ScheduledTriggerContext",
     "SearchFiltersRequest",
     "SearchNodeResult",
     "SearchNodeResultData",

vellum_ai-1.8.6/src/vellum/client/types/api_actor_type_enum.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ApiActorTypeEnum = typing.Union[
+    typing.Literal["WORKSPACE_API_KEY", "ENVIRONMENT_API_KEY", "JWT", "SERVICE_TOKEN"], typing.Any
+]

vellum_ai-1.8.6/src/vellum/client/types/api_request_parent_context.py

@@ -0,0 +1,42 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .api_actor_type_enum import ApiActorTypeEnum
+
+
+class ApiRequestParentContext(UniversalBaseModel):
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
+    type: typing.Literal["API_REQUEST"] = "API_REQUEST"
+    span_id: str
+    api_actor_id: typing.Optional[str] = None
+    api_actor_type: typing.Optional[ApiActorTypeEnum] = None
+    api_actor_label: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(ApiRequestParentContext)

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/external_parent_context.py

@@ -25,8 +25,10 @@ class ExternalParentContext(UniversalBaseModel):
 
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

vellum_ai-1.8.6/src/vellum/client/types/integration_trigger_context.py

@@ -0,0 +1,38 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+
+
+class IntegrationTriggerContext(UniversalBaseModel):
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
+    type: typing.Literal["INTEGRATION"] = "INTEGRATION"
+    span_id: str
+
+    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
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(IntegrationTriggerContext)

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_fulfilled_event.py

@@ -34,8 +34,10 @@ class NodeExecutionFulfilledEvent(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_initiated_event.py

@@ -34,8 +34,10 @@ class NodeExecutionInitiatedEvent(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_paused_event.py

@@ -34,8 +34,10 @@ class NodeExecutionPausedEvent(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_rejected_event.py

@@ -34,8 +34,10 @@ class NodeExecutionRejectedEvent(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_resumed_event.py

@@ -34,8 +34,10 @@ class NodeExecutionResumedEvent(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_span.py

@@ -34,8 +34,10 @@ class NodeExecutionSpan(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_execution_streaming_event.py

@@ -34,8 +34,10 @@ class NodeExecutionStreamingEvent(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001

{vellum_ai-1.8.5 → vellum_ai-1.8.6}/src/vellum/client/types/node_parent_context.py

@@ -28,7 +28,9 @@ class NodeParentContext(UniversalBaseModel):
 
 from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
 from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
 from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
 from .span_link import SpanLink  # noqa: E402, F401, I001
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001