Fix opensearch tests

Author: strawgate

key-value/key-value-aio/tests/stores/opensearch/test_opensearch.py

@@ -197,7 +197,7 @@ async def test_value_stored_as_f_object(self, store: OpenSearchStore, opensearch
         doc_id = store._get_document_id(key="test_key")  # pyright: ignore[reportPrivateUsage]
 
         response = await opensearch_client.get(index=index_name, id=doc_id)
-        assert response.body["_source"] == snapshot(
+        assert response["_source"] == snapshot(
             {
                 "value": {"flat": {"name": "Alice", "age": 30}},
                 "created_at": IsStr(min_length=20, max_length=40),
@@ -207,7 +207,7 @@ async def test_value_stored_as_f_object(self, store: OpenSearchStore, opensearch
         # Test with TTL
         await store.put(collection="test", key="test_key", value={"name": "Bob", "age": 25}, ttl=10)
         response = await opensearch_client.get(index=index_name, id=doc_id)
-        assert response.body["_source"] == snapshot(
+        assert response["_source"] == snapshot(
             {
                 "value": {"flat": {"name": "Bob", "age": 25}},
                 "created_at": IsStr(min_length=20, max_length=40),

key-value/key-value-sync/tests/code_gen/stores/opensearch/test_opensearch.py

@@ -187,14 +187,14 @@ def test_value_stored_as_f_object(self, store: OpenSearchStore, opensearch_clien
         doc_id = store._get_document_id(key="test_key")  # pyright: ignore[reportPrivateUsage]
 
         response = opensearch_client.get(index=index_name, id=doc_id)
-        assert response.body["_source"] == snapshot(
+        assert response["_source"] == snapshot(
             {"value": {"flat": {"name": "Alice", "age": 30}}, "created_at": IsStr(min_length=20, max_length=40)}
         )
 
         # Test with TTL
         store.put(collection="test", key="test_key", value={"name": "Bob", "age": 25}, ttl=10)
         response = opensearch_client.get(index=index_name, id=doc_id)
-        assert response.body["_source"] == snapshot(
+        assert response["_source"] == snapshot(
             {
                 "value": {"flat": {"name": "Bob", "age": 25}},
                 "created_at": IsStr(min_length=20, max_length=40),