spire-ai 0.1.3

AI-native SDK for SpireDB — RAG, code search, agents
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

//! SpireAI — High-level AI SDK for SpireDB.
//!
//! Build RAG pipelines, semantic code search, and agent memory
//! on top of SpireDB with minimal boilerplate.
//!
//! # Quick Start
//!
//! ```rust,no_run
//! use spire_ai::prelude::*;
//!
//! #[derive(Doc, Serialize, Deserialize, Clone)]
//! struct Article {
//!     #[id]
//!     slug: String,
//!     title: String,
//!     content: String,
//! }
//!
//! #[tokio::main]
//! async fn main() -> spire_ai::Result<()> {
//!     let spire = Spire::connect("http://127.0.0.1:50051").await?;
//!     let articles = spire.collection::<Article>("articles");
//!     articles.ensure().await?;
//!
//!     articles.insert(&Article {
//!         slug: "hello".into(),
//!         title: "Hello World".into(),
//!         content: "An introduction to SpireAI.".into(),
//!     }).await?;
//!
//!     let hits = articles.search("introduction").run().await?;
//!     for hit in hits {
//!         println!("{}: {}", hit.score, hit.doc.title);
//!     }
//!     Ok(())
//! }
//! ```

pub mod client;
pub mod collection;
pub mod document;
pub mod embedding;
pub mod error;
pub mod llm;
pub mod rag;
pub mod search;
pub mod types;
pub mod watch;

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

pub mod filecache;

pub mod agent;
pub mod tool;

// Re-exports
pub use client::{Spire, SpireBuilder};
pub use collection::Collection;
pub use document::Doc;
pub use error::{Error, Result};
pub use filecache::FileCache;
pub use search::{Filter, Hit, Search};
pub use types::{IndexResult, IngestResult};
pub use watch::{Change, WatchStream};

#[cfg(feature = "macros")]
pub use spire_ai_macros::Doc;

/// Prelude — import everything you need with `use spire_ai::prelude::*`.
pub mod prelude {
    pub use crate::client::{Spire, SpireBuilder};
    pub use crate::collection::Collection;
    pub use crate::document::Doc;
    pub use crate::embedding::Embedder;
    pub use crate::error::{Error, Result};
    pub use crate::rag::chunker::Chunk;
    pub use crate::rag::{RagBuilder, RagPipeline, ScoredChunk};
    pub use crate::search::{Filter, Hit, Search};
    pub use crate::{Change, WatchStream};

    #[cfg(feature = "code")]
    pub use crate::code::{CodeChunk, CodeContext, CodeIndex};

    pub use crate::filecache::{CacheStats, FileCache, ReadResult};

    pub use crate::agent::{AgentLoop, AgentLoopConfig, AgentMemory};
    pub use crate::llm::{ChatMessage, ChatResponse, ToolDef};
    pub use crate::tool::{Tool, ToolRegistry, ToolResult};

    #[cfg(feature = "macros")]
    pub use spire_ai_macros::Doc;

    pub use serde::{Deserialize, Serialize};
}