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

Refactor

Browse Source
This commit is contained in:
shamoon
2025-04-21 11:34:09 -07:00
parent 4e23a072d4
commit de6e43738c
2 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/paperless/ai/llm_classifier.py → src/paperless/ai/ai_classifier.py
View File

@@ -4,7 +4,7 @@ import logging
from documents.models import Document
from paperless.ai.client import run_llm_query
logger = logging.getLogger("paperless.ai.llm_classifier")
logger = logging.getLogger("paperless.ai.ai_classifier")
def get_ai_document_classification(document: Document) -> dict:
@@ -50,14 +50,14 @@ def get_ai_document_classification(document: Document) -> dict:
try:
result = run_llm_query(prompt)
suggestions = parse_llm_classification_response(result)
suggestions = parse_ai_classification_response(result)
return suggestions or {}
except Exception:
logger.exception("Error during LLM classification: %s", exc_info=True)
return {}
def parse_llm_classification_response(text: str) -> dict:
def parse_ai_classification_response(text: str) -> dict:
"""
Parses LLM output and ensures it conforms to expected schema.
"""
@@ -77,4 +77,9 @@ def parse_llm_classification_response(text: str) -> dict:
}
except json.JSONDecodeError:
# fallback: try to extract JSON manually?
logger.exception(
"Failed to parse LLM classification response: %s",
text,
exc_info=True,
)
return {}