Cargo.toml 2.77 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

16
17
18
19
[workspace]
# empty workspace to exclude from top level workspace
# excluded due to pyo3 extension module build issues

20
[package]
Neelay Shah's avatar
Neelay Shah committed
21
name = "dynamo-py3"
22
version = "0.4.1"
23
24
25
edition = "2021"
authors = ["NVIDIA"]
license = "Apache-2.0"
Neelay Shah's avatar
Neelay Shah committed
26
27
homepage = "https://github.com/ai-dynamo/dynamo"
repository = "https://github.com/ai-dynamo/dynamo.git"
28
29
30
31
32

[lib]
path = "rust/lib.rs"
name = "_core"
# "cdylib" is necessary to produce a shared library for Python to import from.
33
34
# "rlib" is necessary to support doctests.
crate-type = ["cdylib", "rlib"]
35

36
37
[features]
default = []
Ryan Olson's avatar
Ryan Olson committed
38
block-manager = ["dynamo-llm/block-manager", "dep:dlpark", "dep:cudarc"]
Neelay Shah's avatar
Neelay Shah committed
39

40
[dependencies]
41
dynamo-llm = { path = "../../llm" }
Neelay Shah's avatar
Neelay Shah committed
42
dynamo-runtime = { path = "../../runtime" }
43

44
anyhow = { version = "1" }
45
dynamo-async-openai = { path = "../../async-openai" }
46
47
async-stream = { version = "0.3" }
async-trait = { version = "0.1" }
Ryan Olson's avatar
Ryan Olson committed
48
derive-getters = "0.5"
49
either = { version = "1.13", features = ["serde"] }
50
51
52
53
54
futures = { version = "0.3" }
once_cell = { version = "1.20.3" }
serde = { version = "1" }
serde_json = { version = "1.0.138" }
thiserror = { version = "2.0" }
55
tokio = { version = "1.46.0", features = ["full"] }
56
tokio-stream = { version = "0" }
Ryan Olson's avatar
Ryan Olson committed
57
58
tokio-util = { version = "0.7", features = ["rt"] }
tracing = { version = "0"  }
59
tracing-subscriber = { version = "0.3", features = ["fmt", "env-filter"] }
Ryan Olson's avatar
Ryan Olson committed
60
uuid = { version = "1.17", features = ["v4", "serde"] }
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

# "extension-module" tells pyo3 we want to build an extension module (skips linking against libpython.so)
# "abi3-py39" tells pyo3 (and maturin) to build using the stable ABI with minimum Python version 3.9
pyo3 = { version = "0.23.4", default-features = false, features = [
  "macros",
  "experimental-async",
  "experimental-inspect",
  "extension-module",
  "py-clone",
] }

pyo3-async-runtimes = { version = "0.23.0", default-features = false, features = [
  "attributes",
  "testing",
  "tokio-runtime",
  "unstable-streams",
] }

79
pythonize = "0.23"
80
81

dlpark = { version = "0.5", features = ["pyo3", "half"], optional = true }
Ryan Olson's avatar
Ryan Olson committed
82
cudarc = { version = "0.16.2", features = ["cuda-12020"], optional = true }
83

Ryan Olson's avatar
Ryan Olson committed
84
85
86

[dev-dependencies]
rstest = "0.25"