daimon 0.16.0

A Rust-native AI agent 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
51
52
53
54
55

//! Retrieval-Augmented Generation (RAG) abstractions.
//!
//! The [`Retriever`] trait models any system that fetches relevant documents
//! for a query (vector stores, full-text search, hybrid). [`RetrieverTool`]
//! wraps a retriever as a [`Tool`](crate::tool::Tool) so agents can search
//! on demand.
//!
//! ## Plugin System
//!
//! For a pluggable storage backend, implement [`VectorStore`] for your
//! vector database, then compose with [`SimpleKnowledgeBase`] for a
//! complete embedding-backed retrieval pipeline.
//!
//! ```ignore
//! use daimon::retriever::{VectorStore, SimpleKnowledgeBase, InMemoryVectorStoreBackend, Document};
//!
//! let store = InMemoryVectorStoreBackend::new();
//! let kb = SimpleKnowledgeBase::new(embedding_model, store);
//!
//! kb.ingest(vec![Document::new("relevant text")]).await?;
//! let results = kb.search("query", 5).await?;
//! ```

pub mod in_memory;
pub mod in_memory_store;
pub mod knowledge_base;
mod traits;
mod types;
mod tool;
pub mod vector_store;

#[cfg(feature = "qdrant")]
pub mod qdrant;

#[cfg(feature = "pgvector")]
pub mod pgvector {
    //! pgvector-backed vector store — re-exported from `daimon-plugin-pgvector`.
    pub use daimon_plugin_pgvector::*;
}

#[cfg(feature = "opensearch")]
pub mod opensearch {
    //! OpenSearch k-NN backed vector store — re-exported from `daimon-plugin-opensearch`.
    pub use daimon_plugin_opensearch::*;
}

pub use in_memory::InMemoryVectorStore;
pub use in_memory_store::InMemoryVectorStoreBackend;
pub use knowledge_base::{
    ErasedKnowledgeBase, KnowledgeBase, SharedKnowledgeBase, SimpleKnowledgeBase,
};
pub use traits::{ErasedRetriever, Retriever, SharedRetriever};
pub use tool::RetrieverTool;
pub use types::Document;
pub use vector_store::{ErasedVectorStore, ScoredDocument, SharedVectorStore, VectorStore};