LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 08:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

(feat:attch) store file mime type

Browse Source
This commit is contained in:
ManishMadan2882
2025-04-07 20:16:03 +05:30
parent 1f3d1cc73e
commit 0c1138179b
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
application/worker.py
View File

@@ -328,6 +328,7 @@ def attachment_worker(self, directory, file_info, user):
"""
import datetime
import os
import mimetypes
from application.utils import num_tokens_from_string
mongo = MongoDB.get_client()
@@ -361,6 +362,8 @@ def attachment_worker(self, directory, file_info, user):
file_path_relative = f"{settings.UPLOAD_FOLDER}/{user}/attachments/{attachment_id}/{filename}"
mime_type = mimetypes.guess_type(file_path)[0] or 'application/octet-stream'
doc_id = ObjectId(attachment_id)
attachments_collection.insert_one({
"_id": doc_id,
@@ -368,6 +371,7 @@ def attachment_worker(self, directory, file_info, user):
"path": file_path_relative,
"content": content,
"token_count": token_count,
"mime_type": mime_type,
"date": datetime.datetime.now(),
})
@@ -380,7 +384,8 @@ def attachment_worker(self, directory, file_info, user):
"filename": filename,
"path": file_path_relative,
"token_count": token_count,
"attachment_id": attachment_id
"attachment_id": attachment_id,
"mime_type": mime_type
}
else:
logging.warning("No content was extracted from the file",
@@ -388,4 +393,4 @@ def attachment_worker(self, directory, file_info, user):
raise ValueError("No content was extracted from the file")
except Exception as e:
logging.error(f"Error processing file {filename}: {e}", extra={"user": user}, exc_info=True)
raise
raise