Cargo.toml 2.03 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# 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 = []
15
metrics = []
16
bench = ["dep:clap", "dep:indicatif", "dep:serde_json", "dynamo-runtime/integration", "dep:plotters"]
17
18
19

[dependencies]
# repo
20
dynamo-runtime = { workspace = true }
21
22
23
24
25
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
derive_builder = { workspace = true }
28
derive-getters = { workspace = true }
29
30
31
prometheus = { workspace = true }
rand = { workspace = true }
serde = { workspace = true }
Yan Ru Pei's avatar
Yan Ru Pei committed
32
serde_json = { workspace = true, optional = true }
33
34
35
36
thiserror = { workspace = true }
tokio = { workspace = true }
tokio-util = { workspace = true }
tracing = { workspace = true }
37
uuid = { workspace = true }
38
validator = { workspace = true }
39
40
xxhash-rust = { workspace = true }

Yan Ru Pei's avatar
Yan Ru Pei committed
41
42
43
44
# dependencies
flume = "0.12.0"
parking_lot = { workspace = true }

45
46
47
# bench (optional)
clap = { version = "4.5", features = ["derive"], optional = true }
indicatif = { version = "0.18.0", optional = true }
48
plotters = { version = "0.3", optional = true, default-features = false, features = ["svg_backend", "line_series", "point_series", "full_palette"] }
49
rustc-hash = "2.1.1"
50
51

[dev-dependencies]
Yan Ru Pei's avatar
Yan Ru Pei committed
52
dynamo-bench = { path = "../bench" }
53
54
55
56
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
57
dynamo-mocker = { workspace = true }
58
dynamo-tokens = { workspace = true }
Yan Ru Pei's avatar
Yan Ru Pei committed
59
60
minstant = "0.1.7"

61
62
63
64
[[bench]]
name = "kv_indexer_bench"
harness = false
required-features = ["bench"]
Yan Ru Pei's avatar
Yan Ru Pei committed
65
66
67
68
69

[[bench]]
name = "mooncake_bench"
harness = false
required-features = ["bench"]
70
71
72
73
74

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