athena-python-docx 0.21.6__tar.gz → 0.21.7__tar.gz

{athena_python_docx-0.21.6 → athena_python_docx-0.21.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: athena-python-docx
-Version: 0.21.6
+Version: 0.21.7
 Summary: Drop-in replacement for python-docx that connects to Athena's DOCX Studio collaborative document stack
 Project-URL: Homepage, https://athenaintelligence.ai
 Author-email: Athena Intelligence <engineering@athenaintelligence.ai>

{athena_python_docx-0.21.6 → athena_python_docx-0.21.7}/docx/__init__.py

@@ -6,7 +6,7 @@ See CLAUDE.md for the API parity contract.
 
 from __future__ import annotations
 
-__version__ = "0.21.6"
+__version__ = "0.21.7"
 
 from docx.api import Document
 from docx._buffer import discard_all, flush_all

{athena_python_docx-0.21.6 → athena_python_docx-0.21.7}/docx/document.py

@@ -1222,7 +1222,23 @@ class Document:
             ),
         )
         if isinstance(result, dict):
-            return result
+            # Tolerate an older docx-studio deployment that returned only
+            # ``{"matches": [...]}``. The SDK's public contract promises
+            # ``count`` and ``truncated`` too; normalizing here prevents a
+            # model script from crashing with ``KeyError: 'count'`` during a
+            # rolling deploy. The server now supplies the true total count,
+            # while this fallback can conservatively report the returned size.
+            matches_obj: object = result.get("matches", [])
+            matches: list[object] = matches_obj if isinstance(matches_obj, list) else []
+            count_obj: object = result.get("count", len(matches))
+            count: int = int(count_obj) if isinstance(count_obj, (int, float)) else len(matches)
+            truncated_obj: object = result.get("truncated", count > len(matches))
+            return {
+                **result,
+                "count": count,
+                "matches": matches,
+                "truncated": bool(truncated_obj),
+            }
         return {"count": 0, "matches": [], "truncated": False}
 
     @athena_extension(

{athena_python_docx-0.21.6 → athena_python_docx-0.21.7}/docx/table.py

@@ -1477,8 +1477,9 @@ class _Cell:
 
     def _cell_info_at(self, row: int, col: int) -> dict | None:
         """Raw ``tables.getCells`` for (row, col). Returns ``None`` when
-        DOCX Studio reports no cell at that grid position — either an
-        out-of-bounds query or a secondary slot covered by a colspan."""
+        DOCX Studio reports no cell at that grid position. Current servers
+        resolve a secondary slot covered by a colspan to its master cell;
+        older servers returned ``None`` for that case."""
         nid: str = self._table._fresh_node_id()
         result: object = run_sync(
             self._table._session.doc.tables.get_cells(
@@ -1500,8 +1501,8 @@ class _Cell:
     def _cell_info(self) -> dict | None:
         """Return the cell info backing ``(self._row, self._col)``.
 
-        When ``self`` is the secondary slot of a horizontal merge, real
-        DOCX Studio returns an empty cell list (only the master is
+        When ``self`` is the secondary slot of a horizontal merge, older
+        DOCX Studio deployments return an empty cell list (only the master is
         addressable by its top-left position). Mirror python-docx's
         ``_Cell``-on-merge semantics by walking back along the row to
         find the first cell whose ``colspan`` covers our grid column,

{athena_python_docx-0.21.6 → athena_python_docx-0.21.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "athena-python-docx"
-version = "0.21.6"
+version = "0.21.7"
 description = "Drop-in replacement for python-docx that connects to Athena's DOCX Studio collaborative document stack"
 readme = "README.md"
 license = "MIT"
@@ -56,3 +56,9 @@ markers = [
 [tool.ruff]
 line-length = 100
 target-version = "py311"
+
+[tool.ruff.lint]
+# Keep the SDK's historical Ruff contract explicit. Ruff 0.16 expanded its
+# implicit defaults, which otherwise turns an unpinned CI upgrade into hundreds
+# of unrelated findings. Broader rule families should be adopted deliberately.
+select = ["E4", "E7", "E9", "F"]

{athena_python_docx-0.21.6 → athena_python_docx-0.21.7}/tests/test_find_replace_session_open.py

@@ -231,3 +231,22 @@ def test_find_text_returns_server_context(fake_sd_methods: Any) -> None:
     assert result["truncated"] is True
     assert session.open.await_count == 1
     assert session.send_command.await_count == 1
+
+
+def test_find_text_normalizes_legacy_matches_only_response(
+    fake_sd_methods: Any,
+) -> None:
+    """Rolling deploys must not make ``result['count']`` raise KeyError."""
+    session = _make_unopened_session(fake_sd_methods)
+    session.send_command.return_value = {
+        "matches": [{"text": "Buyer"}, {"text": "Buyer"}],
+    }
+    doc = _make_unopened_document(session)
+
+    result = doc.find_text("Buyer")
+
+    assert result == {
+        "count": 2,
+        "matches": [{"text": "Buyer"}, {"text": "Buyer"}],
+        "truncated": False,
+    }