Small logging fixes

Author: strawgate

key-value/key-value-aio/src/key_value/aio/adapters/pydantic/base.py

@@ -68,7 +68,6 @@ def _validate_model(self, value: dict[str, Any]) -> T | None:
                         extra={
                             "model_type": self._get_model_type_name(),
                             "error": "missing 'items' wrapper",
-                            "value_preview": str(value)[:200],
                         },
                         exc_info=False,
                     )
@@ -91,7 +90,6 @@ def _validate_model(self, value: dict[str, Any]) -> T | None:
                     "model_type": self._get_model_type_name(),
                     "error_count": len(error_details),
                     "errors": error_details,
-                    "value_preview": str(value)[:200],
                 },
                 exc_info=True,
             )

key-value/key-value-aio/tests/adapters/test_pydantic.py

@@ -1,4 +1,5 @@
 from datetime import datetime, timezone
+from logging import LogRecord
 
 import pytest
 from inline_snapshot import snapshot
@@ -47,6 +48,20 @@ class Order(BaseModel):
 TEST_KEY_2: str = "test_key_2"
 
 
+def model_type_from_log_record(record: LogRecord) -> str:
+    if not hasattr(record, "model_type"):
+        msg = "Log record does not have a model_type attribute"
+        raise ValueError(msg)
+    return record.model_type  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
+
+
+def errors_from_log_record(record: LogRecord) -> list[str]:
+    if not hasattr(record, "errors"):
+        msg = "Log record does not have an errors attribute"
+        raise ValueError(msg)
+    return record.errors  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
+
+
 class TestPydanticAdapter:
     @pytest.fixture
     async def store(self) -> MemoryStore:
