Fix docstrings

Author: b-long

src/otdf_python/kas_client.py

@@ -692,5 +692,5 @@ def _unwrap_with_connect_rpc(self, key_access, signed_token, policy_json):  # no
             raise SDKException(f"Connect RPC rewrap failed: {e}")
 
     def get_key_cache(self) -> KASKeyCache:
-        """Returns the key cache"""
+        """Returns the KAS key cache used for storing and retrieving encryption keys."""
         return self.cache

src/otdf_python/kas_key_cache.py

@@ -38,7 +38,7 @@ def store(self, kas_info) -> None:
             self._cache[cache_key] = kas_info
 
     def set(self, key, value):
-        """Backwards compatibility method"""
+        """Store a key-value pair in the cache."""
         with self._lock:
             self._cache[key] = value
 

src/otdf_python/nanotdf.py

@@ -51,7 +51,7 @@ def _create_policy_object(self, attributes: list[str]) -> PolicyObject:
         return PolicyObject(uuid=policy_uuid, body=body)
 
     def _serialize_policy_object(self, obj):
-        """Custom serializer for policy objects to match otdfctl format."""
+        """Custom NanoTDF serializer to convert to compatible JSON format."""
         from otdf_python.policy_object import PolicyBody, AttributeObject
 
         if isinstance(obj, PolicyBody):
@@ -123,7 +123,7 @@ def _prepare_policy_data(self, config: NanoTDFConfig) -> tuple[bytes, str]:
         return policy_body, policy_type
 
     def _prepare_encryption_key(self, config: NanoTDFConfig) -> bytes:
-        """Get encryption key from config or generate a new one."""
+        """Get encryption key from config if provided as hex string, otherwise generate a new random key."""
         key = None
         if (
             config.cipher
@@ -438,7 +438,7 @@ def _handle_legacy_key_config(
         return key, config
 
     def create_nanotdf(self, data: bytes, config: dict | NanoTDFConfig) -> bytes:
-        """Legacy method for backwards compatibility with existing tests"""
+        """Create a NanoTDF from input data using the provided configuration."""
         if len(data) > self.K_MAX_TDF_SIZE:
             raise NanoTDFMaxSizeLimit("exceeds max size for nano tdf")
 
@@ -512,7 +512,7 @@ def _extract_key_for_reading(
     def read_nanotdf(
         self, nanotdf_bytes: bytes, config: dict | NanoTDFConfig | None = None
     ) -> bytes:
-        """Legacy method for backwards compatibility with existing tests"""
+        """Read and decrypt a NanoTDF, returning the original plaintext data."""
         output = BytesIO()
         from otdf_python.header import Header  # Local import to avoid circular import
 

src/otdf_python/sdk_exceptions.py

@@ -9,8 +9,8 @@ def __init__(self, message, cause=None):
         super().__init__(message, cause)
 
 
-class KasBadRequestException(SDKException):
-    """Thrown when the KAS returns a bad request response."""
+class KASBadRequestException(SDKException):
+    """Thrown when the KAS returns a bad request response or other client request errors."""
 
     def __init__(self, message):
         super().__init__(message)

src/otdf_python/tdf.py

@@ -150,7 +150,7 @@ def _build_policy_json(self, config: TDFConfig) -> str:
             return _json.dumps(policy, default=self._serialize_policy_object)
 
     def _serialize_policy_object(self, obj):
-        """Custom serializer for policy objects to match otdfctl format."""
+        """Custom TDF serializer to convert to compatible JSON format."""
         from otdf_python.policy_object import PolicyBody, AttributeObject
 
         if isinstance(obj, PolicyBody):