feat: add agent extraction tool

Author: radu-mocanu

pyproject.toml

@@ -5,8 +5,7 @@ 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.4.0, <2.5.0",
-    "uipath-runtime>=0.4.0, <0.5.0",
+      "uipath-runtime>=0.4.0, <0.5.0",
     "langgraph>=1.0.0, <2.0.0",
     "langchain-core>=1.2.5, <2.0.0",
     "aiosqlite==0.21.0",
@@ -21,6 +20,7 @@ dependencies = [
     "jsonpath-ng>=1.7.0",
     "mcp==1.24.0",
     "langchain-mcp-adapters==0.2.1",
+      "uipath==2.4.11.dev1010893714",
 ]
 
 classifiers = [
@@ -115,3 +115,6 @@ name = "testpypi"
 url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
 explicit = true
+
+[tool.uv.sources]
+uipath = { index = "testpypi" }

src/uipath_langchain/agent/tools/__init__.py

@@ -8,6 +8,7 @@
 from .tool_factory import (
     create_tools_from_resources,
 )
+from .extraction_tool import create_ixp_extraction_tool
 from .tool_node import ToolWrapperMixin, UiPathToolNode, create_tool_node
 
 __all__ = [
@@ -18,6 +19,7 @@
     "create_integration_tool",
     "create_escalation_tool",
     "create_mcp_tools",
+    "create_ixp_extraction_tool",
     "UiPathToolNode",
     "ToolWrapperMixin",
 ]

src/uipath_langchain/agent/tools/extraction_tool.py

@@ -0,0 +1,55 @@
+"""Process tool creation for UiPath process execution."""
+
+from typing import Any
+
+from langchain_core.tools import StructuredTool
+from langgraph.types import interrupt
+from uipath.agent.models.agent import AgentIxpExtractionResourceConfig
+from uipath.eval.mocks import mockable
+from uipath.platform.common import DocumentExtraction
+from uipath.platform.attachments import Attachment
+from uipath.platform.documents import ExtractionResponseIXP
+
+from .structured_tool_with_output_type import StructuredToolWithOutputType
+from .utils import sanitize_tool_name
+
+
+def create_ixp_extraction_tool(resource: AgentIxpExtractionResourceConfig) -> StructuredTool:
+    """Uses interrupt() to suspend graph execution until data is extracted (handled by runtime)."""
+    tool_name: str = sanitize_tool_name(resource.name)
+    project_name = resource.properties.project_name
+    version_tag = resource.properties.version_tag
+
+    @mockable(
+        name=resource.name,
+        description=resource.description,
+        input_schema=Attachment.model_json_schema(),
+        output_schema=ExtractionResponseIXP.model_json_schema(),
+        example_calls=resource.properties.example_calls,
+    )
+    async def escalation_tool_fn(attachment: Attachment):
+        # current workaround. DocumentExtraction model should support attachment_id and use the
+        # start_ixp_extraction_from_attachment sdk method once support is added
+        from uipath.platform import UiPath
+        uipath = UiPath()
+        attachment_local_file_path = await uipath.attachments.download_async(
+            key=attachment.id,
+            destination_path=attachment.full_name
+        )
+        return interrupt(
+            DocumentExtraction(
+                project_name=project_name,
+                tag=version_tag,
+                file_path=attachment_local_file_path,
+            )
+        )
+
+    tool = StructuredToolWithOutputType(
+        name=tool_name,
+        description=resource.description,
+        args_schema=Attachment,
+        coroutine=escalation_tool_fn,
+        output_type=ExtractionResponseIXP,
+    )
+
+    return tool