clone/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-04-24 14:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

deadlock

Browse Source
This commit is contained in:
Ahmed Ibrahim
2025-08-04 15:12:53 -07:00
parent a362ad00ce
commit 7dec04ae4f
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
codex-rs/mcp-server/src/tool_handlers/stream_conversation.rs
View File

@@ -17,7 +17,7 @@ pub(crate) async fn handle_stream_conversation(
let session_id = conversation_id.0;
// Ensure the session exists and enable streaming
// Ensure the session exists
let conv = get_session(session_id, message_processor.conversation_map()).await;
if conv.is_none() {
@@ -28,11 +28,7 @@ pub(crate) async fn handle_stream_conversation(
return;
}
if let Some(conv) = conv {
conv.lock().await.set_streaming(true).await;
}
// Acknowledge the stream request
// Acknowledge the stream request first so the client can start listening
message_processor
.send_response_with_optional_error(
id,
@@ -42,6 +38,13 @@ pub(crate) async fn handle_stream_conversation(
Some(false),
)
.await;
// Now enable streaming and send initial state
if let Some(conv) = conv {
tokio::spawn(async move {
conv.lock().await.set_streaming(true).await;
});
}
}
/// Handles cancellation for ConversationStream by disabling streaming for the session.