cognis 0.3.2

Cognis umbrella crate: agent builder, multi-agent orchestration, memory, middleware (rate limit, retry, PII, prompt caching), built-in tools, and re-exports of cognis-core, cognis-graph, cognis-llm, and cognis-rag.
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

//! Per-agent lifecycle hooks.
//!
//! Implementations register via [`AgentBuilder::with_lifecycle`]. The
//! agent invokes them around `run()`. Cheap to ignore — the default
//! impls do nothing.

use async_trait::async_trait;
use uuid::Uuid;

use cognis_core::{CognisError, Message};

use super::AgentResponse;

/// Lifecycle hook surface. All methods have no-op defaults; override only
/// what you need.
#[async_trait]
pub trait AgentLifecycle: Send + Sync {
    /// Fires before the agent's graph runs.
    async fn on_start(&self, run_id: Uuid, input: &Message) {
        let _ = (run_id, input);
    }
    /// Fires when the run completes successfully.
    async fn on_stop(&self, run_id: Uuid, response: &AgentResponse) {
        let _ = (run_id, response);
    }
    /// Fires when the run errors.
    async fn on_error(&self, run_id: Uuid, error: &CognisError) {
        let _ = (run_id, error);
    }
}

/// Convenience: a closure-based lifecycle that observes only `on_start`.
pub struct OnStart<F: Fn(Uuid, &Message) + Send + Sync>(pub F);

#[async_trait]
impl<F: Fn(Uuid, &Message) + Send + Sync> AgentLifecycle for OnStart<F> {
    async fn on_start(&self, run_id: Uuid, input: &Message) {
        (self.0)(run_id, input)
    }
}