Cargo.toml 2 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

[package]
name = "dynamo-kv-router"
description = "KV Router - Radix tree for LLM KV cache routing"
version.workspace = true
edition.workspace = true
authors.workspace = true
license.workspace = true
repository.workspace = true

[features]
default = []
metrics = ["dep:dynamo-runtime"]
16
bench = ["dep:clap", "dep:indicatif", "dep:serde_json", "dynamo-runtime/integration", "dep:uuid", "dep:plotters"]
17
18
19
20
21
22
23
24
25

[dependencies]
# repo
dynamo-runtime = { workspace = true, optional = true }
dynamo-tokens = { workspace = true }

# workspace
anyhow = { workspace = true }
async-trait = { workspace = true }
Yan Ru Pei's avatar
Yan Ru Pei committed
26
dashmap = { workspace = true }
27
28
29
prometheus = { workspace = true }
rand = { workspace = true }
serde = { workspace = true }
Yan Ru Pei's avatar
Yan Ru Pei committed
30
serde_json = { workspace = true, optional = true }
31
32
33
34
35
36
thiserror = { workspace = true }
tokio = { workspace = true }
tokio-util = { workspace = true }
tracing = { workspace = true }
xxhash-rust = { workspace = true }

Yan Ru Pei's avatar
Yan Ru Pei committed
37
38
39
40
# dependencies
flume = "0.12.0"
parking_lot = { workspace = true }

41
42
43
# bench (optional)
clap = { version = "4.5", features = ["derive"], optional = true }
indicatif = { version = "0.18.0", optional = true }
Yan Ru Pei's avatar
Yan Ru Pei committed
44
uuid = { workspace = true, optional = true }
45
plotters = { version = "0.3", optional = true, default-features = false, features = ["svg_backend", "line_series", "point_series", "full_palette"] }
46
rustc-hash = "2.1.1"
47
48

[dev-dependencies]
Yan Ru Pei's avatar
Yan Ru Pei committed
49
dynamo-bench = { path = "../bench" }
50
51
52
53
rstest = "0.18.2"
rstest_reuse = "0.7.0"
serde_json = { workspace = true }
tokio = { workspace = true, features = ["rt", "macros", "time"] }
Yan Ru Pei's avatar
Yan Ru Pei committed
54
dynamo-mocker = { workspace = true }
55
dynamo-tokens = { workspace = true }
Yan Ru Pei's avatar
Yan Ru Pei committed
56
57
58
minstant = "0.1.7"
futures = "0.3"

59
60
61
62
63

[[bench]]
name = "radix_tree_microbench"
harness = false
required-features = ["bench"]
64
65
66
67
68

[[bench]]
name = "kv_indexer_bench"
harness = false
required-features = ["bench"]
Yan Ru Pei's avatar
Yan Ru Pei committed
69
70
71
72
73

[[bench]]
name = "mooncake_bench"
harness = false
required-features = ["bench"]