mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-06 22:12:12 +02:00
7fb0707933
Scoped codemod over surfsense_backend/app (excluding routes/, Wave E): renames search_space_id -> workspace_id, search_space -> workspace, SearchSpace -> Workspace across services, utils, tasks, agents, gateway, event_bus, notifications, podcasts, automations, observability params, and prompt .md files. Also flips the camelCase payload key searchSpaceId -> workspaceId (no backend reader; hard cutover). Preserved carve-outs (verbatim): Celery task names "delete_search_space_background" and "ai_sort_search_space" (wire names), and the OTel/metric key "search_space.id" (dashboards depend on it). Enum values 'SEARCH_SPACE' and SearchSourceConnector untouched.
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""Pure presentation logic for document-processing notifications."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import hashlib
|
|
from datetime import UTC, datetime
|
|
from typing import Any
|
|
|
|
from app.notifications.service.messages.text import format_title
|
|
|
|
|
|
def operation_id(document_type: str, filename: str, workspace_id: int) -> str:
|
|
"""Build a unique id for a document processing run."""
|
|
timestamp = datetime.now(UTC).strftime("%Y%m%d_%H%M%S_%f")
|
|
filename_hash = hashlib.md5(filename.encode()).hexdigest()[:8]
|
|
return f"doc_{document_type}_{workspace_id}_{timestamp}_{filename_hash}"
|
|
|
|
|
|
def started_title(document_name: str) -> str:
|
|
"""Title shown when document processing is queued."""
|
|
return format_title("Processing: ", document_name)
|
|
|
|
|
|
def progress(
|
|
stage: str,
|
|
stage_message: str | None = None,
|
|
chunks_count: int | None = None,
|
|
) -> tuple[str, dict[str, Any]]:
|
|
"""Compute the progress message and metadata updates for a processing run."""
|
|
stage_messages = {
|
|
"parsing": "Reading your file",
|
|
"chunking": "Preparing for search",
|
|
"embedding": "Preparing for search",
|
|
"storing": "Finalizing",
|
|
}
|
|
|
|
message = stage_message or stage_messages.get(stage, "Processing")
|
|
|
|
metadata_updates: dict[str, Any] = {"processing_stage": stage}
|
|
if chunks_count is not None:
|
|
metadata_updates["chunks_count"] = chunks_count
|
|
|
|
return message, metadata_updates
|
|
|
|
|
|
def completion(
|
|
document_name: str,
|
|
error_message: str | None = None,
|
|
document_id: int | None = None,
|
|
chunks_count: int | None = None,
|
|
) -> tuple[str, str, str, dict[str, Any]]:
|
|
"""Compute the final title, message, status, and metadata for a finished run."""
|
|
if error_message:
|
|
title = format_title("Failed: ", document_name)
|
|
message = f"Processing failed: {error_message}"
|
|
status = "failed"
|
|
else:
|
|
title = format_title("Ready: ", document_name)
|
|
message = "Now searchable!"
|
|
status = "completed"
|
|
|
|
metadata_updates: dict[str, Any] = {
|
|
"processing_stage": "completed" if not error_message else "failed",
|
|
"error_message": error_message,
|
|
}
|
|
if document_id is not None:
|
|
metadata_updates["document_id"] = document_id
|
|
if chunks_count is not None:
|
|
metadata_updates["chunks_count"] = chunks_count
|
|
|
|
return title, message, status, metadata_updates
|