vellum-ai 0.12.17__py3-none-any.whl → 0.13.1__py3-none-any.whl

vellum/workflows/workflows/base.py

@@ -188,6 +188,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
         external_inputs: Optional[ExternalInputsArg] = None,
         cancel_signal: Optional[ThreadingEvent] = None,
         node_output_mocks: Optional[List[BaseOutputs]] = None,
+        max_concurrency: Optional[int] = None,
     ) -> TerminalWorkflowEvent:
         """
         Invoke a Workflow, returning the last event emitted, which should be one of:
@@ -215,6 +216,11 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
 
         node_output_mocks: Optional[List[Outputs]] = None
             A list of Outputs to mock for Nodes during Workflow Execution.
+
+        max_concurrency: Optional[int] = None
+            The max number of concurrent threads to run the Workflow with. If not provided, the Workflow will run
+            without limiting concurrency. This configuration only applies to the current Workflow and not to any
+            subworkflows or nodes that utilizes threads.
         """
 
         events = WorkflowRunner(
@@ -226,6 +232,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             cancel_signal=cancel_signal,
             node_output_mocks=node_output_mocks,
             parent_context=self._context.parent_context,
+            max_concurrency=max_concurrency,
         ).stream()
         first_event: Optional[Union[WorkflowExecutionInitiatedEvent, WorkflowExecutionResumedEvent]] = None
         last_event = None
@@ -289,6 +296,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
         external_inputs: Optional[ExternalInputsArg] = None,
         cancel_signal: Optional[ThreadingEvent] = None,
         node_output_mocks: Optional[List[BaseOutputs]] = None,
+        max_concurrency: Optional[int] = None,
     ) -> WorkflowEventStream:
         """
         Invoke a Workflow, yielding events as they are emitted.
@@ -317,6 +325,11 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
 
         node_output_mocks: Optional[List[Outputs]] = None
             A list of Outputs to mock for Nodes during Workflow Execution.
+
+        max_concurrency: Optional[int] = None
+            The max number of concurrent threads to run the Workflow with. If not provided, the Workflow will run
+            without limiting concurrency. This configuration only applies to the current Workflow and not to any
+            subworkflows or nodes that utilizes threads.
         """
 
         should_yield = event_filter or workflow_event_filter
@@ -329,6 +342,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             cancel_signal=cancel_signal,
             node_output_mocks=node_output_mocks,
             parent_context=self.context.parent_context,
+            max_concurrency=max_concurrency,
         ).stream():
             if should_yield(self.__class__, event):
                 yield event
@@ -437,40 +451,8 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             raise ValueError(f"No workflows found in {module_path}")
         elif len(workflows) > 1:
             raise ValueError(f"Multiple workflows found in {module_path}")
-        try:
-            BaseWorkflow.import_node_display(module_path)
-        except ModuleNotFoundError:
-            pass
-
         return workflows[0]
 
-    @staticmethod
-    def import_node_display(module_path):
-        # Import the nodes package
-        nodes_package = importlib.import_module(f"{module_path}.display.nodes")
-        # Use the loader to get the code
-        if hasattr(nodes_package, "__spec__") and nodes_package.__spec__ and nodes_package.__spec__.loader:
-            loader = nodes_package.__spec__.loader
-
-            # Check if the loader has a code attribute
-            if hasattr(loader, "code"):
-                code = loader.code
-
-                # Parse the code to find import statements
-                import_lines = [line.strip() for line in code.splitlines() if line.startswith("from ")]
-
-                # Import each module specified in the code
-                for line in import_lines:
-                    try:
-                        # Extract module name from the import line
-                        module_name = line.split(" ")[1]
-                        full_module_path = f"{module_path}.display.nodes{module_name}"
-                        importlib.import_module(full_module_path)
-                    except Exception:
-                        continue
-        # Also import from workflow.py
-        importlib.import_module(f"{module_path}.display.workflow")
-
 
 WorkflowExecutionInitiatedBody.model_rebuild()
 WorkflowExecutionFulfilledBody.model_rebuild()

{vellum_ai-0.12.17.dist-info → vellum_ai-0.13.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.12.17
+Version: 0.13.1
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0