fix: improve codegen string transformations and remove OpenSearch bulk operations

- Add visit_Constant method to codegen script to transform string literals
  - Replace py-key-value-aio with py-key-value-sync in error messages
  - Remove [async] extras from opensearch-py package install instructions
- Remove bulk write and delete operations from OpenSearch store
  - Remove _put_managed_entries method
  - Remove _delete_managed_entries method
  - Remove new_bulk_action import
- Regenerate all sync code with corrected package references

This fixes the codegen check failures where error messages incorrectly
pointed users to install the async package instead of the sync package.

Co-authored-by: William Easton <[email protected]>

Author: github-actions[bot]

key-value/key-value-aio/src/key_value/aio/stores/opensearch/store.py

@@ -1,7 +1,6 @@
 import contextlib
 import logging
 from collections.abc import Sequence
-from datetime import datetime
 from typing import Any, overload
 
 from key_value.shared.errors import DeserializationError, SerializationError
@@ -30,7 +29,7 @@
     BaseEnumerateKeysStore,
     BaseStore,
 )
-from key_value.aio.stores.opensearch.utils import LessCapableJsonSerializer, new_bulk_action
+from key_value.aio.stores.opensearch.utils import LessCapableJsonSerializer
 
 try:
     from opensearchpy import AsyncOpenSearch
@@ -372,39 +371,6 @@ async def _put_managed_entry(
             msg = f"Failed to serialize document: {e}"
             raise SerializationError(message=msg) from e
 
-    @override
-    async def _put_managed_entries(
-        self,
-        *,
-        collection: str,
-        keys: Sequence[str],
-        managed_entries: Sequence[ManagedEntry],
-        ttl: float | None,
-        created_at: datetime,
-        expires_at: datetime | None,
-    ) -> None:
-        if not keys:
-            return
-
-        operations: list[dict[str, Any] | str] = []
-
-        index_name: str = self._get_index_name(collection=collection)
-
-        for key, managed_entry in zip(keys, managed_entries, strict=True):
-            document_id: str = self._get_document_id(key=key)
-
-            index_action: dict[str, Any] = new_bulk_action(action="index", index=index_name, document_id=document_id)
-
-            document: dict[str, Any] = self._serializer.dump_dict(entry=managed_entry, key=key, collection=collection)
-
-            operations.extend([index_action, document])
-
-        try:
-            _ = await self._client.bulk(body=operations, params={"refresh": "true"})  # type: ignore[reportUnknownVariableType]
-        except Exception as e:
-            msg = f"Failed to serialize bulk operations: {e}"
-            raise SerializationError(message=msg) from e
-
     @override
     async def _delete_managed_entry(self, *, key: str, collection: str) -> bool:
         index_name: str = self._get_index_name(collection=collection)
@@ -422,37 +388,6 @@ async def _delete_managed_entry(self, *, key: str, collection: str) -> bool:
 
         return result == "deleted"
 
-    @override
-    async def _delete_managed_entries(self, *, keys: Sequence[str], collection: str) -> int:
-        if not keys:
-            return 0
-
-        operations: list[dict[str, Any]] = []
-
-        for key in keys:
-            index_name, document_id = self._get_destination(collection=collection, key=key)
-
-            delete_action: dict[str, Any] = new_bulk_action(action="delete", index=index_name, document_id=document_id)
-
-            operations.append(delete_action)
-
-        try:
-            opensearch_response = await self._client.bulk(body=operations)
-        except Exception:
-            return 0
-
-        body: dict[str, Any] = get_body_from_response(response=opensearch_response)
-
-        # Count successful deletions
-        deleted_count = 0
-        items = body.get("items", [])
-        for item in items:
-            delete_result = item.get("delete", {})
-            if delete_result.get("result") == "deleted":
-                deleted_count += 1
-
-        return deleted_count
-
     @override
     async def _get_collection_keys(self, *, collection: str, limit: int | None = None) -> list[str]:
         """Get up to 10,000 keys in the specified collection (eventually consistent)."""

key-value/key-value-sync/src/key_value/sync/code_gen/stores/disk/multi_store.py

@@ -16,7 +16,7 @@
     from diskcache import Cache
     from pathvalidate import sanitize_filename
 except ImportError as e:
-    msg = "DiskStore requires py-key-value-aio[disk]"
+    msg = "DiskStore requires py-key-value-sync[disk]"
     raise ImportError(msg) from e
 
 CacheFactory = Callable[[str], Cache]

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

@@ -14,7 +14,7 @@
 try:
     from diskcache import Cache
 except ImportError as e:
-    msg = "DiskStore requires py-key-value-aio[disk]"
+    msg = "DiskStore requires py-key-value-sync[disk]"
     raise ImportError(msg) from e
 
 

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

@@ -38,7 +38,7 @@
         get_source_from_body,
     )
 except ImportError as e:
-    msg = "ElasticsearchStore requires py-key-value-aio[elasticsearch]"
+    msg = "ElasticsearchStore requires py-key-value-sync[elasticsearch]"
     raise ImportError(msg) from e
 
 logger = logging.getLogger(__name__)

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

