vellum-workflow-server 1.9.6.post2__tar.gz → 1.9.7.post1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (35) hide show
  1. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/PKG-INFO +2 -2
  2. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/pyproject.toml +2 -2
  3. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/tests/test_workflow_view.py +24 -24
  4. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/workflow_view.py +2 -1
  5. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/core/executor.py +3 -1
  6. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/README.md +0 -0
  7. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/__init__.py +0 -0
  8. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/__init__.py +0 -0
  9. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/auth_middleware.py +0 -0
  10. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/healthz_view.py +0 -0
  11. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/status_view.py +0 -0
  12. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/tests/__init__.py +0 -0
  13. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/tests/test_input_display_mapping.py +0 -0
  14. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py +0 -0
  15. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/code_exec_runner.py +0 -0
  16. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/config.py +0 -0
  17. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/core/__init__.py +0 -0
  18. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/core/cancel_workflow.py +0 -0
  19. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/core/events.py +0 -0
  20. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/core/utils.py +0 -0
  21. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/core/workflow_executor_context.py +0 -0
  22. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/logging_config.py +0 -0
  23. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/server.py +0 -0
  24. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/start.py +0 -0
  25. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/__init__.py +0 -0
  26. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/exit_handler.py +0 -0
  27. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/log_proxy.py +0 -0
  28. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/oom_killer.py +0 -0
  29. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/sentry.py +0 -0
  30. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/system_utils.py +0 -0
  31. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/tests/__init__.py +0 -0
  32. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/tests/test_sentry_integration.py +0 -0
  33. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/tests/test_system_utils.py +0 -0
  34. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/tests/test_utils.py +0 -0
  35. {vellum_workflow_server-1.9.6.post2 → vellum_workflow_server-1.9.7.post1}/src/workflow_server/utils/utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vellum-workflow-server
3
- Version: 1.9.6.post2
3
+ Version: 1.9.7.post1
4
4
  Summary:
5
5
  License: AGPL
6
6
  Requires-Python: >=3.9.0,<4
@@ -29,7 +29,7 @@ Requires-Dist: pyjwt (==2.10.0)
29
29
  Requires-Dist: python-dotenv (==1.0.1)
30
30
  Requires-Dist: retrying (==1.3.4)
31
31
  Requires-Dist: sentry-sdk[flask] (==2.20.0)
32
- Requires-Dist: vellum-ai (==1.9.6)
32
+ Requires-Dist: vellum-ai (==1.9.7)
33
33
  Description-Content-Type: text/markdown
34
34
 
35
35
  # Vellum Workflow Runner Server
@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
3
3
 
4
4
  [tool.poetry]
5
5
  name = "vellum-workflow-server"
6
- version = "1.9.6.post2"
6
+ version = "1.9.7.post1"
7
7
  description = ""
8
8
  readme = "README.md"
9
9
  authors = []
@@ -45,7 +45,7 @@ flask = "2.3.3"
45
45
  orderly-set = "5.2.2"
46
46
  pebble = "5.0.7"
47
47
  gunicorn = "23.0.0"
48
- vellum-ai = "1.9.6"
48
+ vellum-ai = "1.9.7"
49
49
  python-dotenv = "1.0.1"
50
50
  retrying = "1.3.4"
51
51
  sentry-sdk = {extras = ["flask"], version = "2.20.0"}
@@ -63,11 +63,11 @@ class TestNode(BaseNode):
63
63
  "comment": {"expanded": True, "value": "A test node for processing data."},
64
64
  "position": {"x": 0.0, "y": 0.0},
65
65
  },
66
- "id": "7a8b251d-f5ca-462a-b293-071d219460fb",
66
+ "id": "6f4c9178-9f46-4723-bcb7-0bd59db54eca",
67
67
  "label": "Test Node",
68
68
  "outputs": [],
69
- "ports": [{"id": "a3a0eefd-45d0-4f13-8c58-a836a9f7f9ed", "name": "default", "type": "DEFAULT"}],
70
- "trigger": {"id": "a022e36c-9852-4772-9be3-3c6c147fd811", "merge_behavior": "AWAIT_ATTRIBUTES"},
69
+ "ports": [{"id": "4394823f-79a8-4dbc-99ae-06a1df6c7408", "name": "default", "type": "DEFAULT"}],
70
+ "trigger": {"id": "07240af1-67c6-4460-b53d-53f0b0f1b90e", "merge_behavior": "AWAIT_ATTRIBUTES"},
71
71
  "type": "GENERIC",
72
72
  }
73
73
 
@@ -127,11 +127,11 @@ class SomeOtherNode(BaseNode):
127
127
  "comment": {"expanded": True, "value": "This is Some Node."},
128
128
  "position": {"x": 0.0, "y": 0.0},
