From c4278f4784e31a0f0819478292864e5111faf452 Mon Sep 17 00:00:00 2001 From: Vaclav Cerny Date: Wed, 4 Jun 2025 14:13:00 +0200 Subject: [PATCH] fix description vs classification mismatch --- backend/open_webui/retrieval/loaders/main.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/backend/open_webui/retrieval/loaders/main.py b/backend/open_webui/retrieval/loaders/main.py index c4730ea709..817d9ddbcf 100644 --- a/backend/open_webui/retrieval/loaders/main.py +++ b/backend/open_webui/retrieval/loaders/main.py @@ -148,13 +148,13 @@ class DoclingLoader: params = { "image_export_mode": "placeholder", - "table_mode": "accurate", + "table_mode": "accurate" } if self.params: - if self.params.get("do_picture_classification"): - params["do_picture_classification"] = self.params.get( - "do_picture_classification" + if self.params.get("do_picture_description"): + params["do_picture_description"] = self.params.get( + "do_picture_description" ) if self.params.get("ocr_engine") and self.params.get("ocr_lang"): @@ -291,7 +291,7 @@ class Loader: params={ "ocr_engine": self.kwargs.get("DOCLING_OCR_ENGINE"), "ocr_lang": self.kwargs.get("DOCLING_OCR_LANG"), - "do_picture_classification": self.kwargs.get( + "do_picture_description": self.kwargs.get( "DOCLING_DO_PICTURE_DESCRIPTION" ), "picture_description_local": (