uipath 2.1.111__py3-none-any.whl → 2.1.113__py3-none-any.whl

uipath/_services/documents_service.py

@@ -1,7 +1,8 @@
 import asyncio
 import time
+from contextlib import nullcontext
 from pathlib import Path
-from typing import Any, Awaitable, Callable, Dict, List, Optional, Set, Tuple
+from typing import Any, Awaitable, Callable, Dict, List, Optional, Set, Tuple, Union
 from uuid import UUID
 
 from httpx._types import FileContent
@@ -13,6 +14,8 @@ from .._utils import Endpoint
 from ..models.documents import (
     ActionPriority,
     ExtractionResponse,
+    ExtractionResponseIXP,
+    ProjectType,
     ValidatedResult,
     ValidationAction,
 )
@@ -41,11 +44,13 @@ class DocumentsService(FolderContext, BaseService):
             "X-UiPath-Internal-ConsumptionSourceType": "CodedAgents",
         }
 
-    def _get_project_id_by_name(self, project_name: str) -> str:
+    def _get_project_id_by_name(
+        self, project_name: str, project_type: ProjectType
+    ) -> str:
         response = self.request(
             "GET",
             url=Endpoint("/du_/api/framework/projects"),
-            params={"api-version": 1.1, "type": "IXP"},
+            params={"api-version": 1.1, "type": project_type.value},
             headers=self._get_common_headers(),
         )
 
@@ -58,11 +63,13 @@ class DocumentsService(FolderContext, BaseService):
         except StopIteration:
             raise ValueError(f"Project '{project_name}' not found.") from None
 
-    async def _get_project_id_by_name_async(self, project_name: str) -> str:
+    async def _get_project_id_by_name_async(
+        self, project_name: str, project_type: ProjectType
+    ) -> str:
         response = await self.request_async(
             "GET",
             url=Endpoint("/du_/api/framework/projects"),
-            params={"api-version": 1.1, "type": "IXP"},
+            params={"api-version": 1.1, "type": project_type.value},
             headers=self._get_common_headers(),
         )
 
@@ -93,8 +100,10 @@ class DocumentsService(FolderContext, BaseService):
         )
         return {tag["name"] for tag in response.json().get("tags", [])}
 
-    def _get_project_id_and_validate_tag(self, project_name: str, tag: str) -> str:
-        project_id = self._get_project_id_by_name(project_name)
+    def _get_project_id_and_validate_tag(
+        self, project_name: str, project_type: ProjectType, tag: str
+    ) -> str:
+        project_id = self._get_project_id_by_name(project_name, project_type)
         tags = self._get_project_tags(project_id)
         if tag not in tags:
             raise ValueError(
@@ -104,9 +113,11 @@ class DocumentsService(FolderContext, BaseService):
         return project_id
 
     async def _get_project_id_and_validate_tag_async(
-        self, project_name: str, tag: str
+        self, project_name: str, project_type: ProjectType, tag: str
     ) -> str:
-        project_id = await self._get_project_id_by_name_async(project_name)
+        project_id = await self._get_project_id_by_name_async(
+            project_name, project_type
+        )
         tags = await self._get_project_tags_async(project_id)
         if tag not in tags:
             raise ValueError(
@@ -147,16 +158,71 @@ class DocumentsService(FolderContext, BaseService):
             )
         ).json()["documentId"]
 
