From b825f85bf2dd03d5bdad93c0ebb278d2c5edb94f Mon Sep 17 00:00:00 2001 From: Christophe Bornet Date: Thu, 21 Aug 2025 16:07:46 +0200 Subject: [PATCH] fix(standard-tests): fix `BaseStoreAsyncTests.test_set_values_is_idempotent` (#32638) The async version of the test should use the `ayield_keys` method instead of `yield_keys`. Otherwise tools such as `blockbuster` may trigger on a blocking call. --- .../langchain_tests/integration_tests/base_store.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/standard-tests/langchain_tests/integration_tests/base_store.py b/libs/standard-tests/langchain_tests/integration_tests/base_store.py index cba4c245e1d..e7f6812901d 100644 --- a/libs/standard-tests/langchain_tests/integration_tests/base_store.py +++ b/libs/standard-tests/langchain_tests/integration_tests/base_store.py @@ -239,7 +239,7 @@ class BaseStoreAsyncTests(BaseStandardTests, Generic[V]): await kv_store.amset(key_value_pairs) await kv_store.amset(key_value_pairs) assert await kv_store.amget(["foo", "bar"]) == [foo, bar] - assert sorted(kv_store.yield_keys()) == ["bar", "foo"] + assert sorted([key async for key in kv_store.ayield_keys()]) == ["bar", "foo"] async def test_get_can_get_same_value( self, kv_store: BaseStore[str, V], three_values: tuple[V, V, V]