forked from vlwkaos/ir
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCargo.toml
More file actions
58 lines (51 loc) · 1.68 KB
/
Copy pathCargo.toml
File metadata and controls
58 lines (51 loc) · 1.68 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
[package]
name = "ir-search"
version = "0.15.0"
edition = "2024"
description = "Local markdown semantic search with hybrid BM25+vector retrieval and LLM reranking"
license = "MIT"
repository = "https://github.com/vlwkaos/ir"
[features]
default = ["llama-metal", "llama-openmp"]
llama-metal = ["llama-cpp-2/metal"]
llama-openmp = ["llama-cpp-2/openmp"]
llama-cuda = ["llama-cpp-2/cuda"]
llama-rocm = ["llama-cpp-2/rocm"]
llama-vulkan = ["llama-cpp-2/vulkan"]
[[bin]]
name = "ir"
path = "src/main.rs"
[dependencies]
clap = { version = "4", features = ["derive"] }
rusqlite = { version = "0.32", features = ["bundled", "vtab"] }
sqlite-vec = "0.1"
serde = { version = "1", features = ["derive"] }
serde_yaml = "0.9"
serde_json = "1"
ignore = "0.4"
sha2 = "0.10"
hex = "0.4"
indicatif = "0.17"
dialoguer = "0.11"
libc = "0.2"
colored = "2"
thiserror = "2"
chrono = { version = "0.4", features = ["serde"] }
dirs = "5"
hf-hub = { version = "0.5.0", default-features = false, features = ["ureq"] }
rmcp = { version = "1.3.0", features = ["server", "macros", "transport-io", "transport-streamable-http-server"] }
tokio = { version = "1.51.0", features = ["macros", "rt", "rt-multi-thread", "io-std", "net", "signal"] }
axum = { version = "0.8.8", default-features = false, features = ["http1", "tokio"] }
tower-http = { version = "0.6.8", features = ["cors"] }
schemars = "1.2.1"
regex = "1"
[dev-dependencies]
tempfile = "3"
[target.'cfg(target_os = "macos")'.dependencies]
llama-cpp-2 = { version = "0.1.137", default-features = false }
[target.'cfg(not(target_os = "macos"))'.dependencies]
llama-cpp-2 = { version = "0.1.137", default-features = false }
[profile.release]
opt-level = 3
lto = true
codegen-units = 1