vellum-workflow-server 1.3.11__py3-none-any.whl → 1.4.0__py3-none-any.whl

{vellum_workflow_server-1.3.11.dist-info → vellum_workflow_server-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.3.11
+Version: 1.4.0
 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.11)
+Requires-Dist: vellum-ai (==1.4.0)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.3.11.dist-info → vellum_workflow_server-1.4.0.dist-info}/RECORD

@@ -8,11 +8,11 @@ workflow_server/api/tests/test_workflow_view.py,sha256=RlAw1tHeIlnOXGrFQN-w3EOLP
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=Qo8u6mPyRCmE2jamY1yIh8l44hgo4-Nwlq03z61ND5g,27031
 workflow_server/api/workflow_view.py,sha256=Ufc49WJ4LON3FFxzY2MeP4q7uQyorMNMu9zLRuJvdp4,21523
 workflow_server/code_exec_runner.py,sha256=lBnMIorPZL8zZBye6TjeCIs06WTJM7P2HR07B1fjJJI,2533
-workflow_server/config.py,sha256=DyTty8NrAwvtx-esM3KthnpsNh-nKdWNlovWQOgiGpg,1417
+workflow_server/config.py,sha256=qmmTr6ty3ZN5LDOFs3TfUxYshYe6Mmn_LanplHHeE9Q,1796
 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=ZEA18S4vCZbolupRyoyRCG8O4wkpfKtv8vmvklH0kj4,17678
+workflow_server/core/executor.py,sha256=srpxx2xkjAE0QNBo1sfQV6Ovwyq3tWQdjicTySxSrpc,18092
 workflow_server/core/utils.py,sha256=aIpSINstLGslP2PIoDLM82_1GlJ1uC_0AIrP-V7Yobo,3230
 workflow_server/core/workflow_executor_context.py,sha256=w3OhV_AXpgh7AxpjEsc0vo-IJypgJcgr5DXJCqGptOU,1587
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
@@ -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=0Nq6du8o-iBtTrip9_wgHES53JSiJbVdSXaBnPobw3s,6930
 workflow_server/utils/utils.py,sha256=ZPoM1Suhid22dpB8oEFLux8wx-9iyzmSfWuYxSCrgWk,4774
-vellum_workflow_server-1.3.11.dist-info/METADATA,sha256=wnPS4p1OybwF8GbXPDa5SR-qKVRW-B2c0cCX9B8zN5s,2269
-vellum_workflow_server-1.3.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.3.11.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.3.11.dist-info/RECORD,,
+vellum_workflow_server-1.4.0.dist-info/METADATA,sha256=Xj4ZNLbf9KnNGWKA5Hm9EFX8dCdhxc5BS5TthF5z3Us,2267
+vellum_workflow_server-1.4.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.4.0.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.4.0.dist-info/RECORD,,

workflow_server/config.py

@@ -30,6 +30,12 @@ VELLUM_API_URL_PORT = os.getenv("VELLUM_API_URL_PORT", 8000)
 CONCURRENCY = int(os.getenv("CONCURRENCY", "8"))
 ENABLE_PROCESS_WRAPPER = os.getenv("ENABLE_PROCESS_WRAPPER", "true").lower() == "true"
 
+# This controls if we should just load a module already available in the py path for running
+# a workflow. Currently used for running agent builder end to end locally
+LOCAL_WORKFLOW_MODULE = os.getenv("LOCAL_WORKFLOW_MODULE")
+# The deployment name to match against when using local mode so you can still run your normal workflow
+LOCAL_DEPLOYMENT = os.getenv("LOCAL_DEPLOYMENT")
+
 
 def is_development() -> bool:
     return os.getenv("FLASK_ENV", "local") == "local"

workflow_server/core/executor.py

@@ -36,6 +36,7 @@ from vellum.workflows.state.base import BaseState, StateMeta
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.state.store import EmptyStore
 from vellum.workflows.workflows.event_filters import all_workflow_event_filter
+from workflow_server.config import LOCAL_DEPLOYMENT, LOCAL_WORKFLOW_MODULE
 from workflow_server.core.cancel_workflow import CancelWorkflowWatcherThread
 from workflow_server.core.events import (
     SPAN_ID_EVENT,
@@ -393,6 +394,13 @@ def _create_workflow_context(executor_context: BaseExecutorContext) -> WorkflowC
 
 
 def _get_file_namespace(executor_context: BaseExecutorContext) -> str:
+    if (
+        LOCAL_WORKFLOW_MODULE
+        and hasattr(executor_context.execution_context.parent_context, "deployment_name")
+        and LOCAL_DEPLOYMENT == executor_context.execution_context.parent_context.deployment_name  # type: ignore
+    ):
+        return LOCAL_WORKFLOW_MODULE
+
     return str(executor_context.execution_id) or "".join(
         random.choice(string.ascii_letters + string.digits) for i in range(14)
     )
@@ -488,7 +496,8 @@ def _get_run_from_node(executor_context: WorkflowExecutorContext, workflow: Base
 def _gather_workflow(context: WorkflowExecutorContext) -> Tuple[BaseWorkflow, str]:
     try:
         namespace = _get_file_namespace(context)
-        sys.meta_path.append(VirtualFileFinder(context.files, namespace))
+        if namespace != LOCAL_WORKFLOW_MODULE:
+            sys.meta_path.append(VirtualFileFinder(context.files, namespace))
         workflow = _create_workflow(
             executor_context=context,
             namespace=namespace,