129
129
  },
130
- "id": "1e559c2e-db82-41f0-9ceb-5e89b0c5a0a3",
130
+ "id": "89e84bac-5a5f-4f64-8083-7d3ebec98be1",
131
131
  "label": "Some Node",
132
132
  "outputs": [],
133
- "ports": [{"id": "48e39e97-5fd4-471e-b4f2-51d3baf06456", "name": "default", "type": "DEFAULT"}],
134
- "trigger": {"id": "e3381fb7-61fc-4c46-ae8e-51fc463b6a59", "merge_behavior": "AWAIT_ATTRIBUTES"},
133
+ "ports": [{"id": "2983ea5c-1d29-483a-b896-53098f5de4f1", "name": "default", "type": "DEFAULT"}],
134
+ "trigger": {"id": "6996efb0-5a20-4719-8835-34fe6552764a", "merge_behavior": "AWAIT_ATTRIBUTES"},
135
135
  "type": "GENERIC",
136
136
  }
137
137
 
@@ -150,11 +150,11 @@ class SomeOtherNode(BaseNode):
150
150
  "comment": {"expanded": True, "value": "This is Some Other Node."},
151
151
  "position": {"x": 0.0, "y": 0.0},
152
152
  },
153
- "id": "7aee541b-b245-4c8a-9137-3e4631d5100c",
153
+ "id": "3cdbba02-8a34-4e0f-8b94-770a944dcaa3",
154
154
  "label": "Some Other Node",
155
155
  "outputs": [],
156
- "ports": [{"id": "fb66b46a-d970-4bc9-83ea-70c154c57ddd", "name": "default", "type": "DEFAULT"}],
157
- "trigger": {"id": "13fa2714-20b3-4bc3-ab79-621a188e3bfa", "merge_behavior": "AWAIT_ATTRIBUTES"},
156
+ "ports": [{"id": "1839bde5-2ad4-4723-b21b-2c55fa833a7a", "name": "default", "type": "DEFAULT"}],
157
+ "trigger": {"id": "c36df8a8-5624-45be-99c9-826cf511a951", "merge_behavior": "AWAIT_ATTRIBUTES"},
158
158
  "type": "GENERIC",
159
159
  }
160
160
 
@@ -222,11 +222,11 @@ class HelperClass:
222
222
  "comment": {"expanded": True, "value": "Processes input data."},
223
223
  "position": {"x": 0.0, "y": 0.0},
224
224
  },
225
- "id": "f92c09f0-0434-46cb-829d-a73f801d6343",
225
+ "id": "7121bcb9-98a1-4907-bf9b-9734d773fd15",
226
226
  "label": "Processing Node",
227
227
  "outputs": [],
228
- "ports": [{"id": "abaa2984-b312-4491-b069-e689759f72c8", "name": "default", "type": "DEFAULT"}],
229
- "trigger": {"id": "35378c2b-f089-44af-ac37-efe4ea42c817", "merge_behavior": "AWAIT_ATTRIBUTES"},
228
+ "ports": [{"id": "de27da74-30e9-4e7b-95c2-92bdfc5bf042", "name": "default", "type": "DEFAULT"}],
229
+ "trigger": {"id": "e02bd85e-8b03-4b21-8b3e-f411042334ce", "merge_behavior": "AWAIT_ATTRIBUTES"},
230
230
  "type": "GENERIC",
231
231
  }
232
232
 
@@ -240,11 +240,11 @@ class HelperClass:
240
240
  "comment": {"expanded": True, "value": "Transforms data format."},
241
241
  "position": {"x": 0.0, "y": 0.0},
242
242
  },
243
- "id": "09ca32f7-c8f2-4469-97e5-1f288f85127a",
243
+ "id": "6a785cb0-f631-4f03-94c6-e82331c14c1a",
244
244
  "label": "Transformation Node",
245
245
  "outputs": [],
246
- "ports": [{"id": "88778117-fbfc-4b44-964b-5a4994aa2f24", "name": "default", "type": "DEFAULT"}],
247
- "trigger": {"id": "5d096263-7fbf-490a-83b7-e441852b5fb6", "merge_behavior": "AWAIT_ATTRIBUTES"},
246
+ "ports": [{"id": "67a13ea0-fd6b-44dc-af46-c72da06aa11f", "name": "default", "type": "DEFAULT"}],
247
+ "trigger": {"id": "08d4e317-baa8-478f-b278-99362e50e6b4", "merge_behavior": "AWAIT_ATTRIBUTES"},
248
248
  "type": "GENERIC",
249
249
  }
250
250
 
@@ -306,11 +306,11 @@ class BrokenNode(BaseNode)
306
306
  "comment": {"expanded": True, "value": "This is Some Node."},
307
307
  "position": {"x": 0.0, "y": 0.0},
