cognis-core 0.2.0

Core traits and types for the Cognis LLM framework
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
50

use std::sync::Arc;

use async_trait::async_trait;
use serde_json::Value;

use crate::error::{CognisError, Result};

use super::base::Runnable;
use super::config::RunnableConfig;

/// Maps a runnable over each element of an input array.
///
/// Input must be `Value::Array`. Returns `Value::Array` of results.
pub struct RunnableEach {
    bound: Arc<dyn Runnable>,
}

impl RunnableEach {
    pub fn new(bound: Arc<dyn Runnable>) -> Self {
        Self { bound }
    }
}

#[async_trait]
impl Runnable for RunnableEach {
    fn name(&self) -> &str {
        "RunnableEach"
    }

    async fn invoke(&self, input: Value, config: Option<&RunnableConfig>) -> Result<Value> {
        let arr = input.as_array().ok_or_else(|| CognisError::TypeMismatch {
            expected: "Array".into(),
            got: match &input {
                Value::Null => "Null",
                Value::Bool(_) => "Bool",
                Value::Number(_) => "Number",
                Value::String(_) => "String",
                Value::Object(_) => "Object",
                Value::Array(_) => unreachable!(),
            }
            .into(),
        })?;

        let mut results = Vec::with_capacity(arr.len());
        for item in arr {
            results.push(self.bound.invoke(item.clone(), config).await?);
        }
        Ok(Value::Array(results))
    }
}