fix: configurable state file

Author: cristipufu

packages/uipath-llamaindex/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.3.0"
+version = "0.4.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.4.0, <2.5.0",
-    "uipath-runtime>=0.4.0, <0.5.0",
+    "uipath>=2.5.0, <2.6.0",
+    "uipath-runtime>=0.5.0, <0.6.0",
 ]
 classifiers = [
     "Intended Audience :: Developers",
@@ -58,7 +58,9 @@ dev = [
     "pytest>=7.4.0",
     "pytest-cov>=4.1.0",
     "pytest-mock>=3.11.1",
-    "pre-commit>=4.1.0",
+    "pre-commit>=4.5.1",
+    "filelock>=3.20.3",
+    "virtualenv>=20.36.1",
     "pytest-asyncio>=1.0.0",
     "numpy>=1.24.0",
 ]

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

@@ -67,9 +67,16 @@ def _setup_instrumentation(self, trace_manager: UiPathTraceManager | None) -> No
 
     def _get_storage_path(self) -> str:
         """Get the storage path for workflow state."""
+        if self.context.state_file_path is not None:
+            return self.context.state_file_path
+
         if self.context.runtime_dir and self.context.state_file:
             path = os.path.join(self.context.runtime_dir, self.context.state_file)
-            if not self.context.resume and self.context.job_id is None:
+            if (
+                not self.context.resume
+                and self.context.job_id is None
+                and not self.context.keep_state_file
+            ):
                 # If not resuming and no job id, delete the previous state file
                 if os.path.exists(path):
                     os.remove(path)