fix: add warmup method to factory protocol

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-runtime"
-version = "0.6.2"
+version = "0.7.0"
 description = "Runtime abstractions and interfaces for building agents and automation scripts in the UiPath ecosystem"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/runtime/factory.py

@@ -26,6 +26,10 @@ class UiPathRuntimeFactoryProtocol(
 ):
     """Protocol for discovering and creating UiPath runtime instances."""
 
+    async def warmup(self) -> None:
+        """Pre-load modules and resources to reduce cold start latency."""
+        ...
+
     def discover_entrypoints(self) -> list[str]:
         """Discover all runtime entrypoints."""
         ...

tests/test_factory.py

@@ -66,6 +66,9 @@ async def dispose(self) -> None:
 class CreatorWithKwargs:
     """Implementation with kwargs."""
 
+    async def warmup(self) -> None:
+        pass
+
     def discover_entrypoints(self) -> list[str]:
         return ["main.py"]
 

tests/test_registry.py

@@ -71,6 +71,9 @@ def __init__(self, context: Optional[UiPathRuntimeContext] = None):
         self.context = context
         self.name = "functions"
 
+    async def warmup(self) -> None:
+        pass
+
     def discover_entrypoints(self) -> list[str]:
         return ["main.py", "handler.py"]
 
@@ -96,6 +99,9 @@ def __init__(self, context: Optional[UiPathRuntimeContext] = None):
         self.context = context
         self.name = "langgraph"
 
+    async def warmup(self) -> None:
+        pass
+
     def discover_entrypoints(self) -> list[str]:
         return ["agent", "workflow"]
 
@@ -121,6 +127,9 @@ def __init__(self, context: Optional[UiPathRuntimeContext] = None):
         self.context = context
         self.name = "llamaindex"
 
+    async def warmup(self) -> None:
+        pass
+
     def discover_entrypoints(self) -> list[str]:
         return ["chatbot", "rag"]