use std::sync::Arc;
use super::SessionTask;
use super::SessionTaskContext;
use crate::product::agent::codex::TurnContext;
use crate::product::agent::state::TaskKind;
use crate::product::protocol::user_input::UserInput;
use async_trait::async_trait;
use tokio_util::sync::CancellationToken;
#[derive(Clone, Copy, Default)]
pub(crate) struct CompactTask;
#[async_trait]
impl SessionTask for CompactTask {
fn kind(&self) -> TaskKind {
TaskKind::Compact
}
async fn run(
self: Arc<Self>,
session: Arc<SessionTaskContext>,
ctx: Arc<TurnContext>,
input: Vec<UserInput>,
_cancellation_token: CancellationToken,
) -> Option<String> {
let session = session.clone_session();
let runtime_capabilities = ctx.runtime.runtime_capabilities();
if crate::product::agent::compact::should_use_remote_compact_task(
session.as_ref(),
&runtime_capabilities,
) {
let _ = session.services.otel_manager.counter(
"codex.task.compact",
1,
&[("type", "remote")],
);
crate::product::agent::compact_remote::run_remote_compact_task(session, ctx).await
} else {
let _ = session.services.otel_manager.counter(
"codex.task.compact",
1,
&[("type", "local")],
);
crate::product::agent::compact::run_compact_task(session, ctx, input).await
}
None
}
}