feat(pptx): partition_pptx() accepts strategy arg

test_unstructured/partition/pptx/test_pptx.py

@@ -261,7 +261,7 @@ def test_partition_pptx_malformed():
         assert element.metadata.filename == "fake-power-point-malformed.pptx"
 
 
-# == DescribePptxPartitionerMetadataBehaviors ====================================================
+# == metadata behaviors ==========================================================================
 
 
 def test_partition_pptx_metadata_date(mocker: MockFixture):
@@ -357,7 +357,7 @@ def test_partition_pptx_raises_TypeError_for_invalid_languages():
         partition_pptx(example_doc_path("fake-power-point.pptx"), languages="eng")  # type: ignore
 
 
-# == DescribePptxPartitionerDownstreamBehaviors ==================================================
+# == downstream behaviors ========================================================================
 
 
 def test_partition_pptx_with_json():
@@ -714,6 +714,17 @@ def but_it_raises_ValueError_when_neither_a_file_path_or_file_is_provided(
         with pytest.raises(ValueError, match="No PPTX document specified, either `filename` or "):
             opts.pptx_file
 
+    # -- .strategy -------------------------------
+
+    @pytest.mark.parametrize("arg_value", ["fast", "hi_res"])
+    def it_knows_which_partitioning_strategy_to_use(
+        self, arg_value: str, opts_args: dict[str, Any]
+    ):
+        opts_args["strategy"] = arg_value
+        opts = _PptxPartitionerOptions(**opts_args)
+
+        assert opts.strategy == arg_value
+
     # -- .table_metadata -------------------------
 
     def it_can_create_table_metadata(self, opts_args: dict[str, Any]):
@@ -776,4 +787,5 @@ def opts_args(self) -> dict[str, Any]:
             "infer_table_structure": True,
             "metadata_file_path": None,
             "metadata_last_modified": None,
+            "strategy": "fast",
         }

unstructured/partition/pptx.py

@@ -39,6 +39,7 @@
     is_possible_narrative_text,
     is_possible_title,
 )
+from unstructured.partition.utils.constants import PartitionStrategy
 from unstructured.utils import lazyproperty
 
 DETECTION_ORIGIN = "pptx"
@@ -49,15 +50,17 @@
 @add_chunking_strategy
 def partition_pptx(
     filename: Optional[str] = None,
+    *,
     file: Optional[IO[bytes]] = None,
+    date_from_file_object: bool = False,
+    detect_language_per_element: bool = False,
     include_page_breaks: bool = True,
-    metadata_filename: Optional[str] = None,
-    metadata_last_modified: Optional[str] = None,
     include_slide_notes: bool = False,
     infer_table_structure: bool = True,
     languages: Optional[list[str]] = ["auto"],
-    detect_language_per_element: bool = False,
-    date_from_file_object: bool = False,
+    metadata_filename: Optional[str] = None,
+    metadata_last_modified: Optional[str] = None,
+    strategy: str = PartitionStrategy.FAST,
     **kwargs: Any,
 ) -> list[Element]:
     """Partition PowerPoint document in .pptx format into its document elements.
@@ -104,6 +107,7 @@ def partition_pptx(
         infer_table_structure=infer_table_structure,
         metadata_file_path=metadata_filename,
         metadata_last_modified=metadata_last_modified,
+        strategy=strategy,
     )
 
     elements = _PptxPartitioner.iter_presentation_elements(opts)
@@ -314,6 +318,7 @@ def __init__(
         infer_table_structure: bool,
         metadata_file_path: Optional[str],
         metadata_last_modified: Optional[str],
+        strategy: str,
     ):
         self._date_from_file_object = date_from_file_object
         self._file = file
@@ -323,6 +328,8 @@ def __init__(
         self._infer_table_structure = infer_table_structure
         self._metadata_file_path = metadata_file_path
         self._metadata_last_modified = metadata_last_modified
+        self._strategy = strategy
+        # -- options object maintains page-number state --
         self._page_counter = 0
 
     @lazyproperty
@@ -414,6 +421,19 @@ def pptx_file(self) -> str | IO[bytes]:
             "No PPTX document specified, either `filename` or `file` argument must be provided"
         )
 
+    @lazyproperty
+    def strategy(self) -> str:
+        """The requested partitioning strategy.
+
+        This indicates whether the partitioner should undertake expensive operations like inference
+        and OCR to produce a more thorough and/or accurate partitioning of the document.
+
+        Can take several values but for PPTX purposes there is only "hi_res" and not "hi_res".
+        Depending on the picture-partitioner used, images may only be OCR'ed and added to the
+        element-stream when this partitioning strategy is "hi_res".
+        """
+        return self._strategy
+
     def table_metadata(self, text_as_html: str | None):
         """ElementMetadata instance suitable for use with Table element."""
         element_metadata = ElementMetadata(