vellum-workflow-server 1.3.10.post1__tar.gz → 1.3.11__tar.gz

{vellum_workflow_server-1.3.10.post1 → vellum_workflow_server-1.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.3.10.post1
+Version: 1.3.11
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -29,7 +29,7 @@ Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.0.1)
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==1.3.10)
+Requires-Dist: vellum-ai (==1.3.11)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.3.10.post1 → vellum_workflow_server-1.3.11}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.3.10.post1"
+version = "1.3.11"
 description = ""
 readme = "README.md"
 authors = []
@@ -45,7 +45,7 @@ flask = "2.3.3"
 orderly-set = "5.2.2"
 pebble = "5.0.7"
 gunicorn = "23.0.0"
-vellum-ai = "1.3.10"
+vellum-ai = "1.3.11"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.3.10.post1 → vellum_workflow_server-1.3.11}/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.post1 → vellum_workflow_server-1.3.11}/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