retab 0.0.35__py3-none-any.whl → 0.0.37__py3-none-any.whl

{uiform → retab}/resources/consensus/responses.py

@@ -1,33 +1,20 @@
-import json
-from pathlib import Path
-from typing import Any, AsyncGenerator, Generator, TypeVar, Generic, Optional, Union, List, Sequence, cast
+from typing import Optional, TypeVar, Union
 
-from openai.types.chat.chat_completion_reasoning_effort import ChatCompletionReasoningEffort
-from openai.types.chat.parsed_chat_completion import ParsedChatCompletionMessage
 from openai.types.responses.response import Response
-from openai.types.responses.response_input_param import ResponseInputParam, ResponseInputItemParam
-from openai.types.responses.response_output_item import ResponseOutputItem
-from openai.types.shared_params.response_format_json_schema import ResponseFormatJSONSchema
+from openai.types.responses.response_input_param import ResponseInputParam
+from openai.types.responses.response_text_config_param import ResponseTextConfigParam
+from openai.types.shared_params.reasoning import Reasoning
 from pydantic import BaseModel
 
 from ..._resource import AsyncAPIResource, SyncAPIResource
 from ..._utils.ai_models import assert_valid_model_extraction
-from ..._utils.json_schema import load_json_schema, unflatten_dict
-from ..._utils.responses import convert_to_openai_format, convert_from_openai_format, parse_openai_responses_response
-from ..._utils.stream_context_managers import as_async_context_manager, as_context_manager
-from ...types.chat import ChatCompletionUiformMessage
-from ...types.completions import UiChatResponseCreateRequest, UiChatCompletionsRequest
-from ...types.documents.extractions import UiParsedChatCompletion, UiParsedChatCompletionChunk, UiParsedChoice, UiResponse
-from ...types.standards import PreparedRequest
+from ...types.completions import UiChatResponseCreateRequest
+from ...types.documents.extractions import UiResponse
 from ...types.schemas.object import Schema
+from ...types.standards import PreparedRequest
 
-from typing import Optional, Union
-from openai.types.shared_params.reasoning import Reasoning
-from openai.types.responses.response_input_param import ResponseInputParam
-from openai.types.responses.response_text_config_param import ResponseTextConfigParam
-from openai.types.shared_params.response_format_json_schema import ResponseFormatJSONSchema
+T = TypeVar("T", bound=BaseModel)
 
