ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix description vs classification mismatch

Browse source
This commit is contained in:
Vaclav Cerny 2025-06-04 14:13:00 +02:00
parent 8644e81a1c
commit c4278f4784
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
backend/open_webui/retrieval/loaders/main.py
View file

@ -148,13 +148,13 @@ class DoclingLoader:
params = { params = {
"image_export_mode": "placeholder", "image_export_mode": "placeholder",
"table_mode": "accurate", "table_mode": "accurate"
} }
if self.params: if self.params:
if self.params.get("do_picture_classification"): if self.params.get("do_picture_description"):
params["do_picture_classification"] = self.params.get( params["do_picture_description"] = self.params.get(
"do_picture_classification" "do_picture_description"
) )
if self.params.get("ocr_engine") and self.params.get("ocr_lang"): if self.params.get("ocr_engine") and self.params.get("ocr_lang"):
@ -291,7 +291,7 @@ class Loader:
params={ params={
"ocr_engine": self.kwargs.get("DOCLING_OCR_ENGINE"), "ocr_engine": self.kwargs.get("DOCLING_OCR_ENGINE"),
"ocr_lang": self.kwargs.get("DOCLING_OCR_LANG"), "ocr_lang": self.kwargs.get("DOCLING_OCR_LANG"),
"do_picture_classification": self.kwargs.get( "do_picture_description": self.kwargs.get(
"DOCLING_DO_PICTURE_DESCRIPTION" "DOCLING_DO_PICTURE_DESCRIPTION"
), ),
"picture_description_local": ( "picture_description_local": (