a3s-code-core 3.4.0

A3S Code Core - Embeddable AI agent library with tool execution
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

//! Host-facing HITL control.
//!
//! Tool execution uses the lower-level confirmation runtime; this module owns
//! the public session control surface for listing, resolving, and cancelling
//! pending confirmations.

use super::AgentSession;
use crate::error::{CodeError, Result};
use crate::hitl::PendingConfirmationInfo;

pub(super) struct HitlControl<'a> {
    session: &'a AgentSession,
}

impl<'a> HitlControl<'a> {
    pub(super) fn from_session(session: &'a AgentSession) -> Self {
        Self { session }
    }

    pub(super) async fn pending_confirmations(&self) -> Vec<PendingConfirmationInfo> {
        match &self.session.config.confirmation_manager {
            Some(manager) => manager.pending_confirmations().await,
            None => Vec::new(),
        }
    }

    pub(super) async fn confirm_tool_use(
        &self,
        tool_id: &str,
        approved: bool,
        reason: Option<String>,
    ) -> Result<bool> {
        let Some(manager) = &self.session.config.confirmation_manager else {
            return Ok(false);
        };

        manager
            .confirm(tool_id, approved, reason)
            .await
            .map_err(CodeError::Session)
    }

    pub(super) async fn cancel_confirmations(&self) -> usize {
        match &self.session.config.confirmation_manager {
            Some(manager) => manager.cancel_all().await,
            None => 0,
        }
    }
}