Merge pull request #127 from UiPath/fix/update_runtime_contracts

fix: update runtime contracts

Author: cristipufu

packages/uipath-llamaindex/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.4.1"
+version = "0.5.0"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
@@ -10,8 +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.5.0, <2.6.0",
-    "uipath-runtime>=0.5.0, <0.6.0",
+    "uipath>=2.6.0, <2.7.0",
+    "uipath-runtime>=0.6.0, <0.7.0",
 ]
 classifiers = [
     "Intended Audience :: Developers",

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

@@ -12,7 +12,9 @@
 from uipath.runtime import (
     UiPathResumableRuntime,
     UiPathRuntimeContext,
+    UiPathRuntimeFactorySettings,
     UiPathRuntimeProtocol,
+    UiPathRuntimeStorageProtocol,
 )
 from uipath.runtime.errors import UiPathErrorCategory
 from workflows import Workflow
@@ -211,27 +213,20 @@ def discover_entrypoints(self) -> list[str]:
             return []
         return config.entrypoints
 
-    async def discover_runtimes(self) -> list[UiPathRuntimeProtocol]:
+    async def get_storage(self) -> UiPathRuntimeStorageProtocol | None:
         """
-        Discover runtime instances for all entrypoints.
-
-        Returns:
-            List of LlamaIndexRuntime instances, one per entrypoint
+        Get the shared storage instance.
         """
-        entrypoints = self.discover_entrypoints()
+        return await self._get_storage()
 
-        runtimes: list[UiPathRuntimeProtocol] = []
-        for entrypoint in entrypoints:
-            workflow = await self._resolve_workflow(entrypoint)
-
-            runtime = await self._create_runtime_instance(
-                workflow=workflow,
-                runtime_id=entrypoint,
-                entrypoint=entrypoint,
-            )
-            runtimes.append(runtime)
+    async def get_settings(self) -> UiPathRuntimeFactorySettings | None:
+        """
+        Get the factory settings.
 
-        return runtimes
+        Returns:
+            Factory settings
+        """
+        return None
 
     async def _create_runtime_instance(
         self,