+    def _get_document_type_id(
+        self,
+        project_id: str,
+        document_type_name: Optional[str],
+        project_type: ProjectType,
+    ) -> str:
+        if project_type == ProjectType.IXP:
+            return str(UUID(int=0))
+
+        response = self.request(
+            "GET",
+            url=Endpoint(f"/du_/api/framework/projects/{project_id}/document-types"),
+            params={"api-version": 1.1},
+            headers=self._get_common_headers(),
+        )
+
+        try:
+            return next(
+                extractor["id"]
+                for extractor in response.json().get("documentTypes", [])
+                if extractor["name"].lower() == document_type_name.lower()  # type: ignore
+            )
+        except StopIteration:
+            raise ValueError(
+                f"Document type '{document_type_name}' not found."
+            ) from None
+
+    async def _get_document_type_id_async(
+        self,
+        project_id: str,
+        document_type_name: Optional[str],
+        project_type: ProjectType,
+    ) -> str:
+        if project_type == ProjectType.IXP:
+            return str(UUID(int=0))
+
+        response = await self.request_async(
+            "GET",
+            url=Endpoint(f"/du_/api/framework/projects/{project_id}/document-types"),
+            params={"api-version": 1.1},
+            headers=self._get_common_headers(),
+        )
+
+        try:
+            return next(
+                extractor["id"]
+                for extractor in response.json().get("documentTypes", [])
+                if extractor["name"].lower() == document_type_name.lower()  # type: ignore
+            )
+        except StopIteration:
+            raise ValueError(
+                f"Document type '{document_type_name}' not found."
+            ) from None
+
     def _start_extraction(
         self,
         project_id: str,
         tag: str,
+        document_type_id: str,
         document_id: str,
     ) -> str:
         return self.request(
             "POST",
             url=Endpoint(
-                f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/extraction/start"
+                f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/extraction/start"
             ),
             params={"api-version": 1.1},
             headers=self._get_common_headers(),
@@ -167,13 +233,14 @@ class DocumentsService(FolderContext, BaseService):
         self,
         project_id: str,
         tag: str,
+        document_type_id: str,
         document_id: str,
     ) -> str:
         return (
             await self.request_async(
                 "POST",
                 url=Endpoint(
-                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/extraction/start"
+                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/extraction/start"
                 ),
                 params={"api-version": 1.1},
                 headers=self._get_common_headers(),
@@ -183,7 +250,7 @@ class DocumentsService(FolderContext, BaseService):
 
     def _wait_for_operation(
         self,
-        result_getter: Callable[[], Tuple[str, Any]],
+        result_getter: Callable[[], Tuple[Any, Optional[Any], Optional[Any]]],
         wait_statuses: List[str],
         success_status: str,
     ) -> Any:
@@ -195,19 +262,23 @@ class DocumentsService(FolderContext, BaseService):
             status in wait_statuses
             and (time.monotonic() - start_time) < POLLING_TIMEOUT
         ):
-            status, result = result_getter()
+            status, error, result = result_getter()
             time.sleep(POLLING_INTERVAL)
 
         if status != success_status:
             if time.monotonic() - start_time >= POLLING_TIMEOUT:
                 raise TimeoutError("Operation timed out.")
-            raise RuntimeError(f"Operation failed with status: {status}")
+            raise RuntimeError(
+                f"Operation failed with status: {status}, error: {error}"
+            )
 
         return result
 
     async def _wait_for_operation_async(
         self,
-        result_getter: Callable[[], Awaitable[Tuple[str, Any]]],
+        result_getter: Callable[
+            [], Awaitable[Tuple[Any, Optional[Any], Optional[Any]]]
+        ],
         wait_statuses: List[str],
         success_status: str,
     ) -> Any:
@@ -219,55 +290,80 @@ class DocumentsService(FolderContext, BaseService):
             status in wait_statuses
             and (time.monotonic() - start_time) < POLLING_TIMEOUT
         ):
-            status, result = await result_getter()
+            status, error, result = await result_getter()
             await asyncio.sleep(POLLING_INTERVAL)
 
         if status != success_status:
             if time.monotonic() - start_time >= POLLING_TIMEOUT:
                 raise TimeoutError("Operation timed out.")
-            raise RuntimeError(f"Operation failed with status: {status}")
+            raise RuntimeError(
+                f"Operation failed with status: {status}, error: {error}"
+            )
 
         return result
 
     def _wait_for_extraction(
-        self, project_id: str, tag: str, operation_id: str
-    ) -> ExtractionResponse:
-        extraction_response = self._wait_for_operation(
-            result_getter=lambda: (
-                (
-                    result := self.request(
-                        method="GET",
-                        url=Endpoint(
-                            f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/extraction/result/{operation_id}"
-                        ),
-                        params={"api-version": 1.1},
-                        headers=self._get_common_headers(),
-                    ).json()
-                )["status"],
+        self,
+        project_id: str,
+        tag: str,
+        document_type_id: str,
+        operation_id: str,
+        project_type: ProjectType,
+    ) -> Union[ExtractionResponse, ExtractionResponseIXP]:
+        def result_getter() -> Tuple[str, str, Any]:
+            result = self.request(
+                method="GET",
+                url=Endpoint(
+                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/extraction/result/{operation_id}"
+                ),
+                params={"api-version": 1.1},
+                headers=self._get_common_headers(),
+            ).json()
+            return (
+                result["status"],
+                result.get("error", None),
                 result.get("result", None),
-            ),
+            )
+
+        extraction_response = self._wait_for_operation(
+            result_getter=result_getter,
             wait_statuses=["NotStarted", "Running"],
             success_status="Succeeded",
         )
 
         extraction_response["projectId"] = project_id
         extraction_response["tag"] = tag
