add option to select cache keys

expectedparrot · Jan 26, 2025 · b23f733 · b23f733
1 parent 54a7f5e
commit b23f733
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/edsl/coop/coop.py b/edsl/coop/coop.py
@@ -549,6 +549,7 @@ def remote_cache_create_many(
     def remote_cache_get(
         self,
         exclude_keys: Optional[list[str]] = None,
+        select_keys: Optional[list[str]] = None,
     ) -> list[CacheEntry]:
         """
         Get all remote cache entries.
@@ -560,10 +561,12 @@ def remote_cache_get(
         """
         if exclude_keys is None:
             exclude_keys = []
+        if select_keys is None:
+            select_keys = []
         response = self._send_server_request(
             uri="api/v0/remote-cache/get-many",
             method="POST",
-            payload={"keys": exclude_keys},
+            payload={"keys": exclude_keys, "selected_keys": select_keys},
             timeout=40,
         )
         self._resolve_server_response(response)