westfarn
de84ff1b00
Add uv packaging, env-driven Django settings, OLLAMA_BASE_URL (LAN GPU host), DatabaseStorage for prompt/document blobs, compose/CI/deploy workflows mirroring scha, and ASGI gunicorn+UvicornWorker for WebSockets.
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from django.db.models.signals import post_save, post_delete
|
|
from django.dispatch import receiver
|
|
from django.conf import settings
|
|
import os
|
|
|
|
from chat_backend.models import Document
|
|
|
|
|
|
def _rag_init_skipped() -> bool:
|
|
return os.environ.get("SKIP_RAG_INIT", "").lower() in {"1", "true", "yes"}
|
|
|
|
|
|
@receiver(post_save, sender=Document)
|
|
def update_vector_on_save(sender, instance, **kwargs):
|
|
"""Update vector store when documents are saved"""
|
|
if _rag_init_skipped():
|
|
return
|
|
if not kwargs.get("created", False):
|
|
return
|
|
try:
|
|
from .services.rag_services import AsyncRAGService
|
|
|
|
rag_service = AsyncRAGService()
|
|
rag_service.ingest_documents()
|
|
except Exception as exc:
|
|
print(f"Skipping vector update on Document save: {exc}")
|
|
|
|
|
|
@receiver(post_delete, sender=Document)
|
|
def delete_vector_on_remove(sender, instance, **kwargs):
|
|
"""Handle document deletion by re-indexing the whole workspace"""
|
|
if _rag_init_skipped():
|
|
return
|
|
try:
|
|
from .services.rag_services import AsyncRAGService
|
|
|
|
rag_service = AsyncRAGService()
|
|
rag_service.ingest_documents()
|
|
except Exception as exc:
|
|
print(f"Skipping vector update on Document delete: {exc}")
|