vellum-workflow-server 0.14.89__py3-none-any.whl → 0.14.89.post1__py3-none-any.whl

{vellum_workflow_server-0.14.89.dist-info → vellum_workflow_server-0.14.89.post1.dist-info}/METADATA

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

{vellum_workflow_server-0.14.89.dist-info → vellum_workflow_server-0.14.89.post1.dist-info}/RECORD

@@ -12,9 +12,9 @@ workflow_server/config.py,sha256=DyTty8NrAwvtx-esM3KthnpsNh-nKdWNlovWQOgiGpg,141
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=Ffkc3mzmrdMEUcD-sHfEhX4IwVrka-E--SxKA1dUfIU,2185
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=RMEp2TeTQZ2X4JveNNCc2nDG9GEZ9sgQlL-nY-gUCyE,16515
+workflow_server/core/executor.py,sha256=fvI6OS_oMS0GVEKZ6vCp_rlBACDA10DczG28mTcd0G4,16569
 workflow_server/core/utils.py,sha256=lgzxkAEjEXPxGXXQlUYTYuCdHht-eDJJmHj5AhEb3_o,1500
-workflow_server/core/workflow_executor_context.py,sha256=a-v48GJbOWUh4JIf_bNwDX-BvfKkg4xwRSPEyRVQmp4,1373
+workflow_server/core/workflow_executor_context.py,sha256=kqbHDAteyeyB5-ISJKS2d3Bafm7mndhIjovCqNNnFPE,1486
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
 workflow_server/start.py,sha256=qpIg0SgIgz8RNyc8Cu9LxyzXdOXZRv9qq3M3uSBbgD0,2180
 workflow_server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,7 +28,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRj
 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=Wqqn-1l2ugkGgy5paWWdt0AVxAyPMQCYcnRSSOMjXlA,4355
-vellum_workflow_server-0.14.89.dist-info/METADATA,sha256=pyKz7BTDy0LSngv9lRujbbeIQd2L34jpn0jTq6HQgN8,2271
-vellum_workflow_server-0.14.89.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.89.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.89.dist-info/RECORD,,
+vellum_workflow_server-0.14.89.post1.dist-info/METADATA,sha256=cnlmUZqQf__3fsEVAp5x4kHFajGmxt0YE8Phi9DD3X0,2277
+vellum_workflow_server-0.14.89.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-0.14.89.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-0.14.89.post1.dist-info/RECORD,,

workflow_server/core/executor.py

@@ -352,6 +352,7 @@ def _create_workflow_context(executor_context: BaseExecutorContext) -> WorkflowC
         vellum_client=Vellum(
             api_key=executor_context.environment_api_key,
             environment=environment,
+            api_version=executor_context.api_version,
         ),
         execution_context=executor_context.execution_context,
         generated_files=executor_context.files,

workflow_server/core/workflow_executor_context.py

@@ -3,6 +3,7 @@ from uuid import UUID
 from typing import Any, Optional
 
 from vellum.client.core import UniversalBaseModel
+from vellum.client.types.api_version_enum import ApiVersionEnum
 from vellum.workflows.context import ExecutionContext
 
 DEFAULT_TIMEOUT_SECONDS = 60 * 30
@@ -14,6 +15,7 @@ class BaseExecutorContext(UniversalBaseModel):
     timeout: int = DEFAULT_TIMEOUT_SECONDS
     files: dict[str, str]
     environment_api_key: str
+    api_version: Optional[ApiVersionEnum] = None
     execution_id: UUID
     module: str
     execution_context: ExecutionContext = field(default_factory=ExecutionContext)