fix: storage scoped to runtime id

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.1.8"
+version = "0.2.0"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
@@ -10,7 +10,8 @@ dependencies = [
     "llama-index-embeddings-azure-openai>=0.4.1",
     "llama-index-llms-azure-openai>=0.4.2",
     "openinference-instrumentation-llama-index>=4.3.9",
-    "uipath>=2.2.26, <2.3.0",
+    "uipath>=2.3.0, <2.4.0",
+    "uipath-runtime>=0.3.1, <0.4.0",
 ]
 classifiers = [
     "Intended Audience :: Developers",
@@ -58,6 +59,7 @@ dev = [
     "pytest-cov>=4.1.0",
     "pytest-mock>=3.11.1",
     "pre-commit>=4.1.0",
+    "pytest-asyncio>=1.0.0",
     "numpy>=1.24.0",
 ]
 
@@ -98,10 +100,11 @@ disallow_untyped_defs = false
 testpaths = ["tests"]
 python_files = "test_*.py"
 addopts = "-ra -q"
+asyncio_default_fixture_loop_scope = "function"
+asyncio_mode = "auto"
 
 [[tool.uv.index]]
 name = "testpypi"
 url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
 explicit = true
-

src/uipath_llamaindex/runtime/factory.py

@@ -260,6 +260,7 @@ async def _create_runtime_instance(
             delegate=base_runtime,
             storage=storage,
             trigger_manager=trigger_manager,
+            runtime_id=runtime_id,
         )
 
     async def new_runtime(

src/uipath_llamaindex/runtime/runtime.py

@@ -138,7 +138,6 @@ async def _run_workflow(
         Core workflow execution logic used by both execute() and stream().
         """
         workflow_input = input or {}
-
         is_resuming = bool(options and options.resume)
 
         if is_resuming:
@@ -154,7 +153,11 @@ async def _run_workflow(
         if is_resuming:
             handler: WorkflowHandler = self.workflow.run(ctx=self._context)
             if workflow_input:
-                handler.ctx.send_event(HumanResponseEvent(**workflow_input))
+                handler.ctx.send_event(
+                    HumanResponseEvent(
+                        **workflow_input.get(self.runtime_id, workflow_input)
+                    )
+                )
             else:
                 handler.ctx.send_event(BreakpointResumeEvent())
         else:
@@ -267,13 +270,14 @@ def _create_suspended_result(
             if hasattr(event, "_data") and "prefix" in event._data:
                 prefix = event._data["prefix"]
 
+            resume_map = {self.runtime_id: prefix or ""}
             return UiPathRuntimeResult(
-                output=prefix,
+                output=resume_map,
                 status=UiPathRuntimeStatus.SUSPENDED,
             )
 
         return UiPathRuntimeResult(
-            output=event,
+            output={self.runtime_id: event},
             status=UiPathRuntimeStatus.SUSPENDED,
         )