[fix] Handling of WorkflowDone exception

llama-index-core/llama_index/core/workflow/workflow.py

@@ -228,6 +228,8 @@ async def _task(
                             try:
                                 new_ev = await instrumented_step(**kwargs)
                                 break  # exit the retrying loop
+                            except WorkflowDone:
+                                raise
                             except Exception as e:
                                 if config.retry_policy is None:
                                     raise WorkflowRuntimeError(
@@ -256,6 +258,8 @@ async def _task(
                             new_ev = await asyncio.get_event_loop().run_in_executor(
                                 None, run_task
                             )
+                        except WorkflowDone:
+                            raise
                         except Exception as e:
                             raise WorkflowRuntimeError(
                                 f"Error in step '{name}': {e!s}"

llama-index-core/tests/workflow/test_streaming.py

@@ -66,7 +66,9 @@ async def step(self, ctx: Context, ev: StartEvent) -> StopEvent:
             assert ev.test_param == "foo"
 
     # Make sure the await actually caught the exception
-    with pytest.raises(ValueError, match="The step raised an error!"):
+    with pytest.raises(
+        WorkflowRuntimeError, match="Error in step 'step': The step raised an error!"
+    ):
         await r
 
 

llama-index-core/tests/workflow/test_workflow.py

@@ -406,7 +406,9 @@ async def step(self, ev: StartEvent) -> StopEvent:
             raise ValueError("The step raised an error!")
 
     workflow = DummyWorkflow()
-    with pytest.raises(ValueError, match="The step raised an error!"):
+    with pytest.raises(
+        WorkflowRuntimeError, match="Error in step 'step': The step raised an error!"
+    ):
         await workflow.run()
 
 
@@ -418,7 +420,9 @@ async def step(self, ev: StartEvent) -> StopEvent:
             raise ValueError("The step raised an error!")
 
     workflow = DummyWorkflow()
-    with pytest.raises(ValueError, match="The step raised an error!"):
+    with pytest.raises(
+        WorkflowRuntimeError, match="Error in step 'step': The step raised an error!"
+    ):
         handler = workflow.run(stepwise=True)
         await handler.run_step()
 

llama-index-integrations/embeddings/llama-index-embeddings-nebius/pyproject.toml

@@ -27,11 +27,12 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-embeddings-nebius"
 readme = "README.md"
-version = "0.3.0"
+version = "0.3.1"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
 llama-index-core = "^0.12.0"
+llama-index-embeddings-openai = "^0.3.0"
 
 [tool.poetry.group.dev.dependencies]
 ipython = "8.10.0"