Первый коммит

app/modules/rag/job_store.py

@@ -0,0 +1,66 @@
+from dataclasses import dataclass
+from uuid import uuid4
+
+from app.modules.rag.repository import RagRepository
+from app.schemas.common import ErrorPayload, ModuleName
+from app.schemas.indexing import IndexJobStatus
+
+
+@dataclass
+class IndexJob:
+    index_job_id: str
+    rag_session_id: str
+    status: IndexJobStatus = IndexJobStatus.QUEUED
+    indexed_files: int = 0
+    failed_files: int = 0
+    error: ErrorPayload | None = None
+
+
+class IndexJobStore:
+    def __init__(self, repository: RagRepository) -> None:
+        self._repo = repository
+
+    def create(self, rag_session_id: str) -> IndexJob:
+        job = IndexJob(index_job_id=str(uuid4()), rag_session_id=rag_session_id)
+        self._repo.create_job(job.index_job_id, rag_session_id, job.status.value)
+        return job
+
+    def get(self, index_job_id: str) -> IndexJob | None:
+        row = self._repo.get_job(index_job_id)
+        if not row:
+            return None
+        payload = None
+        if row.error_code:
+            module = ModuleName.RAG
+            if row.error_module:
+                try:
+                    module = ModuleName(row.error_module)
+                except ValueError:
+                    module = ModuleName.RAG
+            payload = ErrorPayload(
+                code=row.error_code,
+                desc=row.error_desc or "",
+                module=module,
+            )
+        return IndexJob(
+            index_job_id=row.index_job_id,
+            rag_session_id=row.rag_session_id,
+            status=IndexJobStatus(row.status),
+            indexed_files=row.indexed_files,
+            failed_files=row.failed_files,
+            error=payload,
+        )
+
+    def save(self, job: IndexJob) -> None:
+        error_code = job.error.code if job.error else None
+        error_desc = job.error.desc if job.error else None
+        error_module = job.error.module.value if job.error else None
+        self._repo.update_job(
+            job.index_job_id,
+            status=job.status.value,
+            indexed_files=job.indexed_files,
+            failed_files=job.failed_files,
+            error_code=error_code,
+            error_desc=error_desc,
+            error_module=error_module,
+        )