+        extraction_response["documentTypeId"] = document_type_id
+
+        if project_type == ProjectType.IXP:
+            return ExtractionResponseIXP.model_validate(extraction_response)
+
         return ExtractionResponse.model_validate(extraction_response)
 
     async def _wait_for_extraction_async(
-        self, project_id: str, tag: str, operation_id: str
-    ) -> ExtractionResponse:
-        async def result_getter() -> Tuple[str, Any]:
-            result = await self.request_async(
-                method="GET",
-                url=Endpoint(
-                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/extraction/result/{operation_id}"
-                ),
-                params={"api-version": 1.1},
-                headers=self._get_common_headers(),
+        self,
+        project_id: str,
+        tag: str,
+        document_type_id: str,
+        operation_id: str,
+        project_type: ProjectType,
+    ) -> Union[ExtractionResponse, ExtractionResponseIXP]:
+        async def result_getter() -> Tuple[str, str, Any]:
+            result = (
+                await self.request_async(
+                    method="GET",
+                    url=Endpoint(
+                        f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/extraction/result/{operation_id}"
+                    ),
+                    params={"api-version": 1.1},
+                    headers=self._get_common_headers(),
+                )
+            ).json()
+            return (
+                result["status"],
+                result.get("error", None),
+                result.get("result", None),
             )
-            json_result = result.json()
-            return json_result["status"], json_result.get("result", None)
 
         extraction_response = await self._wait_for_operation_async(
             result_getter=result_getter,
@@ -277,6 +373,11 @@ class DocumentsService(FolderContext, BaseService):
 
         extraction_response["projectId"] = project_id
         extraction_response["tag"] = tag
+        extraction_response["documentTypeId"] = document_type_id
+
+        if project_type == ProjectType.IXP:
+            return ExtractionResponseIXP.model_validate(extraction_response)
+
         return ExtractionResponse.model_validate(extraction_response)
 
     @traced(name="documents_extract", run_type="uipath")
@@ -286,14 +387,18 @@ class DocumentsService(FolderContext, BaseService):
         tag: str,
         file: Optional[FileContent] = None,
         file_path: Optional[str] = None,
-    ) -> ExtractionResponse:
+        project_type: ProjectType = ProjectType.IXP,
+        document_type_name: Optional[str] = None,
+    ) -> Union[ExtractionResponse, ExtractionResponseIXP]:
         """Extract predicted data from a document using an IXP project.
 
         Args:
-            project_name (str): Name of the IXP project. Details about IXP projects can be found in the [official documentation](https://docs.uipath.com/ixp/automation-cloud/latest/overview/managing-projects#creating-a-new-project).
+            project_name (str): Name of the [IXP](https://docs.uipath.com/ixp/automation-cloud/latest/overview/managing-projects#creating-a-new-project)/[DU Modern](https://docs.uipath.com/document-understanding/automation-cloud/latest/user-guide/about-document-understanding) project.
             tag (str): Tag of the published project version.
             file (FileContent, optional): The document file to be processed.
             file_path (str, optional): Path to the document file to be processed.
+            project_type (ProjectType, optional): Type of the project. Defaults to `ProjectType.IXP`.
+            document_type_name (str, optional): Document type name associated with the extractor to be used for extraction. Required if `project_type` is `ProjectType.MODERN`.
 
         Note:
             Either `file` or `file_path` must be provided, but not both.
@@ -302,38 +407,63 @@ class DocumentsService(FolderContext, BaseService):
             ExtractionResponse: The extraction result containing predicted data.
 
         Examples:
+            IXP projects:
             ```python
             with open("path/to/document.pdf", "rb") as file:
                 extraction_response = service.extract(
-                    project_name="MyProject",
+                    project_name="MyIXPProjectName",
                     tag="live",
                     file=file,
                 )
             ```
+
+            DU Modern projects:
+            ```python
+            with open("path/to/document.pdf", "rb") as file:
+                extraction_response = service.extract(
+                    project_name="MyModernProjectName",
+                    tag="Production",
+                    file=file,
+                    project_type=ProjectType.MODERN,
+                    document_type_name="Receipts",
+                )
+            ```
         """
         if file is None and file_path is None:
             raise ValueError("Either `file` or `file_path` must be provided")
         if file is not None and file_path is not None:
             raise ValueError("`file` and `file_path` are mutually exclusive")
