fix typo and types

Author: Coniferish

_test_unstructured_client/unit/test_split_pdf_hook.py

@@ -14,6 +14,7 @@
 
 from unstructured_client._hooks.custom import form_utils, pdf_utils, request_utils
 from unstructured_client._hooks.custom.form_utils import (
+    FormData,
     PARTITION_FORM_CONCURRENCY_LEVEL_KEY,
     PARTITION_FORM_STARTING_PAGE_NUMBER_KEY,
     PARTITION_FORM_PAGE_RANGE_KEY,
@@ -442,7 +443,7 @@ def test_unit_get_split_pdf_cache_tmp_data_dir_uses_dir_from_form_data(mock_path
     # -- Create the form_data
     dir_key = form_utils.PARTITION_FORM_SPLIT_CACHE_TMP_DATA_DIR_KEY # -- "split_pdf_cache_tmp_data_dir"
     mock_dir = "/mock/dir"
-    form_data = {dir_key: mock_dir}  
+    form_data: FormData = {dir_key: mock_dir}  
 
     # -- Mock the Path object in form_utils
     mock_path_instance = MagicMock()
@@ -457,7 +458,7 @@ def test_unit_get_split_pdf_cache_tmp_data_dir_uses_dir_from_form_data(mock_path
     )
 
     assert dir_key == "split_pdf_cache_tmp_data_dir"
-    assert form_data.get(dir_key) == "/mock/existing/dir"
+    assert form_data.get(dir_key) == "/mock/dir"
     mock_path.assert_called_once_with(mock_dir)
     mock_path_instance.exists.assert_called_once()
     assert result == str(Path(mock_dir).resolve())

src/unstructured_client/_hooks/custom/form_utils.py

@@ -2,7 +2,7 @@
 
 import logging
 from pathlib import Path
-from typing import TYPE_CHECKING, Mapping
+from typing import TYPE_CHECKING
 from typing_extensions import TypeAlias
 
 from requests_toolbelt.multipart.decoder import MultipartDecoder  # type: ignore
@@ -14,7 +14,7 @@
     from typing import Union
 
 logger = logging.getLogger(UNSTRUCTURED_CLIENT_LOGGER_NAME)
-FormData: TypeAlias = "Mapping[str, Union[str, shared.Files, list[str]]]"
+FormData: TypeAlias = "dict[str, Union[str, shared.Files, list[str]]]"
 
 PARTITION_FORM_FILES_KEY = "files"
 PARTITION_FORM_SPLIT_PDF_PAGE_KEY = "split_pdf_page"