vellum-workflow-server 1.3.10.post1__py3-none-any.whl → 1.3.10.post2__py3-none-any.whl

{vellum_workflow_server-1.3.10.post1.dist-info → vellum_workflow_server-1.3.10.post2.dist-info}/METADATA

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

{vellum_workflow_server-1.3.10.post1.dist-info → vellum_workflow_server-1.3.10.post2.dist-info}/RECORD

@@ -26,9 +26,9 @@ workflow_server/utils/system_utils.py,sha256=SVO7ofsoEXl7XduBtpiwHKMx7DTPmdC-uH6
 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_system_utils.py,sha256=_4GwXvVvU5BrATxUEWwQIPg0bzQXMWBtiBmjP8MTxJM,4314
-workflow_server/utils/tests/test_utils.py,sha256=qwK5Rmy3RQyjtlUrYAuGuDlBeRzZKsf1yS-y2IpUizQ,6452
-workflow_server/utils/utils.py,sha256=zcjnY6tSPf90VZOBm8w0NfxYWPQYAL9TCNlza231Odg,4535
-vellum_workflow_server-1.3.10.post1.dist-info/METADATA,sha256=8Ge-op2CDdZR4WgrpXRnZdFYPT_3kOd8gu3qUCgjY0A,2275
-vellum_workflow_server-1.3.10.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.3.10.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.3.10.post1.dist-info/RECORD,,
+workflow_server/utils/tests/test_utils.py,sha256=0Nq6du8o-iBtTrip9_wgHES53JSiJbVdSXaBnPobw3s,6930
+workflow_server/utils/utils.py,sha256=ZPoM1Suhid22dpB8oEFLux8wx-9iyzmSfWuYxSCrgWk,4774
+vellum_workflow_server-1.3.10.post2.dist-info/METADATA,sha256=Loq_8QdX4WYdHgzTukQT5hrvMcnJ5qo7nOr9NF85guw,2275
+vellum_workflow_server-1.3.10.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.3.10.post2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.3.10.post2.dist-info/RECORD,,

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",

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