@@ -18,7 +18,7 @@
     import keyring
     from keyring.errors import PasswordDeleteError
 except ImportError as e:
-    msg = "KeyringStore requires py-key-value-aio[keyring]"
+    msg = "KeyringStore requires py-key-value-sync[keyring]"
     raise ImportError(msg) from e
 
 DEFAULT_KEYCHAIN_SERVICE = "py-key-value"

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

@@ -21,7 +21,7 @@
 try:
     from cachetools import TLRUCache
 except ImportError as e:
-    msg = "MemoryStore requires py-key-value-aio[memory]"
+    msg = "MemoryStore requires py-key-value-sync[memory]"
     raise ImportError(msg) from e
 
 

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

@@ -21,7 +21,7 @@
     from pymongo.database import Database
     from pymongo.results import DeleteResult  # noqa: TC002
 except ImportError as e:
-    msg = "MongoDBStore requires py-key-value-aio[mongodb]"
+    msg = "MongoDBStore requires py-key-value-sync[mongodb]"
     raise ImportError(msg) from e
 
 DEFAULT_DB = "kv-store-adapter"

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

@@ -4,7 +4,6 @@
 import contextlib
 import logging
 from collections.abc import Sequence
-from datetime import datetime
 from typing import Any, overload
 
 from key_value.shared.errors import DeserializationError, SerializationError
@@ -23,7 +22,7 @@
     BaseEnumerateKeysStore,
     BaseStore,
 )
-from key_value.sync.code_gen.stores.opensearch.utils import LessCapableJsonSerializer, new_bulk_action
+from key_value.sync.code_gen.stores.opensearch.utils import LessCapableJsonSerializer
 
 try:
     from opensearchpy import OpenSearch
@@ -326,39 +325,6 @@ def _put_managed_entry(self, *, key: str, collection: str, managed_entry: Manage
             msg = f"Failed to serialize document: {e}"
             raise SerializationError(message=msg) from e
 
-    @override
-    def _put_managed_entries(
-        self,
-        *,
-        collection: str,
-        keys: Sequence[str],
-        managed_entries: Sequence[ManagedEntry],
-        ttl: float | None,
-        created_at: datetime,
-        expires_at: datetime | None,
-    ) -> None:
-        if not keys:
-            return
-
-        operations: list[dict[str, Any] | str] = []
-
-        index_name: str = self._get_index_name(collection=collection)
-
-        for key, managed_entry in zip(keys, managed_entries, strict=True):
-            document_id: str = self._get_document_id(key=key)
-
-            index_action: dict[str, Any] = new_bulk_action(action="index", index=index_name, document_id=document_id)
-
-            document: dict[str, Any] = self._serializer.dump_dict(entry=managed_entry, key=key, collection=collection)
-
-            operations.extend([index_action, document])
-
-        try:
-            _ = self._client.bulk(body=operations, params={"refresh": "true"})  # type: ignore[reportUnknownVariableType]
-        except Exception as e:
-            msg = f"Failed to serialize bulk operations: {e}"
-            raise SerializationError(message=msg) from e
-
     @override
     def _delete_managed_entry(self, *, key: str, collection: str) -> bool:
         index_name: str = self._get_index_name(collection=collection)
@@ -376,37 +342,6 @@ def _delete_managed_entry(self, *, key: str, collection: str) -> bool:
 
         return result == "deleted"
 
-    @override
-    def _delete_managed_entries(self, *, keys: Sequence[str], collection: str) -> int:
-        if not keys:
-            return 0
-
-        operations: list[dict[str, Any]] = []
-
-        for key in keys:
-            (index_name, document_id) = self._get_destination(collection=collection, key=key)
-
-            delete_action: dict[str, Any] = new_bulk_action(action="delete", index=index_name, document_id=document_id)
-
-            operations.append(delete_action)
-
-        try:
-            opensearch_response = self._client.bulk(body=operations)
-        except Exception:
-            return 0
-
-        body: dict[str, Any] = get_body_from_response(response=opensearch_response)
-
-        # Count successful deletions
-        deleted_count = 0
-        items = body.get("items", [])
-        for item in items:
-            delete_result = item.get("delete", {})
-            if delete_result.get("result") == "deleted":
-                deleted_count += 1
-
-        return deleted_count
-
     @override
     def _get_collection_keys(self, *, collection: str, limit: int | None = None) -> list[str]:
         """Get up to 10,000 keys in the specified collection (eventually consistent)."""

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

@@ -18,7 +18,7 @@
 try:
     from redis import Redis
 except ImportError as e:
-    msg = "RedisStore requires py-key-value-aio[redis]"
+    msg = "RedisStore requires py-key-value-sync[redis]"
     raise ImportError(msg) from e
 
 DEFAULT_PAGE_SIZE = 10000

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

@@ -16,7 +16,7 @@
 try:
     from rocksdict import Options, Rdict, WriteBatch
 except ImportError as e:
-    msg = "RocksDBStore requires py-key-value-aio[rocksdb]"
+    msg = "RocksDBStore requires py-key-value-sync[rocksdb]"
     raise ImportError(msg) from e