use super::ToolRegistry;
impl ToolRegistry {
pub fn metrics_collector(&self) -> std::sync::Arc<crate::metrics::MetricsCollector> {
self.metrics.clone()
}
pub fn tool_call_count(&self) -> u64 {
self.tool_call_counter
.load(std::sync::atomic::Ordering::Relaxed)
}
pub fn pty_poll_count(&self) -> u64 {
self.pty_poll_counter
.load(std::sync::atomic::Ordering::Relaxed)
}
#[allow(dead_code)]
pub(crate) fn increment_tool_calls(&self) {
self.tool_call_counter
.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
}
#[allow(dead_code)]
pub(crate) fn increment_pty_polls(&self) {
self.pty_poll_counter
.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
}
}