vellum-workflow-server 1.7.1__tar.gz → 1.7.2__tar.gz

{vellum_workflow_server-1.7.1 → vellum_workflow_server-1.7.2}/PKG-INFO

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

{vellum_workflow_server-1.7.1 → vellum_workflow_server-1.7.2}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.7.1"
+version = "1.7.2"
 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.7.1"
+vellum-ai = "1.7.2"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.7.1 → vellum_workflow_server-1.7.2}/src/workflow_server/api/workflow_view.py

@@ -7,8 +7,6 @@ from multiprocessing import Process, Queue, set_start_method
 import os
 import pkgutil
 from queue import Empty
-import random
-import string
 import sys
 from threading import Event as ThreadingEvent
 import time
@@ -35,7 +33,12 @@ from workflow_server.core.events import (
     VembdaExecutionInitiatedBody,
     VembdaExecutionInitiatedEvent,
 )
-from workflow_server.core.executor import stream_node_process_timeout, stream_workflow, stream_workflow_process_timeout
+from workflow_server.core.executor import (
+    get_random_namespace,
+    stream_node_process_timeout,
+    stream_workflow,
+    stream_workflow_process_timeout,
+)
 from workflow_server.core.utils import (
     create_vembda_rejected_event,
     is_events_emitting_enabled,
@@ -434,7 +437,7 @@ def serialize_route() -> Response:
     client = create_vellum_client(api_key=workspace_api_key)
 
     # Generate a unique namespace for this serialization request
-    namespace = "".join(random.choice(string.ascii_letters + string.digits) for i in range(14))
+    namespace = get_random_namespace()
     virtual_finder = VirtualFileFinder(files, namespace)
     try:
         sys.meta_path.append(virtual_finder)

{vellum_workflow_server-1.7.1 → vellum_workflow_server-1.7.2}/src/workflow_server/core/executor.py

@@ -31,7 +31,6 @@ from vellum.workflows.nodes import BaseNode
 from vellum.workflows.nodes.mocks import MockNodeExecution
 from vellum.workflows.resolvers.base import BaseWorkflowResolver
 from vellum.workflows.resolvers.resolver import VellumResolver
-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
@@ -400,9 +399,14 @@ def _get_file_namespace(executor_context: BaseExecutorContext) -> str:
     ):
         return LOCAL_WORKFLOW_MODULE
 
-    return str(executor_context.execution_id) or "".join(
-        random.choice(string.ascii_letters + string.digits) for i in range(14)
-    )
+    if executor_context.execution_id:
+        return str(executor_context.execution_id)
+
+    return get_random_namespace()
+
+
+def get_random_namespace() -> str:
+    return "workflow_tmp_" + "".join(random.choice(string.ascii_letters + string.digits) for i in range(14))
 
 
 def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext) -> dict:
@@ -467,29 +471,6 @@ but found {inputs_module.Inputs.__class__.__name__}""",
     return inputs_module.Inputs(**executor_context.inputs)
 
 
-def _get_workflow_state(
-    executor_context: BaseExecutorContext, workflow_inputs: Optional[BaseInputs]
-) -> Optional[BaseState]:
-    namespace = _get_file_namespace(executor_context)
-    State = importlib.import_module(f"{namespace}.state").State if executor_context.files.get("state.py") else None
-
-    if not State:
-        return None
-
-    if not issubclass(State, BaseState):
-        return None
-
-    if executor_context.state:
-        return State(**executor_context.state)
-
-    if workflow_inputs:
-        return State(
-            meta=StateMeta(workflow_inputs=workflow_inputs),
-        )
-
-    return State()
-
-
 def _get_run_from_node(executor_context: WorkflowExecutorContext, workflow: BaseWorkflow) -> Optional[Type[BaseNode]]:
     if not executor_context.node_id:
         return None