vellum-workflow-server 1.7.5.post1__py3-none-any.whl → 1.7.5.post3__py3-none-any.whl

{vellum_workflow_server-1.7.5.post1.dist-info → vellum_workflow_server-1.7.5.post3.dist-info}/METADATA

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

{vellum_workflow_server-1.7.5.post1.dist-info → vellum_workflow_server-1.7.5.post3.dist-info}/RECORD

@@ -6,29 +6,29 @@ workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
 workflow_server/api/tests/test_workflow_view.py,sha256=XfsW8_hFKN5rzp2UFVviVoLW-8FD-nvD6mSZHnIyrWM,30483
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=l4sfma1uNB2CtFjxKTIQ2SSn-PpyxaSWJZyy7VYSDYM,37311
-workflow_server/api/workflow_view.py,sha256=WJC8wfdgqIDk5LLdRN2rom1M4qKvqHroxXpCjYczmjg,19859
+workflow_server/api/workflow_view.py,sha256=u58JmPlAnN-Ktyh0lchNzachZ-36jMPd-4PXwX_umOw,19897
 workflow_server/code_exec_runner.py,sha256=DLNNrinCRbnkSvlqVvSZ1wv_etI7r_kKAXNPGMj3jBk,2196
-workflow_server/config.py,sha256=CKU7qKC5k1_M1AFOJn_nUqntA46KPk_Z7fblfQaGC-0,1877
+workflow_server/config.py,sha256=cUdI_lEovV7e7lwCkGJ1eM9R4OZVJw5R5zT1eG1SzQQ,2122
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=QcEeYUIrxq4pub-z9BlGi5fLI3gVRml-56rMCW7j5Hc,2212
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
 workflow_server/core/executor.py,sha256=IazlinBkJ9kTqhB-6iiCozB2mblc0CshDEChVml_CZ4,17630
 workflow_server/core/utils.py,sha256=si0NB4Suurc-mn8NYdn59xM9CkPrfOP1aWEVrZvifDI,1929
-workflow_server/core/workflow_executor_context.py,sha256=rSjgkWenRGpFOwioRXNnUhGnEI7isOB-Qk7Tw-2JwnM,4521
+workflow_server/core/workflow_executor_context.py,sha256=Q2R0T2KkYZ1z52v8erDMysJfxSODbzPhDxBxX--k4Zw,3202
 workflow_server/server.py,sha256=lhHPmK1PhRZd6eCkj1C0acK3YwaApZgoPHghMChw0fc,1461
 workflow_server/start.py,sha256=WHZEsLen8mtGX-wF3xUbB91_NpC_GBRtOEi87OR1rv4,2360
 workflow_server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/utils/exit_handler.py,sha256=_FacDVi4zc3bfTA3D2mJsISePlJ8jpLrnGVo5-xZQFs,743
 workflow_server/utils/log_proxy.py,sha256=nugi6fOgAYKX2X9DIc39TG366rsmmDUPoEtG3gzma_Y,3088
 workflow_server/utils/oom_killer.py,sha256=dzaqSzi0jQ3MvALwwiYIO9r6VWLa5Ln9AY6l11WEexo,3050
-workflow_server/utils/sentry.py,sha256=pqx3X_4W3yOzmz8QMJYUEi39skIKWtrTN5nyFhaPkbk,1597
+workflow_server/utils/sentry.py,sha256=pmGDoaFhJwUprjP_Vmz6bETitqKQulJ0vwRP-gYb2w4,2145
 workflow_server/utils/system_utils.py,sha256=3jNv113zRkKJ0928i2Vm6TqFHrDulteQu1kjseP2B0Y,3271
 workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRjRq9JY6Z5ShLZyx_N_L0-FU6OI,2100
+workflow_server/utils/tests/test_sentry_integration.py,sha256=14PfuW8AaQNNtqLmBs16EPe5T3f_iTI7YJMCRtiboZk,4502
 workflow_server/utils/tests/test_system_utils.py,sha256=_4GwXvVvU5BrATxUEWwQIPg0bzQXMWBtiBmjP8MTxJM,4314
 workflow_server/utils/tests/test_utils.py,sha256=0Nq6du8o-iBtTrip9_wgHES53JSiJbVdSXaBnPobw3s,6930
 workflow_server/utils/utils.py,sha256=m7iMJtor5SQLWu7jlJw-X5Q3nmbq69BCxTMv6qnFYrA,4835
