92994e8b83
Backend (studio/backend/): - core/rag/: parsers (PDF/TXT/MD/DOCX/HTML via pypdf/python-docx/bs4), recursive token-aware chunker, embeddings singleton via FastSentenceTransformer.from_pretrained(for_inference=True), Qdrant local vector store, bm25s lexical index, RRF hybrid retrieval, spawn-subprocess ingestion job with SSE progress, optional CrossEncoder reranker (off-by-default). - routes/rag.py: KB CRUD, doc upload (KB + per-thread), doc list/delete, ingestion SSE, hybrid+rerank search, thread-index list/clear. - routes/chat_history.py: purge thread RAG artifacts on thread delete and clear-all (rag_documents has no FK cascade to chat_threads so uploads work on un-persisted threads). - studio.db gains 4 RAG tables; storage_roots gains rag_*() helpers. - auth/authentication.py: get_current_subject_sse accepts ?token=... so EventSource can stream ingestion progress. Frontend (studio/frontend/): - features/rag/: api client, Zustand store, hooks, dropzone, KB list, doc rows, ingestion-progress, thread-index list components. - Settings dialog gains a Knowledge Bases tab (master/detail + thread documents list); /knowledge-bases deep-links to it. - features/chat/: per-thread ragSource/enableRerank/ragTopK state in chat-runtime-store; Retrieval section in chat-settings-sheet with KB DropdownMenu (active highlight + per-row trash), thread doc list with Clear-thread-index button, RAG Top K slider, reranker toggle; chat-adapter retrieves before /v1/chat/completions and injects hits as a system block; shared-composer + button routes documents into pendingDocs (auto-uploads, send blocked while indexing).
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""
|
|
API Routes
|
|
"""
|
|
|
|
from routes.training import router as training_router
|
|
from routes.models import router as models_router
|
|
from routes.inference import router as inference_router
|
|
from routes.inference import studio_router as inference_studio_router
|
|
from routes.datasets import router as datasets_router
|
|
from routes.auth import router as auth_router
|
|
from routes.data_recipe import router as data_recipe_router
|
|
from routes.export import router as export_router
|
|
from routes.training_history import router as training_history_router
|
|
from routes.chat_history import router as chat_history_router
|
|
from routes.providers import router as providers_router
|
|
from routes.rag import router as rag_router
|
|
|
|
__all__ = [
|
|
"training_router",
|
|
"models_router",
|
|
"inference_router",
|
|
"inference_studio_router",
|
|
"datasets_router",
|
|
"auth_router",
|
|
"data_recipe_router",
|
|
"export_router",
|
|
"training_history_router",
|
|
"chat_history_router",
|
|
"providers_router",
|
|
"rag_router",
|
|
]
|