fix: improve input validation to allow quotes for strategy parameter

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.0.81.dev
+
+* Update `strategy` parameter to allow `'` and `"` as input surrounding the value.
+
 ## 0.0.80
 
 * Bump to `unstructured` 0.15.10

prepline_general/api/models/form_params.py

@@ -156,7 +156,7 @@ def as_form(
                 description="The strategy to use for partitioning PDF/image. Options are fast, hi_res, auto. Default: auto",
                 examples=["auto", "hi_res"],
             ),
-            BeforeValidator(SmartValueParser[str]().value_or_first_element),
+            BeforeValidator(SmartValueParser[str]().literal_value_stripped_or_first_element),
         ] = "auto",
         extract_image_block_types: Annotated[
             List[str],

prepline_general/api/utils.py

@@ -94,6 +94,17 @@ def value_or_first_element(self, value: Union[T, list[T]]) -> list[T] | T | None
             return [_cast_to_type(elem, container_elems_class) for elem in value]
         return _cast_to_type(value, origin_class)  # noqa
 
+    def literal_value_stripped_or_first_element(self, value: str) -> str | None:
+        """Returns the value itself for literal strings and strips quotation characters.
+
+        Args:
+            value (Union[T, List[str]]): value to cast to a type T or return as is
+        """
+        origin_class, container_elems_class = self._get_origin_container_classes()
+        value = value.replace("'", "")
+        value = value.replace('"', "")
+        return _cast_to_type(value, origin_class)
+
     def _get_origin_container_classes(self) -> tuple[type, type | None]:
         """Extracts class (and container class if it's a list) from a type hint
 

test_general/test_utils.py → test_general/api/test_utils.py

@@ -30,3 +30,23 @@
 def test_smart_value_parser(desired_type: type, value_to_parse: Any, expected_result: Any):
     parsed_value = SmartValueParser[desired_type]().value_or_first_element(value_to_parse)
     assert expected_result == parsed_value
+
+
+@pytest.mark.parametrize(
+    "desired_type, value_to_parse, expected_result",
+    [
+        (str, "fast", "fast"),
+        (str, "'fast'", "fast"),
+        (str, '"fast"', "fast"),
+        (str, "!fast", "!fast"),
+        (str, "fa'st", "fast"),
+        (str, "fast''''''", "fast"),
+    ],
+)
+def test_literal_value_stripped_or_first_element(
+    desired_type: type, value_to_parse: Any, expected_result: Any
+):
+    parsed_value = SmartValueParser[desired_type]().literal_value_stripped_or_first_element(
+        value_to_parse
+    )
+    assert expected_result == parsed_value