+        if project_type == ProjectType.MODERN and document_type_name is None:
+            raise ValueError(
+                "`document_type_name` must be provided when `project_type` is `ProjectType.MODERN`"
+            )
 
         project_id = self._get_project_id_and_validate_tag(
-            project_name=project_name, tag=tag
+            project_name=project_name, project_type=project_type, tag=tag
         )
 
-        if file_path is not None:
-            with open(Path(file_path), "rb") as handle:
-                document_id = self._start_digitization(
-                    project_id=project_id, file=handle
-                )
-        else:
-            document_id = self._start_digitization(project_id=project_id, file=file)  # type: ignore
+        with open(Path(file_path), "rb") if file_path else nullcontext(file) as handle:
+            document_id = self._start_digitization(project_id=project_id, file=handle)  # type: ignore
+
+        document_type_id = self._get_document_type_id(
+            project_id=project_id,
+            document_type_name=document_type_name,
+            project_type=project_type,
+        )
 
         operation_id = self._start_extraction(
-            project_id=project_id, tag=tag, document_id=document_id
+            project_id=project_id,
+            tag=tag,
+            document_type_id=document_type_id,
+            document_id=document_id,
         )
 
         return self._wait_for_extraction(
-            project_id=project_id, tag=tag, operation_id=operation_id
+            project_id=project_id,
+            tag=tag,
+            document_type_id=document_type_id,
+            operation_id=operation_id,
+            project_type=project_type,
         )
 
     @traced(name="documents_extract_async", run_type="uipath")
@@ -343,40 +473,55 @@ class DocumentsService(FolderContext, BaseService):
         tag: str,
         file: Optional[FileContent] = None,
         file_path: Optional[str] = None,
-    ) -> ExtractionResponse:
-        """Asynchronously extract predicted data from a document using an IXP project."""
+        project_type: ProjectType = ProjectType.IXP,
+        document_type_name: Optional[str] = None,
+    ) -> Union[ExtractionResponse, ExtractionResponseIXP]:
+        """Asynchronously version of the [`extract`][uipath._services.documents_service.DocumentsService.extract] method."""
         if file is None and file_path is None:
             raise ValueError("Either `file` or `file_path` must be provided")
         if file is not None and file_path is not None:
             raise ValueError("`file` and `file_path` are mutually exclusive")
+        if project_type == ProjectType.MODERN and document_type_name is None:
+            raise ValueError(
+                "`document_type_name` must be provided when `project_type` is `ProjectType.MODERN`"
+            )
 
         project_id = await self._get_project_id_and_validate_tag_async(
-            project_name=project_name, tag=tag
+            project_name=project_name, project_type=project_type, tag=tag
         )
 
-        if file_path is not None:
-            with open(Path(file_path), "rb") as handle:
-                document_id = await self._start_digitization_async(
-                    project_id=project_id, file=handle
-                )
-        else:
+        with open(Path(file_path), "rb") if file_path else nullcontext(file) as handle:
             document_id = await self._start_digitization_async(
                 project_id=project_id,
-                file=file,  # type: ignore
+                file=handle,  # type: ignore
             )
 
