nucliadb-utils 2.45.0.post370__py3-none-any.whl → 2.45.0.post372__py3-none-any.whl

nucliadb_utils/asyncio_utils.py

@@ -69,6 +69,12 @@ async def run_concurrently(
     Runs a list of coroutines concurrently, with a maximum number of tasks running.
     Returns the results of the coroutines in the order they were scheduled.
     """
+    if len(tasks) == 0:
+        return []
+
+    if len(tasks) == 1:
+        return [await tasks[0]]
+
     runner = ConcurrentRunner(max_tasks=max_concurrent)
     for task in tasks:
         runner.schedule(task)

nucliadb_utils/const.py

@@ -76,6 +76,7 @@ class Features:
     EXPERIMENTAL_KB = "nucliadb_experimental_kb"
     READ_REPLICA_SEARCHES = "nucliadb_read_replica_searches"
     VERSIONED_PRIVATE_PREDICT = "nucliadb_versioned_private_predict"
+    PREDICT_QUERY_ENDPOINT = "nucliadb_predict_query_endpoint"
     BACK_PRESSURE = "nucliadb_back_pressure"
     REBALANCE_KB = "nucliadb_rebalance_kb"
     CORS_MIDDLEWARE = "nucliadb_cors_middleware_enabled"

nucliadb_utils/featureflagging.py

@@ -53,6 +53,10 @@ DEFAULT_FLAG_DATA: dict[str, Any] = {
         "rollout": 0,
         "variants": {"environment": ["local"]},
     },
+    const.Features.PREDICT_QUERY_ENDPOINT: {
+        "rollout": 0,
+        "variants": {"environment": ["local"]},
+    },
     const.Features.BACK_PRESSURE: {
         "rollout": 0,
         "variants": {"environment": ["local"]},

nucliadb_utils/tests/unit/test_asyncio_utils.py

@@ -30,7 +30,8 @@ async def test_run_concurrently():
         await asyncio.sleep(random.uniform(0.0, 0.5))
         return value
 
-    results = await run_concurrently(mycoro(i) for i in range(10))
+    tasks = [mycoro(i) for i in range(10)]
+    results = await run_concurrently(tasks)
     assert results == list(range(10))
 
 

{nucliadb_utils-2.45.0.post370.dist-info → nucliadb_utils-2.45.0.post372.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb-utils
-Version: 2.45.0.post370
+Version: 2.45.0.post372
 Home-page: https://nuclia.com
 License: BSD
 Classifier: Development Status :: 4 - Beta
@@ -21,8 +21,8 @@ Requires-Dist: nats-py[nkeys] >=2.6.0
 Requires-Dist: pyjwt >=2.4.0
 Requires-Dist: memorylru >=1.1.2
 Requires-Dist: mrflagly
-Requires-Dist: nucliadb-protos >=2.45.0.post370
-Requires-Dist: nucliadb-telemetry >=2.45.0.post370
+Requires-Dist: nucliadb-protos >=2.45.0.post372
+Requires-Dist: nucliadb-telemetry >=2.45.0.post372
 Provides-Extra: cache
 Requires-Dist: redis >=4.3.4 ; extra == 'cache'
 Requires-Dist: orjson >=3.6.7 ; extra == 'cache'

{nucliadb_utils-2.45.0.post370.dist-info → nucliadb_utils-2.45.0.post372.dist-info}/RECORD

@@ -1,10 +1,10 @@
 nucliadb_utils/__init__.py,sha256=EvBCH1iTODe-AgXm48aj4kVUt_Std3PeL8QnwimR5wI,895
-nucliadb_utils/asyncio_utils.py,sha256=i4u0NP32zx0GWB7L5bc5eSWc-zGIxDMP7rQZhoYxRxo,2794
+nucliadb_utils/asyncio_utils.py,sha256=9A0xwlw9E4HYAxiPtLlOzurqiTlCWNqvXrpDl1q7szE,2894
 nucliadb_utils/authentication.py,sha256=HTj6kFFrfdN4-DW_MNqldAPqNPdPwGHnSWl-SigiMvk,6071
-nucliadb_utils/const.py,sha256=yd7We31W7-B7zI-Wt2x5QDTXb78gBNeNXpoyK1IOKb0,2412
+nucliadb_utils/const.py,sha256=ug8iK4GCfHo_7bkYShgDLWvL9WYhWso8fFxIImCEoWQ,2475
 nucliadb_utils/debug.py,sha256=saSfh_CDQoQl-35KCyqef5hdB_OVdrIEnlmWnZU18vg,2470
 nucliadb_utils/exceptions.py,sha256=y_3wk77WLVUtdo-5FtbBsdSkCtK_DsJkdWb5BoPn3qo,1094
-nucliadb_utils/featureflagging.py,sha256=Rxr999pbYCA5QTzyubxUft2MSA4Tlc_gTYd35YDIBkQ,2826
+nucliadb_utils/featureflagging.py,sha256=7g_HcyiUUaEakh5fTe6q7tN7EsbcrZYZjDVntgP_FA0,2948
 nucliadb_utils/grpc.py,sha256=USXwPRuCJiSyLf0JW4isIKZne6od8gM_GGLWaHHjknk,3336
 nucliadb_utils/helpers.py,sha256=fOL6eImdvKO3NV39ymmo2UOCT-GAK1dfXKoMKdzdmFo,1599
 nucliadb_utils/indexing.py,sha256=ELJ9bTnrxcb6tmO77HzzO1SDLPmKXTc_nBiLnJocz5I,3462
@@ -49,7 +49,7 @@ nucliadb_utils/tests/nats.py,sha256=lgRe6YH9LSoI7XgcyKAC2VTSAtuu8EeMve0jWWC_kOY,
 nucliadb_utils/tests/pg.py,sha256=HBpvaNDs9T_L55tvJCJTPnsCDrB8ehI_9HRYz6SPWNE,1819
 nucliadb_utils/tests/s3.py,sha256=YB8QqDaBXxyhHonEHmeBbRRDmvB7sTOaKBSi8KBGokg,2330
 nucliadb_utils/tests/unit/__init__.py,sha256=itSI7dtTwFP55YMX4iK7JzdMHS5CQVUiB1XzQu4UBh8,833
-nucliadb_utils/tests/unit/test_asyncio_utils.py,sha256=Dfs-lEa6ZGp6m-dOmhtB8PGuQQ9CUOfVbX-YvUEfdrc,2339
+nucliadb_utils/tests/unit/test_asyncio_utils.py,sha256=ETAkzQ-QLCEZW9ZtAslzoS0xRFMlN2-bqibUjJm-gRo,2359
 nucliadb_utils/tests/unit/test_authentication.py,sha256=clxj4mmmUnHbUh9dvvtoA2XNruZSxRPEXD68MXXzgzU,5361
 nucliadb_utils/tests/unit/test_helpers.py,sha256=w0Yxo1NteBZi_iSRVZ3vv4J-o3Z7SJIz1q6yyXGzQLg,1574
 nucliadb_utils/tests/unit/test_nats.py,sha256=4lLQOucQGwL9lubTfNFqXhXtua9u4lTlWjF_lFCYORc,4436
@@ -63,8 +63,8 @@ nucliadb_utils/tests/unit/storages/test_aws.py,sha256=GCsB_jwCUNV3Ogt8TZZEmNKAHv
 nucliadb_utils/tests/unit/storages/test_gcs.py,sha256=2XzJwgNpfjVGjtE-QdZhu3ayuT1EMEXINdM-_SatPCY,3554
 nucliadb_utils/tests/unit/storages/test_pg.py,sha256=sJfUttMSzq8W1XYolAUcMxl_R5HcEzb5fpCklPeMJiY,17000
 nucliadb_utils/tests/unit/storages/test_storage.py,sha256=VFpRq6Q6BjnIrBQCumYzR8DQUacwhxt5CzTKSlqqD24,6892
-nucliadb_utils-2.45.0.post370.dist-info/METADATA,sha256=urKc4N32LOXKJ6cq-eXXhKQsQ5Rqd_qivMR7prFVLeo,1978
-nucliadb_utils-2.45.0.post370.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-nucliadb_utils-2.45.0.post370.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-2.45.0.post370.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-2.45.0.post370.dist-info/RECORD,,
+nucliadb_utils-2.45.0.post372.dist-info/METADATA,sha256=X9DUy1vVZDi_BAAvITgb9Cvt8sPuMgjLRPtxfusL38U,1978
+nucliadb_utils-2.45.0.post372.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+nucliadb_utils-2.45.0.post372.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-2.45.0.post372.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-2.45.0.post372.dist-info/RECORD,,