lib.rs 1.37 KB
Newer Older
1
// SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
// 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 engine_routes;
28
pub mod error;
29
pub mod health_check;
30
pub mod local_endpoint_registry;
31
32
pub mod system_status_server;
pub use system_status_server::SystemStatusServerInfo;
33
pub mod distributed;
34
pub mod instances;
35
pub mod logging;
36
pub mod metrics;
37
pub mod nvtx;
Ryan Olson's avatar
Ryan Olson committed
38
pub mod pipeline;
39
pub mod prelude;
Ryan Olson's avatar
Ryan Olson committed
40
pub mod protocols;
Ryan Olson's avatar
Ryan Olson committed
41
pub mod runnable;
Ryan Olson's avatar
Ryan Olson committed
42
43
pub mod runtime;
pub mod service;
44
pub mod slug;
45
pub mod storage;
46
pub mod system_health;
Ryan Olson's avatar
Ryan Olson committed
47
pub mod traits;
Ryan Olson's avatar
Ryan Olson committed
48
pub mod transports;
Ryan Olson's avatar
Ryan Olson committed
49
pub mod utils;
Ryan Olson's avatar
Ryan Olson committed
50
51
pub mod worker;

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

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

use config::HealthStatus;