vellum-workflow-server 0.14.72.post1__tar.gz → 0.14.72.post2__tar.gz

{vellum_workflow_server-0.14.72.post1 → vellum_workflow_server-0.14.72.post2}/PKG-INFO

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

{vellum_workflow_server-0.14.72.post1 → vellum_workflow_server-0.14.72.post2}/pyproject.toml

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

{vellum_workflow_server-0.14.72.post1 → vellum_workflow_server-0.14.72.post2}/src/workflow_server/api/workflow_view.py

@@ -15,6 +15,7 @@ from typing import Any, Dict, Generator, Iterator, Union, cast
 
 from flask import Blueprint, Response, current_app as app, request, stream_with_context
 from pydantic import ValidationError
+import sentry_sdk
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
 from vellum_ee.workflows.display.types import WorkflowDisplayContext
 
@@ -112,6 +113,7 @@ def stream_workflow_route() -> Response:
         )
         increment_process_count(1)
     except Exception as e:
+        sentry_sdk.set_tag("vellum_trace_id", str(context.trace_id))
         logger.exception(e)
 
         vembda_fulfilled_event = VembdaExecutionFulfilledEvent(
@@ -215,6 +217,7 @@ def stream_workflow_route() -> Response:
 
                 continue
             except Exception as e:
+                sentry_sdk.set_tag("vellum_trace_id", str(context.trace_id))
                 logger.exception(e)
                 break
 
@@ -327,6 +330,7 @@ def stream_node_route() -> Response:
                         parent=None,
                     )
                     yield vembda_fulfilled_event.model_dump(mode="json")
+                    sentry_sdk.set_tag("vellum_trace_id", str(context.trace_id))
                     app.logger.exception(stream_future.exception())
                     break
                 else:
@@ -401,6 +405,7 @@ def get_version_route() -> tuple[dict, int]:
 
         resp["nodes"] = nodes
     except Exception as e:
+        sentry_sdk.set_tag("vellum_trace_id", "unknown")
         logger.exception(f"Failed to discover nodes: {str(e)}")
         resp["nodes"] = []
 
@@ -409,10 +414,15 @@ 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": uuid4(),
+        "trace_id": trace_id,
         "request_start_time": time.time_ns(),
     }
 
@@ -420,10 +430,15 @@ 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": uuid4(),
+        "trace_id": trace_id,
         "request_start_time": time.time_ns(),
     }
 

{vellum_workflow_server-0.14.72.post1 → vellum_workflow_server-0.14.72.post2}/src/workflow_server/code_exec_runner.py

@@ -5,6 +5,8 @@ import os
 from uuid import uuid4
 from typing import Optional
 
+import sentry_sdk
+
 from vellum.workflows.exceptions import WorkflowInitializationException
 from workflow_server.api.workflow_view import get_workflow_request_context
 from workflow_server.core.events import (
@@ -69,6 +71,7 @@ def run_code_exec_stream() -> None:
 
             print(f"{_EVENT_LINE}{json.dumps(fulfilled_event)}")  # noqa: T201
     except Exception as e:
+        sentry_sdk.set_tag("vellum_trace_id", str(context.trace_id) if context else "unknown")
         logger.exception(e)
 
         vembda_fulfilled_event = VembdaExecutionFulfilledEvent(

{vellum_workflow_server-0.14.72.post1 → vellum_workflow_server-0.14.72.post2}/src/workflow_server/core/executor.py

@@ -15,6 +15,7 @@ from uuid import uuid4
 from typing import Any, Callable, Generator, Iterator, Optional, Tuple, Type
 
 from pebble import concurrent
+import sentry_sdk
 from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
@@ -66,6 +67,7 @@ def _stream_node_wrapper(executor_context: NodeExecutorContext, queue: Queue) ->
         for event in stream_node(executor_context=executor_context):
             queue.put(event)
     except Exception as e:
+        sentry_sdk.set_tag("vellum_trace_id", str(executor_context.trace_id))
         logger.exception(e)
         queue.put(
             VembdaExecutionFulfilledEvent(
@@ -104,6 +106,7 @@ def _stream_workflow_wrapper(executor_context: WorkflowExecutorContext, queue: Q
             ).model_dump(mode="json")
         )
     except Exception as e:
+        sentry_sdk.set_tag("vellum_trace_id", str(executor_context.trace_id))
         logger.exception(e)
         queue.put(
             VembdaExecutionFulfilledEvent(
@@ -209,6 +212,7 @@ def stream_workflow(
                         event.body.display_context = display_context
 
                 if event.name.endswith("rejected") and event.body.error.code.name == "INTERNAL_ERROR":  # type: ignore
+                    sentry_sdk.set_tag("vellum_trace_id", str(executor_context.trace_id))
                     logger.exception(event.body.error.message)  # type: ignore
                     event.body.error = WorkflowError(  # type: ignore
                         code=WorkflowErrorCode.INTERNAL_ERROR.value, message="Internal Error"  # type: ignore
@@ -458,6 +462,7 @@ def _gather_workflow(context: WorkflowExecutorContext) -> Tuple[BaseWorkflow, st
         )
         return workflow, namespace
     except Exception as e:
+        sentry_sdk.set_tag("vellum_trace_id", str(context.trace_id))
         logger.exception("Failed to initialize Workflow")
         raise WorkflowInitializationException(f"Failed to initialize workflow: {e}") from e
 
@@ -466,5 +471,6 @@ def _gather_display_context(workflow: BaseWorkflow, namespace: str) -> Optional[
     try:
         return BaseWorkflowDisplay.gather_event_display_context(namespace, workflow.__class__)
     except Exception:
+        sentry_sdk.set_tag("vellum_trace_id", "unknown")
         logger.exception("Unable to Parse Workflow Display Context")
         return None