+        document_type_id = await self._get_document_type_id_async(
+            project_id=project_id,
+            document_type_name=document_type_name,
+            project_type=project_type,
+        )
+
         operation_id = await self._start_extraction_async(
-            project_id=project_id, tag=tag, document_id=document_id
+            project_id=project_id,
+            tag=tag,
+            document_type_id=document_type_id,
+            document_id=document_id,
         )
 
         return await self._wait_for_extraction_async(
-            project_id=project_id, tag=tag, operation_id=operation_id
+            project_id=project_id,
+            tag=tag,
+            document_type_id=document_type_id,
+            operation_id=operation_id,
+            project_type=project_type,
         )
 
     def _start_validation(
         self,
         project_id: str,
         tag: str,
+        document_type_id: str,
         action_title: str,
         action_priority: ActionPriority,
         action_catalog: str,
@@ -388,7 +533,7 @@ class DocumentsService(FolderContext, BaseService):
         return self.request(
             "POST",
             url=Endpoint(
-                f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/validation/start"
+                f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/validation/start"
             ),
             params={"api-version": 1.1},
             headers=self._get_common_headers(),
@@ -409,6 +554,7 @@ class DocumentsService(FolderContext, BaseService):
         self,
         project_id: str,
         tag: str,
+        document_type_id: str,
         action_title: str,
         action_priority: ActionPriority,
         action_catalog: str,
@@ -421,7 +567,7 @@ class DocumentsService(FolderContext, BaseService):
             await self.request_async(
                 "POST",
                 url=Endpoint(
-                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/validation/start"
+                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/validation/start"
                 ),
                 params={"api-version": 1.1},
                 headers=self._get_common_headers(),
@@ -440,25 +586,25 @@ class DocumentsService(FolderContext, BaseService):
         ).json()["operationId"]
 
     def _get_validation_result(
-        self, project_id: str, tag: str, operation_id: str
+        self, project_id: str, tag: str, document_type_id: str, operation_id: str
     ) -> Dict:  # type: ignore
         return self.request(
             method="GET",
             url=Endpoint(
-                f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/validation/result/{operation_id}"
+                f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/validation/result/{operation_id}"
             ),
             params={"api-version": 1.1},
             headers=self._get_common_headers(),
         ).json()
 
     async def _get_validation_result_async(
-        self, project_id: str, tag: str, operation_id: str
+        self, project_id: str, tag: str, document_type_id: str, operation_id: str
     ) -> Dict:  # type: ignore
         return (
             await self.request_async(
                 method="GET",
                 url=Endpoint(
-                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{UUID(int=0)}/validation/result/{operation_id}"
+                    f"/du_/api/framework/projects/{project_id}/{tag}/document-types/{document_type_id}/validation/result/{operation_id}"
                 ),
                 params={"api-version": 1.1},
                 headers=self._get_common_headers(),
@@ -466,43 +612,58 @@ class DocumentsService(FolderContext, BaseService):
         ).json()
 
     def _wait_for_create_validation_action(
-        self, project_id: str, tag: str, operation_id: str
+        self, project_id: str, tag: str, document_type_id: str, operation_id: str
     ) -> ValidationAction:
-        response = self._wait_for_operation(
-            lambda: (
-                (
-                    result := self._get_validation_result(
-                        project_id=project_id, tag=tag, operation_id=operation_id
-                    )
-                )["status"],
+        def result_getter() -> Tuple[Any, Optional[Any], Optional[Any]]:
+            result = self._get_validation_result(
+                project_id=project_id,
+                tag=tag,
+                document_type_id=document_type_id,
+                operation_id=operation_id,
+            )
+            return (
+                result["status"],
+                result.get("error", None),
                 result.get("result", None),
-            ),
+            )
+
+        response = self._wait_for_operation(
+            result_getter=result_getter,
             wait_statuses=["NotStarted", "Running"],
             success_status="Succeeded",
         )
 
         response["projectId"] = project_id
         response["tag"] = tag
+        response["documentTypeId"] = document_type_id
         response["operationId"] = operation_id
         return ValidationAction.model_validate(response)
 
     async def _wait_for_create_validation_action_async(
-        self, project_id: str, tag: str, operation_id: str
+        self, project_id: str, tag: str, document_type_id: str, operation_id: str
     ) -> ValidationAction:
-        async def result_getter() -> Tuple[str, Any]:
+        async def result_getter_async() -> Tuple[Any, Optional[Any], Optional[Any]]:
             result = await self._get_validation_result_async(
-                project_id=project_id, tag=tag, operation_id=operation_id
+                project_id=project_id,
+                tag=tag,
+                document_type_id=document_type_id,
+                operation_id=operation_id,
+            )
+            return (
+                result["status"],
+                result.get("error", None),
+                result.get("result", None),
             )
-            return result["status"], result.get("result", None)
 
         response = await self._wait_for_operation_async(
-            result_getter=result_getter,
+            result_getter=result_getter_async,
             wait_statuses=["NotStarted", "Running"],
             success_status="Succeeded",
         )
 
         response["projectId"] = project_id
         response["tag"] = tag
+        response["documentTypeId"] = document_type_id
         response["operationId"] = operation_id
         return ValidationAction.model_validate(response)
 
@@ -546,7 +707,8 @@ class DocumentsService(FolderContext, BaseService):
         """
         operation_id = self._start_validation(
             project_id=extraction_response.project_id,
-            tag=extraction_response.tag,  # should I validate tag again?
+            tag=extraction_response.tag,
+            document_type_id=extraction_response.document_type_id,
             action_title=action_title,
             action_priority=action_priority,
             action_catalog=action_catalog,
@@ -559,6 +721,7 @@ class DocumentsService(FolderContext, BaseService):
         return self._wait_for_create_validation_action(
             project_id=extraction_response.project_id,
             tag=extraction_response.tag,
+            document_type_id=extraction_response.document_type_id,
             operation_id=operation_id,
         )
 
@@ -573,11 +736,11 @@ class DocumentsService(FolderContext, BaseService):
         storage_bucket_directory_path: str,
         extraction_response: ExtractionResponse,
     ) -> ValidationAction:
-        """Asynchronously create a validation action for a document based on the extraction response."""
-        # Add reference to sync method docstring
+        """Asynchronous version of the [`create_validation_action`][uipath._services.documents_service.DocumentsService.create_validation_action] method."""
         operation_id = await self._start_validation_async(
             project_id=extraction_response.project_id,
-            tag=extraction_response.tag,  # should I validate tag again?
+            tag=extraction_response.tag,
+            document_type_id=extraction_response.document_type_id,
             action_title=action_title,
             action_priority=action_priority,
             action_catalog=action_catalog,
@@ -590,6 +753,7 @@ class DocumentsService(FolderContext, BaseService):
         return await self._wait_for_create_validation_action_async(
             project_id=extraction_response.project_id,
             tag=extraction_response.tag,
+            document_type_id=extraction_response.document_type_id,
             operation_id=operation_id,
         )
 
@@ -613,17 +777,22 @@ class DocumentsService(FolderContext, BaseService):
             validated_result = service.get_validation_result(validation_action)
             ```
         """
-        response = self._wait_for_operation(
-            result_getter=lambda: (
-                (
-                    result := self._get_validation_result(
-                        project_id=validation_action.project_id,
-                        tag=validation_action.tag,
-                        operation_id=validation_action.operation_id,
-                    )
-                )["result"]["actionStatus"],
+
+        def result_getter() -> Tuple[str, None, Any]:
+            result = self._get_validation_result(
+                project_id=validation_action.project_id,
+                tag=validation_action.tag,
+                document_type_id=validation_action.document_type_id,
+                operation_id=validation_action.operation_id,
+            )
+            return (
+                result["result"]["actionStatus"],
+                None,
                 result["result"].get("validatedExtractionResults", None),
-            ),
+            )
+
+        response = self._wait_for_operation(
+            result_getter=result_getter,
             wait_statuses=["Unassigned", "Pending"],
             success_status="Completed",
         )
@@ -634,16 +803,19 @@ class DocumentsService(FolderContext, BaseService):
     async def get_validation_result_async(
         self, validation_action: ValidationAction
     ) -> ValidatedResult:
-        """Asynchronously get the result of a validation action."""
+        """Asynchronous version of the [`get_validation_result`][uipath._services.documents_service.DocumentsService.get_validation_result] method."""
 
-        async def result_getter() -> Tuple[str, Any]:
+        async def result_getter() -> Tuple[str, None, Any]:
             result = await self._get_validation_result_async(
                 project_id=validation_action.project_id,
                 tag=validation_action.tag,
+                document_type_id=validation_action.document_type_id,
                 operation_id=validation_action.operation_id,
             )
-            return result["result"]["actionStatus"], result["result"].get(
-                "validatedExtractionResults", None
+            return (
+                result["result"]["actionStatus"],
+                None,
+                result["result"].get("validatedExtractionResults", None),
             )
 
         response = await self._wait_for_operation_async(