Merge pull request #841 from i-dot-ai/bugfix/REDBOX-566-errors-logged…

…-on-file-upload Fix errors on file upload.
i-dot-ai · Jul 24, 2024 · 0802d93 · 0802d93
2 parents 679870c + 5d84cb0
commit 0802d93
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/django_app/redbox_app/redbox_core/client.py b/django_app/redbox_app/redbox_core/client.py
@@ -59,9 +59,9 @@ class CoreChatResponse:
 @dataclass_json(undefined=Undefined.EXCLUDE)
 @dataclass(frozen=True)
 class FileStatus:
-    processing_status: Literal[
-        "uploaded", "parsing", "chunking", "embedding", "indexing", "complete", "unknown", "errored"
-    ]
+    processing_status: (
+        Literal["uploaded", "parsing", "chunking", "embedding", "indexing", "complete", "unknown", "errored"] | None
+    )
 
 
 @dataclass_json(undefined=Undefined.EXCLUDE)

diff --git a/django_app/redbox_app/redbox_core/views/document_views.py b/django_app/redbox_app/redbox_core/views/document_views.py
@@ -193,7 +193,7 @@ def file_status_api_view(request: HttpRequest) -> JsonResponse:
         logger.error("Error getting file object information - no file ID provided %s.")
         return JsonResponse({"status": StatusEnum.unknown.label})
     try:
-        file = get_object_or_404(File, id=file_id)
+        file: File = get_object_or_404(File, id=file_id)
     except File.DoesNotExist as ex:
         logger.exception("File object information not found in django - file does not exist %s.", file_id, exc_info=ex)
         return JsonResponse({"status": StatusEnum.unknown.label})
@@ -205,6 +205,6 @@ def file_status_api_view(request: HttpRequest) -> JsonResponse:
             file.status = StatusEnum.unknown.label
             file.save()
         return JsonResponse({"status": file.status})
-    file.status = core_file_status_response.processing_status
+    file.status = core_file_status_response.processing_status or StatusEnum.unknown.name
     file.save()
     return JsonResponse({"status": file.get_status_text()})
diff --git a/redbox-core/redbox/models/file.py b/redbox-core/redbox/models/file.py
@@ -148,7 +148,7 @@ class FileStatus(BaseModel):
     """Status of a file."""
 
     file_uuid: UUID
-    processing_status: ProcessingStatusEnum
+    processing_status: ProcessingStatusEnum | None
     chunk_statuses: None = Field(default=None, description="deprecated, see processing_status")