-vellum_workflow_server-1.7.5.post1.dist-info/METADATA,sha256=AS79hT9zNyxoVj6nYGoLmTvDLmND6YvsDTgtaaTT_3I,2273
-vellum_workflow_server-1.7.5.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.7.5.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.7.5.post1.dist-info/RECORD,,
+vellum_workflow_server-1.7.5.post3.dist-info/METADATA,sha256=GCLbJTODw2-YDRMx97PmGb58OHPfp0PID_VPlLbPgOs,2273
+vellum_workflow_server-1.7.5.post3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.7.5.post3.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.7.5.post3.dist-info/RECORD,,

workflow_server/api/workflow_view.py

@@ -22,6 +22,7 @@ from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.nodes import BaseNode
+from vellum.workflows.vellum_client import create_vellum_client
 from workflow_server.config import ENABLE_PROCESS_WRAPPER, MEMORY_LIMIT_MB
 from workflow_server.core.events import (
     SPAN_ID_EVENT,
@@ -47,7 +48,6 @@ from workflow_server.core.workflow_executor_context import (
     DEFAULT_TIMEOUT_SECONDS,
     NodeExecutorContext,
     WorkflowExecutorContext,
-    create_vellum_client,
 )
 from workflow_server.utils.oom_killer import get_is_oom_killed
 from workflow_server.utils.system_utils import (

workflow_server/config.py

@@ -27,6 +27,11 @@ MEMORY_LIMIT_MB = int(os.getenv("MEMORY_LIMIT_MB", "2048"))
 PORT = os.getenv("PORT", "8000")
 VELLUM_API_URL_HOST = os.getenv("VELLUM_API_URL_HOST", "localhost")
 VELLUM_API_URL_PORT = os.getenv("VELLUM_API_URL_PORT", 8000)
+
+# Set VELLUM_API_URL for SDK when using local Vellum API
+if os.getenv("USE_LOCAL_VELLUM_API") == "true" and "VELLUM_API_URL" not in os.environ:
+    os.environ["VELLUM_API_URL"] = "http://{}:{}".format(VELLUM_API_URL_HOST, VELLUM_API_URL_PORT)
+
 CONCURRENCY = int(os.getenv("CONCURRENCY", "8"))
 CONTAINER_IMAGE = os.getenv("CONTAINER_IMAGE", "python-workflow-runtime:latest")
 ENABLE_PROCESS_WRAPPER = os.getenv("ENABLE_PROCESS_WRAPPER", "true").lower() == "true"

workflow_server/core/workflow_executor_context.py

@@ -9,54 +9,15 @@ from typing_extensions import Self
 from flask import has_request_context, request
 from pydantic import Field, field_validator, model_validator
 
-from vellum import ApiVersionEnum, Vellum, VellumEnvironment
+from vellum import ApiVersionEnum, Vellum
 from vellum.client.core import UniversalBaseModel
 from vellum.workflows.context import ExecutionContext
-from workflow_server.config import IS_VPC, VELLUM_API_URL_HOST, VELLUM_API_URL_PORT
+from vellum.workflows.vellum_client import create_vellum_client
 from workflow_server.utils.utils import convert_json_inputs_to_vellum
 
 DEFAULT_TIMEOUT_SECONDS = int(os.getenv("MAX_WORKFLOW_RUNTIME_SECONDS", 1800))
 
 
-def create_vellum_client(
-    api_key: str,
-    api_version: Optional[ApiVersionEnum] = None,
-) -> Vellum:
-    """
-    Create a VellumClient with proper environment configuration.
-
-    Args:
-        api_key: The API key for the Vellum client
-        api_version: Optional API version to use
-
-    Returns:
-        Configured Vellum client instance
-
-    Note: Ideally we replace this with `vellum.workflows.vellum_client.create_vellum_client`
-    """
-    if IS_VPC:
-        environment = VellumEnvironment(
-            default=os.getenv("VELLUM_DEFAULT_API_URL", VellumEnvironment.PRODUCTION.default),
-            documents=os.getenv("VELLUM_DOCUMENTS_API_URL", VellumEnvironment.PRODUCTION.documents),
-            predict=os.getenv("VELLUM_PREDICT_API_URL", VellumEnvironment.PRODUCTION.predict),
-        )
-    elif os.getenv("USE_LOCAL_VELLUM_API") == "true":
-        VELLUM_API_URL = f"http://{VELLUM_API_URL_HOST}:{VELLUM_API_URL_PORT}"
-        environment = VellumEnvironment(
-            default=VELLUM_API_URL,
-            documents=VELLUM_API_URL,
-            predict=VELLUM_API_URL,
-        )
-    else:
-        environment = VellumEnvironment.PRODUCTION
-
-    return Vellum(
-        api_key=api_key,
-        environment=environment,
-        api_version=api_version,
-    )
-
-
 class BaseExecutorContext(UniversalBaseModel):
     inputs: dict = Field(default_factory=dict)
     state: Optional[dict] = None

workflow_server/utils/sentry.py

@@ -37,11 +37,32 @@ def _tag_trace_id(event: dict) -> None:
     event["tags"]["vellum_trace_id"] = trace_id
 
 
+def _apply_custom_tags(event: dict, hint: dict) -> None:
+    """
+    Extracts 'sentry_tags' from logger exception's extra data and adds them to the event tags.
+
+    Modifies the event dictionary in place.
+    """
+    record = hint.get("log_record")
+    if not record:
+        return
+
+    sentry_tags = getattr(record, "sentry_tags", None)
+    if not sentry_tags or not isinstance(sentry_tags, dict):
+        return
+
+    if "tags" not in event:
+        event["tags"] = {}
+
+    event["tags"].update(sentry_tags)
+
+
 def before_send(event: dict, hint: dict) -> Optional[dict]:
     if "exc_info" in hint:
         _, _, _ = hint["exc_info"]
 
     _tag_trace_id(event)
+    _apply_custom_tags(event, hint)
 
     return event
 

workflow_server/utils/tests/test_sentry_integration.py

@@ -1,4 +1,5 @@
 import pytest
+import logging
 from uuid import uuid4
 
 from workflow_server.server import create_app
@@ -67,3 +68,76 @@ class Workflow(BaseWorkflow):
 
         # AND the trace_id is tagged
         assert event["tags"]["vellum_trace_id"] == trace_id
+
+
+def test_sentry_integration_applies_custom_tags_from_logger_extra(monkeypatch, mock_sentry_capture_envelope):
+    """Test that Sentry events include custom tags from logger exception extra data."""
+
+    # GIVEN sentry is configured
+    monkeypatch.setenv("SENTRY_DSN", "https://test-dsn@sentry.io/1234567890")
+
+    # AND we have a function that will log with custom sentry_tags when called
+    def mock_get_version():
+        logger = logging.getLogger(__name__)
+        try:
+            raise Exception("Test exception with custom tags")
+        except Exception:
+            logger.exception(
+                "Failed during workflow execution",
+                extra={
+                    "sentry_tags": {
+                        "operation": "stream",
+                        "test_tag": "test_value",
+                        "numeric_tag": "12345",
+                    }
+                },
+            )
+            raise
+
+    monkeypatch.setattr("workflow_server.api.workflow_view.get_version", mock_get_version)
+
+    # AND we have a valid request body
+    body = {
+        "execution_id": str(uuid4()),
+        "inputs": [],
+        "environment_api_key": "test",
+        "module": "workflow",
+        "timeout": 360,
+        "files": {
+            "__init__.py": "",
+            "workflow.py": """\
+from vellum.workflows import BaseWorkflow
+
+class Workflow(BaseWorkflow):
+    pass
+""",
+        },
+        "execution_context": {
+            "trace_id": str(uuid4()),
+            "parent_context": {
+                "type": "API_REQUEST",
+                "span_id": str(uuid4()),
+                "parent": None,
+            },
+        },
+    }
+
+    # WHEN we call an endpoint that triggers the error
+    flask_app = create_app()
+
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/stream", json=body)
+
+        # THEN we get a 500 error
+        assert response.status_code == 500
+
+        # AND sentry captures the error
+        assert mock_sentry_capture_envelope.call_count == 1
+        envelope = mock_sentry_capture_envelope.call_args[0][0]
+        event = envelope.get_event()
+
+        # AND the custom tags are included in the event
+        assert "tags" in event
+        assert event["tags"]["operation"] == "stream"
+        assert event["tags"]["test_tag"] == "test_value"
+        assert event["tags"]["numeric_tag"] == "12345"