Browse Source

fix description vs classification mismatch

Vaclav Cerny 4 tháng trước cách đây
mục cha
commit
c4278f4784
1 tập tin đã thay đổi với 5 bổ sung5 xóa
  1. 5 5
      backend/open_webui/retrieval/loaders/main.py

+ 5 - 5
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": (