lib.rs 1.32 KB
Newer Older
1
2
// SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
Ryan Olson's avatar
Ryan Olson committed
3

Neelay Shah's avatar
Neelay Shah committed
4
//! Dynamo
Ryan Olson's avatar
Ryan Olson committed
5
6
7
8

#![allow(dead_code)]
#![allow(unused_imports)]

9
10
11
12
13
14
15
16
17
18
19
use std::{
    collections::HashMap,
    sync::{Arc, OnceLock, Weak},
};

pub use anyhow::{
    Context as ErrorContext, Error, Ok as OK, Result, anyhow as error, bail as raise,
};

use async_once_cell::OnceCell;

20
pub mod config;
Ryan Olson's avatar
Ryan Olson committed
21
22
23
pub use config::RuntimeConfig;

pub mod component;
24
pub mod compute;
25
pub mod discovery;
Ryan Olson's avatar
Ryan Olson committed
26
pub mod engine;
27
pub mod health_check;
28
pub mod local_endpoint_registry;
29
30
pub mod system_status_server;
pub use system_status_server::SystemStatusServerInfo;
31
pub mod distributed;
32
pub mod instances;
33
pub mod logging;
34
pub mod metrics;
Ryan Olson's avatar
Ryan Olson committed
35
pub mod pipeline;
36
pub mod prelude;
Ryan Olson's avatar
Ryan Olson committed
37
pub mod protocols;
Ryan Olson's avatar
Ryan Olson committed
38
pub mod runnable;
Ryan Olson's avatar
Ryan Olson committed
39
40
pub mod runtime;
pub mod service;
41
pub mod slug;
42
pub mod storage;
43
pub mod system_health;
Ryan Olson's avatar
Ryan Olson committed
44
pub mod traits;
Ryan Olson's avatar
Ryan Olson committed
45
pub mod transports;
Ryan Olson's avatar
Ryan Olson committed
46
pub mod utils;
Ryan Olson's avatar
Ryan Olson committed
47
48
pub mod worker;

49
pub use distributed::{DistributedRuntime, distributed_test_utils};
Ryan Olson's avatar
Ryan Olson committed
50
pub use futures::stream;
51
pub use metrics::MetricsRegistry;
52
pub use runtime::Runtime;
53
pub use system_health::{HealthCheckTarget, SystemHealth};
Ryan Olson's avatar
Ryan Olson committed
54
55
pub use tokio_util::sync::CancellationToken;
pub use worker::Worker;
56
57
58
59
60

use component::Endpoint;
use utils::GracefulShutdownTracker;

use config::HealthStatus;