Function build_session_capabilities

core/src/agent_api/capabilities.rs:41–114

(input: SessionCapabilityInput<'_>)

}

pub(super) fn build_session_capabilities(input: SessionCapabilityInput<'_>) -> SessionCapabilities {
    let artifact_limits = input.opts.artifact_store_limits.unwrap_or_default();
    let workspace_services = input
        .opts
        .workspace_services
        .clone()
        .unwrap_or_else(|| crate::workspace::WorkspaceServices::local(input.workspace));
    let tool_executor = Arc::new(
        ToolExecutor::new_with_workspace_services_and_artifact_limits(
            input.workspace.display().to_string(),
            workspace_services,
            artifact_limits,
        ),
    );
    let trace_sink = match input.opts.retention_limits.and_then(|l| l.max_trace_events) {
        Some(cap) => crate::trace::InMemoryTraceSink::with_max_events(cap),
        None => crate::trace::InMemoryTraceSink::new(),
    };
    tool_executor.set_trace_sink(Arc::new(trace_sink.clone()));

    if let Some(ref search_config) = input.code_config.search {
        tool_executor
            .registry()
            .set_search_config(search_config.clone());
    }

    let subagent_tasks = Arc::new(
        match input
            .opts
            .retention_limits
            .and_then(|l| l.max_terminal_subagent_tasks)
        {
            Some(cap) => {
                crate::subagent_task_tracker::InMemorySubagentTaskTracker::with_max_terminal_tasks(
                    cap,
                )
            }
            None => crate::subagent_task_tracker::InMemorySubagentTaskTracker::new(),
        },
    );
    let agent_registry = register_task_capability(
        input.code_config,
        input.opts,
        input.workspace,
        Arc::clone(&input.llm_client),
        &tool_executor,
        Arc::clone(&subagent_tasks),
    );

    // Register generate_object tool (structured JSON output)
    crate::tools::register_generate_object(tool_executor.registry(), Arc::clone(&input.llm_client));

    register_mcp_capabilities(
        &tool_executor,
        input.opts,
        input.global_mcp,
        input.cached_global_mcp_tools,
    );