halo-agent 0.1.0

Halo Agent and Workflow execution kernel
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

#![allow(async_fn_in_trait)]

//! Short-term conversation history contracts.
//!
//! The two records have independent responsibilities and live in their
//! nearest module. Each module exposes a small `Store` contract with the same
//! CRUD shape; there is no cross-record storage trait. Chat.state carries the
//! complete terminal Agent/Workflow state, so there is no third process record.

pub type Error = crate::Error;

/// Top-level owner of a short-term conversation record.
///
/// This is intentionally separate from `memory::Type`, whose values describe
/// long-term memory records (`summary`, `fact`, and `preference`).
#[derive(Clone, Debug, serde::Deserialize, Eq, PartialEq, serde::Serialize)]
#[serde(rename_all = "lowercase")]
pub enum Type {
    Agent,
    Workflow,
}

pub mod chat;
pub mod session;

/// Unified short-term memory contract.
///
/// Session and Chat remain separate records internally, but callers configure
/// one short-memory store. A backend decides how those two records are stored
/// and exposes them only through this composite contract.
pub trait Store: Send + Sync {
    type Session: session::Store;
    type Chat: chat::Store;

    fn session(&self) -> &Self::Session;
    fn chat(&self) -> &Self::Chat;
}

/// Small adapter for composing independent Session and Chat implementations.
/// Backends may use this adapter, but application code should normally use a
/// backend's own `short::...::new` constructor.
#[derive(Clone)]
pub struct Pair<S, C> {
    session: std::sync::Arc<S>,
    chat: std::sync::Arc<C>,
}

impl<S, C> Pair<S, C> {
    pub fn new(session: S, chat: C) -> Self {
        Self {
            session: std::sync::Arc::new(session),
            chat: std::sync::Arc::new(chat),
        }
    }
}

impl<S, C> Store for Pair<S, C>
where
    S: session::Store,
    C: chat::Store,
{
    type Session = S;
    type Chat = C;

    fn session(&self) -> &Self::Session {
        self.session.as_ref()
    }

    fn chat(&self) -> &Self::Chat {
        self.chat.as_ref()
    }
}