wip: uipath-platform

pyproject.toml

@@ -5,8 +5,10 @@ description = "Python SDK that enables developers to build and deploy LangGraph
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
-    "uipath>=2.8.46, <2.9.0",
-    "uipath-runtime>=0.9.0, <0.10.0",
+    "uipath==2.9.0.dev1013544947",
+    "uipath-platform==0.0.1.dev1000020019",
+    "uipath-core>=0.5.0, <0.6.0",
+    "uipath-runtime==0.9.0.dev1000950339",
     "langgraph>=1.0.0, <2.0.0",
     "langchain-core>=1.2.11, <2.0.0",
     "langgraph-checkpoint-sqlite>=3.0.3, <4.0.0",
@@ -116,3 +118,9 @@ name = "testpypi"
 url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
 explicit = true
+
+[tool.uv.sources]
+uipath-platform = { index = "testpypi" }
+uipath-core = { index = "testpypi" }
+uipath = {index = "testpypi"}
+uipath-runtime = { index = "testpypi" }

src/uipath_langchain/agent/tools/internal_tools/batch_transform_tool.py

@@ -12,8 +12,11 @@
 )
 from uipath.eval.mocks import mockable
 from uipath.platform import UiPath
-from uipath.platform.common import CreateBatchTransform, UiPathConfig
-from uipath.platform.common.interrupt_models import WaitEphemeralIndex
+from uipath.platform.common import (
+    CreateBatchTransform,
+    UiPathConfig,
+    WaitEphemeralIndex,
+)
 from uipath.platform.context_grounding import (
     BatchTransformOutputColumn,
     EphemeralIndexUsage,

src/uipath_langchain/agent/tools/internal_tools/deeprag_tool.py

@@ -12,8 +12,7 @@
 )
 from uipath.eval.mocks import mockable
 from uipath.platform import UiPath
-from uipath.platform.common import CreateDeepRag
-from uipath.platform.common.interrupt_models import WaitEphemeralIndex
+from uipath.platform.common import CreateDeepRag, WaitEphemeralIndex
 from uipath.platform.context_grounding import (
     CitationMode,
     EphemeralIndexUsage,

src/uipath_langchain/chat/bedrock.py

@@ -7,8 +7,7 @@
 from langchain_core.messages import BaseMessage
 from langchain_core.outputs import ChatGenerationChunk, ChatResult
 from tenacity import AsyncRetrying, Retrying
-from uipath._utils import resource_override
-from uipath.utils import EndpointManager
+from uipath.platform.common import EndpointManager, resource_override
 
 from .header_capture import HeaderCapture
 from .retryers.bedrock import AsyncBedrockRetryer, BedrockRetryer

src/uipath_langchain/chat/models.py

@@ -17,7 +17,7 @@
 from langchain_core.runnables import Runnable
 from langchain_openai.chat_models import AzureChatOpenAI
 from pydantic import BaseModel
-from uipath.utils import EndpointManager
+from uipath.platform.common import EndpointManager
 
 from uipath_langchain._utils._request_mixin import UiPathRequestMixin
 

src/uipath_langchain/chat/openai.py

@@ -5,9 +5,11 @@
 import httpx
 from langchain_openai import AzureChatOpenAI
 from pydantic import PrivateAttr
-from uipath._utils import resource_override
-from uipath._utils._ssl_context import get_httpx_client_kwargs
-from uipath.utils import EndpointManager
+from uipath.platform.common import (
+    EndpointManager,
+    get_httpx_client_kwargs,
+    resource_override,
+)
 
 from .supported_models import OpenAIModels
 from .types import APIFlavor, LLMProvider

src/uipath_langchain/chat/vertex.py

@@ -13,7 +13,7 @@
 from tenacity import AsyncRetrying, Retrying
 from uipath._utils import resource_override
 from uipath._utils._ssl_context import get_httpx_client_kwargs
-from uipath.utils import EndpointManager
+from uipath.platform.common import EndpointManager
 
 from .header_capture import HeaderCapture
 from .retryers.vertex import AsyncVertexRetryer, VertexRetryer

src/uipath_langchain/embeddings/embeddings.py

@@ -4,8 +4,7 @@
 import httpx
 from langchain_openai.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
 from pydantic import Field
-from uipath._utils._ssl_context import get_httpx_client_kwargs
-from uipath.utils import EndpointManager
+from uipath.platform.common import EndpointManager, get_httpx_client_kwargs
 
 from uipath_langchain._utils._request_mixin import UiPathRequestMixin
 

src/uipath_langchain/runtime/storage.py

@@ -6,7 +6,7 @@
 from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
 from pydantic import BaseModel
 from uipath.core.serialization import serialize_json
-from uipath.runtime import UiPathResumeTrigger
+from uipath.core.triggers import UiPathResumeTrigger
 
 
 class SqliteResumableStorage:

tests/runtime/test_resumable.py

@@ -7,12 +7,14 @@
 from langgraph.graph import END, START, StateGraph
 from langgraph.types import interrupt
 from uipath.core.errors import ErrorCategory, UiPathPendingTriggerError
-from uipath.runtime import (
-    UiPathExecuteOptions,
-    UiPathResumableRuntime,
+from uipath.core.triggers import (
     UiPathResumeTrigger,
     UiPathResumeTriggerName,
     UiPathResumeTriggerType,
+)
+from uipath.runtime import (
+    UiPathExecuteOptions,
+    UiPathResumableRuntime,
     UiPathRuntimeStatus,
 )
 

tests/runtime/test_storage.py

@@ -7,7 +7,7 @@
 
 import pytest
 from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
-from uipath.runtime import (
+from uipath.core.triggers import (
     UiPathResumeTrigger,
     UiPathResumeTriggerName,
     UiPathResumeTriggerType,