Cutlery/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-11-04 03:27:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

Changeup logging

Browse Source
This commit is contained in:
shamoon 2025-04-19 21:01:54 -07:00
parent 56b1c7adeb
commit 65b8a74166
No known key found for this signature in database
2 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/documents/ai/client.py
View File

@ -7,9 +7,20 @@ logger = logging.getLogger("paperless.ai.client")
def run_llm_query(prompt: str) -> str:
if settings.LLM_BACKEND == "ollama":
return _run_ollama_query(prompt)
return _run_openai_query(prompt)
logger.debug(
"Running LLM query against %s with model %s",
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:

5
src/documents/ai/llm_classifier.py
View File

@ -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: