vellum-workflow-server 0.14.72.post4__tar.gz → 0.14.72.post6__tar.gz

{vellum_workflow_server-0.14.72.post4 → vellum_workflow_server-0.14.72.post6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 0.14.72.post4
+Version: 0.14.72.post6
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4

{vellum_workflow_server-0.14.72.post4 → vellum_workflow_server-0.14.72.post6}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "0.14.72.post4"
+version = "0.14.72.post6"
 description = ""
 readme = "README.md"
 authors = []

{vellum_workflow_server-0.14.72.post4 → vellum_workflow_server-0.14.72.post6}/src/workflow_server/api/workflow_view.py

@@ -398,7 +398,7 @@ def get_version_route() -> tuple[dict, int]:
                             node_display_class = get_node_display_class(obj)
                             exec_config_raw = node_display_class().serialize(WorkflowDisplayContext())
                             exec_config = cast(Dict[str, Any], exec_config_raw)
-                            module = exec_config["definition"]["module"]
+                            config_module = exec_config["definition"]["module"]
                             label = exec_config["label"]
                             description = (
                                 exec_config["display_data"]["comment"]["value"]
@@ -408,7 +408,7 @@ def get_version_route() -> tuple[dict, int]:
                             nodes.append(
                                 {
                                     "id": str(uuid4()),
-                                    "module": module,
+                                    "module": config_module,
                                     "name": obj.__name__,
                                     "label": pascal_to_title_case(label),
                                     "description": description,
@@ -428,16 +428,9 @@ def get_version_route() -> tuple[dict, int]:
 
 
 def get_workflow_request_context(data: dict) -> WorkflowExecutorContext:
-    # not sure if this is the filter we want to pass forward?
-    trace_id = uuid4()
-    if "execution_context" in data and isinstance(data["execution_context"], dict):
-        if "trace_id" in data["execution_context"]:
-            trace_id = data["execution_context"]["trace_id"]
-
     context_data = {
         **data,
         "inputs": convert_json_inputs_to_vellum(data.get("inputs") or []),
-        "trace_id": trace_id,
         "request_start_time": time.time_ns(),
     }
 
@@ -445,15 +438,9 @@ def get_workflow_request_context(data: dict) -> WorkflowExecutorContext:
 
 
 def get_node_request_context(data: dict) -> NodeExecutorContext:
-    trace_id = uuid4()
-    if "execution_context" in data and isinstance(data["execution_context"], dict):
-        if "trace_id" in data["execution_context"]:
-            trace_id = data["execution_context"]["trace_id"]
-
     context_data = {
         **data,
         "inputs": convert_json_inputs_to_vellum(data["inputs"]),
-        "trace_id": trace_id,
         "request_start_time": time.time_ns(),
     }
 

{vellum_workflow_server-0.14.72.post4 → vellum_workflow_server-0.14.72.post6}/src/workflow_server/core/workflow_executor_context.py

@@ -15,7 +15,6 @@ class BaseExecutorContext(UniversalBaseModel):
     files: dict[str, str]
     environment_api_key: str
     execution_id: UUID
-    trace_id: UUID
     module: str
     execution_context: ExecutionContext = field(default_factory=ExecutionContext)
     request_start_time: int
@@ -28,6 +27,10 @@ class BaseExecutorContext(UniversalBaseModel):
     def container_overhead_latency(self) -> int:
         return self.stream_start_time - self.request_start_time if self.stream_start_time else -1
 
+    @property
+    def trace_id(self) -> UUID:
+        return self.execution_context.trace_id
+
     def __hash__(self) -> int:
         # do we think we need anything else for a unique hash for caching?
         return hash(str(self.execution_id))