-T = TypeVar('T', bound=BaseModel)
 
 class BaseResponsesMixin:
     def prepare_create(
@@ -55,35 +42,20 @@ class BaseResponsesMixin:
 
         if instructions is None:
             instructions = schema_obj.developer_system_prompt
-        
+
         # Create the request object based on the UiChatResponseCreateRequest model
-        data = UiChatResponseCreateRequest(
+        request = UiChatResponseCreateRequest(
             model=model,
             input=input,
             temperature=temperature,
             stream=False,
-            reasoning=reasoning, 
+            reasoning=reasoning,
             n_consensus=n_consensus,
-            text={
-                "format": {
-                    "type": "json_schema", 
-                    "name": schema_obj.id, 
-                    "schema": schema_obj.inference_json_schema, 
-                    "strict": True
-                }
-            },
+            text={"format": {"type": "json_schema", "name": schema_obj.id, "schema": schema_obj.inference_json_schema, "strict": True}},
             instructions=instructions,
         )
 
-        # Validate the request data
-        ui_chat_response_create_request = UiChatResponseCreateRequest.model_validate(data)
-
-        return PreparedRequest(
-            method="POST", 
-            url="/v1/responses", 
-            data=ui_chat_response_create_request.model_dump(), 
-            idempotency_key=idempotency_key
-        )
+        return PreparedRequest(method="POST", url="/v1/responses", data=request.model_dump(), idempotency_key=idempotency_key)
 
     def prepare_parse(
         self,
@@ -106,42 +78,24 @@ class BaseResponsesMixin:
 
         if instructions is None:
             instructions = schema_obj.developer_system_prompt
-        
+
         # Create the request object based on the UiChatResponseCreateRequest model
-        data = UiChatResponseCreateRequest(
+        request = UiChatResponseCreateRequest(
             model=model,
             input=input,
             temperature=temperature,
             stream=False,
-            reasoning=reasoning, 
+            reasoning=reasoning,
             n_consensus=n_consensus,
-            text={
-                "format": {
-                    "type": "json_schema", 
-                    "name": schema_obj.id, 
-                    "schema": schema_obj.inference_json_schema, 
-                    "strict": True
-                }
-            },
+            text={"format": {"type": "json_schema", "name": schema_obj.id, "schema": schema_obj.inference_json_schema, "strict": True}},
             instructions=instructions,
         )
 
-        # Validate the request data
-        ui_chat_response_create_request = UiChatResponseCreateRequest.model_validate(data)
-
-        return PreparedRequest(
-            method="POST", 
-            url="/v1/responses", 
-            data=ui_chat_response_create_request.model_dump(), 
-            idempotency_key=idempotency_key
-        )
-        
-
         return PreparedRequest(
-            method="POST", 
-            url="/v1/completions", 
-            data=ui_chat_completions_request.model_dump(), 
-            idempotency_key=idempotency_key
+            method="POST",
+            url="/v1/responses",
+            data=request.model_dump(),
+            idempotency_key=idempotency_key,
         )
 
 
@@ -161,7 +115,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
     ) -> Response:
         """
         Create a completion using the UiForm API with OpenAI Responses API compatible interface.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -187,9 +141,9 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
 
         result = self._client._prepared_request(request)
         response = UiResponse.model_validate(result)
-        
+
         return response
-            
+
     def parse(
         self,
         model: str,
@@ -203,7 +157,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
     ) -> Response:
         """
         Parse content using the UiForm API with OpenAI Responses API compatible interface.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -213,7 +167,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Response: OpenAI Responses API compatible response with parsed content
         """
@@ -230,7 +184,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
 
         result = self._client._prepared_request(request)
         response = UiResponse.model_validate(result)
-        
+
         return response
 
 
@@ -250,7 +204,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
     ) -> UiResponse:
         """
         Create a completion using the UiForm API asynchronously with OpenAI Responses API compatible interface.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -260,7 +214,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Response: OpenAI Responses API compatible response
         """
@@ -278,9 +232,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
         result = await self._client._prepared_request(request)
         response = UiResponse.model_validate(result)
         return response
-        
 
-            
     async def parse(
         self,
         model: str,
@@ -294,7 +246,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
     ) -> UiResponse:
         """
         Parse content using the UiForm API asynchronously with OpenAI Responses API compatible interface.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -304,7 +256,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Response: OpenAI Responses API compatible response with parsed content
         """
@@ -322,4 +274,3 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
         result = await self._client._prepared_request(request)
         response = UiResponse.model_validate(result)
         return response
-        

{uiform → retab}/resources/consensus/responses_stream.py

@@ -1,33 +1,20 @@
-import json
-from pathlib import Path
-from typing import Any, AsyncGenerator, Generator, TypeVar, Generic, Optional, Union, List, Sequence, cast
-
-from openai.types.chat.chat_completion_reasoning_effort import ChatCompletionReasoningEffort
-from openai.types.chat.parsed_chat_completion import ParsedChatCompletionMessage
-from openai.types.responses.response import Response
-from openai.types.responses.response_input_param import ResponseInputParam, ResponseInputItemParam
-from openai.types.responses.response_output_item import ResponseOutputItem
-from openai.types.shared_params.response_format_json_schema import ResponseFormatJSONSchema
+from typing import AsyncGenerator, Generator, Optional, TypeVar, Union
+
+from openai.types.responses.response_input_param import ResponseInputParam
+from openai.types.responses.response_text_config_param import ResponseTextConfigParam
+from openai.types.shared_params.reasoning import Reasoning
 from pydantic import BaseModel
 
 from ..._resource import AsyncAPIResource, SyncAPIResource
 from ..._utils.ai_models import assert_valid_model_extraction
-from ..._utils.json_schema import load_json_schema, unflatten_dict
-from ..._utils.responses import convert_to_openai_format, convert_from_openai_format, parse_openai_responses_response
 from ..._utils.stream_context_managers import as_async_context_manager, as_context_manager
-from ...types.chat import ChatCompletionUiformMessage
-from ...types.completions import UiChatResponseCreateRequest, UiChatCompletionsRequest
-from ...types.documents.extractions import UiParsedChatCompletion, UiParsedChatCompletionChunk, UiParsedChoice, UiResponse
-from ...types.standards import PreparedRequest
+from ...types.completions import UiChatResponseCreateRequest
+from ...types.documents.extractions import UiResponse
 from ...types.schemas.object import Schema
+from ...types.standards import PreparedRequest
 
-from typing import Optional, Union
-from openai.types.shared_params.reasoning import Reasoning
-from openai.types.responses.response_input_param import ResponseInputParam
-from openai.types.responses.response_text_config_param import ResponseTextConfigParam
-from openai.types.shared_params.response_format_json_schema import ResponseFormatJSONSchema
+T = TypeVar("T", bound=BaseModel)
 
-T = TypeVar('T', bound=BaseModel)
 
 class BaseResponsesMixin:
     def prepare_create(
@@ -56,35 +43,20 @@ class BaseResponsesMixin:
 
         if instructions is None:
             instructions = schema_obj.developer_system_prompt
-        
+
         # Create the request object based on the UiChatResponseCreateRequest model
-        data = UiChatResponseCreateRequest(
+        request = UiChatResponseCreateRequest(
             model=model,
             input=input,
             temperature=temperature,
             stream=stream,
-            reasoning=reasoning, 
+            reasoning=reasoning,
             n_consensus=n_consensus,
-            text={
-                "format": {
-                    "type": "json_schema", 
-                    "name": schema_obj.id, 
-                    "schema": schema_obj.inference_json_schema, 
-                    "strict": True
-                }
-            },
+            text={"format": {"type": "json_schema", "name": schema_obj.id, "schema": schema_obj.inference_json_schema, "strict": True}},
             instructions=instructions,
         )
 
-        # Validate the request data
-        ui_chat_response_create_request = UiChatResponseCreateRequest.model_validate(data)
-
-        return PreparedRequest(
-            method="POST", 
-            url="/v1/responses", 
-            data=ui_chat_response_create_request.model_dump(), 
-            idempotency_key=idempotency_key
-        )
+        return PreparedRequest(method="POST", url="/v1/responses", data=request.model_dump(), idempotency_key=idempotency_key)
 
     def prepare_parse(
         self,
@@ -108,42 +80,24 @@ class BaseResponsesMixin:
 
         if instructions is None:
             instructions = schema_obj.developer_system_prompt
-        
+
         # Create the request object based on the UiChatResponseCreateRequest model
-        data = UiChatResponseCreateRequest(
+        request = UiChatResponseCreateRequest(
             model=model,
             input=input,
             temperature=temperature,
             stream=stream,
-            reasoning=reasoning, 
+            reasoning=reasoning,
             n_consensus=n_consensus,
-            text={
-                "format": {
-                    "type": "json_schema", 
-                    "name": schema_obj.id, 
-                    "schema": schema_obj.inference_json_schema, 
-                    "strict": True
-                }
-            },
+            text={"format": {"type": "json_schema", "name": schema_obj.id, "schema": schema_obj.inference_json_schema, "strict": True}},
             instructions=instructions,
         )
 
-        # Validate the request data
-        ui_chat_response_create_request = UiChatResponseCreateRequest.model_validate(data)
-
-        return PreparedRequest(
-            method="POST", 
-            url="/v1/responses", 
-            data=ui_chat_response_create_request.model_dump(), 
-            idempotency_key=idempotency_key
-        )
-        
-
         return PreparedRequest(
-            method="POST", 
-            url="/v1/completions", 
-            data=ui_chat_completions_request.model_dump(), 
-            idempotency_key=idempotency_key
+            method="POST",
+            url="/v1/responses",
+            data=request.model_dump(),
+            idempotency_key=idempotency_key,
         )
 
 
@@ -164,7 +118,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
     ) -> Generator[UiResponse, None, None]:
         """
         Create a completion using the UiForm API with streaming enabled.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -174,7 +128,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Generator[UiResponse]: Stream of responses
 
@@ -218,7 +172,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
     ) -> Generator[UiResponse, None, None]:
         """
         Parse content using the UiForm API with streaming enabled.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -228,7 +182,7 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Generator[UiResponse]: Stream of parsed responses
 
@@ -259,7 +213,6 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             yield response
 
 
-
 class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
     """UiForm Responses API compatible with OpenAI Responses API for async usage"""
 
@@ -277,7 +230,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
     ) -> AsyncGenerator[UiResponse, None]:
         """
         Create a completion using the UiForm API asynchronously with streaming enabled.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -287,7 +240,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             AsyncGenerator[UiResponse]: Async stream of responses
 
@@ -316,7 +269,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
                 continue
             response = UiResponse.model_validate(chunk_json)
             yield response
-        
+
     @as_async_context_manager
     async def stream_parse(
         self,
@@ -331,7 +284,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
     ) -> AsyncGenerator[UiResponse, None]:
         """
         Parse content using the UiForm API asynchronously with streaming enabled.
-        
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -341,7 +294,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             AsyncGenerator[UiResponse]: Async stream of parsed responses
 
@@ -370,4 +323,3 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
                 continue
             response = UiResponse.model_validate(chunk_json)
             yield response
-        

{uiform → retab}/resources/documents/client.py

@@ -1,18 +1,20 @@
 from io import IOBase
 from pathlib import Path
-from typing import Any, Literal
+from typing import Any
 
 import PIL.Image
 from pydantic import HttpUrl
+from pydantic_core import PydanticUndefined
 
 from ..._resource import AsyncAPIResource, SyncAPIResource
+from ..._utils.json_schema import load_json_schema
 from ..._utils.mime import convert_mime_data_to_pil_image, prepare_mime_document
-from ...types.documents.create_messages import DocumentCreateMessageRequest, DocumentMessage, DocumentCreateInputRequest
+from ...types.documents.create_messages import DocumentCreateInputRequest, DocumentCreateMessageRequest, DocumentMessage
+from ...types.browser_canvas import BrowserCanvas
 from ...types.mime import MIMEData
 from ...types.modalities import Modality
 from ...types.standards import PreparedRequest
 from .extractions import AsyncExtractions, Extractions
-from ..._utils.json_schema import load_json_schema
 
 
 class BaseDocumentsMixin:
@@ -20,21 +22,18 @@ class BaseDocumentsMixin:
         self,
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
         modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
+        image_resolution_dpi: int = PydanticUndefined,  # type: ignore[assignment]
+        browser_canvas: BrowserCanvas = PydanticUndefined,  # type: ignore[assignment]
         idempotency_key: str | None = None,
     ) -> PreparedRequest:
         mime_document = prepare_mime_document(document)
-        data: dict[str, Any] = {
-            "document": mime_document.model_dump(),
-            "modality": modality,
-        }
-        if image_resolution_dpi:
-            data["image_resolution_dpi"] = image_resolution_dpi
-        if browser_canvas:
-            data["browser_canvas"] = browser_canvas
-
-        loading_request = DocumentCreateMessageRequest.model_validate(data)
+
+        loading_request = DocumentCreateMessageRequest(
+            document=mime_document,
+            modality=modality,
+            image_resolution_dpi=image_resolution_dpi,
+            browser_canvas=browser_canvas,
+        )
         return PreparedRequest(method="POST", url="/v1/documents/create_messages", data=loading_request.model_dump(), idempotency_key=idempotency_key)
 
     def _prepare_create_inputs(
@@ -42,24 +41,20 @@ class BaseDocumentsMixin:
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
         json_schema: dict[str, Any] | Path | str,
         modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
+        image_resolution_dpi: int = PydanticUndefined,  # type: ignore[assignment],
+        browser_canvas: BrowserCanvas = PydanticUndefined,  # type: ignore[assignment],
         idempotency_key: str | None = None,
     ) -> PreparedRequest:
         mime_document = prepare_mime_document(document)
         loaded_schema = load_json_schema(json_schema)
 
-        data: dict[str, Any] = {
-            "document": mime_document.model_dump(),
-            "modality": modality,
-            "json_schema": loaded_schema,
-        }
-        if image_resolution_dpi:
-            data["image_resolution_dpi"] = image_resolution_dpi
-        if browser_canvas:
-            data["browser_canvas"] = browser_canvas
-
-        loading_request = DocumentCreateInputRequest.model_validate(data)
+        loading_request = DocumentCreateInputRequest(
+            document=mime_document,
+            modality=modality,
+            json_schema=loaded_schema,
+            image_resolution_dpi=image_resolution_dpi,
+            browser_canvas=browser_canvas,
+        )
         return PreparedRequest(method="POST", url="/v1/documents/create_inputs", data=loading_request.model_dump(), idempotency_key=idempotency_key)
 
     def _prepare_correct_image_orientation(self, document: Path | str | IOBase | MIMEData | PIL.Image.Image) -> PreparedRequest:
@@ -105,15 +100,15 @@ class Documents(SyncAPIResource, BaseDocumentsMixin):
         """
         request = self._prepare_correct_image_orientation(document)
         response = self._client._prepared_request(request)
-        mime_response = MIMEData.model_validate(response['document'])
+        mime_response = MIMEData.model_validate(response["document"])
         return convert_mime_data_to_pil_image(mime_response)
 
     def create_messages(
         self,
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
         modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
+        image_resolution_dpi: int = PydanticUndefined,  # type: ignore[assignment]
+        browser_canvas: BrowserCanvas = PydanticUndefined,  # type: ignore[assignment]
         idempotency_key: str | None = None,
     ) -> DocumentMessage:
         """
@@ -131,7 +126,9 @@ class Documents(SyncAPIResource, BaseDocumentsMixin):
         Raises:
             UiformAPIError: If the API request fails.
         """
-        request = self._prepare_create_messages(document, modality, image_resolution_dpi, browser_canvas, idempotency_key)
+        request = self._prepare_create_messages(
+            document=document, modality=modality, image_resolution_dpi=image_resolution_dpi, browser_canvas=browser_canvas, idempotency_key=idempotency_key
+        )
         response = self._client._prepared_request(request)
         return DocumentMessage.model_validate(response)
 
@@ -140,8 +137,8 @@ class Documents(SyncAPIResource, BaseDocumentsMixin):
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
         json_schema: dict[str, Any] | Path | str,
         modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
+        image_resolution_dpi: int = PydanticUndefined,  # type: ignore[assignment]
+        browser_canvas: BrowserCanvas = PydanticUndefined,  # type: ignore[assignment]
         idempotency_key: str | None = None,
     ) -> DocumentMessage:
         """
@@ -160,7 +157,14 @@ class Documents(SyncAPIResource, BaseDocumentsMixin):
         Raises:
             UiformAPIError: If the API request fails.
         """
-        request = self._prepare_create_inputs(document, json_schema, modality, image_resolution_dpi, browser_canvas, idempotency_key)
+        request = self._prepare_create_inputs(
+            document=document,
+            json_schema=json_schema,
+            modality=modality,
+            image_resolution_dpi=image_resolution_dpi,
+            browser_canvas=browser_canvas,
+            idempotency_key=idempotency_key,
+        )
         response = self._client._prepared_request(request)
         return DocumentMessage.model_validate(response)
 
@@ -176,8 +180,8 @@ class AsyncDocuments(AsyncAPIResource, BaseDocumentsMixin):
         self,
         document: Path | str | IOBase | MIMEData | PIL.Image.Image,
         modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
+        image_resolution_dpi: int = PydanticUndefined,  # type: ignore[assignment]
+        browser_canvas: BrowserCanvas = PydanticUndefined,  # type: ignore[assignment]
         idempotency_key: str | None = None,
     ) -> DocumentMessage:
         """
@@ -193,10 +197,13 @@ class AsyncDocuments(AsyncAPIResource, BaseDocumentsMixin):
         Raises:
             UiformAPIError: If the API request fails.
         """
