chalk-remote-call-python 1.10.10__tar.gz → 1.10.11__tar.gz

{chalk_remote_call_python-1.10.10/chalk_remote_call_python.egg-info → chalk_remote_call_python-1.10.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalk-remote-call-python
-Version: 1.10.10
+Version: 1.10.11
 Summary: Chalk remote call Python runtime interface client
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

{chalk_remote_call_python-1.10.10 → chalk_remote_call_python-1.10.11}/chalk-remote-call-rs/Cargo.lock

@@ -511,7 +511,7 @@ dependencies = [
 [[package]]
 name = "chalk-queue"
 version = "0.0.0"
-source = "git+https://github.com/chalk-ai/chalk-private.git?rev=1460144ea659debbaee9dccadcd4d7dbf383ccf8#1460144ea659debbaee9dccadcd4d7dbf383ccf8"
+source = "git+https://github.com/chalk-ai/chalk-private.git?rev=6156b025245e28f409bd6c115cfa158ed60203ea#6156b025245e28f409bd6c115cfa158ed60203ea"
 dependencies = [
  "base64 0.22.1",
  "redis",

{chalk_remote_call_python-1.10.10 → chalk_remote_call_python-1.10.11}/chalk-remote-call-rs/chalk-remote-call-server/Cargo.toml

@@ -45,7 +45,7 @@ futures-util = "0.3"
 # `[patch.crates-io]` here it builds against vanilla crates.io redis (it only
 # uses standard APIs). Metrics and StatsD share a revision so their client types agree.
 [target.'cfg(target_os = "linux")'.dependencies]
-chalk-queue = { git = "https://github.com/chalk-ai/chalk-private.git", rev = "1460144ea659debbaee9dccadcd4d7dbf383ccf8" }
+chalk-queue = { git = "https://github.com/chalk-ai/chalk-private.git", rev = "6156b025245e28f409bd6c115cfa158ed60203ea" }
 chalk_metrics = { git = "https://github.com/chalk-ai/chalk-private.git", rev = "a0f8932ec10d3b8340af3e98cf63b800d60b4926" }
 chalk_statsd = { git = "https://github.com/chalk-ai/chalk-private.git", rev = "a0f8932ec10d3b8340af3e98cf63b800d60b4926" }
 

{chalk_remote_call_python-1.10.10 → chalk_remote_call_python-1.10.11}/chalk-remote-call-rs/chalk-remote-call-server/src/self_consumer.rs

@@ -42,7 +42,7 @@ use std::str::FromStr;
 use std::sync::{Arc, Mutex};
 use std::time::{Duration, Instant};
 
-use chalk_queue::semaphore::renew_many;
+use chalk_queue::semaphore::{release_many, renew_many};
 use chalk_queue::stream_writer;
 use chalk_queue::types::{DeliveryAttempt, QueuedCall};
 use chalk_queue::{
@@ -802,7 +802,7 @@ async fn run_stream_dispatch_loop(
     input_queue: BoundFunctionQueue,
     legacy_queue: Option<BoundFunctionQueue>,
     sem: chalk_queue::DistributedSemaphore,
-    conn: ConnectionManager,
+    mut conn: ConnectionManager,
     consumer_name: String,
     live_calls: LiveCalls,
     handler: Arc<PythonHandler>,
@@ -877,23 +877,23 @@ async fn run_stream_dispatch_loop(
             }
         }
 
+        // Return unused reservations in one round-trip before dispatching work.
+        // Transferring the tokens prevents Drop from also releasing each lease.
+        let used = deliveries.len().min(leases.len());
+        let unused_tokens = leases
+            .split_off(used)
+            .into_iter()
+            .map(SemaphoreLease::into_token)
+            .collect::<Vec<_>>();
+        let _ = release_many(&mut conn, sem.key(), &unused_tokens).await;
+
         if deliveries.is_empty() {
-            for lease in leases {
-                let _ = lease.release().await;
-            }
             tokio::time::sleep(backoff).await;
             backoff = (backoff * 2).min(config.poll_backoff_max);
             continue;
         }
 
         backoff = config.poll_backoff_initial;
-        // `dequeue_many` can return fewer records than requested. Only retain
-        // one lease per record and immediately return the unneeded capacity.
-        let used = deliveries.len().min(leases.len());
-        let unused = leases.split_off(used);
-        for lease in unused {
-            let _ = lease.release().await;
-        }
         for (delivery, lease) in deliveries.into_iter().zip(leases) {
             dispatch_delivery(
                 queue.clone(),

chalk_remote_call_python-1.10.11/chalk_remote_call/_version.py

@@ -0,0 +1 @@
+__version__ = "1.10.11"

{chalk_remote_call_python-1.10.10 → chalk_remote_call_python-1.10.11/chalk_remote_call_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalk-remote-call-python
-Version: 1.10.10
+Version: 1.10.11
 Summary: Chalk remote call Python runtime interface client
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

chalk_remote_call_python-1.10.10/chalk_remote_call/_version.py

@@ -1 +0,0 @@
-__version__ = "1.10.10"