]> git.ipfire.org Git - thirdparty/paperless-ngx.git/commitdiff
Some logging, error handling
authorshamoon <4887959+shamoon@users.noreply.github.com>
Sun, 20 Apr 2025 03:42:14 +0000 (20:42 -0700)
committershamoon <4887959+shamoon@users.noreply.github.com>
Wed, 2 Jul 2025 18:00:58 +0000 (11:00 -0700)
src/documents/ai/client.py
src/documents/ai/llm_classifier.py

index 588b45bf3dd511e5214dd4ccd50ea4b3d9f574d5..4d3b85ec0c00d33c06ff23696fdc9ba2c6bb9a45 100644 (file)
@@ -1,6 +1,10 @@
+import logging
+
 import httpx
 from django.conf import settings
 
+logger = logging.getLogger("paperless.ai.client")
+
 
 def run_llm_query(prompt: str) -> str:
     if settings.LLM_BACKEND == "ollama":
index a6809c15e17776bbd5443a5a5cc33c44a41e9dff..53fad9148eea9be515b631ecf5ec77e02dd4de06 100644 (file)
@@ -33,12 +33,14 @@ def get_ai_document_classification(document: Document) -> dict:
     """
 
     try:
+        logger.debug(f"LLM classification prompt: {prompt}")
         result = run_llm_query(prompt)
+        logger.debug(f"LLM classification result: {result}")
         suggestions = parse_llm_classification_response(result)
-        return suggestions
-    except Exception as e:
-        logger.error(f"Error during LLM classification: {e}")
-        return None
+        return suggestions or {}
+    except Exception:
+        logger.exception("Error during LLM classification: %s", exc_info=True)
+        return {}
 
 
 def parse_llm_classification_response(text: str) -> dict: