ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Studio: move RAG deps to dedicated rag.txt and install in normal path

Browse source 
no-torch-runtime.txt is only consumed in NO_TORCH (Intel Mac
GGUF-only) mode, so qdrant-client / bm25s / pymupdf etc. were never
installed in the normal install path. Split them into rag.txt and
add a step to install_python_stack.py that installs it after studio
deps, skipped only when NO_TORCH is set.
This commit is contained in:
Roland Tannous 2026-05-24 16:23:26 +04:00
commit 6c694d2ef8
3 changed files with 40 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

13
studio/install_python_stack.py
View file

@ -915,6 +915,8 @@ def install_python_stack() -> int:
base_total -= 1 # triton step is skipped on macOS
if not IS_WINDOWS and not IS_MACOS and not NO_TORCH:
base_total += 3
if not NO_TORCH:
base_total += 1 # studio RAG deps (rag.txt)
_TOTAL = (base_total - 1) if skip_base else base_total
# 1. Try to use uv for faster installs (must happen before pip upgrade
@ -1203,6 +1205,17 @@ def install_python_stack() -> int:
req = REQ_ROOT / "studio.txt",
)
# 8b. RAG dependencies (vector store + lexical index + layout-aware
# parsers). Skipped in no-torch mode because RAG embeddings go
# through sentence-transformers, which requires torch.
if not NO_TORCH:
_progress("rag deps")
pip_install(
"Installing RAG dependencies",
"--no-cache-dir",
req = REQ_ROOT / "rag.txt",
)
# 9. Data-designer dependencies
_progress("data designer deps")
pip_install(