diff --git a/codex-rs/core/src/session/mcp.rs b/codex-rs/core/src/session/mcp.rs index caf6a27924..4c8b13c9e8 100644 --- a/codex-rs/core/src/session/mcp.rs +++ b/codex-rs/core/src/session/mcp.rs @@ -154,6 +154,7 @@ impl Session { } /// Publishes changed MCP state, waiting for any refresh already in progress. + #[tracing::instrument(name = "mcp.runtime.refresh_if_dirty", skip_all)] pub(crate) async fn refresh_mcp_if_dirty(self: &Arc) { let Ok(_refresh) = self.mcp_refresh.acquire().await else { error!("MCP runtime refresh semaphore closed"); diff --git a/codex-rs/core/src/session/mcp_refresh.rs b/codex-rs/core/src/session/mcp_refresh.rs index 0b21fc480e..19cf2f8b9b 100644 --- a/codex-rs/core/src/session/mcp_refresh.rs +++ b/codex-rs/core/src/session/mcp_refresh.rs @@ -30,6 +30,7 @@ impl McpRefresh { self.pending.swap(false, Ordering::AcqRel) } + #[tracing::instrument(name = "mcp.runtime.refresh_wait", skip_all)] pub(super) async fn acquire(&self) -> Result, AcquireError> { self.gate.acquire().await }