Skip to main content

atomr_accel_agents/
lib.rs

1//! Agentic / LLM GPU actor blueprints on atomr-accel-cuda.
2//!
3//! ```ignore
4//! use atomr_accel_agents::prelude::*;
5//! ```
6//!
7//! - [`shared_state::SharedGpuStateCoordinator`] — write-token
8//!   coordination over a `ManagedRef<f32>` world buffer.
9//! - [`embedding_cache::EmbeddingCache`] — LRU cache of
10//!   `(input hash) -> Vec<f32>`.
11//! - [`rag::RagPipeline`] — retrieval + generate harness.
12//! - [`vector_index::CpuVectorIndex`] — top-k cosine similarity
13//!   over a flat host index.
14//! - [`langgraph_nodes::LangGraphGpuActor`] — DAG executor with
15//!   cycle detection.
16
17pub mod embedding_cache;
18pub mod langgraph_nodes;
19pub mod rag;
20pub mod shared_state;
21pub mod vector_index;
22
23pub mod prelude {
24    //! Canonical re-exports. `use atomr_accel_agents::prelude::*;`.
25    pub use crate::embedding_cache::{
26        CacheStats, EmbeddingCache, EmbeddingCacheConfig, EmbeddingCacheMsg,
27    };
28    pub use crate::langgraph_nodes::{
29        LangGraphGpuActor, NodeEntry, NodeGraph, NodeGraphMsg, NodeId,
30    };
31    pub use crate::rag::{RagAnswer, RagConfig, RagMsg, RagPipeline, RagQuery};
32    pub use crate::shared_state::{
33        SharedGpuStateCoordinator, SharedStateMsg, SharedStateStats, WriteToken,
34    };
35    pub use crate::vector_index::{CpuVectorIndex, VectorEntry, VectorIndexMsg};
36}