fix: use public Context.get_step_context() for debug breakpoints

replaces private Context._create_internal() with the new public
Context.get_step_context() API from llama-index-workflows>=2.18.0

Author: radu-mocanu

packages/uipath-llamaindex/pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.5.10"
+version = "0.5.11"
 description = "Python SDK that enables developers to build and deploy LlamaIndex agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
     "aiosqlite>=0.21.0",
     "llama-index>=0.14.13, <1.0.0",
-    "llama-index-workflows>=2.14.1, <3.0.0",
+    "llama-index-workflows>=2.18.0, <3.0.0",
     "llama-index-embeddings-azure-openai>=0.4.1",
     "llama-index-llms-azure-openai>=0.4.2",
     "openinference-instrumentation-llama-index>=4.3.9",

packages/uipath-llamaindex/src/uipath_llamaindex/runtime/breakpoints.py

@@ -71,15 +71,13 @@ def make_wrapper(
     """
     Return a wrapped step function that pauses on breakpoints.
 
-    The wrapper creates an InternalContext via ``Context._create_internal``
-    to call ``wait_for_event``. This works because the wrapper executes
-    inside the step worker where the framework has already set the
-    ``StepWorkerStateContextVar``.
+    Uses ``Context.get_step_context()`` to retrieve the current step's
+    context without requiring the step to declare a ``ctx`` parameter.
     """
 
     @functools.wraps(original)
     async def wrapper(self, *args: Any, **kwargs: Any) -> Any:
-        ctx = Context._create_internal(workflow=self)
+        ctx = Context.get_step_context()
 
         bp_event = BreakpointEvent(
             breakpoint_node=step_name,

packages/uipath-llamaindex/src/uipath_llamaindex/runtime/schema.py

@@ -136,7 +136,7 @@ def get_entrypoints_schema(workflow: Workflow) -> dict[str, Any]:
     else:
         # For custom StopEvent subclasses, extract their Pydantic schema
         try:
-            output_schema = stop_event_class.model_json_schema()
+            output_schema = stop_event_class.model_json_schema()  # type: ignore[attr-defined]
             # Resolve references and handle nullable types
             unpacked_output, _ = transform_references(output_schema)
             schema["output"]["properties"] = transform_nullable_types(