cognis-core 0.3.0

Foundation traits and types for Cognis: typed Runnable<I, O>, Message, errors, streaming, prompts, output parsers, callbacks. Zero external integrations.
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

//! Identity runnable.

use async_trait::async_trait;

use crate::runnable::{Runnable, RunnableConfig};
use crate::Result;

/// Identity runnable: outputs its input unchanged. Generic over `I`,
/// implementing `Runnable<I, I>` for any `I: Send + 'static`.
#[derive(Debug, Default, Clone, Copy)]
pub struct Passthrough;

impl Passthrough {
    /// Construct a `Passthrough`.
    pub fn new() -> Self {
        Self
    }
}

#[async_trait]
impl<I> Runnable<I, I> for Passthrough
where
    I: Send + 'static,
{
    async fn invoke(&self, input: I, _: RunnableConfig) -> Result<I> {
        Ok(input)
    }
    fn name(&self) -> &str {
        "Passthrough"
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn passes_through() {
        let p = Passthrough::new();
        let v: u32 = p.invoke(42, RunnableConfig::default()).await.unwrap();
        assert_eq!(v, 42);
    }
}