cognee-vector 0.1.3

Vector-store abstraction with brute-force, LanceDB, and pgvector adapters for cognee.
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

//! Vector database abstraction for Cognee-Rust.
//!
//! Provides vector storage and similarity search for embeddings.

/// Pure-Rust in-memory brute-force vector database (OSS default fallback).
pub mod brute_force_vector_db;
/// Error types for vector database operations.
pub mod error;
/// Data models for vector points, search results, and collection configuration.
pub mod models;
/// Vector database trait definition.
pub mod vector_db_trait;

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

/// Embedded file-backed vector store via the LanceDB crate.
///
/// Disabled on Android — the LanceDB + Arrow native stack does not
/// cross-compile cleanly to mobile targets, so `BruteForceVectorDB`
/// remains the Android default.
#[cfg(not(target_os = "android"))]
pub mod lancedb_adapter;

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

pub use brute_force_vector_db::BruteForceVectorDB;
pub use error::{VectorDBError, VectorDBResult};
pub use models::{
    CollectionConfig, DATASET_ID_KEY, DATASET_IDS_KEY, DistanceMetric, SearchResult, VectorPoint,
};
pub use vector_db_trait::VectorDB;

#[cfg(feature = "pgvector")]
pub use pgvector_adapter::PgVectorAdapter;

#[cfg(not(target_os = "android"))]
pub use lancedb_adapter::LanceDbAdapter;

#[cfg(feature = "testing")]
pub use mock_vector_db::MockVectorDB;