| use codex_app_server_protocol::DynamicToolCallOutputContentItem; |
| use codex_app_server_protocol::ServerNotification; |
| use codex_app_server_protocol::ThreadItem; |
| use codex_app_server_protocol::UserInput; |
| use codex_protocol::models::ContentItem; |
| use codex_protocol::models::FunctionCallOutputBody; |
| use codex_protocol::models::FunctionCallOutputContentItem; |
| use codex_protocol::models::ResponseItem; |
|
|
| pub(crate) fn without_notification_media(notification: ServerNotification) -> ServerNotification { |
| match notification { |
| ServerNotification::ItemStarted(mut notification) => { |
| notification.item = without_thread_item_media(notification.item); |
| ServerNotification::ItemStarted(notification) |
| } |
| ServerNotification::ItemCompleted(mut notification) => { |
| notification.item = without_thread_item_media(notification.item); |
| ServerNotification::ItemCompleted(notification) |
| } |
| ServerNotification::RawResponseItemCompleted(mut notification) => { |
| match &mut notification.item { |
| ResponseItem::Message { content, .. } => { |
| content.retain(|item| match item { |
| ContentItem::InputImage { .. } | ContentItem::InputAudio { .. } => false, |
| ContentItem::InputText { .. } | ContentItem::OutputText { .. } => true, |
| }); |
| } |
| ResponseItem::FunctionCallOutput { output, .. } |
| | ResponseItem::CustomToolCallOutput { output, .. } => { |
| if let Some(items) = output.content_items_mut() { |
| items.retain(|item| match item { |
| FunctionCallOutputContentItem::InputImage { .. } |
| | FunctionCallOutputContentItem::InputAudio { .. } => false, |
| FunctionCallOutputContentItem::InputText { .. } |
| | FunctionCallOutputContentItem::EncryptedContent { .. } => true, |
| }); |
| } |
| } |
| ResponseItem::ImageGenerationCall { result, .. } => result.clear(), |
| ResponseItem::AdditionalTools { .. } |
| | ResponseItem::AgentMessage { .. } |
| | ResponseItem::Reasoning { .. } |
| | ResponseItem::LocalShellCall { .. } |
| | ResponseItem::FunctionCall { .. } |
| | ResponseItem::ToolSearchCall { .. } |
| | ResponseItem::CustomToolCall { .. } |
| | ResponseItem::ToolSearchOutput { .. } |
| | ResponseItem::WebSearchCall { .. } |
| | ResponseItem::Compaction { .. } |
| | ResponseItem::ConfigurationUpdate { .. } |
| | ResponseItem::CompactionTrigger { .. } |
| | ResponseItem::ContextCompaction { .. } |
| | ResponseItem::Other => {} |
| } |
| ServerNotification::RawResponseItemCompleted(notification) |
| } |
| ServerNotification::Error(_) |
| | ServerNotification::ThreadStarted(_) |
| | ServerNotification::ThreadStatusChanged(_) |
| | ServerNotification::ThreadArchived(_) |
| | ServerNotification::ThreadDeleted(_) |
| | ServerNotification::ThreadUnarchived(_) |
| | ServerNotification::ThreadClosed(_) |
| | ServerNotification::ThreadReverted(_) |
| | ServerNotification::SkillsChanged(_) |
| | ServerNotification::ThreadNameUpdated(_) |
| | ServerNotification::ThreadAttachmentUpdated(_) |
| | ServerNotification::ThreadGoalUpdated(_) |
| | ServerNotification::ThreadGoalCleared(_) |
| | ServerNotification::ThreadQueueChanged(_) |
| | ServerNotification::ProjectChanged(_) |
| | ServerNotification::ThreadProjectUpdated(_) |
| | ServerNotification::EnvironmentConnected(_) |
| | ServerNotification::EnvironmentDisconnected(_) |
| | ServerNotification::ThreadSettingsUpdated(_) |
| | ServerNotification::ThreadTokenUsageUpdated(_) |
| | ServerNotification::TurnStarted(_) |
| | ServerNotification::HookStarted(_) |
| | ServerNotification::TurnCompleted(_) |
| | ServerNotification::HookCompleted(_) |
| | ServerNotification::TurnDiffUpdated(_) |
| | ServerNotification::TurnPlanUpdated(_) |
| | ServerNotification::ItemGuardianApprovalReviewStarted(_) |
| | ServerNotification::ItemGuardianApprovalReviewCompleted(_) |
| | ServerNotification::StrictReviewRequired(_) |
| | ServerNotification::RawResponseCompleted(_) |
| | ServerNotification::AgentMessageDelta(_) |
| | ServerNotification::PlanDelta(_) |
| | ServerNotification::CommandExecOutputDelta(_) |
| | ServerNotification::ProcessOutputDelta(_) |
| | ServerNotification::ProcessExited(_) |
| | ServerNotification::CommandExecutionOutputDelta(_) |
| | ServerNotification::TerminalInteraction(_) |
| | ServerNotification::FileChangeOutputDelta(_) |
| | ServerNotification::FileChangePatchUpdated(_) |
| | ServerNotification::ServerRequestResolved(_) |
| | ServerNotification::McpToolCallProgress(_) |
| | ServerNotification::McpServerOauthLoginCompleted(_) |
| | ServerNotification::McpServerStatusUpdated(_) |
| | ServerNotification::McpServerEventStream(_) |
| | ServerNotification::AccountUpdated(_) |
| | ServerNotification::AccountRateLimitsUpdated(_) |
| | ServerNotification::AppListUpdated(_) |
| | ServerNotification::RemoteControlStatusChanged(_) |
| | ServerNotification::ExternalAgentConfigImportProgress(_) |
| | ServerNotification::ExternalAgentConfigImportCompleted(_) |
| | ServerNotification::FsChanged(_) |
| | ServerNotification::ReasoningSummaryTextDelta(_) |
| | ServerNotification::ReasoningSummaryPartAdded(_) |
| | ServerNotification::ReasoningTextDelta(_) |
| | ServerNotification::ContextCompacted(_) |
| | ServerNotification::ModelRerouted(_) |
| | ServerNotification::ModelVerification(_) |
| | ServerNotification::AuthRecoveryStarted(_) |
| | ServerNotification::AuthRecoveryCompleted(_) |
| | ServerNotification::TurnModerationMetadata(_) |
| | ServerNotification::ModelSafetyBufferingUpdated(_) |
| | ServerNotification::Warning(_) |
| | ServerNotification::GuardianWarning(_) |
| | ServerNotification::DeprecationNotice(_) |
| | ServerNotification::ConfigWarning(_) |
| | ServerNotification::FuzzyFileSearchSessionUpdated(_) |
| | ServerNotification::FuzzyFileSearchSessionCompleted(_) |
| | ServerNotification::ThreadRealtimeStarted(_) |
| | ServerNotification::ThreadRealtimeItemAdded(_) |
| | ServerNotification::ThreadRealtimeItemStarted(_) |
| | ServerNotification::ThreadRealtimeItemTranscriptDelta(_) |
| | ServerNotification::ThreadRealtimeItemCompleted(_) |
| | ServerNotification::ThreadRealtimeTranscriptDelta(_) |
| | ServerNotification::ThreadRealtimeTranscriptDone(_) |
| | ServerNotification::ThreadRealtimeOutputAudioDelta(_) |
| | ServerNotification::ThreadRealtimeSdp(_) |
| | ServerNotification::ThreadRealtimeError(_) |
| | ServerNotification::ThreadRealtimeClosed(_) |
| | ServerNotification::WindowsWorldWritableWarning(_) |
| | ServerNotification::WindowsSandboxSetupCompleted(_) |
| | ServerNotification::AccountLoginCompleted(_) => notification, |
| } |
| } |
|
|
| fn without_thread_item_media(mut item: ThreadItem) -> ThreadItem { |
| match &mut item { |
| ThreadItem::UserMessage { content, .. } => { |
| content.retain(|item| match item { |
| UserInput::Image { .. } | UserInput::Audio { .. } => false, |
| UserInput::Text { .. } |
| | UserInput::LocalImage { .. } |
| | UserInput::LocalAudio { .. } |
| | UserInput::Skill { .. } |
| | UserInput::Mention { .. } => true, |
| }); |
| } |
| ThreadItem::DynamicToolCall { |
| content_items: Some(content_items), |
| .. |
| } => { |
| content_items.retain(|item| match item { |
| DynamicToolCallOutputContentItem::InputImage { .. } |
| | DynamicToolCallOutputContentItem::InputAudio { .. } => false, |
| DynamicToolCallOutputContentItem::InputText { .. } => true, |
| }); |
| } |
| ThreadItem::FunctionCallOutput { |
| output: FunctionCallOutputBody::ContentItems(items), |
| .. |
| } => { |
| items.retain(|item| match item { |
| FunctionCallOutputContentItem::InputImage { .. } |
| | FunctionCallOutputContentItem::InputAudio { .. } => false, |
| FunctionCallOutputContentItem::InputText { .. } |
| | FunctionCallOutputContentItem::EncryptedContent { .. } => true, |
| }); |
| } |
| ThreadItem::McpToolCall { |
| result: Some(result), |
| .. |
| } => { |
| |
| |
| result.content.retain(|item| { |
| !matches!( |
| item.get("type").and_then(serde_json::Value::as_str), |
| Some("image" | "audio") |
| ) && item |
| .get("resource") |
| .and_then(|resource| resource.get("blob")) |
| .is_none() |
| }); |
| } |
| ThreadItem::ImageGeneration(item) => item.result.clear(), |
| ThreadItem::HookPrompt { .. } |
| | ThreadItem::AgentMessage { .. } |
| | ThreadItem::Plan { .. } |
| | ThreadItem::Reasoning { .. } |
| | ThreadItem::FunctionCallOutput { |
| output: FunctionCallOutputBody::Text(_), |
| .. |
| } |
| | ThreadItem::CommandExecution { .. } |
| | ThreadItem::FileChange { .. } |
| | ThreadItem::McpToolCall { result: None, .. } |
| | ThreadItem::DynamicToolCall { |
| content_items: None, |
| .. |
| } |
| | ThreadItem::CollabAgentToolCall { .. } |
| | ThreadItem::SubAgentActivity { .. } |
| | ThreadItem::WebSearch(_) |
| | ThreadItem::ImageView { .. } |
| | ThreadItem::Sleep(_) |
| | ThreadItem::EnteredReviewMode { .. } |
| | ThreadItem::ExitedReviewMode { .. } |
| | ThreadItem::ContextCompaction { .. } => {} |
| } |
| item |
| } |
|
|
