retab 0.0.41__py3-none-any.whl → 0.0.43__py3-none-any.whl

retab/resources/processors/automations/endpoints.py

@@ -1,6 +1,6 @@
-from typing import Literal, Optional
+from typing import Any, Literal, Optional
 
-from pydantic_core import PydanticUndefined
+from ....types.standards import FieldUnset
 
 from ...._resource import AsyncAPIResource, SyncAPIResource
 from ....utils.ai_models import assert_valid_model_extraction
@@ -15,22 +15,27 @@ class EndpointsMixin:
         name: str,
         webhook_url: str,
         model: str = "gpt-4o-mini",
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> PreparedRequest:
         assert_valid_model_extraction(model)
 
-        request = Endpoint(
-            processor_id=processor_id,
-            name=name,
-            webhook_url=webhook_url,
-            webhook_headers=webhook_headers,
-            need_validation=need_validation,
-        )
-        return PreparedRequest(method="POST", url="/v1/processors/automations/endpoints", data=request.model_dump(mode="json"))
+        endpoint_dict: dict[str, Any] = {
+            'processor_id': processor_id,
+            'name': name,
+            'webhook_url': webhook_url,
+        }
+        if webhook_headers is not FieldUnset:
+            endpoint_dict['webhook_headers'] = webhook_headers
+        if need_validation is not FieldUnset:
+            endpoint_dict['need_validation'] = need_validation
+
+        request = Endpoint(**endpoint_dict)
+        return PreparedRequest(method="POST", url="/v1/processors/automations/endpoints", data=request.model_dump(mode="json", exclude_unset=True))
 
     def prepare_list(
         self,
+        processor_id: str,
         before: Optional[str] = None,
         after: Optional[str] = None,
         limit: Optional[int] = 10,
@@ -40,6 +45,7 @@ class EndpointsMixin:
         webhook_url: Optional[str] = None,
     ) -> PreparedRequest:
         params = {
+            "processor_id": processor_id,
             "before": before,
             "after": after,
             "limit": limit,
@@ -66,20 +72,26 @@ class EndpointsMixin:
     def prepare_update(
         self,
         endpoint_id: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        name: str = FieldUnset,
+        default_language: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> PreparedRequest:
-        request = UpdateEndpointRequest(
-            name=name,
-            default_language=default_language,
-            webhook_url=webhook_url,
-            webhook_headers=webhook_headers,
-            need_validation=need_validation,
-        )
-        return PreparedRequest(method="PUT", url=f"/v1/processors/automations/endpoints/{endpoint_id}", data=request.model_dump(mode="json"))
+        update_dict: dict[str, Any] = {}
+        if name is not FieldUnset:
+            update_dict['name'] = name
+        if default_language is not FieldUnset:
+            update_dict['default_language'] = default_language
+        if webhook_url is not FieldUnset:
+            update_dict['webhook_url'] = webhook_url
+        if webhook_headers is not FieldUnset:
+            update_dict['webhook_headers'] = webhook_headers
+        if need_validation is not FieldUnset:
+            update_dict['need_validation'] = need_validation
+
+        request = UpdateEndpointRequest(**update_dict)
+        return PreparedRequest(method="PUT", url=f"/v1/processors/automations/endpoints/{endpoint_id}", data=request.model_dump(mode="json", exclude_unset=True))
 
     def prepare_delete(self, endpoint_id: str) -> PreparedRequest:
         return PreparedRequest(method="DELETE", url=f"/v1/processors/automations/endpoints/{endpoint_id}")
@@ -93,8 +105,8 @@ class Endpoints(SyncAPIResource, EndpointsMixin):
         processor_id: str,
         name: str,
         webhook_url: str,
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Endpoint:
         """Create a new endpoint configuration.
 
@@ -120,11 +132,12 @@ class Endpoints(SyncAPIResource, EndpointsMixin):
             need_validation=need_validation,
         )
         response = self._client._prepared_request(request)
-        print(f"Endpoint Created. Url: https://www.retab.dev/dashboard/processors/automations/{response['id']}")
+        print(f"Endpoint Created. Url: https://www.retab.com/dashboard/processors/automations/{response['id']}")
         return Endpoint.model_validate(response)
 
     def list(
         self,
+        processor_id: str,
         before: Optional[str] = None,
         after: Optional[str] = None,
         limit: Optional[int] = 10,
@@ -145,7 +158,7 @@ class Endpoints(SyncAPIResource, EndpointsMixin):
         Returns:
             ListEndpoints: Paginated list of endpoint configurations with metadata
         """
-        request = self.prepare_list(before, after, limit, order, name, webhook_url)
+        request = self.prepare_list(processor_id, before, after, limit, order, name, webhook_url)
         response = self._client._prepared_request(request)
         return ListEndpoints.model_validate(response)
 
@@ -165,11 +178,11 @@ class Endpoints(SyncAPIResource, EndpointsMixin):
     def update(
         self,
         endpoint_id: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        name: str = FieldUnset,
+        default_language: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Endpoint:
         """Update an endpoint configuration.
 
@@ -218,8 +231,8 @@ class AsyncEndpoints(AsyncAPIResource, EndpointsMixin):
         processor_id: str,
         name: str,
         webhook_url: str,
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Endpoint:
         request = self.prepare_create(
             processor_id=processor_id,
@@ -229,12 +242,13 @@ class AsyncEndpoints(AsyncAPIResource, EndpointsMixin):
             need_validation=need_validation,
         )
         response = await self._client._prepared_request(request)
-        print(f"Endpoint Created. Url: https://www.retab.dev/dashboard/processors/automations/{response['id']}")
+        print(f"Endpoint Created. Url: https://www.retab.com/dashboard/processors/automations/{response['id']}")
 
         return Endpoint.model_validate(response)
 
     async def list(
         self,
+        processor_id: str,
         before: Optional[str] = None,
         after: Optional[str] = None,
         limit: Optional[int] = 10,
@@ -242,7 +256,7 @@ class AsyncEndpoints(AsyncAPIResource, EndpointsMixin):
         name: Optional[str] = None,
         webhook_url: Optional[str] = None,
     ) -> ListEndpoints:
-        request = self.prepare_list(before, after, limit, order, name, webhook_url)
+        request = self.prepare_list(processor_id, before, after, limit, order, name, webhook_url)
         response = await self._client._prepared_request(request)
         return ListEndpoints.model_validate(response)
 
@@ -254,11 +268,11 @@ class AsyncEndpoints(AsyncAPIResource, EndpointsMixin):
     async def update(
         self,
         endpoint_id: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        name: str = FieldUnset,
+        default_language: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Endpoint:
         request = self.prepare_update(
             endpoint_id=endpoint_id,

retab/resources/processors/automations/links.py

@@ -1,10 +1,8 @@
 from typing import Any, Literal, Optional
 
-from pydantic_core import PydanticUndefined
-
 from ...._resource import AsyncAPIResource, SyncAPIResource
 from ....types.automations.links import Link, ListLinks, UpdateLinkRequest
-from ....types.standards import PreparedRequest
+from ....types.standards import PreparedRequest, FieldUnset
 
 
 class LinksMixin:
@@ -15,19 +13,24 @@ class LinksMixin:
         processor_id: str,
         name: str,
         webhook_url: str,
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        password: str | None = PydanticUndefined,  # type: ignore[assignment]
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        password: str | None = FieldUnset,
     ) -> PreparedRequest:
-        request = Link(
-            processor_id=processor_id,
-            name=name,
-            webhook_url=webhook_url,
-            webhook_headers=webhook_headers,
-            need_validation=need_validation,
-            password=password,
-        )
-        return PreparedRequest(method="POST", url=self.links_base_url, data=request.model_dump(mode="json"))
+        link_dict: dict[str, Any] = {
+            'processor_id': processor_id,
+            'name': name,
+            'webhook_url': webhook_url,
+        }
+        if webhook_headers is not FieldUnset:
+            link_dict['webhook_headers'] = webhook_headers
+        if need_validation is not FieldUnset:
+            link_dict['need_validation'] = need_validation
+        if password is not FieldUnset:
+            link_dict['password'] = password
+
+        request = Link(**link_dict)
+        return PreparedRequest(method="POST", url=self.links_base_url, data=request.model_dump(mode="json", exclude_unset=True))
 
     def prepare_list(
         self,
@@ -66,19 +69,25 @@ class LinksMixin:
     def prepare_update(
         self,
         link_id: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        password: str | None = PydanticUndefined,  # type: ignore[assignment]
+        name: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        password: str | None = FieldUnset,
     ) -> PreparedRequest:
-        request = UpdateLinkRequest(
-            name=name,
-            webhook_url=webhook_url,
-            webhook_headers=webhook_headers,
-            need_validation=need_validation,
-            password=password,
-        )
+        update_dict: dict[str, Any] = {}
+        if name is not FieldUnset:
+            update_dict['name'] = name
+        if webhook_url is not FieldUnset:
+            update_dict['webhook_url'] = webhook_url
+        if webhook_headers is not FieldUnset:
+            update_dict['webhook_headers'] = webhook_headers
+        if need_validation is not FieldUnset:
+            update_dict['need_validation'] = need_validation
+        if password is not FieldUnset:
+            update_dict['password'] = password
+
+        request = UpdateLinkRequest(**update_dict)
         return PreparedRequest(method="PUT", url=f"{self.links_base_url}/{link_id}", data=request.model_dump(mode="json", exclude_unset=True, exclude_defaults=True))
 
     def prepare_delete(self, link_id: str) -> PreparedRequest:
@@ -96,9 +105,9 @@ class Links(SyncAPIResource, LinksMixin):
         processor_id: str,
         name: str,
         webhook_url: str,
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        password: str | None = PydanticUndefined,  # type: ignore[assignment]
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        password: str | None = FieldUnset,
     ) -> Link:
         """Create a new extraction link configuration.
 
@@ -128,7 +137,7 @@ class Links(SyncAPIResource, LinksMixin):
         )
         response = self._client._prepared_request(request)
 
-        print(f"Link Created. Url: https://www.retab.dev/dashboard/processors/automations/{response['id']}")
+        print(f"Link Created. Url: https://www.retab.com/dashboard/processors/automations/{response['id']}")
         return Link.model_validate(response)
 
     def list(
@@ -174,11 +183,11 @@ class Links(SyncAPIResource, LinksMixin):
     def update(
         self,
         link_id: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        password: str | None = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        name: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        password: str | None = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Link:
         """Update an extraction link configuration.
 
@@ -235,9 +244,9 @@ class AsyncLinks(AsyncAPIResource, LinksMixin):
         processor_id: str,
         name: str,
         webhook_url: str,
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        password: str | None = PydanticUndefined,  # type: ignore[assignment]
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        password: str | None = FieldUnset,
     ) -> Link:
         request = self.prepare_create(
             processor_id=processor_id,
@@ -248,7 +257,7 @@ class AsyncLinks(AsyncAPIResource, LinksMixin):
             password=password,
         )
         response = await self._client._prepared_request(request)
-        print(f"Link Created. Url: https://www.retab.dev/dashboard/processors/automations/{response['id']}")
+        print(f"Link Created. Url: https://www.retab.com/dashboard/processors/automations/{response['id']}")
         return Link.model_validate(response)
 
     async def list(
@@ -272,11 +281,11 @@ class AsyncLinks(AsyncAPIResource, LinksMixin):
     async def update(
         self,
         link_id: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        password: str | None = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        name: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        password: str | None = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Link:
         request = self.prepare_update(
             link_id=link_id,

retab/resources/processors/automations/logs.py

@@ -147,7 +147,7 @@ class Logs(SyncAPIResource, LogsMixin):
         request = self.prepare_rerun(processor_id, log_id)
         response = self._client._prepared_request(request)
 
-        print(f"Webhook call run successfully. Log available at https://www.retab.dev/dashboard/processors/{processor_id}/logs/{log_id}")
+        print(f"Webhook call run successfully. Log available at https://www.retab.com/dashboard/processors/{processor_id}/logs/{log_id}")
 
         return ExternalRequestLog.model_validate(response)
 
@@ -217,6 +217,6 @@ class AsyncLogs(AsyncAPIResource, LogsMixin):
         request = self.prepare_rerun(processor_id, log_id)
         response = await self._client._prepared_request(request)
 
-        print(f"Webhook call run successfully. Log available at https://www.retab.dev/dashboard/processors/{processor_id}/logs/{log_id}")
+        print(f"Webhook call run successfully. Log available at https://www.retab.com/dashboard/processors/{processor_id}/logs/{log_id}")
 
         return ExternalRequestLog.model_validate(response)