apunkt/ktx
1
0
Fork 0
mirror of https://github.com/Kaelio/ktx.git synced 2026-06-10 08:05:14 +02:00
Code Issues Projects Releases Packages Wiki Activity

build: make local embedding dependencies optional

Browse source
This commit is contained in:
Andrey Avtomonov 2026-05-11 09:54:37 +02:00
parent 131b904229
commit 5461b53f89
2 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
python/ktx-daemon/pyproject.toml
View file

@ -15,15 +15,19 @@ dependencies = [
"psycopg[binary]>=3.2.0",
"pydantic>=2.9.0",
"requests>=2.32.0",
"sentence-transformers>=5.1.1",
"sqlglot>=26",
"torch>=2.2.0",
"uvicorn[standard]>=0.32.0",
]
[project.scripts]
ktx-daemon = "ktx_daemon.__main__:main"
[project.optional-dependencies]
local-embeddings = [
"sentence-transformers>=5.1.1",
"torch>=2.2.0",
]
[project.urls]
Homepage = "https://github.com/kaelio/ktx"
Repository = "https://github.com/kaelio/ktx"

13
uv.lock generated
View file

@ -452,11 +452,15 @@ dependencies = [
{ name = "psycopg", extra = ["binary"] },
{ name = "pydantic" },
{ name = "requests" },
{ name = "sentence-transformers" },
{ name = "sqlglot" },
{ name = "uvicorn", extra = ["standard"] },
]
[package.optional-dependencies]
local-embeddings = [
{ name = "sentence-transformers" },
{ name = "torch", version = "2.11.0", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "sys_platform == 'darwin'" },
{ name = "torch", version = "2.11.0+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "sys_platform != 'darwin'" },
{ name = "uvicorn", extra = ["standard"] },
]
[package.dev-dependencies]
@ -476,11 +480,12 @@ requires-dist = [
{ name = "psycopg", extras = ["binary"], specifier = ">=3.2.0" },
{ name = "pydantic", specifier = ">=2.9.0" },
{ name = "requests", specifier = ">=2.32.0" },
{ name = "sentence-transformers", specifier = ">=5.1.1" },
{ name = "sentence-transformers", marker = "extra == 'local-embeddings'", specifier = ">=5.1.1" },
{ name = "sqlglot", specifier = ">=26" },
{ name = "torch", specifier = ">=2.2.0", index = "https://download.pytorch.org/whl/cpu" },
{ name = "torch", marker = "extra == 'local-embeddings'", specifier = ">=2.2.0", index = "https://download.pytorch.org/whl/cpu" },
{ name = "uvicorn", extras = ["standard"], specifier = ">=0.32.0" },
]
provides-extras = ["local-embeddings"]
[package.metadata.requires-dev]
dev = [