Skip to main content

somatize_core/
lib.rs

1//! Core types and traits for the Soma computational graph runtime.
2//!
3//! This crate defines the contracts that all other crates depend on:
4//! - [`Filter`] — the computation unit (fit/forward)
5//! - [`Value`] — typed data flowing between filters (Tensor, JSON, Bytes)
6//! - [`Graph`] — DAG of filter nodes and edges
7//! - [`CacheKey`] / [`CacheStore`] — content-addressable caching
8//! - [`DataStore`] — abstraction for moving data between workers
9//! - [`Schema`] — dtype + shape for compile-time validation
10//! - [`Event`] — runtime lifecycle events
11
12pub mod cache;
13pub mod error;
14pub mod event;
15pub mod filter;
16pub mod graph;
17pub mod schema;
18pub mod search;
19pub mod store;
20pub mod strategy;
21pub mod study;
22pub mod util;
23pub mod value;
24pub mod virtual_value;
25
26// Re-export core types for convenience.
27pub use cache::{CacheKey, CacheStore, CacheTier, EntryMeta, Origin};
28pub use error::{Result, SomaError};
29pub use event::{Event, MetricRecord, PlanSummary, RunId, StudyId, TrialId};
30pub use filter::{Distribution, Filter, FilterKind, FilterMeta, RemoteTarget, StreamMode};
31pub use graph::{Edge, EdgeKind, Graph, Node, NodeId};
32pub use schema::{DataType, Dimension, Schema};
33pub use search::{Scale, SearchDimension, SearchSpace, Searchable};
34#[cfg(feature = "s3")]
35pub use store::S3DataStore;
36#[cfg(feature = "zarr")]
37pub use store::ZarrStore;
38pub use store::{
39    DataRef, DataStore, LocalDataStore, StorageConfig, StoreMeta, StreamCache, StreamFormat,
40    slice_tensor_rows,
41};
42pub use strategy::{
43    ClientSelection, CommunicationProtocol, ExploitStrategy, ExploreStrategy, FederatedAggregation,
44    GradientAggregation, Partition, TrainingStrategy,
45};
46pub use study::{Direction, Objective, PruningStrategy, SearchStrategy, Study, Trial, TrialState};
47pub use value::Value;
48pub use virtual_value::{ValueStatus, VirtualValue};
49
50// Re-export derive macro
51pub use somatize_macros::SomaFilter;