1
0
Эх сурвалжийг харах

fix description vs classification mismatch

Vaclav Cerny 4 сар өмнө
parent
commit
c4278f4784

+ 5 - 5
backend/open_webui/retrieval/loaders/main.py

@@ -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"):
-                    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"):
                 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": (