35 lines
1.2 KiB
Rust
35 lines
1.2 KiB
Rust
|
pub mod batch;
|
||
|
|
pub mod cache;
|
||
|
|
pub mod config;
|
||
|
|
pub mod error;
|
||
|
|
pub mod metrics;
|
||
|
|
pub mod providers;
|
||
|
|
pub mod service;
|
||
|
|
pub mod store;
|
||
|
|
pub mod traits;
|
||
|
|
|
||
|
|
#[cfg(feature = "memory-cache")]
|
||
|
|
pub use cache::MemoryCache;
|
||
|
|
#[cfg(feature = "persistent-cache")]
|
||
|
|
pub use cache::PersistentCache;
|
||
|
|
pub use config::{BatchConfig, CacheConfig, EmbeddingConfig, ProviderConfig};
|
||
|
|
pub use error::{EmbeddingError, Result};
|
||
|
|
#[cfg(feature = "cohere-provider")]
|
||
|
|
pub use providers::cohere::{CohereModel, CohereProvider};
|
||
|
|
#[cfg(feature = "fastembed-provider")]
|
||
|
|
pub use providers::fastembed::{FastEmbedModel, FastEmbedProvider};
|
||
|
|
#[cfg(feature = "huggingface-provider")]
|
||
|
|
pub use providers::huggingface::{HuggingFaceModel, HuggingFaceProvider};
|
||
|
|
#[cfg(feature = "ollama-provider")]
|
||
|
|
pub use providers::ollama::{OllamaModel, OllamaProvider};
|
||
|
|
#[cfg(feature = "openai-provider")]
|
||
|
|
pub use providers::openai::{OpenAiModel, OpenAiProvider};
|
||
|
|
#[cfg(feature = "voyage-provider")]
|
||
|
|
pub use providers::voyage::{VoyageModel, VoyageProvider};
|
||
|
|
pub use service::EmbeddingService;
|
||
|
|
pub use store::*;
|
||
|
|
pub use traits::{
|
||
|
|
cosine_similarity, euclidean_distance, normalize_embedding, Embedding, EmbeddingOptions,
|
||
|
|
EmbeddingProvider, EmbeddingResult, ProviderInfo,
|
||
|
|
};
|