Changeup logging

src/documents/ai/client.py

@@ -7,9 +7,20 @@ logger = logging.getLogger("paperless.ai.client")
 
 
 def run_llm_query(prompt: str) -> str:
-    if settings.LLM_BACKEND == "ollama":
+    logger.debug(
-        return _run_ollama_query(prompt)
+        "Running LLM query against %s with model %s",
-    return _run_openai_query(prompt)
+        settings.LLM_BACKEND,
+        settings.LLM_MODEL,
+    )
+    match settings.LLM_BACKEND:
+        case "openai":
+            result = _run_openai_query(prompt)
+        case "ollama":
+            result = _run_ollama_query(prompt)
+        case _:
+            raise ValueError(f"Unsupported LLM backend: {settings.LLM_BACKEND}")
+    logger.debug("LLM query result: %s", result)
+    return result
 
 
 def _run_ollama_query(prompt: str) -> str:

src/documents/ai/llm_classifier.py

@@ -15,6 +15,9 @@ def get_ai_document_classification(document: Document) -> dict:
     filename = document.filename or ""
     content = document.content or ""
 
+    # Limit the content to 10k characters
+    content = content[:10000]
+
     prompt = f"""
     You are a document classification assistant. Based on the content below, return a JSON object suggesting the following classification fields:
     - title: A descriptive title for the document
@@ -33,9 +36,7 @@ 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 or {}
     except Exception: