vellum-workflow-server 1.3.10__tar.gz → 1.3.10.post2__tar.gz

{vellum_workflow_server-1.3.10 → vellum_workflow_server-1.3.10.post2}/PKG-INFO

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

{vellum_workflow_server-1.3.10 → vellum_workflow_server-1.3.10.post2}/pyproject.toml

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

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

@@ -20,6 +20,7 @@ from vellum_ee.workflows.display.utils.events import event_enricher
 from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
+from vellum import Vellum
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.emitters.base import BaseWorkflowEmitter
 from vellum.workflows.emitters.vellum_emitter import VellumEmitter
@@ -233,6 +234,7 @@ def stream_workflow(
                 yield _dump_event(
                     event=event,
                     executor_context=executor_context,
+                    client=workflow.context.vellum_client,
                 )
         except Exception as e:
             logger.exception("Failed to generate event from Workflow Stream")
@@ -396,9 +398,9 @@ def _get_file_namespace(executor_context: BaseExecutorContext) -> str:
     )
 
 
-def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext) -> dict:
+def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext, client: Vellum) -> dict:
     module_base = executor_context.module.split(".")
-    dump = event.model_dump(mode="json", context={"event_enricher": event_enricher})
+    dump = event.model_dump(mode="json", context={"event_enricher": lambda event: event_enricher(event, client)})
     if dump["name"] in {
         "workflow.execution.initiated",
         "workflow.execution.fulfilled",

{vellum_workflow_server-1.3.10 → vellum_workflow_server-1.3.10.post2}/src/workflow_server/utils/tests/test_utils.py

@@ -7,8 +7,10 @@ from vellum import (
     SearchResultDocument,
     StringVellumValue,
     VellumAudio,
+    VellumDocument,
     VellumError,
     VellumImage,
+    VellumVideo,
 )
 from workflow_server.utils.utils import (
     convert_json_inputs_to_vellum,
@@ -50,6 +52,14 @@ from workflow_server.utils.utils import (
             {"type": "AUDIO", "name": "audio", "value": {"src": "https://example.com/audio.mp3"}},
             {"audio": VellumAudio(src="https://example.com/audio.mp3")},
         ),
+        (
+            {"type": "VIDEO", "name": "video", "value": {"src": "https://example.com/video.mp4"}},
+            {"video": VellumVideo(src="https://example.com/video.mp4")},
+        ),
+        (
+            {"type": "DOCUMENT", "name": "document", "value": {"src": "https://example.com/document.pdf"}},
+            {"document": VellumDocument(src="https://example.com/document.pdf")},
+        ),
         (
             {
                 "type": "SEARCH_RESULTS",
@@ -117,6 +127,8 @@ from workflow_server.utils.utils import (
         "function_call",
         "image",
         "audio",
+        "video",
+        "document",
         "search_results",
         "error",
         "array",

{vellum_workflow_server-1.3.10 → vellum_workflow_server-1.3.10.post2}/src/workflow_server/utils/utils.py

@@ -11,7 +11,9 @@ from vellum import (
     FunctionCallInput,
     SearchResultsInput,
     VellumAudio,
+    VellumDocument,
     VellumImage,
+    VellumVideo,
 )
 from workflow_server.config import is_development
 
@@ -46,6 +48,10 @@ def convert_json_inputs_to_vellum(inputs: List[dict]) -> dict:
             vellum_inputs[name] = VellumImage.model_validate(value)
         elif type == "AUDIO":
             vellum_inputs[name] = VellumAudio.model_validate(value)
+        elif type == "VIDEO":
+            vellum_inputs[name] = VellumVideo.model_validate(value)
+        elif type == "DOCUMENT":
+            vellum_inputs[name] = VellumDocument.model_validate(value)
         else:
             vellum_inputs[name] = value