vellum-ai 1.7.7__py3-none-any.whl → 1.7.9__py3-none-any.whl

vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -29,6 +29,7 @@ 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 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 (
     EdgeDisplay,
@@ -473,19 +474,34 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
             )
 
         # Return as a list with a single trigger object matching Django schema
-        trigger_id = uuid4_from_hash(f"{trigger_class.__module__} | {trigger_class.__qualname__}")
+        trigger_id = uuid4_from_hash(trigger_class.__qualname__)
 
-        return cast(
+        # Serialize trigger attributes like node outputs
+        attribute_references = trigger_class.attribute_references().values()
+        trigger_attributes: JsonArray = cast(
             JsonArray,
             [
-                {
-                    "id": str(trigger_id),
-                    "type": trigger_type.value,
-                    "attributes": [],
-                }
+                cast(
+                    JsonObject,
+                    {
+                        "id": str(reference.id),
+                        "name": reference.name,
+                        "type": primitive_type_to_vellum_variable_type(reference),
+                        "value": None,
+                    },
+                )
+                for reference in sorted(attribute_references, key=lambda ref: ref.name)
             ],
         )
 
+        trigger_data: JsonObject = {
+            "id": str(trigger_id),
+            "type": trigger_type.value,
+            "attributes": trigger_attributes,
+        }
+
+        return cast(JsonArray, [trigger_data])
+
     def _serialize_edge_display_data(self, edge_display: EdgeDisplay) -> Optional[JsonObject]:
         """Serialize edge display data, returning None if no display data is present."""
         if edge_display.z_index is not None: