chalk-remote-call-python 1.8.0__tar.gz → 1.8.1__tar.gz

{chalk_remote_call_python-1.8.0 → chalk_remote_call_python-1.8.1}/PKG-INFO

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

{chalk_remote_call_python-1.8.0 → chalk_remote_call_python-1.8.1}/chalk-remote-call-rs/chalk-remote-call-server/src/metrics.rs

@@ -1,6 +1,6 @@
 //! Metrics emission for the self-consuming shim, mirroring the catalog
 //! consumer / fnq-server: build a `chalk_metrics` publishing pipeline at startup
-//! and emit `FunctionCallDequeued` per executed call.
+//! and emit per-call dequeue + processing-latency metrics.
 //!
 //! The External Function "Function Calls Dequeued" chart reads TimescaleDB
 //! `metrics1`, which is fed only by the chalk_metrics PubSub bus (statsd goes to
@@ -42,9 +42,7 @@ pub async fn init_pipeline() -> Option<Arc<dyn MetricsPipeline>> {
 
     let publishers: Vec<Arc<dyn BusPublisher>> = build_bus_publisher().await.into_iter().collect();
     if publishers.is_empty() {
-        warn!(
-            "no metrics bus publisher configured; FunctionCallDequeued will not reach the dashboard"
-        );
+        warn!("no metrics bus publisher configured; function call metrics will not reach the dashboard");
     }
 
     let metrics_pipeline: Arc<dyn MetricsPipeline> =
@@ -106,24 +104,37 @@ async fn build_bus_publisher() -> Option<Arc<dyn BusPublisher>> {
     }
 }
 
-/// Emit one `FunctionCallDequeued` count for a completed call, tagged with the
-/// function name and success/failure — mirroring the catalog consumer's emission
-/// (minus the scaling-group revision tag, which the shim does not carry; the
-/// dashboard filters by function name only).
-pub fn record_dequeue(pipeline: &Arc<dyn MetricsPipeline>, function_name: &str, success: bool) {
+/// Emit completed-call metrics tagged with function name and success/failure.
+///
+/// `FunctionCallProcessingLatency` is defined as a sketch in metrics.cue, which
+/// is emitted through the histogram API in chalk_metrics. The shim does not
+/// carry the scaling-group revision tag used by the legacy catalog consumer;
+/// the dashboard filters by function name.
+pub fn record_completed_call(
+    pipeline: &Arc<dyn MetricsPipeline>,
+    function_name: &str,
+    success: bool,
+    elapsed: Duration,
+) {
     let status = if success {
         MetricStatus::Success
     } else {
         MetricStatus::Failure
     };
-    if let Err(e) = pipeline.count(
-        WellKnownMetricName::FunctionCallDequeued,
-        1.0,
-        vec![
-            TagValue::FunctionName(FunctionName(function_name.to_string())),
-            TagValue::Status(status),
-        ],
-    ) {
+    let tags = vec![
+        TagValue::FunctionName(FunctionName(function_name.to_string())),
+        TagValue::Status(status.clone()),
+    ];
+
+    if let Err(e) = pipeline.count(WellKnownMetricName::FunctionCallDequeued, 1.0, tags.clone()) {
         warn!(error = %e, function = %function_name, "failed to emit FunctionCallDequeued metric");
     }
+
+    if let Err(e) = pipeline.histogram(
+        WellKnownMetricName::FunctionCallProcessingLatency,
+        elapsed.as_secs_f64() * 1000.0,
+        tags,
+    ) {
+        warn!(error = %e, function = %function_name, "failed to emit FunctionCallProcessingLatency metric");
+    }
 }

{chalk_remote_call_python-1.8.0 → chalk_remote_call_python-1.8.1}/chalk-remote-call-rs/chalk-remote-call-server/src/self_consumer.rs

@@ -30,7 +30,7 @@
 
 use std::panic::AssertUnwindSafe;
 use std::sync::{Arc, Mutex};
-use std::time::Duration;
+use std::time::{Duration, Instant};
 
 use chalk_queue::semaphore::renew_many;
 use chalk_queue::stream_writer;
@@ -404,6 +404,7 @@ async fn coalesced_execute_and_write(
     let ipc_bytes = work.call.feather_bytes()?;
     let function_name = work.call.name.clone();
     let metadata = work.metadata.otel_headers.clone();
+    let started = Instant::now();
 
     let _ = queue.update_status(call_uuid, "running").await;
     info!(call_id = %work.call_id, function = %function_name, "running coalesced queued call");
@@ -425,11 +426,11 @@ async fn coalesced_execute_and_write(
 
     let response = rx.recv().await;
 
-    // Record the dequeue once the batch has produced this caller's result,
-    // before the stream writes below — parity with the streaming path.
+    // Record metrics once the batch has produced this caller's result, before
+    // stream writes below — parity with the streaming path.
     let success = matches!(response, Some(Ok(_)));
     if let Some(p) = metrics_pipeline {
-        crate::metrics::record_dequeue(p, &function_name, success);
+        crate::metrics::record_completed_call(p, &function_name, success, started.elapsed());
     }
 
     match response {
@@ -487,6 +488,7 @@ async fn execute_and_write(
     let ipc_bytes = work.call.feather_bytes()?;
     let function_name = work.call.name.clone();
     let metadata = work.metadata.otel_headers.clone();
+    let started = Instant::now();
 
     let _ = queue.update_status(call_uuid, "running").await;
     info!(call_id = %work.call_id, function = %function_name, "running queued call");
@@ -501,7 +503,7 @@ async fn execute_and_write(
     let buffer = ChunkBuffer::new().into_shared();
     let handler_fut = handler.call_into_buffer(
         ipc_bytes,
-        function_name,
+        function_name.clone(),
         metadata,
         PEER.to_string(),
         buffer.clone(),
@@ -537,12 +539,11 @@ async fn execute_and_write(
         }
     };
 
-    // Emit FunctionCallDequeued as soon as execution finishes — before the
-    // result-stream writes below — so the dequeue metric is recorded for every
-    // executed call regardless of any downstream Redis write failure. Mirrors
-    // the catalog consumer's per-call emission.
+    // Emit metrics as soon as execution finishes, before result-stream writes,
+    // so every executed call is recorded regardless of downstream Redis write
+    // failure. Mirrors the catalog consumer's per-call emission.
     if let Some(p) = metrics_pipeline {
-        crate::metrics::record_dequeue(p, &work.call.name, result.is_ok());
+        crate::metrics::record_completed_call(p, &function_name, result.is_ok(), started.elapsed());
     }
 
     // Flush whatever the handler produced after the loop exits (on the fast path

{chalk_remote_call_python-1.8.0 → chalk_remote_call_python-1.8.1}/chalk-remote-call-rs/chalk-remote-call-server/src/server.rs

@@ -71,7 +71,7 @@ pub async fn run_server(
     #[cfg(target_os = "linux")]
     if let Some(cfg) = crate::self_consumer::SelfConsumerConfig::from_env() {
         // Build the chalk_metrics pipeline once (mirrors the fnq-server/cc
-        // main init) so the self-consumer can emit FunctionCallDequeued.
+        // main init) so the self-consumer can emit function-call metrics.
         let pipeline = crate::metrics::init_pipeline().await;
         let consumer_handler = python_handler.clone();
         let consumer_coalescing = coalescing_queue.clone();

chalk_remote_call_python-1.8.1/chalk_remote_call/_version.py

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

{chalk_remote_call_python-1.8.0 → chalk_remote_call_python-1.8.1}/chalk_remote_call_python.egg-info/PKG-INFO

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

chalk_remote_call_python-1.8.0/chalk_remote_call/_version.py

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