fusillade 16.8.2

A daemon implementation for sending batched LLM requests with efficient request coalescing
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

//! Batching system for HTTP requests with retry logic and concurrency control.
//!
//! This crate provides 'managers' that accept submitted HTTP requests, and provides an API for
//! checking their status over time. Behind the scenes, a daemon processes these requests in
//! batches, retrying failed requests with exponential backoff and enforcing concurrency limits
//!
//! Batching system with PostgreSQL storage and background daemon for processing requests.

pub mod batch;
pub mod daemon;
pub mod error;
pub mod http;
pub mod manager;
pub mod processor;
pub mod request;
pub mod response_step;

// Re-export commonly used types
pub use batch::*;
pub use daemon::{Daemon, DaemonConfig, ModelEscalationConfig};
pub use error::{FusilladeError, Result};
pub use http::{HttpClient, HttpResponse, MockHttpClient, ReqwestHttpClient, StreamReassembler};
pub use manager::postgres::{PoolProvider, PostgresRequestManager, TestDbPools};
#[cfg(feature = "postgres")]
pub use manager::response_step::PostgresResponseStepManager;
pub use manager::{CreateSingleRequestBatchInput, DaemonExecutor, Storage};
pub use processor::{CancellationFuture, DefaultRequestProcessor, RequestProcessor, ShouldRetry};
pub use request::*;
pub use response_step::{
    CreateStepInput, ResponseStep, ResponseStepStore, StepId, StepKind, StepState,
};

/// Get the fusillade database migrator
///
/// Returns a migrator that can be run against a connection pool.
#[cfg(feature = "postgres")]
pub fn migrator() -> sqlx::migrate::Migrator {
    sqlx::migrate!("./migrations")
}