308
308
  },
309
- "id": "1e559c2e-db82-41f0-9ceb-5e89b0c5a0a3",
309
+ "id": "a2706730-074b-4ea3-968a-25e68af1caed",
310
310
  "label": "Some Node",
311
311
  "outputs": [],
312
- "ports": [{"id": "48e39e97-5fd4-471e-b4f2-51d3baf06456", "name": "default", "type": "DEFAULT"}],
313
- "trigger": {"id": "e3381fb7-61fc-4c46-ae8e-51fc463b6a59", "merge_behavior": "AWAIT_ATTRIBUTES"},
312
+ "ports": [{"id": "e0ee3653-e071-4b91-9dfc-5e1dca9c665b", "name": "default", "type": "DEFAULT"}],
313
+ "trigger": {"id": "8d931b01-30ca-4c0d-b1b7-7c18379c83e6", "merge_behavior": "AWAIT_ATTRIBUTES"},
314
314
  "type": "GENERIC",
315
315
  }
316
316
 
@@ -371,12 +371,12 @@ class MyAdditionNode(BaseNode):
371
371
  "adornments": None,
372
372
  "attributes": [
373
373
  {
374
- "id": "aed3bcbb-d243-4a77-bb5e-409e9a28e868",
374
+ "id": "4223b340-447f-46c2-b35d-30ef16c5ae17",
375
375
  "name": "arg1",
376
376
  "value": None,
377
377
  },
378
378
  {
379
- "id": "9225d225-a41b-4642-8964-f28f58dcf4bf",
379
+ "id": "1de0f46a-95f6-4cd0-bb0f-e2414054d507",
380
380
  "name": "arg2",
381
381
  "value": None,
382
382
  },
@@ -387,11 +387,11 @@ class MyAdditionNode(BaseNode):
387
387
  "comment": {"expanded": True, "value": "Custom node that performs simple addition."},
388
388
  "position": {"x": 0.0, "y": 0.0},
389
389
  },
390
- "id": "195cd69d-3d2d-41e4-a432-16c433cb8d34",
390
+ "id": "2464b610-fb6d-495b-b17c-933ee147f19f",
391
391
  "label": "My Addition Node",
392
- "outputs": [{"id": "3d8e40cb-2aa8-44bd-ae6a-708a9fbc4779", "name": "result", "type": "NUMBER", "value": None}],
393
- "ports": [{"id": "9a9e4ef6-febf-4093-a515-217bbb1373db", "name": "default", "type": "DEFAULT"}],
394
- "trigger": {"id": "a5298668-d808-4a45-a62e-790943948e8a", "merge_behavior": "AWAIT_ATTRIBUTES"},
392
+ "outputs": [{"id": "f39d85c9-e7bf-45e1-bb67-f16225db0118", "name": "result", "type": "NUMBER", "value": None}],
393
+ "ports": [{"id": "bc489295-cd8a-4aa2-88bb-34446374100d", "name": "default", "type": "DEFAULT"}],
394
+ "trigger": {"id": "ff580cad-73d6-44fe-8f2c-4b8dc990ee70", "merge_behavior": "AWAIT_ATTRIBUTES"},
395
395
  "type": "GENERIC",
396
396
  "should_file_merge": True,
397
397
  }
@@ -528,6 +528,7 @@ def serialize_route() -> Response:
528
528
  files = data.get("files", {})
529
529
  workspace_api_key = data.get("workspace_api_key")
530
530
  is_new_server = data.get("is_new_server", False)
531
+ module = data.get("module")
531
532
 
532
533
  if not files:
533
534
  return Response(
@@ -540,7 +541,7 @@ def serialize_route() -> Response:
540
541
 
541
542
  # Generate a unique namespace for this serialization request
542
543
  namespace = get_random_namespace()
543
- virtual_finder = VirtualFileFinder(files, namespace)
544
+ virtual_finder = VirtualFileFinder(files, namespace, source_module=module)
544
545
 
545
546
  headers = {
546
547
  "X-Vellum-Is-New-Server": str(is_new_server).lower(),
@@ -359,7 +359,9 @@ def _call_stream(
359
359
  def _create_workflow(executor_context: BaseExecutorContext) -> Tuple[BaseWorkflow, str]:
360
360
  namespace = _get_file_namespace(executor_context)
361
361
  if namespace != LOCAL_WORKFLOW_MODULE:
362
- sys.meta_path.append(VirtualFileFinder(executor_context.files, namespace))
362
+ sys.meta_path.append(
363
+ VirtualFileFinder(executor_context.files, namespace, source_module=executor_context.module)
364
+ )
363
365
 
364
366
  workflow_context = _create_workflow_context(executor_context)
365
367
  Workflow = BaseWorkflow.load_from_module(namespace)