-        request = self._prepare_create_messages(document, modality, image_resolution_dpi, browser_canvas, idempotency_key)
-        assert request.data is not None
-        print(request.data.keys())
-        print(request.data)
+        request = self._prepare_create_messages(
+            document=document,
+            modality=modality,
+            image_resolution_dpi=image_resolution_dpi,
+            browser_canvas=browser_canvas,
+            idempotency_key=idempotency_key,
+        )
         response = await self._client._prepared_request(request)
         return DocumentMessage.model_validate(response)
 
@@ -205,8 +212,8 @@ class AsyncDocuments(AsyncAPIResource, BaseDocumentsMixin):
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
         json_schema: dict[str, Any] | Path | str,
         modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
+        image_resolution_dpi: int = PydanticUndefined,  # type: ignore[assignment]
+        browser_canvas: BrowserCanvas = PydanticUndefined,  # type: ignore[assignment]
         idempotency_key: str | None = None,
     ) -> DocumentMessage:
         """
@@ -225,7 +232,14 @@ class AsyncDocuments(AsyncAPIResource, BaseDocumentsMixin):
         Raises:
             UiformAPIError: If the API request fails.
         """
-        request = self._prepare_create_inputs(document, json_schema, modality, image_resolution_dpi, browser_canvas, idempotency_key)
+        request = self._prepare_create_inputs(
+            document=document,
+            json_schema=json_schema,
+            modality=modality,
+            image_resolution_dpi=image_resolution_dpi,
+            browser_canvas=browser_canvas,
+            idempotency_key=idempotency_key,
+        )
         response = await self._client._prepared_request(request)
         return DocumentMessage.model_validate(response)
 
@@ -251,5 +265,5 @@ class AsyncDocuments(AsyncAPIResource, BaseDocumentsMixin):
         """
         request = self._prepare_correct_image_orientation(document)
         response = await self._client._prepared_request(request)
-        mime_response = MIMEData.model_validate(response['document'])
+        mime_response = MIMEData.model_validate(response["document"])
         return convert_mime_data_to_pil_image(mime_response)