@@ -166,9 +181,11 @@ async def test_validation_error_logging(
         record = caplog.records[0]
         assert record.levelname == "ERROR"
         assert "Validation failed" in record.message
-        assert record.model_type == "Pydantic model"
-        assert record.error_count == 1
-        assert "is_admin" in str(record.errors)
+        assert model_type_from_log_record(record) == "Pydantic model"
+
+        errors = errors_from_log_record(record)
+        assert len(errors) == 1
+        assert "is_admin" in errors[0]
 
     async def test_list_validation_error_logging(
         self, product_list_adapter: PydanticAdapter[list[Product]], store: MemoryStore, caplog: pytest.LogCaptureFixture
@@ -190,5 +207,7 @@ async def test_list_validation_error_logging(
         record = caplog.records[0]
         assert record.levelname == "ERROR"
         assert "Missing 'items' wrapper" in record.message
-        assert record.model_type == "Pydantic model"
-        assert "missing 'items' wrapper" in record.error
+        assert model_type_from_log_record(record) == "Pydantic model"
+        errors = errors_from_log_record(record)
+        assert len(errors) == 1
+        assert "missing 'items' wrapper" in errors[0]

key-value/key-value-sync/src/key_value/sync/code_gen/adapters/pydantic/base.py

@@ -1,6 +1,7 @@
 # WARNING: this file is auto-generated by 'build_sync_library.py'
 # from the original file 'base.py'
 # DO NOT CHANGE! Change the original file instead.
+import logging
 from abc import ABC, abstractmethod
 from collections.abc import Sequence
 from typing import Any, Generic, SupportsFloat, TypeVar, overload
@@ -12,6 +13,8 @@
 
 from key_value.sync.code_gen.protocols.key_value import KeyValue
 
+logger = logging.getLogger(__name__)
+
 T = TypeVar("T")
 
 
@@ -60,6 +63,14 @@ def _validate_model(self, value: dict[str, Any]) -> T | None:
                     if self._raise_on_validation_error:
                         msg = f"Invalid {self._get_model_type_name()} payload: missing 'items' wrapper"
                         raise DeserializationError(msg)
+
+                    # Log the missing 'items' wrapper when not raising
+                    logger.error(
+                        "Missing 'items' wrapper for list %s",
+                        self._get_model_type_name(),
+                        extra={"model_type": self._get_model_type_name(), "error": "missing 'items' wrapper"},
+                        exc_info=False,
+                    )
                     return None
                 return self._type_adapter.validate_python(value["items"])
 
@@ -69,6 +80,15 @@ def _validate_model(self, value: dict[str, Any]) -> T | None:
                 details = e.errors(include_input=False)
                 msg = f"Invalid {self._get_model_type_name()}: {details}"
                 raise DeserializationError(msg) from e
+
+            # Log the validation error when not raising
+            error_details = e.errors(include_input=False)
+            logger.error(
+                "Validation failed for %s",
+                self._get_model_type_name(),
+                extra={"model_type": self._get_model_type_name(), "error_count": len(error_details), "errors": error_details},
+                exc_info=True,
+            )
             return None
 
     def _serialize_model(self, value: T) -> dict[str, Any]:

key-value/key-value-sync/src/key_value/sync/code_gen/stores/elasticsearch/store.py

@@ -1,6 +1,7 @@
 # WARNING: this file is auto-generated by 'build_sync_library.py'
 # from the original file 'store.py'
 # DO NOT CHANGE! Change the original file instead.
+import logging
 from collections.abc import Sequence
 from datetime import datetime
 from typing import Any, overload
@@ -37,6 +38,8 @@
     msg = "ElasticsearchStore requires py-key-value-aio[elasticsearch]"
     raise ImportError(msg) from e
 
+logger = logging.getLogger(__name__)
+
 DEFAULT_INDEX_PREFIX = "kv_store"
 
 # You might think the `string` field should be a text/keyword field
@@ -249,7 +252,15 @@ def _get_managed_entries(self, *, collection: str, keys: Sequence[str]) -> list[
                 entries_by_id[doc_id] = None
                 continue
 
-            entries_by_id[doc_id] = source_to_managed_entry(source=source)
+            try:
+                entries_by_id[doc_id] = source_to_managed_entry(source=source)
+            except DeserializationError as e:
+                logger.error(
+                    "Failed to deserialize Elasticsearch document in batch operation",
+                    extra={"collection": collection, "document_id": doc_id, "error": str(e)},
+                    exc_info=True,
+                )
+                entries_by_id[doc_id] = None
 
         # Return entries in the same order as input keys
         return [entries_by_id.get(document_id) for document_id in document_ids]

key-value/key-value-sync/tests/code_gen/adapters/test_pydantic.py

@@ -2,6 +2,7 @@
 # from the original file 'test_pydantic.py'
 # DO NOT CHANGE! Change the original file instead.
 from datetime import datetime, timezone
+from logging import LogRecord
 
 import pytest
 from inline_snapshot import snapshot
@@ -50,6 +51,20 @@ class Order(BaseModel):
 TEST_KEY_2: str = "test_key_2"
 
 
+def model_type_from_log_record(record: LogRecord) -> str:
+    if not hasattr(record, "model_type"):
+        msg = "Log record does not have a model_type attribute"
+        raise ValueError(msg)
+    return record.model_type  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
+
+
+def errors_from_log_record(record: LogRecord) -> list[str]:
+    if not hasattr(record, "errors"):
+        msg = "Log record does not have an errors attribute"
+        raise ValueError(msg)
+    return record.errors  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType, reportAttributeAccessIssue]
+
+
 class TestPydanticAdapter:
     @pytest.fixture
     def store(self) -> MemoryStore:
@@ -148,3 +163,54 @@ def test_complex_adapter_with_list(self, product_list_adapter: PydanticAdapter[l
 
         assert product_list_adapter.delete(collection=TEST_COLLECTION, key=TEST_KEY)
         assert product_list_adapter.get(collection=TEST_COLLECTION, key=TEST_KEY) is None
+
+    def test_validation_error_logging(
+        self, user_adapter: PydanticAdapter[User], updated_user_adapter: PydanticAdapter[UpdatedUser], caplog: pytest.LogCaptureFixture
+    ):
+        """Test that validation errors are logged when raise_on_validation_error=False."""
+        import logging
+
+        # Store a User, then try to retrieve as UpdatedUser (missing is_admin field)
+        user_adapter.put(collection=TEST_COLLECTION, key=TEST_KEY, value=SAMPLE_USER)
+
+        with caplog.at_level(logging.ERROR):
+            updated_user = updated_user_adapter.get(collection=TEST_COLLECTION, key=TEST_KEY)
+
+        # Should return None due to validation failure
+        assert updated_user is None
+
+        # Check that an error was logged
+        assert len(caplog.records) == 1
+        record = caplog.records[0]
+        assert record.levelname == "ERROR"
+        assert "Validation failed" in record.message
+        assert model_type_from_log_record(record) == "Pydantic model"
+
+        errors = errors_from_log_record(record)
+        assert len(errors) == 1
+        assert "is_admin" in errors[0]
+
+    def test_list_validation_error_logging(
+        self, product_list_adapter: PydanticAdapter[list[Product]], store: MemoryStore, caplog: pytest.LogCaptureFixture
+    ):
+        """Test that missing 'items' wrapper is logged for list models."""
+        import logging
+
+        # Manually store invalid data (missing 'items' wrapper)
+        store.put(collection=TEST_COLLECTION, key=TEST_KEY, value={"invalid": "data"})
+
+        with caplog.at_level(logging.ERROR):
+            result = product_list_adapter.get(collection=TEST_COLLECTION, key=TEST_KEY)
+
+        # Should return None due to missing 'items' wrapper
+        assert result is None
+
+        # Check that an error was logged
+        assert len(caplog.records) == 1
+        record = caplog.records[0]
+        assert record.levelname == "ERROR"
+        assert "Missing 'items' wrapper" in record.message
+        assert model_type_from_log_record(record) == "Pydantic model"
+        errors = errors_from_log_record(record)
+        assert len(errors) == 1
+        assert "missing 'items' wrapper" in errors[0]