query_pdf_info: Default to all queries.

- Introduced `ALL_PDF_INFO_QUERIES` constant for predefined query sets.
- Updated synchronous and asynchronous `query_pdf_info` methods to use
  `ALL_PDF_INFO_QUERIES` as the default value for `queries`.
- Adjusted type definitions and imports accordingly.
- Updated tests to cover new defaults.

Assisted-by: Codex

Author: datalogics-kam

src/pdfrest/client.py

@@ -45,7 +45,7 @@
     TiffPdfRestPayload,
     UploadURLs,
 )
-from .types import PdfInfoQuery
+from .types import ALL_PDF_INFO_QUERIES, PdfInfoQuery
 
 DEFAULT_BASE_URL = "https://api.pdfrest.com"
 API_KEY_ENV_VAR = "PDFREST_API_KEY"
@@ -1420,7 +1420,7 @@ def query_pdf_info(
         self,
         file: PdfRestFile | Sequence[PdfRestFile],
         *,
-        queries: Sequence[PdfInfoQuery] | PdfInfoQuery,
+        queries: Sequence[PdfInfoQuery] | PdfInfoQuery = ALL_PDF_INFO_QUERIES,
         extra_query: Query | None = None,
         extra_headers: AnyMapping | None = None,
         extra_body: Body | None = None,
@@ -1687,7 +1687,7 @@ async def query_pdf_info(
         self,
         file: PdfRestFile | Sequence[PdfRestFile],
         *,
-        queries: Sequence[PdfInfoQuery] | PdfInfoQuery,
+        queries: Sequence[PdfInfoQuery] | PdfInfoQuery = ALL_PDF_INFO_QUERIES,
         extra_query: Query | None = None,
         extra_headers: AnyMapping | None = None,
         extra_body: Body | None = None,

src/pdfrest/models/_validators.py

@@ -1,5 +1,5 @@
 """Public import surface for shared pdfrest types."""
 
-from .public import PdfInfoQuery
+from .public import ALL_PDF_INFO_QUERIES, PdfInfoQuery
 
-__all__ = ["PdfInfoQuery"]
+__all__ = ["ALL_PDF_INFO_QUERIES", "PdfInfoQuery"]

src/pdfrest/types/__init__.py

@@ -2,9 +2,9 @@
 
 from __future__ import annotations
 
-from typing import Literal
+from typing import Literal, cast, get_args
 
-__all__ = ("PdfInfoQuery",)
+__all__ = ("ALL_PDF_INFO_QUERIES", "PdfInfoQuery")
 
 PdfInfoQuery = Literal[
     "tagged",
@@ -39,3 +39,7 @@
     "pdfx_claim",
     "requires_password_to_open",
 ]
+
+ALL_PDF_INFO_QUERIES: tuple[PdfInfoQuery, ...] = cast(
+    tuple[PdfInfoQuery, ...], get_args(PdfInfoQuery)
+)

src/pdfrest/types/public.py

@@ -7,7 +7,7 @@
 from pdfrest import AsyncPdfRestClient, PdfRestApiError, PdfRestClient
 from pdfrest.models import PdfRestFile, PdfRestInfoResponse
 from pdfrest.models._internal import PdfInfoPayload
-from pdfrest.types import PdfInfoQuery
+from pdfrest.types import ALL_PDF_INFO_QUERIES, PdfInfoQuery
 
 from ..resources import get_test_resource_path
 
@@ -19,6 +19,7 @@ def _allowed_queries() -> tuple[PdfInfoQuery, ...]:
 
 
 ALLOWED_QUERIES: tuple[PdfInfoQuery, ...] = _allowed_queries()
+assert ALLOWED_QUERIES == ALL_PDF_INFO_QUERIES
 
 
 EXPECTED_VALUES: dict[PdfInfoQuery, Any] = {

tests/live/test_live_pdf_info.py

@@ -9,7 +9,7 @@
 
 from pdfrest import AsyncPdfRestClient, PdfRestClient
 from pdfrest.models import PdfRestFileID, PdfRestInfoResponse
-from pdfrest.types import PdfInfoQuery
+from pdfrest.types import ALL_PDF_INFO_QUERIES, PdfInfoQuery
 
 from .graphics_test_helpers import ASYNC_API_KEY, VALID_API_KEY, make_pdf_file
 
@@ -61,6 +61,39 @@ def handler(request: httpx.Request) -> httpx.Response:
     assert response.all_queries_processed is True
 
 
+def test_query_pdf_info_default_queries(monkeypatch: pytest.MonkeyPatch) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+    input_file = make_pdf_file(str(PdfRestFileID.generate()))
+    expected_serialized = ",".join(ALL_PDF_INFO_QUERIES)
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        if request.method != "POST" or request.url.path != "/pdf-info":
+            msg = f"Unexpected request {request.method} {request.url}"
+            raise AssertionError(msg)
+        payload = json.loads(request.content.decode("utf-8"))
+        assert payload == {
+            "id": str(input_file.id),
+            "queries": expected_serialized,
+        }
+        return httpx.Response(
+            200,
+            json={
+                "inputId": str(input_file.id),
+                "page_count": 1,
+                "tagged": False,
+                "allQueriesProcessed": True,
+            },
+        )
+
+    transport = httpx.MockTransport(handler)
+    with PdfRestClient(api_key=VALID_API_KEY, transport=transport) as client:
+        response = client.query_pdf_info(input_file)
+
+    assert isinstance(response, PdfRestInfoResponse)
+    assert response.page_count == 1
+    assert response.tagged is False
+
+
 def test_query_pdf_info_request_customization(monkeypatch: pytest.MonkeyPatch) -> None:
     monkeypatch.delenv("PDFREST_API_KEY", raising=False)
     resource_id = PdfRestFileID.generate()