paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-08-05 18:58:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update ai_classifier.py

Browse Source
This commit is contained in:
shamoon
2025-07-15 14:42:56 -07:00
parent 3583470856
commit da2ac19193
1 changed files with 8 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/paperless_ai/ai_classifier.py
View File

@@ -1,4 +1,3 @@
import json
import logging
from django.contrib.auth.models import User
@@ -76,18 +75,14 @@ def get_context_for_document(
def parse_ai_response(raw: dict) -> dict:
try:
return {
"title": raw.get("title", ""),
"tags": raw.get("tags", []),
"correspondents": raw.get("correspondents", []),
"document_types": raw.get("document_types", []),
"storage_paths": raw.get("storage_paths", []),
"dates": raw.get("dates", []),
}
except (ValueError, json.JSONDecodeError):
logger.exception("Failed to parse AI response")
return {}
return {
"title": raw.get("title", ""),
"tags": raw.get("tags", []),
"correspondents": raw.get("correspondents", []),
"document_types": raw.get("document_types", []),
"storage_paths": raw.get("storage_paths", []),
"dates": raw.get("dates", []),
}
def get_ai_document_classification(