diff --git a/pyproject.toml b/pyproject.toml
index 9222e41..3ce22ab 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.87"
+version = "0.0.88"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath==2.0.1",
+    "uipath>=2.0.4, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",
diff --git a/src/uipath_langchain/_cli/_runtime/_escalation.py b/src/uipath_langchain/_cli/_runtime/_escalation.py
index b2e2213..c63f7cb 100644
--- a/src/uipath_langchain/_cli/_runtime/_escalation.py
+++ b/src/uipath_langchain/_cli/_runtime/_escalation.py
@@ -4,7 +4,7 @@
 from typing import Any, Dict, Optional, Union
 
 from uipath import UiPath
-from uipath._models.actions import Action
+from uipath.models.actions import Action
 
 logger = logging.getLogger(__name__)
 
diff --git a/src/uipath_langchain/_cli/_runtime/_output.py b/src/uipath_langchain/_cli/_runtime/_output.py
index 1c0bcc6..9ffa1b3 100644
--- a/src/uipath_langchain/_cli/_runtime/_output.py
+++ b/src/uipath_langchain/_cli/_runtime/_output.py
@@ -15,8 +15,8 @@
     UiPathRuntimeResult,
     UiPathRuntimeStatus,
 )
-from uipath._models import CreateAction, InvokeProcess, WaitAction, WaitJob
-from uipath._models.actions import Action
+from uipath.models import CreateAction, InvokeProcess, WaitAction, WaitJob
+from uipath.models.actions import Action
 
 from ._context import LangGraphRuntimeContext
 from ._escalation import Escalation