ambi 0.2.1

A flexible, multi-backend, customizable AI agent framework, entirely based on Rust.
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

// src/agent/pipeline.rs
pub mod chat_runner;
pub mod rag;
pub mod react;

use crate::agent::core::{Agent, AgentState};
use crate::error::Result;
use crate::ContentPart;
use std::pin::Pin;
use std::sync::{Arc, Mutex as StdMutex};
use tokio_stream::wrappers::ReceiverStream;

pub trait Pipeline: Send + Sync {
    fn execute(
        &self,
        agent: &Agent,
        state: &Arc<StdMutex<AgentState>>,
        input: Vec<ContentPart>,
    ) -> impl std::future::Future<Output = Result<String>> + Send;

    fn execute_stream(
        &self,
        agent: &Agent,
        state: &Arc<StdMutex<AgentState>>,
        input: Vec<ContentPart>,
    ) -> impl std::future::Future<Output = Result<Pin<Box<ReceiverStream<Result<String>>>>>> + Send;
}

pub use chat_runner::ChatRunner;