#![warn(missing_docs)]
#![warn(clippy::unwrap_used)]
#![warn(clippy::all)]
pub mod error;
#[cfg(feature = "persistent")]
pub mod persistent;
#[cfg(feature = "persistent-cuda")]
pub mod cuda_bridge;
#[cfg(feature = "actix")]
pub mod actix;
#[cfg(feature = "tower")]
pub mod tower;
#[cfg(feature = "axum")]
pub mod axum;
#[cfg(feature = "arrow")]
pub mod arrow;
#[cfg(feature = "polars")]
pub mod polars;
#[cfg(feature = "grpc")]
pub mod grpc;
#[cfg(feature = "candle")]
pub mod candle;
#[cfg(feature = "config")]
pub mod config;
#[cfg(feature = "tracing-integration")]
pub mod tracing_ext;
#[cfg(feature = "prometheus")]
pub mod metrics;
#[cfg(feature = "graphql")]
pub mod graphql;
pub mod llm;
pub mod streaming;
#[cfg(feature = "candle")]
pub mod ml_bridge;
#[cfg(feature = "enterprise")]
pub mod enterprise;
#[cfg(feature = "ml-bridge")]
pub mod ml_bridge;
pub mod prelude {
pub use crate::error::*;
#[cfg(feature = "persistent")]
pub use crate::persistent::*;
#[cfg(feature = "actix")]
pub use crate::actix::{
BridgeHealth, GpuActorBridge, GpuActorConfig, GpuActorExt, GpuRequest, GpuResponse,
HealthCheck, TypedGpuRequest,
};
#[cfg(all(feature = "actix", feature = "persistent"))]
pub use crate::actix::{
GetStatsCmd, GpuPersistentActor, GpuPersistentActorExt, InjectCmd, PauseCmd,
PersistentActorConfig, PersistentResponseMsg, ResumeCmd, RunStepsCmd, ShutdownCmd,
Subscribe, Unsubscribe,
};
#[cfg(all(feature = "tower", feature = "persistent"))]
pub use crate::tower::{
PersistentKernelLayer, PersistentKernelService, PersistentServiceBuilder,
PersistentServiceConfig, PersistentServiceRequest, PersistentServiceResponse,
};
#[cfg(feature = "tower")]
pub use crate::tower::{
RingKernelLayer, RingKernelService, ServiceBuilder, ServiceConfig, ServiceRequest,
ServiceResponse,
};
#[cfg(feature = "axum")]
pub use crate::axum::{
gpu_handler, health_handler, ApiRequest, ApiResponse, AppState, AxumConfig, HealthResponse,
KernelHealth, RequestStats, RouterBuilder, StatsSnapshot,
};
#[cfg(all(feature = "axum", feature = "persistent"))]
pub use crate::axum::{
impulse_handler, pause_handler, persistent_health_handler, persistent_stats_handler,
resume_handler, step_handler, CommandResponse, ImpulseRequest, PersistentAxumConfig,
PersistentGpuState, PersistentHealthResponse, PersistentStatsResponse, StepRequest,
};
#[cfg(feature = "persistent-cuda")]
pub use crate::cuda_bridge::{CudaPersistentHandle, CudaPersistentHandleBuilder};
#[cfg(feature = "arrow")]
pub use crate::arrow::*;
#[cfg(feature = "polars")]
pub use crate::polars::*;
#[cfg(feature = "grpc")]
pub use crate::grpc::*;
#[cfg(feature = "candle")]
pub use crate::candle::*;
#[cfg(feature = "config")]
pub use crate::config::*;
#[cfg(feature = "prometheus")]
pub use crate::metrics::*;
#[cfg(all(feature = "enterprise", feature = "axum"))]
pub use crate::enterprise::{
enterprise_routes, health_handler as enterprise_health_handler, liveness_handler,
metrics_handler as enterprise_metrics_handler, readiness_handler, stats_handler,
};
#[cfg(all(feature = "enterprise", feature = "tower"))]
pub use crate::enterprise::{
CircuitBreakerFuture, CircuitBreakerLayer, CircuitBreakerService, DegradationFuture,
DegradationLayer, DegradationService,
};
#[cfg(feature = "enterprise")]
pub use crate::enterprise::{
EnterpriseHealthResponse, EnterpriseState, EnterpriseStatsResponse, LivenessResponse,
ReadinessResponse,
};
#[cfg(feature = "graphql")]
pub use crate::graphql::{
create_schema, CommandAck, DynPersistentHandle, GraphQLState, InjectInput,
KernelConfigResponse, KernelEvent, KernelEventType, KernelMutation, KernelQuery,
KernelSchema, KernelStatsResponse, KernelStatus, KernelSubscription, ProgressUpdate,
RunStepsInput,
};
#[cfg(all(feature = "graphql", feature = "axum"))]
pub use crate::graphql::{graphql_handler, graphql_playground, graphql_router};
}